package config

import (
	"os"
	"path/filepath"

	"github.com/goccy/go-yaml"
)

type WebConfig struct {
	Port int    `yaml:"port"`
	Sock string `yaml:"sock"`
}

type DatabaseConfig struct {
	Type string `yaml:"type"`
	Path string `yaml:"path"`
	DSN  string `yaml:"dsn"`
}

type Config struct {
	Web      WebConfig      `yaml:"web"`
	Database DatabaseConfig `yaml:"database"`
}

func defaultConfig() *Config {
	return &Config{
		Web: WebConfig{
			Port: 8080,
			Sock: "/run/lmvpnweb.sock",
		},
		Database: DatabaseConfig{
			Type: "sqlite",
			Path: "data/lmvpn.db",
			DSN:  "",
		},
	}
}

func Load(path string) (*Config, error) {
	dir := filepath.Dir(path)
	if err := os.MkdirAll(dir, 0755); err != nil {
		return nil, err
	}

	cfg := defaultConfig()

	data, err := os.ReadFile(path)
	if err != nil {
		if !os.IsNotExist(err) {
			return nil, err
		}

		if err := saveConfig(path, cfg); err != nil {
			return nil, err
		}
		return cfg, nil
	}

	if err := yaml.Unmarshal(data, cfg); err != nil {
		return nil, err
	}

	if err := saveConfig(path, cfg); err != nil {
		return nil, err
	}
	return cfg, nil
}

func saveConfig(path string, cfg *Config) error {
	data, err := yaml.Marshal(cfg)
	if err != nil {
		return err
	}
	return os.WriteFile(path, data, 0644)
}