oc-discovery/main.go

package main

import (
	"context"
	"log"
	"oc-discovery/conf"
	"oc-discovery/daemons/node"
	"os"
	"os/signal"
	"strings"
	"syscall"

	oclib "cloud.o-forge.io/core/oc-lib"
	beego "github.com/beego/beego/v2/server/web"
	"github.com/beego/beego/v2/server/web/filter/cors"
)

const appname = "oc-discovery"

func main() {
	// Init the oc-lib
	oclib.Init(appname)
	// get the right config file

	o := oclib.GetConfLoader()

	conf.GetConfig().Name = o.GetStringDefault("NAME", "opencloud-demo")
	conf.GetConfig().Hostname = o.GetStringDefault("HOSTNAME", "localhost")
	conf.GetConfig().PSKPath = o.GetStringDefault("PSK_PATH", "./psk/psk")
	conf.GetConfig().NodeEndpointPort = o.GetInt64Default("NODE_ENDPOINT_PORT", 4001)
	conf.GetConfig().PublicKeyPath = o.GetStringDefault("PUBLIC_KEY_PATH", "./pem/public.pem")
	conf.GetConfig().PrivateKeyPath = o.GetStringDefault("PRIVATE_KEY_PATH", "./pem/private.pem")
	conf.GetConfig().IndexerAddresses = o.GetStringDefault("INDEXER_ADDRESSES", "")

	conf.GetConfig().NodeMode = o.GetStringDefault("NODE_MODE", "node")

	// Normal beego init
	beego.BConfig.AppName = appname
	beego.BConfig.Listen.HTTPPort = o.GetIntDefault("port", 8080)
	beego.BConfig.WebConfig.DirectoryIndex = true
	beego.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
	beego.InsertFilter("*", beego.BeforeRouter, cors.Allow(&cors.Options{
		AllowAllOrigins:  true,
		AllowMethods:     []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
		AllowHeaders:     []string{"Origin", "Authorization", "Content-Type"},
		ExposeHeaders:    []string{"Content-Length", "Content-Type"},
		AllowCredentials: true,
	}))

	ctx, stop := signal.NotifyContext(
		context.Background(),
		os.Interrupt,
		syscall.SIGTERM,
	)
	defer stop()

	isNode := strings.Contains(conf.GetConfig().NodeMode, "node")
	isIndexer := strings.Contains(conf.GetConfig().NodeMode, "indexer")

	if n, err := node.InitNode(isNode, isIndexer); err != nil {
		panic(err)
	} else {
		<-ctx.Done() // 👈 the only blocking point
		log.Println("shutting down")
		n.Close()
	}

}