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         "helm.sh/helm/v3/pkg/cli"
  30         "oransc.org/nonrtric/capifcore/internal/discoverserviceapi"
  31         "oransc.org/nonrtric/capifcore/internal/invokermanagementapi"
  32         "oransc.org/nonrtric/capifcore/internal/providermanagementapi"
  33         "oransc.org/nonrtric/capifcore/internal/securityapi"
  34
  35         "github.com/deepmap/oapi-codegen/pkg/middleware"
  36         echomiddleware "github.com/labstack/echo/v4/middleware"
  37         log "github.com/sirupsen/logrus"
  38         "oransc.org/nonrtric/capifcore/internal/discoverservice"
  39         "oransc.org/nonrtric/capifcore/internal/helmmanagement"
  40         "oransc.org/nonrtric/capifcore/internal/invokermanagement"
  41         "oransc.org/nonrtric/capifcore/internal/providermanagement"
  42         "oransc.org/nonrtric/capifcore/internal/publishservice"
  43         "oransc.org/nonrtric/capifcore/internal/publishserviceapi"
  44         security "oransc.org/nonrtric/capifcore/internal/securityservice"
  45 )
  46
  47 var url string
  48 var helmManager helmmanagement.HelmManager
  49 var repoName string
  50
  51 func main() {
  52         var port = flag.Int("port", 8090, "Port for CAPIF Core Function HTTP server")
  53         flag.StringVar(&url, "url", "http://localhost:8080", "ChartMuseum url")
  54         flag.StringVar(&repoName, "repoName", "capifcore", "Repository name")
  55         var logLevelStr = flag.String("loglevel", "Info", "Log level")
  56         flag.Parse()
  57
  58         if loglevel, err := log.ParseLevel(*logLevelStr); err == nil {
  59                 log.SetLevel(loglevel)
  60         }
  61
  62         // Add repo
  63         fmt.Printf("Adding %s to Helm Repo\n", url)
  64         helmManager = helmmanagement.NewHelmManager(cli.New())
  65         err := helmManager.SetUpRepo(repoName, url)
  66         if err != nil {
  67                 log.Fatal(err.Error())
  68         }
  69
  70         go startWebServer(getEcho(), *port)
  71
  72         log.Info("Server started and listening on port: ", *port)
  73
  74         keepServerAlive()
  75 }
  76
  77 func getEcho() *echo.Echo {
  78         // This is how you set up a basic Echo router
  79         e := echo.New()
  80         // Log all requests
  81         e.Use(echomiddleware.Logger())
  82
  83         var group *echo.Group
  84         // Register ProviderManagement
  85         providerManagerSwagger, err := providermanagementapi.GetSwagger()
  86         if err != nil {
  87                 log.Fatalf("Error loading ProviderManagement swagger spec\n: %s", err)
  88         }
  89         providerManagerSwagger.Servers = nil
  90         providerManager := providermanagement.NewProviderManager()
  91         group = e.Group("/api-provider-management/v1")
  92         group.Use(middleware.OapiRequestValidator(providerManagerSwagger))
  93         providermanagementapi.RegisterHandlersWithBaseURL(e, providerManager, "/api-provider-management/v1")
  94
  95         // Register PublishService
  96         publishServiceSwagger, err := publishserviceapi.GetSwagger()
  97         if err != nil {
  98                 log.Fatalf("Error loading PublishService swagger spec\n: %s", err)
  99         }
 100         publishServiceSwagger.Servers = nil
 101         publishService := publishservice.NewPublishService(providerManager, helmManager)
 102         group = e.Group("/published-apis/v1")
 103         group.Use(middleware.OapiRequestValidator(publishServiceSwagger))
 104         publishserviceapi.RegisterHandlersWithBaseURL(e, publishService, "/published-apis/v1")
 105
 106         // Register InvokerManagement
 107         invokerManagerSwagger, err := invokermanagementapi.GetSwagger()
 108         if err != nil {
 109                 log.Fatalf("Error loading InvokerManagement swagger spec\n: %s", err)
 110         }
 111         invokerManagerSwagger.Servers = nil
 112         invokerManager := invokermanagement.NewInvokerManager(publishService)
 113         group = e.Group("/api-invoker-management/v1")
 114         group.Use(middleware.OapiRequestValidator(invokerManagerSwagger))
 115         invokermanagementapi.RegisterHandlersWithBaseURL(e, invokerManager, "/api-invoker-management/v1")
 116
 117         // Register DiscoverService
 118         discoverServiceSwagger, err := discoverserviceapi.GetSwagger()
 119         if err != nil {
 120                 log.Fatalf("Error loading DiscoverService swagger spec\n: %s", err)
 121         }
 122         discoverServiceSwagger.Servers = nil
 123         discoverService := discoverservice.NewDiscoverService(invokerManager)
 124         group = e.Group("/service-apis/v1")
 125         group.Use(middleware.OapiRequestValidator(discoverServiceSwagger))
 126         discoverserviceapi.RegisterHandlersWithBaseURL(e, discoverService, "/service-apis/v1")
 127
 128         // Register Security
 129         securitySwagger, err := publishserviceapi.GetSwagger()
 130         if err != nil {
 131                 log.Fatalf("Error loading Security swagger spec\n: %s", err)
 132         }
 133         securitySwagger.Servers = nil
 134         securityService := security.NewSecurity(providerManager, publishService, invokerManager)
 135         group = e.Group("/capif-security/v1")
 136         group.Use(middleware.OapiRequestValidator(securitySwagger))
 137         securityapi.RegisterHandlersWithBaseURL(e, securityService, "/capif-security/v1")
 138
 139         e.GET("/", hello)
 140
 141         return e
 142 }
 143
 144 func startWebServer(e *echo.Echo, port int) {
 145         e.Logger.Fatal(e.Start(fmt.Sprintf("0.0.0.0:%d", port)))
 146 }
 147
 148 func keepServerAlive() {
 149         forever := make(chan int)
 150         <-forever
 151 }
 152
 153 func hello(c echo.Context) error {
 154         return c.String(http.StatusOK, "Hello, World!\n")
 155 }