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         e := echo.New()
  87         // Log all requests
  88         e.Use(echomiddleware.Logger())
  89
  90         var group *echo.Group
  91         // Register ProviderManagement
  92         providerManagerSwagger, err := providermanagementapi.GetSwagger()
  93         if err != nil {
  94                 log.Fatalf("Error loading ProviderManagement swagger spec\n: %s", err)
  95         }
  96         providerManagerSwagger.Servers = nil
  97         providerManager := providermanagement.NewProviderManager()
  98         group = e.Group("/api-provider-management/v1")
  99         group.Use(middleware.OapiRequestValidator(providerManagerSwagger))
 100         providermanagementapi.RegisterHandlersWithBaseURL(e, providerManager, "/api-provider-management/v1")
 101
 102         // Register EventService
 103         eventServiceSwagger, err := eventsapi.GetSwagger()
 104         if err != nil {
 105                 log.Fatalf("Error loading EventService swagger spec\n: %s", err)
 106         }
 107         eventServiceSwagger.Servers = nil
 108         eventService := eventservice.NewEventService(&http.Client{})
 109         group = e.Group("/capif-events/v1")
 110         group.Use(middleware.OapiRequestValidator(eventServiceSwagger))
 111         eventsapi.RegisterHandlersWithBaseURL(e, eventService, "/capif-events/v1")
 112         eventChannel := eventService.GetNotificationChannel()
 113
 114         // Register PublishService
 115         publishServiceSwagger, err := publishserviceapi.GetSwagger()
 116         if err != nil {
 117                 log.Fatalf("Error loading PublishService swagger spec\n: %s", err)
 118         }
 119         publishServiceSwagger.Servers = nil
 120         publishService := publishservice.NewPublishService(providerManager, helmManager, eventChannel)
 121         group = e.Group("/published-apis/v1")
 122         group.Use(middleware.OapiRequestValidator(publishServiceSwagger))
 123         publishserviceapi.RegisterHandlersWithBaseURL(e, publishService, "/published-apis/v1")
 124
 125         // Register InvokerManagement
 126         invokerManagerSwagger, err := invokermanagementapi.GetSwagger()
 127         if err != nil {
 128                 log.Fatalf("Error loading InvokerManagement swagger spec\n: %s", err)
 129         }
 130         invokerManagerSwagger.Servers = nil
 131         invokerManager := invokermanagement.NewInvokerManager(publishService, eventChannel)
 132         group = e.Group("/api-invoker-management/v1")
 133         group.Use(middleware.OapiRequestValidator(invokerManagerSwagger))
 134         invokermanagementapi.RegisterHandlersWithBaseURL(e, invokerManager, "/api-invoker-management/v1")
 135
 136         // Register DiscoverService
 137         discoverServiceSwagger, err := discoverserviceapi.GetSwagger()
 138         if err != nil {
 139                 log.Fatalf("Error loading DiscoverService swagger spec\n: %s", err)
 140         }
 141         discoverServiceSwagger.Servers = nil
 142         discoverService := discoverservice.NewDiscoverService(invokerManager)
 143         group = e.Group("/service-apis/v1")
 144         group.Use(middleware.OapiRequestValidator(discoverServiceSwagger))
 145         discoverserviceapi.RegisterHandlersWithBaseURL(e, discoverService, "/service-apis/v1")
 146
 147         // Register Security
 148         securitySwagger, err := securityapi.GetSwagger()
 149         if err != nil {
 150                 log.Fatalf("Error loading Security swagger spec\n: %s", err)
 151         }
 152         securitySwagger.Servers = nil
 153         securityService := security.NewSecurity(providerManager, publishService, invokerManager)
 154         group = e.Group("/capif-security/v1")
 155         group.Use(middleware.OapiRequestValidator(securitySwagger))
 156         securityapi.RegisterHandlersWithBaseURL(e, securityService, "/capif-security/v1")
 157
 158         e.GET("/", hello)
 159
 160         e.GET("/swagger/:apiName", getSwagger)
 161
 162         return e
 163 }
 164
 165 func startWebServer(e *echo.Echo, port int) {
 166         e.Logger.Fatal(e.Start(fmt.Sprintf("0.0.0.0:%d", port)))
 167 }
 168
 169 func startHttpsWebServer(e *echo.Echo, port int, certPath string, keyPath string) {
 170         e.Logger.Fatal(e.StartTLS(fmt.Sprintf("0.0.0.0:%d", port), certPath, keyPath))
 171 }
 172
 173 func keepServerAlive() {
 174         forever := make(chan int)
 175         <-forever
 176 }
 177
 178 func hello(c echo.Context) error {
 179         return c.String(http.StatusOK, "Hello, World!")
 180 }
 181
 182 func getSwagger(c echo.Context) error {
 183         var swagger *openapi3.T
 184         var err error
 185         switch api := c.Param("apiName"); api {
 186         case "provider":
 187                 swagger, err = providermanagementapi.GetSwagger()
 188         case "publish":
 189                 swagger, err = publishserviceapi.GetSwagger()
 190         case "invoker":
 191                 swagger, err = invokermanagementapi.GetSwagger()
 192         case "discover":
 193                 swagger, err = discoverserviceapi.GetSwagger()
 194         case "events":
 195                 swagger, err = eventsapi.GetSwagger()
 196         case "security":
 197                 swagger, err = securityapi.GetSwagger()
 198         default:
 199                 return c.JSON(http.StatusBadRequest, getProblemDetails("Invalid API name "+api, http.StatusBadRequest))
 200         }
 201         if err != nil {
 202                 return c.JSON(http.StatusInternalServerError, getProblemDetails("Unable to get swagger for API", http.StatusInternalServerError))
 203         }
 204         return c.JSON(http.StatusOK, swagger)
 205 }
 206
 207 func getProblemDetails(cause string, status int) common29122.ProblemDetails {
 208         return common29122.ProblemDetails{
 209                 Cause:  &cause,
 210                 Status: &status,
 211         }
 212 }