pomme/internal/configuration.go

package internal

import (
	"fmt"
	"os"
	"path/filepath"

	"github.com/adrg/xdg"
	"gopkg.in/yaml.v3"
	"gorm.io/gorm"
)

// User struct represents a user in the database.
type User struct {
	gorm.Model
	Username       string
	Password       string
	HashedPassword string
}

// Response struct represents a json response.
type Response struct {
	Username     string `json:"username,omitempty"`
	Message      string `json:"message,omitempty"`
	HTTPResponse int    `json:"status,omitempty"`
}

// ZoneRequest represents a Zone file request.
type ZoneRequest struct {
	*Zone

	User string `json:"user,omitempty" gorm:"foreignKey:username;references:User"`
}

// Zone struct represents a zonefile in the database.
type Zone struct {
	gorm.Model
	FileName    string `json:"name"`
	RawFileName string `json:"rawfilename"`
	Body        string `json:"body"`
}

type Config struct {
	Server        string
	HashingSecret string
	Port          string
}

var config Config

func ReadConfig() (*Config, error) {
	var (
		data              []byte
		err               error
		defaultConfigPath string
	)

	defaultConfigPath = xdg.ConfigHome + "/pomme/config.yaml"

	if data, err = os.ReadFile(filepath.Clean(defaultConfigPath)); err == nil {
		if err = yaml.Unmarshal(data, &config); err != nil {
			return &Config{}, fmt.Errorf("unable to unmarshal config file: %w", err)
		}

		return &config, nil
	}

	if data, err = os.ReadFile(filepath.Clean("./config.yaml")); err == nil {
		if err = yaml.Unmarshal(data, &config); err != nil {
			return &Config{}, fmt.Errorf("unable to unmarshal config file: %w", err)
		}

		return &config, nil
	}

	return &Config{}, fmt.Errorf("unable to read config file: %w", err)
}