update

develop/qinglong/init.go

@@ -1,14 +1,18 @@
 package qinglong
 
 import (
+	"encoding/json"
 	"errors"
 	"fmt"
+	"reflect"
+	"strings"
 	"time"
 
 	"github.com/beego/beego/v2/adapter/httplib"
 	"github.com/beego/beego/v2/core/logs"
 	"github.com/buger/jsonparser"
 	"github.com/cdle/sillyGirl/core"
+	"github.com/cdle/sillyGirl/im"
 )
 
 type Yaml struct {
@@ -29,11 +33,50 @@ var ENVS = "envs"
 
 func init() {
 	core.ReadYaml(core.ExecPath+"/develop/qinglong/conf/", &Config, "https://raw.githubusercontent.com/cdle/sillyGirl/main/develop/qinglong/conf/config.yaml")
-	token, err := getToken()
+	_, err := getToken()
 	if err == nil {
-		logs.Info("青龙可及%v", token)
+		logs.Info("青龙已连接")
 	}
-
+	core.AddCommand([]core.Function{
+		{
+			Rules: []string{`^env\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 {
+					status := "已启用"
+					if env.Status != 0 {
+						status = "已禁用"
+					}
+					if env.Remarks == "" {
+						env.Remarks = "无"
+					}
+					return fmt.Sprintf("名称：%v\n备注：%v\n状态：%v\n时间：%v\n值：%v", env.Name, env.Remarks, status, env.Timestamp, env.Value)
+				}
+				return nil
+			},
+		},
+		{
+			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("操作成功")
+			},
+		},
+	})
 }
 
 func getToken() (string, error) {
@@ -57,3 +100,75 @@ func getToken() (string, error) {
 	expiration, _ = jsonparser.GetInt(data, "data", "expiration")
 	return token, nil
 }
+
+func req(ps ...interface{}) error {
+	token, err := getToken()
+	if err != nil {
+		return err
+	}
+	method := GET
+	body := []byte{}
+	api := ENVS
+	apd := ""
+	var toParse interface{}
+	for _, p := range ps {
+		switch p.(type) {
+		case string:
+			switch p.(string) {
+			case GET, POST, DELETE, PUT:
+				method = p.(string)
+			case ENVS:
+				api = p.(string)
+			default:
+				apd = p.(string)
+			}
+		case []byte:
+			body = p.([]byte)
+		default:
+			if strings.Contains(reflect.TypeOf(p).String(), "*") {
+				toParse = p
+			} else {
+				fmt.Println(p, "--")
+				body, _ = json.Marshal(p)
+				fmt.Println(string(body), "++")
+			}
+		}
+	}
+	var req *httplib.BeegoHTTPRequest
+	api += apd
+
+	switch method {
+	case GET:
+		req = httplib.Get(Config.Host + "/open/" + api)
+	case POST:
+		req = httplib.Delete(Config.Host + "/open/" + api)
+	case DELETE:
+		req = httplib.Delete(Config.Host + "/open/" + api)
+	case PUT:
+		req = httplib.Put(Config.Host + "/open/" + api)
+	}
+	req.Header("Authorization", fmt.Sprintf("Bearer %s", token))
+	req.Header("Content-Type", "application/json;charset=UTF-8")
+	if method != GET {
+
+		req.Body(body)
+	}
+	fmt.Println(Config.Host + "/open/" + api)
+	fmt.Println(method)
+	fmt.Println(string(body))
+	req.Body(body)
+	data, err := req.Bytes()
+	if err != nil {
+		return err
+	}
+	code, _ := jsonparser.GetInt(data, "code")
+	if code != 200 {
+		return errors.New(string(data))
+	}
+	if toParse != nil {
+		if err := json.Unmarshal(data, toParse); err != nil {
+			return errors.New(fmt.Sprintf("解析错误：%v,%v", err, string(data)))
+		}
+	}
+	return nil
+}