Sillyv2/develop/qinglong/env.go

package qinglong

import (
	"errors"
	"fmt"
	"strings"

	"github.com/cdle/sillyGirl/core"
	"github.com/cdle/sillyGirl/im"
)

type EnvResponse struct {
	Code int   `json:"code"`
	Data []Env `json:"data"`
}

type Env struct {
	Value     string `json:"value,omitempty"`
	ID        string `json:"_id,omitempty"`
	Status    int    `json:"status,omitempty"`
	Name      string `json:"name,omitempty"`
	Timestamp string `json:"timestamp,omitempty"`
	Remarks   string `json:"remarks,omitempty"`
}

func init() {
	core.AddCommand([]core.Function{
		{
			Rules: []string{`^env\s+get\s+([\S]+)$`},
			Handle: func(s im.Sender) interface{} {
				m := s.Get()
				env, err := GetEnv(m)
				if err != nil {
					return err
				}
				if env == nil {
					return "未设置该环境变量"
				}
				if env != nil {
					return formatEnv(env)
				}
				return nil
			},
		},
		{
			Rules: []string{`^env\s+find\s+([\S]+)$`},
			Handle: func(s im.Sender) interface{} {
				m := s.Get()
				envs, err := GetEnvs(m)
				if err != nil {
					return err
				}
				if len(envs) == 0 {
					return "未设置该环境变量"
				}
				es := []string{}
				for _, env := range envs {
					es = append(es, formatEnv(&env))
				}
				return strings.Join(es, "\n\n")
			},
		},
		{
			Rules: []string{`^export\s+([^'"=]+)=['"]?([^=]+?)['"]?$`, `^env\s+set\s+([^'"=]+)=['"]?([^=]+?)['"]?$`},
			Handle: func(s im.Sender) interface{} {
				e := &Env{
					Name:  s.Get(0),
					Value: s.Get(1),
				}
				err := SetEnv(e)
				if err != nil {
					return err
				}
				return fmt.Sprintf("操作成功")
			},
		},
		{
			Rules: []string{`^env\s+del\s+([\S]+)$`},
			Handle: func(s im.Sender) interface{} {
				if err := RemEnv(&Env{ID: s.Get()}); err != nil {
					return err
				}
				return "操作成功"
			},
		},
		{
			Rules: []string{`^env\s+remark\s+([\S]+)\s+([\S]+)$`},
			Handle: func(s im.Sender) interface{} {
				if err := ModEnv(&Env{ID: s.Get(0), Remarks: s.Get(1)}); err != nil {
					return err
				}
				return "操作成功"
			},
		},
	})
}

func GetEnv(name string) (*Env, error) {
	envs, err := GetEnvs("")
	if err != nil {
		return nil, err
	}
	for _, env := range envs {
		if env.Name == name {
			return &env, nil
		}
	}
	return nil, nil
}

func GetEnvs(searchValue string) ([]Env, error) {
	er := EnvResponse{}
	if err := req(ENVS, &er, "?searchValue="+searchValue); err != nil {
		return nil, err
	}
	return er.Data, nil
}

func SetEnv(e *Env) error {
	if e.Name == "JD_COOKIE" {
		return errors.New("不支持的操作")
	}
	envs, err := GetEnvs("")
	if err != nil {
		return err
	}
	for _, env := range envs {
		if env.Name == e.Name {
			if e.Remarks != "" {
				env.Remarks = e.Remarks
			}
			if e.Value != "" {
				env.Value = e.Value
			}
			if e.Name != "" {
				env.Name = e.Name
			}
			env.Timestamp = ""
			return req(PUT, ENVS, env)
		}
	}
	return AddEnv(e)
}

func ModEnv(e *Env) error {
	envs, err := GetEnvs("")
	if err != nil {
		return err
	}
	for _, env := range envs {
		if env.ID == e.ID {
			if e.Remarks != "" {
				env.Remarks = e.Remarks
			}
			if e.Value != "" {
				env.Value = e.Value
			}
			if e.Name != "" {
				env.Name = e.Name
			}
			env.Timestamp = ""
			return req(PUT, ENVS, env)
		}
	}
	return errors.New("找不到环境变量")
}

func AddEnv(e *Env) error {
	return req(POST, ENVS, []Env{*e})
}

func RemEnv(e *Env) error {
	return req(DELETE, ENVS, []byte(`["`+e.ID+`"]`))
}

func formatEnv(env *Env) string {
	status := "已启用"
	if env.Status != 0 {
		status = "已禁用"
	}
	if env.Remarks == "" {
		env.Remarks = "无"
	}
	return fmt.Sprintf("名称：%v\n编号：%v\n备注：%v\n状态：%v\n时间：%v\n值：%v", env.Name, env.ID, env.Remarks, status, env.Timestamp, env.Value)
}