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