Library/models/config.go

package models

import (
	"github.com/go-ini/ini"
	"os"
	"fmt"
)

// Config holds the config struct
type ConfigStruct struct {
	Database struct {
		Host     string
		User     string
		Password string
		Database string
	}

	JWTLoginSecret []byte

	FirstUser User `ini:"User"`
}

var Config = new(ConfigStruct)

// SetConfig initianlises the config and publishes it for other functions to use
func SetConfig() error {

	// File Checks
	if _, err := os.Stat("config.ini"); os.IsNotExist(err) {
		return err
	}

	cfg, err := ini.Load("config.ini")
	if err != nil {
		return err
	}

	err = cfg.MapTo(Config)
	if err != nil {
		return err
	}

	// Check if the first user already exists, aka a user with the ID = 1. If not, insert it
	_, exists, err := GetUserByID(1)
	if err != nil {
		return err
	}

	// If it doesn't exist, create it
	if !exists {
		_, err = CreateUser(Config.FirstUser)
		if err != nil {
			return err
		}
		fmt.Println("Created new user " + Config.FirstUser.Username)
	}

	// JWT secret
	Config.JWTLoginSecret = []byte(cfg.Section("General").Key("JWTSecret").String())

	return nil
}