wujian_develop_tool/config/configFile.go

179 lines
3.4 KiB
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{}
}