wujian_develop_tool/config/configFile.go

package config

import (
	"git.icechen.cn/pkg/wujian_develop_tool/util"
	"io/ioutil"

	"github.com/urfave/cli/v2"

	"gopkg.in/yaml.v3"
)

var DefaultConfigFile = ".drone.yml"

func Init(c *cli.Context) error {
	// 读取配置
	configFileName := c.String("config")
	if configFileName != "" {
		DefaultConfigFile = configFileName
	}
	if err := ReadConfig(); err != nil {
		return err
	}

	appList := c.Args().Slice()
	Config.HasApi = Config.Api.HasApp(appList)
	Config.HasService = Config.Service.HasApp(appList)

	return nil
}

type ConfigFile struct {
	Kind    string      `json:"kind" yaml:"kind"`
	Type    string      `json:"type" yaml:"type"`
	Name    string      `json:"name" yaml:"name"`
	Api     ApiList     `json:"api" yaml:"api"`
	Service ServiceList `json:"service" yaml:"service"`

	HasApi     ApiList     `json:"-" yaml:"-"`
	HasService ServiceList `json:"-" yaml:"-"`
}

type (
	ApiList []Api
	Api     struct {
		Name      string `json:"name" yaml:"name"`
		AliasName string `json:"alias_name" yaml:"aliasName"`
		Root      string `json:"root" yaml:"root"`
		Type      string `json:"type" yaml:"type"`
		Port      string `json:"port" yaml:"port"`
		Host      string `json:"host,omitempty" yaml:"host,omitempty"`
		Path      string `json:"path,omitempty" yaml:"path,omitempty"`
	}
)

type (
	ServiceList []Service
	Service     struct {
		Name      string `json:"name" yaml:"name"`
		AliasName string `json:"alias_name" yaml:"aliasName"`
		Root      string `json:"root" yaml:"root"`
		Type      string `json:"type" yaml:"type"`
		Port      string `json:"port" yaml:"port"`
	}
)

var Config ConfigFile

func ReadConfig() error {
	configFile, err := ioutil.ReadFile(DefaultConfigFile)
	if err != nil {
		return err
	}
	err = yaml.Unmarshal(configFile, &Config)
	if err != nil {
		return err
	}
	return nil
}

func (c *ConfigFile) AppendAPI(api Api) error {
	has := false
	for i, a := range c.Api {
		if a.Name == api.Name {
			c.Api[i] = api
			has = true
			break
		}
	}

	if !has {
		c.Api = append(c.Api, api)
	}

	out, err := yaml.Marshal(c)
	if err != nil {
		return err
	}

	return util.WriteFileWithNotEdit(DefaultConfigFile, out)
}

func (c *ConfigFile) AppendService(service Service) error {
	has := false
	for i, s := range c.Service {
		if s.Name == service.Name {
			c.Service[i] = service
			has = true
			break
		}
	}

	if !has {
		c.Service = append(c.Service, service)
	}

	out, err := yaml.Marshal(c)
	if err != nil {
		return err
	}

	return util.WriteFileWithNotEdit(DefaultConfigFile, out)
}

func (sl ApiList) HasApp(appNameList []string) ApiList {
	if len(appNameList) == 0 {
		return sl
	}

	hasApp := make(ApiList, 0, len(sl))

al:
	for _, api := range sl {
		for _, a := range appNameList {
			if a == api.Name {
				hasApp = append(hasApp, api)
				continue al
			}
		}
	}
	return hasApp
}

func (sl ApiList) GetApp(appName string) Api {
	for _, api := range sl {
		if api.Name == appName {
			return api
		}
	}
	return Api{}
}

func (sl ServiceList) HasApp(appNameList []string) ServiceList {
	if len(appNameList) == 0 {
		return sl
	}

	hasApp := make(ServiceList, 0, len(sl))

sl:
	for _, service := range sl {
		for _, a := range appNameList {
			if a == service.Name {
				hasApp = append(hasApp, service)
				continue sl
			}
		}
	}
	return hasApp
}

func (sl ServiceList) GetApp(appName string) Service {
	for _, service := range sl {
		if service.Name == appName {
			return service
		}
	}
	return Service{}
}