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/getkin/kin-openapi/openapi3"
  29         "github.com/labstack/echo/v4"
  30         "helm.sh/helm/v3/pkg/cli"
  31         "oransc.org/nonrtric/capifcore/internal/common29122"
  32         "oransc.org/nonrtric/capifcore/internal/discoverserviceapi"
  33         "oransc.org/nonrtric/capifcore/internal/eventsapi"
  34         "oransc.org/nonrtric/capifcore/internal/invokermanagementapi"
  35         "oransc.org/nonrtric/capifcore/internal/providermanagementapi"
  36         "oransc.org/nonrtric/capifcore/internal/securityapi"
  37
  38         "github.com/deepmap/oapi-codegen/pkg/middleware"
  39         echomiddleware "github.com/labstack/echo/v4/middleware"
  40         log "github.com/sirupsen/logrus"
  41         "oransc.org/nonrtric/capifcore/internal/discoverservice"
  42         "oransc.org/nonrtric/capifcore/internal/eventservice"
  43         "oransc.org/nonrtric/capifcore/internal/helmmanagement"
  44         "oransc.org/nonrtric/capifcore/internal/invokermanagement"
  45         "oransc.org/nonrtric/capifcore/internal/providermanagement"
  46         "oransc.org/nonrtric/capifcore/internal/publishservice"
  47         "oransc.org/nonrtric/capifcore/internal/publishserviceapi"
  48         security "oransc.org/nonrtric/capifcore/internal/securityservice"
  49 )
  50
  51 var url string
  52 var helmManager helmmanagement.HelmManager
  53 var repoName string
  54
  55 func main() {
  56         var port = flag.Int("port", 8090, "Port for CAPIF Core Function HTTP server")
  57         var secPort = flag.Int("secPort", 4433, "Port for CAPIF Core Function HTTPS server")
  58         flag.StringVar(&url, "chartMuseumUrl", "", "ChartMuseum URL")
  59         flag.StringVar(&repoName, "repoName", "capifcore", "Repository name")
  60         var logLevelStr = flag.String("loglevel", "Info", "Log level")
  61         var certPath = flag.String("certPath", "certs/cert.pem", "Path for server certificate")
  62         var keyPath = flag.String("keyPath", "certs/key.pem", "Path for server private key")
  63
  64         flag.Parse()
  65
  66         if loglevel, err := log.ParseLevel(*logLevelStr); err == nil {
  67                 log.SetLevel(loglevel)
  68         }
  69
  70         // Add repo
  71         helmManager = helmmanagement.NewHelmManager(cli.New())
  72         err := helmManager.SetUpRepo(repoName, url)
  73         if err != nil {
  74                 log.Warnf("No Helm repo added due to: %s", err.Error())
  75         }
  76
  77         go startWebServer(getEcho(), *port)
  78         go startHttpsWebServer(getEcho(), *secPort, *certPath, *keyPath)
  79
  80         log.Info("Server started and listening on port: ", *port)
  81
  82         keepServerAlive()
  83 }
  84
  85 func getEcho() *echo.Echo {
  86         // This is how you set up a basic Echo router
  87         e := echo.New()
  88         // Log all requests
  89         e.Use(echomiddleware.Logger())
  90
  91         var group *echo.Group
  92         // Register ProviderManagement
  93         providerManagerSwagger, err := providermanagementapi.GetSwagger()
  94         if err != nil {
  95                 log.Fatalf("Error loading ProviderManagement swagger spec\n: %s", err)
  96         }
  97         providerManagerSwagger.Servers = nil
  98         providerManager := providermanagement.NewProviderManager()
  99         group = e.Group("/api-provider-management/v1")
 100         group.Use(middleware.OapiRequestValidator(providerManagerSwagger))
 101         providermanagementapi.RegisterHandlersWithBaseURL(e, providerManager, "/api-provider-management/v1")
 102
 103         // Register EventService
 104         eventServiceSwagger, err := eventsapi.GetSwagger()
 105         if err != nil {
 106                 log.Fatalf("Error loading EventService swagger spec\n: %s", err)
 107         }
 108         eventServiceSwagger.Servers = nil
 109         eventService := eventservice.NewEventService(&http.Client{})
 110         group = e.Group("/capif-events/v1")
 111         group.Use(middleware.OapiRequestValidator(eventServiceSwagger))
 112         eventsapi.RegisterHandlersWithBaseURL(e, eventService, "/capif-events/v1")
 113         eventChannel := eventService.GetNotificationChannel()
 114
 115         // Register PublishService
 116         publishServiceSwagger, err := publishserviceapi.GetSwagger()
 117         if err != nil {
 118                 log.Fatalf("Error loading PublishService swagger spec\n: %s", err)
 119         }
 120         publishServiceSwagger.Servers = nil
 121         publishService := publishservice.NewPublishService(providerManager, helmManager, eventChannel)
 122         group = e.Group("/published-apis/v1")
 123         group.Use(middleware.OapiRequestValidator(publishServiceSwagger))
 124         publishserviceapi.RegisterHandlersWithBaseURL(e, publishService, "/published-apis/v1")
 125
 126         // Register InvokerManagement
 127         invokerManagerSwagger, err := invokermanagementapi.GetSwagger()
 128         if err != nil {
 129                 log.Fatalf("Error loading InvokerManagement swagger spec\n: %s", err)
 130         }
 131         invokerManagerSwagger.Servers = nil
 132         invokerManager := invokermanagement.NewInvokerManager(publishService, eventChannel)
 133         group = e.Group("/api-invoker-management/v1")
 134         group.Use(middleware.OapiRequestValidator(invokerManagerSwagger))
 135         invokermanagementapi.RegisterHandlersWithBaseURL(e, invokerManager, "/api-invoker-management/v1")
 136
 137         // Register DiscoverService
 138         discoverServiceSwagger, err := discoverserviceapi.GetSwagger()
 139         if err != nil {
 140                 log.Fatalf("Error loading DiscoverService swagger spec\n: %s", err)
 141         }
 142         discoverServiceSwagger.Servers = nil
 143         discoverService := discoverservice.NewDiscoverService(invokerManager)
 144         group = e.Group("/service-apis/v1")
 145         group.Use(middleware.OapiRequestValidator(discoverServiceSwagger))
 146         discoverserviceapi.RegisterHandlersWithBaseURL(e, discoverService, "/service-apis/v1")
 147
 148         // Register Security
 149         securitySwagger, err := securityapi.GetSwagger()
 150         if err != nil {
 151                 log.Fatalf("Error loading Security swagger spec\n: %s", err)
 152         }
 153         securitySwagger.Servers = nil
 154         securityService := security.NewSecurity(providerManager, publishService, invokerManager)
 155         group = e.Group("/capif-security/v1")
 156         group.Use(middleware.OapiRequestValidator(securitySwagger))
 157         securityapi.RegisterHandlersWithBaseURL(e, securityService, "/capif-security/v1")
 158
 159         e.GET("/", hello)
 160
 161         e.GET("/swagger/:apiName", getSwagger)
 162
 163         return e
 164 }
 165
 166 func startWebServer(e *echo.Echo, port int) {
 167         e.Logger.Fatal(e.Start(fmt.Sprintf("0.0.0.0:%d", port)))
 168 }
 169
 170 func startHttpsWebServer(e *echo.Echo, port int, certPath string, keyPath string) {
 171         e.Logger.Fatal(e.StartTLS(fmt.Sprintf("0.0.0.0:%d", port), certPath, keyPath))
 172 }
 173
 174 func keepServerAlive() {
 175         forever := make(chan int)
 176         <-forever
 177 }
 178
 179 func hello(c echo.Context) error {
 180         return c.String(http.StatusOK, "Hello, World!")
 181 }
 182
 183 func getSwagger(c echo.Context) error {
 184         var swagger *openapi3.T
 185         var err error
 186         switch api := c.Param("apiName"); api {
 187         case "provider":
 188                 swagger, err = providermanagementapi.GetSwagger()
 189         case "publish":
 190                 swagger, err = publishserviceapi.GetSwagger()
 191         case "invoker":
 192                 swagger, err = invokermanagementapi.GetSwagger()
 193         case "discover":
 194                 swagger, err = discoverserviceapi.GetSwagger()
 195         case "events":
 196                 swagger, err = eventsapi.GetSwagger()
 197         case "security":
 198                 swagger, err = securityapi.GetSwagger()
 199         default:
 200                 return c.JSON(http.StatusBadRequest, getProblemDetails("Invalid API name "+api, http.StatusBadRequest))
 201         }
 202         if err != nil {
 203                 return c.JSON(http.StatusInternalServerError, getProblemDetails("Unable to get swagger for API", http.StatusInternalServerError))
 204         }
 205         return c.JSON(http.StatusOK, swagger)
 206 }
 207
 208 func getProblemDetails(cause string, status int) common29122.ProblemDetails {
 209         return common29122.ProblemDetails{
 210                 Cause:  &cause,
 211                 Status: &status,
 212         }
 213 }