cdle
138 lines
2.7 KiB
Go
138 lines
2.7 KiB
Go
package qinglong
|
|
|
|
import (
|
|
"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 "操作成功"
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func GetEnv(searchValue string) (*Env, error) {
|
|
envs, err := GetEnvs(searchValue)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(envs) == 0 {
|
|
return nil, nil
|
|
}
|
|
return &envs[0], 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 {
|
|
es, err := GetEnvs(e.Name)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(es) == 0 {
|
|
return AddEnv(e)
|
|
}
|
|
e.ID = es[0].ID
|
|
return req(PUT, ENVS, *e)
|
|
}
|
|
|
|
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)
|
|
}
|