This commit is contained in:
cdle
2021-09-09 18:05:48 +08:00
parent b64fe087ff
commit b4dac8396b
5 changed files with 164 additions and 49 deletions
+1 -1
View File
@@ -96,7 +96,7 @@ func init() {
}
}
for i := range f {
qinglong.SetEnv(&qinglong.Env{
qinglong.SetEnv(qinglong.Env{
Name: i,
Value: f[i],
})
+114
View File
@@ -1 +1,115 @@
package qinglong
import (
"fmt"
"regexp"
"strings"
"github.com/cdle/sillyGirl/core"
"github.com/cdle/sillyGirl/im"
)
func init() {
core.AddCommand("ql", []core.Function{
{
Rules: []string{`config`},
Admin: true,
Handle: func(_ im.Sender) interface{} {
config, err := GetConfig()
if err != nil {
return err
}
return config
},
},
})
}
func GetConfig() (string, error) {
config := "data"
if err := req(CONFIG, &config, "/config.sh"); err != nil {
return "", err
}
return config, nil
}
func SvaeConfig(content string) error {
if err := req(POST, CONFIG, map[string]interface{}{
"name": "config.sh",
"content": content,
}, "/save"); err != nil {
return err
}
return nil
}
func GetEnvs(searchValue string) ([]Env, error) {
envs := []Env{}
content, err := GetConfig()
if err != nil {
return nil, err
}
lines := strings.Split(content, "\n")
for _, line := range lines {
if searchValue != "" && !strings.Contains(line, searchValue) {
continue
}
for i, pattern := range []string{`^\s*export\s+([^'"=\s]+)=[ '"]?(.*?)['"]?$`, `^\s*#[#\s]*export\s+([^'"=\s]+)=[ '"]?(.*?)['"]?$`, `^\s*([^'"=\s]+)=[ '"]?(.*?)['"]?$`, `^\s*#[#\s]*([^'"=\s]+)=[ '"]?(.*?)['"]?$`} {
if v := regexp.MustCompile(pattern).FindStringSubmatch(line); len(v) > 0 {
e := Env{}
if i == 1 || i == 3 {
e.Status = 1
}
e.Name = v[1]
e.Value = v[2]
envs = append(envs, e)
break
}
}
}
return envs, nil
}
func SetEnv(env Env) error {
config, err := GetConfig()
if err != nil {
return err
}
lines := strings.Split(config, "\n")
set := false
for j, line := range lines {
for i, pattern := range []string{`^\s*export\s+([^'"=\s]+)=[ '"]?(.*?)['"]?$`, `^\s*#[#\s]*export\s+([^'"=\s]+)=[ '"]?(.*?)['"]?$`, `^\s*([^'"=\s]+)=[ '"]?(.*?)['"]?$`, `^\s*#[#\s]*([^'"=\s]+)=[ '"]?(.*?)['"]?$`} {
if v := regexp.MustCompile(pattern).FindStringSubmatch(line); len(v) > 0 {
e := Env{}
if i == 1 || i == 3 {
e.Status = 1
}
e.Name = v[1]
e.Value = v[2]
if env.Name != e.Name {
continue
}
if env.Value != e.Value {
e.Value = env.Value
}
if env.Status != e.Status {
e.Status = env.Status
}
h := ""
if e.Status == 1 {
h = "# "
}
if i <= 1 {
h = "export "
}
lines[j] = h + fmt.Sprintf("%s=\"%s\"", e.Name, e.Value)
set = true
break
}
}
}
if !set {
lines = append(lines, fmt.Sprintf("export %s=\"%s\"", env.Name, env.Value))
}
return SvaeConfig(strings.Join(lines, "\n"))
}
+37 -42
View File
@@ -65,7 +65,7 @@ func init() {
},
},
{
Rules: []string{`env find ?)`},
Rules: []string{`env find ?`},
Admin: true,
Handle: func(s im.Sender) interface{} {
m := s.Get()
@@ -84,15 +84,13 @@ func init() {
},
},
{
Rules: []string{`export ([^'"=]+)=['"]?([^=]+?)['"]?`},
Regex: true,
Rules: []string{`env set ? ?`},
Admin: true,
Handle: func(s im.Sender) interface{} {
e := &Env{
err := SetEnv(Env{
Name: s.Get(0),
Value: s.Get(1),
}
err := SetEnv(e)
})
if err != nil {
return err
}
@@ -135,39 +133,39 @@ 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 {
// 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("")
@@ -205,8 +203,5 @@ func formatEnv(env *Env) string {
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)
return fmt.Sprintf("名称:%v\n备注:%v\n状态:%v\n值:%v", env.Name, env.Remarks, status, env.Value)
}
+10 -4
View File
@@ -31,6 +31,7 @@ var POST = "POST"
var DELETE = "DELETE"
var ENVS = "envs"
var CRONS = "crons"
var CONFIG = "configs"
type Carrier struct {
Get string
@@ -79,6 +80,7 @@ func req(ps ...interface{}) error {
body := []byte{}
api := ENVS
apd := ""
var get *string
var c *Carrier
var toParse interface{}
for _, p := range ps {
@@ -87,7 +89,7 @@ func req(ps ...interface{}) error {
switch p.(string) {
case GET, POST, DELETE, PUT:
method = p.(string)
case ENVS, CRONS:
case ENVS, CRONS, CONFIG:
api = p.(string)
default:
apd = p.(string)
@@ -96,6 +98,8 @@ func req(ps ...interface{}) error {
body = p.([]byte)
case *Carrier:
c = p.(*Carrier)
case *string:
get = p.(*string)
default:
if strings.Contains(reflect.TypeOf(p).String(), "*") {
toParse = p
@@ -135,11 +139,13 @@ func req(ps ...interface{}) error {
}
if toParse != nil {
if err := json.Unmarshal(data, toParse); err != nil {
fmt.Println(err)
return err
}
// fmt.Println(string(data))
// fmt.Println(toParse)
}
if get != nil {
if *get, err = jsonparser.GetString(data, *get); err != nil {
return err
}
}
if c != nil {
c.Value, _ = jsonparser.GetString(data, strings.Split(c.Get, ".")...)