nginx-ui/internal/cert/dns/config_env.go

package dns

import (
	"github.com/0xJacky/Nginx-UI/internal/cert/config"
	"github.com/BurntSushi/toml"
	"log"
	"os"
	"sort"
	"strings"
)

type Configuration struct {
	Credentials map[string]string `json:"credentials"`
	Additional  map[string]string `json:"additional"`
}

type Links struct {
	API      string `json:"api"`
	GoClient string `json:"go_client"`
}

type Config struct {
	Name          string         `json:"name"`
	Code          string         `json:"code"`
	Configuration *Configuration `json:"configuration,omitempty"`
	Links         *Links         `json:"links,omitempty"`
}

var configurations []Config

var configurationMap map[string]Config

func init() {
	filenames, err := config.ListConfigs()
	if err != nil {
		log.Fatalln(err)
	}

	configurationMap = make(map[string]Config)

	for _, filename := range filenames {
		if !strings.HasSuffix(filename, ".toml") {
			continue
		}

		data, err := config.GetConfig(filename)
		if err != nil {
			log.Fatalln(err)
		}

		c := Config{}
		err = toml.Unmarshal(data, &c)
		if err != nil {
			log.Fatalln(err)
		}

		configurationMap[c.Code] = c

		c.Configuration = nil
		c.Links = nil
		configurations = append(configurations, c)
	}

	sort.SliceStable(configurations, func(i, j int) bool {
		leftName := strings.ToLower(configurations[i].Name)
		rightName := strings.ToLower(configurations[j].Name)
		if leftName == rightName {
			return strings.ToLower(configurations[i].Code) < strings.ToLower(configurations[j].Code)
		}
		return leftName < rightName
	})
}

func GetProvidersList() []Config {
	return configurations
}

func GetProvider(code string) (Config, bool) {
	if v, ok := configurationMap[code]; ok {
		return v, ok
	}
	return Config{}, false
}

func (c *Config) SetEnv(configuration Configuration) error {
	if c.Configuration != nil {
		for k := range c.Configuration.Credentials {
			if value, ok := configuration.Credentials[k]; ok {
				err := os.Setenv(k, value)
				if err != nil {
					return err
				}
			}
		}
		for k := range c.Configuration.Additional {
			if value, ok := configuration.Additional[k]; ok {
				err := os.Setenv(k, value)
				if err != nil {
					return err
				}
			}
		}
	}
	return nil
}

func (c *Config) CleanEnv() {
	if c.Configuration != nil {
		for k := range c.Configuration.Credentials {
			_ = os.Unsetenv(k)
		}
		for k := range c.Configuration.Additional {
			_ = os.Unsetenv(k)
		}
	}
}