update
This commit is contained in:
cdle
+23
-86
@@ -1,39 +1,41 @@
|
||||
package qinglong
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"reflect"
|
||||
"strings"
|
||||
|
||||
"github.com/beego/beego/v2/adapter/httplib"
|
||||
"github.com/buger/jsonparser"
|
||||
)
|
||||
|
||||
type EnvResponse struct {
|
||||
Code int `json:"code"`
|
||||
Data []Env `json:"data"`
|
||||
}
|
||||
|
||||
type Env struct {
|
||||
Value string `json:"value"`
|
||||
ID string `json:"_id"`
|
||||
Status int `json:"status"`
|
||||
Name string `json:"name"`
|
||||
Remarks string `json:"remarks"`
|
||||
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 GetEnv(searchValue string) ([]Env, error) {
|
||||
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(PUT, ENVS, &er, "?searchValue="+searchValue); err != nil {
|
||||
if err := req(ENVS, &er, "?searchValue="+searchValue); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return er.Data, nil
|
||||
}
|
||||
|
||||
func SetEnv(e *Env) error {
|
||||
es, err := GetEnv(e.Name)
|
||||
es, err := GetEnvs(e.Name)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
@@ -41,7 +43,7 @@ func SetEnv(e *Env) error {
|
||||
return AddEnv(e)
|
||||
}
|
||||
e.ID = es[0].ID
|
||||
return req(PUT, ENVS, e)
|
||||
return req(PUT, ENVS, *e)
|
||||
}
|
||||
|
||||
func AddEnv(e *Env) error {
|
||||
@@ -49,7 +51,7 @@ func AddEnv(e *Env) error {
|
||||
}
|
||||
|
||||
func RemEnv(e *Env) error {
|
||||
es, err := GetEnv(e.Name)
|
||||
es, err := GetEnvs(e.Name)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
@@ -62,68 +64,3 @@ func RemEnv(e *Env) error {
|
||||
}
|
||||
return req(DELETE, ENVS, td)
|
||||
}
|
||||
|
||||
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:
|
||||
method = p.(string)
|
||||
default:
|
||||
apd = p.(string)
|
||||
}
|
||||
case []byte:
|
||||
body = p.([]byte)
|
||||
default:
|
||||
if strings.Contains(reflect.TypeOf(&p).String(), "*") {
|
||||
toParse = p
|
||||
} else {
|
||||
body, _ = json.Marshal(p)
|
||||
}
|
||||
}
|
||||
}
|
||||
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)
|
||||
}
|
||||
if method != GET && len(body) > 0 {
|
||||
req.Body(body)
|
||||
}
|
||||
req.Header("Authorization", fmt.Sprintf("Bearer %s", token))
|
||||
data, _ := json.Marshal(body)
|
||||
req.Body(data)
|
||||
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
|
||||
}
|
||||
|
||||
+118
-3
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user