capifcore/main.go

   1 // -
   2 //   ========================LICENSE_START=================================
   3 //   O-RAN-SC
   4 //   %%
   5 //   Copyright (C) 2022: Nordix Foundation
   6 //   %%
   7 //   Licensed under the Apache License, Version 2.0 (the "License");
   8 //   you may not use this file except in compliance with the License.
   9 //   You may obtain a copy of the License at
  10 //
  11 //        http://www.apache.org/licenses/LICENSE-2.0
  12 //
  13 //   Unless required by applicable law or agreed to in writing, software
  14 //   distributed under the License is distributed on an "AS IS" BASIS,
  15 //   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16 //   See the License for the specific language governing permissions and
  17 //   limitations under the License.
  18 //   ========================LICENSE_END===================================
  19 //
  20
  21 package main
  22
  23 import (
  24         "flag"
  25         "fmt"
  26         "net/http"
  27
  28         "github.com/labstack/echo/v4"
  29         "oransc.org/nonrtric/capifcore/internal/discoverserviceapi"
  30         "oransc.org/nonrtric/capifcore/internal/invokermanagementapi"
  31         "oransc.org/nonrtric/capifcore/internal/providermanagementapi"
  32         "oransc.org/nonrtric/capifcore/internal/securityapi"
  33
  34         "github.com/deepmap/oapi-codegen/pkg/middleware"
  35         echomiddleware "github.com/labstack/echo/v4/middleware"
  36         log "github.com/sirupsen/logrus"
  37         "oransc.org/nonrtric/capifcore/internal/discoverservice"
  38         "oransc.org/nonrtric/capifcore/internal/helmmanagement"
  39         "oransc.org/nonrtric/capifcore/internal/invokermanagement"
  40         "oransc.org/nonrtric/capifcore/internal/providermanagement"
  41         "oransc.org/nonrtric/capifcore/internal/publishservice"
  42         "oransc.org/nonrtric/capifcore/internal/publishserviceapi"
  43         "oransc.org/nonrtric/capifcore/internal/security"
  44 )
  45
  46 var url string
  47 var helmManager helmmanagement.HelmManager
  48 var repoName string
  49
  50 func main() {
  51         var port = flag.Int("port", 8090, "Port for CAPIF Core Function HTTP server")
  52         flag.StringVar(&url, "url", "http://chartmuseum:8080", "ChartMuseum url")
  53         flag.StringVar(&repoName, "repoName", "local-dev", "Repository name")
  54         var logLevelStr = flag.String("loglevel", "Info", "Log level")
  55         flag.Parse()
  56
  57         if loglevel, err := log.ParseLevel(*logLevelStr); err == nil {
  58                 log.SetLevel(loglevel)
  59         }
  60
  61         // Add repo
  62         fmt.Printf("Adding %s to Helm Repo\n", url)
  63         helmManager = helmmanagement.NewHelmManager()
  64         err := helmManager.AddToRepo(repoName, url)
  65         if err != nil {
  66                 log.Fatal(err.Error())
  67         }
  68
  69         go startWebServer(getEcho(), *port)
  70
  71         log.Info("Server started and listening on port: ", *port)
  72
  73         keepServerAlive()
  74 }
  75
  76 func getEcho() *echo.Echo {
  77         // This is how you set up a basic Echo router
  78         e := echo.New()
  79         // Log all requests
  80         e.Use(echomiddleware.Logger())
  81
  82         var group *echo.Group
  83         // Register ProviderManagement
  84         providerManagerSwagger, err := providermanagementapi.GetSwagger()
  85         if err != nil {
  86                 log.Fatalf("Error loading ProviderManagement swagger spec\n: %s", err)
  87         }
  88         providerManagerSwagger.Servers = nil
  89         providerManager := providermanagement.NewProviderManager()
  90         group = e.Group("/api-provider-management/v1")
  91         group.Use(middleware.OapiRequestValidator(providerManagerSwagger))
  92         providermanagementapi.RegisterHandlersWithBaseURL(e, providerManager, "/api-provider-management/v1")
  93
  94         // Register PublishService
  95         publishServiceSwagger, err := publishserviceapi.GetSwagger()
  96         if err != nil {
  97                 log.Fatalf("Error loading PublishService swagger spec\n: %s", err)
  98         }
  99         publishServiceSwagger.Servers = nil
 100         publishService := publishservice.NewPublishService(providerManager, helmManager)
 101         group = e.Group("/published-apis/v1")
 102         group.Use(middleware.OapiRequestValidator(publishServiceSwagger))
 103         publishserviceapi.RegisterHandlersWithBaseURL(e, publishService, "/published-apis/v1")
 104
 105         // Register DiscoverService
 106         discoverServiceSwagger, err := discoverserviceapi.GetSwagger()
 107         if err != nil {
 108                 log.Fatalf("Error loading DiscoverService swagger spec\n: %s", err)
 109         }
 110         discoverServiceSwagger.Servers = nil
 111         discoverService := discoverservice.NewDiscoverService(publishService)
 112         group = e.Group("/service-apis/v1")
 113         group.Use(middleware.OapiRequestValidator(discoverServiceSwagger))
 114         discoverserviceapi.RegisterHandlersWithBaseURL(e, discoverService, "/service-apis/v1")
 115
 116         // Register InvokerManagement
 117         invokerManagerSwagger, err := invokermanagementapi.GetSwagger()
 118         if err != nil {
 119                 log.Fatalf("Error loading InvokerManagement swagger spec\n: %s", err)
 120         }
 121         invokerManagerSwagger.Servers = nil
 122         invokerManager := invokermanagement.NewInvokerManager(publishService)
 123         group = e.Group("/api-invoker-management/v1")
 124         group.Use(middleware.OapiRequestValidator(invokerManagerSwagger))
 125         invokermanagementapi.RegisterHandlersWithBaseURL(e, invokerManager, "/api-invoker-management/v1")
 126
 127         // Register Security
 128         securitySwagger, err := publishserviceapi.GetSwagger()
 129         if err != nil {
 130                 log.Fatalf("Error loading Security swagger spec\n: %s", err)
 131         }
 132         securitySwagger.Servers = nil
 133         securityService := security.NewSecurity(providerManager, publishService, invokerManager)
 134         group = e.Group("/capif-security/v1")
 135         group.Use(middleware.OapiRequestValidator(securitySwagger))
 136         securityapi.RegisterHandlersWithBaseURL(e, securityService, "/capif-security/v1")
 137
 138         e.GET("/", hello)
 139
 140         return e
 141 }
 142
 143 func startWebServer(e *echo.Echo, port int) {
 144         e.Logger.Fatal(e.Start(fmt.Sprintf("0.0.0.0:%d", port)))
 145 }
 146
 147 func keepServerAlive() {
 148         forever := make(chan int)
 149         <-forever
 150 }
 151
 152 func hello(c echo.Context) error {
 153         return c.String(http.StatusOK, "Hello, World!\n")
 154 }