controller.go

package nucleus

import (
	"code.fbi.h-da.de/cocsn/gosdn/database"
	"code.fbi.h-da.de/cocsn/gosdn/log"
	"code.fbi.h-da.de/cocsn/gosdn/nucleus/interfaces"
	"code.fbi.h-da.de/cocsn/gosdn/restconf/client/ciena"
	"github.com/BurntSushi/toml"
	"os"
)

type controllerConfig struct {
	CliSocket        string
	DatabaseSocket   string
	DatabaseUser     string
	DatabasePassword string
	DatabaseCrypto   bool
	ConfigPath       string
}

type clientConfig struct{
	identifier string
	endpoint string
	username string
	password string
}

type Core struct {
	//Assert type with clients[key].(*MCPClient)
	clients  map[string]interfaces.Client
	database database.Database
	config   controllerConfig
	IsRunning chan bool
}

func (c *Core) Init(socket, configFileController, configFileClient string, IsRunningChannel chan bool) {
	if err := c.readControllerConfig(configFileController); err != nil {
		log.Fatal(err)
	}

	if socket != "localhost:55055" {
		c.config.CliSocket = socket
	}

	c.AttachDatabase()

	c.IsRunning = IsRunningChannel

	if err := c.readClientConfig(configFileClient); err != nil {
		log.Fatal(err)
	}
}

func (c *Core) AttachDatabase() {
	c.database = database.NewDatabaseClient(c.config.DatabaseSocket, c.config.DatabaseUser, c.config.DatabasePassword, c.config.DatabaseCrypto)
}

func (c *Core) Shutdown() {

	<- c.IsRunning
	log.Info("Received shutdown signal. Shutting down")

	f, err := os.Create(c.config.ConfigPath)
	if err != nil {
		log.Fatal(err)
	}
	enc := toml.NewEncoder(f)
	if err := enc.Encode(c.config); err != nil {
		log.Fatal(err)
	}
	log.Info("Shutdown complete")
	os.Exit(0)
}

func (c *Core)readControllerConfig(configFileController string) error {
	if configFileController == "" {
		configFileController = "gosdn.toml"
	}
	if _, err := os.Stat(configFileController); err != nil {
		return err
	}

	c.config = controllerConfig{}
	if _, err := toml.DecodeFile(configFileController, &c.config); err != nil {
		return err
	}

	if c.config.ConfigPath == "" {
		c.config.ConfigPath = configFileController
	}
	return nil
}

func (c *Core)readClientConfig(configFileClient string) error {
	if configFileClient == "" {
		configFileClient = "clients.toml"
	}
	if _,err := os.Stat(configFileClient); err != nil {
		return err
	}
	clients := make([]clientConfig, 0)
	if _,err := toml.DecodeFile(configFileClient, &clients); err != nil {
		return err
	}
	for _,client := range clients {
		c.clients[client.identifier] = ciena.NewMCPClient(client.endpoint, client.username, client.password, &c.database)
	}
	return nil
}