Sansui233_proxypool/config/config.go

package config

import (
	"fmt"
	"io/ioutil"
	"os"

	"github.com/ghodss/yaml"
	"github.com/zu1k/proxypool/pkg/tool"
)

var (
	NeedFetch = true
	Url       = "https://raw.githubusercontent.com/zu1k/proxypool/master/source.yaml"
)

type Source struct {
	Type    string       `json:"type" yaml:"type"`
	Options tool.Options `json:"options" yaml:"options"`
}

type Config struct {
	Domain  string   `json:"domain" yaml:"domain"`
	CFEmail string   `json:"cf_email" yaml:"cf_email"`
	CFKey   string   `json:"cf_key" yaml:"cf_key"`
	Sources []Source `json:"sources" yaml:"sources"`
}

var SourceConfig = Config{}

func Parse(path string) (*Config, error) {
	fileData, err := readFile(path)
	if err != nil {
		return nil, err
	}
	SourceConfig = Config{}
	err = yaml.Unmarshal(fileData, &SourceConfig)
	if err != nil {
		return nil, err
	}
	if domain := os.Getenv("DOMAIN"); domain != "" {
		SourceConfig.Domain = domain
	}
	if cfEmail := os.Getenv("CF_API_EMAIL"); cfEmail != "" {
		SourceConfig.CFEmail = cfEmail
	}
	if cfKey := os.Getenv("CF_API_KEY"); cfKey != "" {
		SourceConfig.CFKey = cfKey
	}
	return &SourceConfig, nil
}

func readFile(path string) ([]byte, error) {
	if _, err := os.Stat(path); os.IsNotExist(err) {
		return nil, err
	}
	data, err := ioutil.ReadFile(path)
	if err != nil {
		return nil, err
	}

	if len(data) == 0 {
		return nil, fmt.Errorf("Configuration file %s is empty", path)
	}

	return data, err
}