controller.go

package nucleus

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

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

type Core struct {
	clients  []interfaces.Client
	database database.Database
	config   controllerConfig
}

func (c *Core) Init(socket, configfile string) {
	if configfile == "" {
		configfile = "gosdn.toml"
	}
	_, err := os.Stat(configfile)
	if err != nil {
		log.Fatal("Config file is missing: ", configfile)
	}

	c.config = controllerConfig{}
	if _, err := toml.DecodeFile(configfile, &c.config); err != nil {
		log.Fatal(err)
	}
	if socket != "localhost:55055" {
		c.config.CliSocket = socket
	}
	if c.config.ConfigPath == "" {
		c.config.ConfigPath = configfile
	}

	c.database = database.NewDatabaseClient(c.config.DatabaseSocket, c.config.DatabaseUser, c.config.DatabasePassword, c.config.DatabaseCrypto)
}

func (c *Core) Shutdown() {
	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)
	}
}