chickliu/Sillyv2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
cdle
2021-09-09 19:28:23 +08:00
parent ae9b3b2916
commit 19804c540e
4 changed files with 166 additions and 149 deletions
Download Patch File Download Diff File Expand all files Collapse all files
develop/qinglong/env.go
+29 -142
View File
@@ -2,11 +2,6 @@ package qinglong
import (
"errors"
"fmt"
"strings"
"github.com/cdle/sillyGirl/core"
"github.com/cdle/sillyGirl/im"
)
type EnvResponse struct {
@@ -23,103 +18,6 @@ type Env struct {
Remarks string `json:"remarks,omitempty"`
}
func init() {
core.AddCommand("ql", []core.Function{
{
Rules: []string{`envs`},
Admin: true,
Handle: func(_ im.Sender) interface{} {
envs, err := GetEnvs("")
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{`env get ?`},
Admin: true,
Handle: func(s im.Sender) interface{} {
name := s.Get()
envs, err := GetEnvs(name)
if err != nil {
return err
}
if len(envs) == 0 {
return "未设置该环境变量"
}
es := []string{}
for _, env := range envs {
if env.Name == name {
es = append(es, formatEnv(&env))
}
}
return strings.Join(es, "\n\n")
},
},
{
Rules: []string{`env find ?`},
Admin: true,
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{`env set ? ?`},
Admin: true,
Handle: func(s im.Sender) interface{} {
err := SetEnv(Env{
Name: s.Get(0),
Value: s.Get(1),
})
if err != nil {
return err
}
return fmt.Sprintf("操作成功")
},
},
{
Rules: []string{`env del ?`},
Admin: true,
Handle: func(s im.Sender) interface{} {
if err := RemEnv(&Env{ID: s.Get()}); err != nil {
return err
}
return "操作成功"
},
},
{
Rules: []string{`env remark ? ?`},
Admin: true,
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 {
@@ -133,39 +31,36 @@ func GetEnv(name string) (*Env, error) {
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 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 SetEnv(e *Env) error {
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("")
@@ -197,11 +92,3 @@ func AddEnv(e *Env) error {
func RemEnv(e *Env) error {
return req(DELETE, ENVS, []byte(`["`+e.ID+`"]`))
}
func formatEnv(env *Env) string {
status := "已启用"
if env.Status != 0 {
status = "已禁用"
}
return fmt.Sprintf("名称：%v\n备注：%v\n状态：%v\n值：%v", env.Name, env.Remarks, status, env.Value)
}