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-10-02 10:40:20 +08:00
parent a5c9cee870
commit 8164f3e161
5 changed files with 42 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.gitignore
+1
View File
@@ -11,3 +11,4 @@ jd_*
data data
dwz dwz
replies replies
qinglongs
develop/qinglong/config.go
+2 -2
View File
@@ -144,14 +144,14 @@ func init() {
func GetConfig() (string, error) { func GetConfig() (string, error) {
config := "data" config := "data"
if err := Req(CONFIG, &config, "/config.sh"); err != nil { if err := Config.Req(CONFIG, &config, "/config.sh"); err != nil {
return "", err return "", err
} }
return config, nil return config, nil
} }
func SvaeConfig(content string) error { func SvaeConfig(content string) error {
if err := Req(POST, CONFIG, map[string]interface{}{ if err := Config.Req(POST, CONFIG, map[string]interface{}{
"name": "config.sh", "name": "config.sh",
"content": content, "content": content,
}, "/save"); err != nil { }, "/save"); err != nil {
develop/qinglong/cron.go
+9 -9
View File
@@ -81,7 +81,7 @@ func init() {
if err != nil { if err != nil {
return err return err
} }
if err := Req(CRONS, PUT, "/run", []byte(fmt.Sprintf(`["%s"]`, cron.ID))); err != nil { if err := Config.Req(CRONS, PUT, "/run", []byte(fmt.Sprintf(`["%s"]`, cron.ID))); err != nil {
return err return err
} }
return fmt.Sprintf("已运行 %v", cron.Name) return fmt.Sprintf("已运行 %v", cron.Name)
@@ -95,7 +95,7 @@ func init() {
if err != nil { if err != nil {
return err return err
} }
if err := Req(CRONS, PUT, "/stop", []byte(fmt.Sprintf(`["%s"]`, cron.ID))); err != nil { if err := Config.Req(CRONS, PUT, "/stop", []byte(fmt.Sprintf(`["%s"]`, cron.ID))); err != nil {
return err return err
} }
return "操作成功" return "操作成功"
@@ -109,7 +109,7 @@ func init() {
if err != nil { if err != nil {
return err return err
} }
if err := Req(CRONS, PUT, "/enable", []byte(fmt.Sprintf(`["%s"]`, cron.ID))); err != nil { if err := Config.Req(CRONS, PUT, "/enable", []byte(fmt.Sprintf(`["%s"]`, cron.ID))); err != nil {
return err return err
} }
return "操作成功" return "操作成功"
@@ -123,7 +123,7 @@ func init() {
if err != nil { if err != nil {
return err return err
} }
if err := Req(CRONS, PUT, "/disable", []byte(fmt.Sprintf(`["%s"]`, cron.ID))); err != nil { if err := Config.Req(CRONS, PUT, "/disable", []byte(fmt.Sprintf(`["%s"]`, cron.ID))); err != nil {
return err return err
} }
return "操作成功" return "操作成功"
@@ -173,7 +173,7 @@ func init() {
if err != nil { if err != nil {
return err return err
} }
if err := Req(CRONS, PUT, "/run", []byte(fmt.Sprintf(`["%s"]`, cron.ID))); err != nil { if err := Config.Req(CRONS, PUT, "/run", []byte(fmt.Sprintf(`["%s"]`, cron.ID))); err != nil {
return err return err
} }
return "操作成功" return "操作成功"
@@ -230,7 +230,7 @@ func init() {
continue continue
} }
if strings.Contains(crons[i].Command, "jd_disable.py") { if strings.Contains(crons[i].Command, "jd_disable.py") {
Req(CRONS, PUT, "/disable", []byte(fmt.Sprintf(`["%s"]`, crons[i].ID))) Config.Req(CRONS, PUT, "/disable", []byte(fmt.Sprintf(`["%s"]`, crons[i].ID)))
continue continue
} }
if task, ok := tasks[crons[i].Name]; ok { if task, ok := tasks[crons[i].Name]; ok {
@@ -241,7 +241,7 @@ func init() {
dup = task dup = task
tasks[crons[i].Name] = crons[i] tasks[crons[i].Name] = crons[i]
} }
if err := Req(CRONS, PUT, "/disable", []byte(fmt.Sprintf(`["%s"]`, dup.ID))); err != nil { if err := Config.Req(CRONS, PUT, "/disable", []byte(fmt.Sprintf(`["%s"]`, dup.ID))); err != nil {
s.Reply(fmt.Sprintf("隐藏 %v %v %v", dup.Name, dup.Command, err)) s.Reply(fmt.Sprintf("隐藏 %v %v %v", dup.Name, dup.Command, err))
} else { } else {
s.Reply(fmt.Sprintf("已隐藏重复任务 %v %v", dup.Name, dup.Command), core.N) s.Reply(fmt.Sprintf("已隐藏重复任务 %v %v", dup.Name, dup.Command), core.N)
@@ -258,7 +258,7 @@ func init() {
func GetCrons(searchValue string) ([]Cron, error) { func GetCrons(searchValue string) ([]Cron, error) {
er := CronResponse{} er := CronResponse{}
if err := Req(CRONS, &er, "?searchValue="+searchValue); err != nil { if err := Config.Req(CRONS, &er, "?searchValue="+searchValue); err != nil {
return nil, err return nil, err
} }
return er.Data, nil return er.Data, nil
@@ -268,7 +268,7 @@ func GetCronLog(id string) (string, error) {
c := &Carrier{ c := &Carrier{
Get: "data", Get: "data",
} }
if err := Req(CRONS, "/"+id+"/log", c); err != nil { if err := Config.Req(CRONS, "/"+id+"/log", c); err != nil {
return "", err return "", err
} }
return c.Value, nil return c.Value, nil
develop/qinglong/env.go
+6 -6
View File
@@ -32,7 +32,7 @@ func GetEnv(id string) (*Env, error) {
func GetEnvs(searchValue string) ([]Env, error) { func GetEnvs(searchValue string) ([]Env, error) {
er := EnvResponse{} er := EnvResponse{}
if err := Req(ENVS, &er, "?searchValue="+searchValue); err != nil { if err := Config.Req(ENVS, &er, "?searchValue="+searchValue); err != nil {
return nil, err return nil, err
} }
return er.Data, nil return er.Data, nil
@@ -54,14 +54,14 @@ func SetEnv(e Env) error {
if e.Name != "" { if e.Name != "" {
env.Name = e.Name env.Name = e.Name
} }
return Req(PUT, ENVS, env) return Config.Req(PUT, ENVS, env)
} }
} }
return AddEnv(e) return AddEnv(e)
} }
func UdpEnv(env Env) error { func UdpEnv(env Env) error {
return Req(PUT, ENVS, env) return Config.Req(PUT, ENVS, env)
} }
func ModEnv(e Env) error { func ModEnv(e Env) error {
@@ -80,16 +80,16 @@ func ModEnv(e Env) error {
if e.Name != "" { if e.Name != "" {
env.Name = e.Name env.Name = e.Name
} }
return Req(PUT, ENVS, env) return Config.Req(PUT, ENVS, env)
} }
} }
return errors.New("找不到环境变量") return errors.New("找不到环境变量")
} }
func AddEnv(e Env) error { func AddEnv(e Env) error {
return Req(POST, ENVS, []Env{e}) return Config.Req(POST, ENVS, []Env{e})
} }
func RemEnv(e *Env) error { func RemEnv(e *Env) error {
return Req(DELETE, ENVS, []byte(`["`+e.ID+`"]`)) return Config.Req(DELETE, ENVS, []byte(`["`+e.ID+`"]`))
} }
develop/qinglong/init.go
+23 -22
View File
@@ -15,16 +15,16 @@ import (
"github.com/cdle/sillyGirl/core" "github.com/cdle/sillyGirl/core"
) )
type Yaml struct { type QingLong struct {
Host string Host string `json:"host"`
ClientID string `yaml:"client_id"` ClientID string `json:"client_id"`
ClientSecret string `yaml:"client_secret"` ClientSecret string `json:"client_secret"`
Token string
} }
var Config Yaml var Config *QingLong
var qinglong = core.NewBucket("qinglong") var qinglong = core.NewBucket("qinglong")
var token string
var expiration int64 var expiration int64
var GET = "GET" var GET = "GET"
var PUT = "PUT" var PUT = "PUT"
@@ -40,23 +40,27 @@ type Carrier struct {
} }
func init() { func init() {
Config = &QingLong{}
Config.Host = qinglong.Get("host") Config.Host = qinglong.Get("host")
Config.ClientID = qinglong.Get("client_id") Config.ClientID = qinglong.Get("client_id")
Config.ClientSecret = qinglong.Get("client_secret") Config.ClientSecret = qinglong.Get("client_secret")
if Config.Host == "" {
return
}
if v := regexp.MustCompile(`^(https?://[\.\w]+:?\d*)`).FindStringSubmatch(Config.Host); len(v) == 2 { if v := regexp.MustCompile(`^(https?://[\.\w]+:?\d*)`).FindStringSubmatch(Config.Host); len(v) == 2 {
Config.Host = v[1] Config.Host = v[1]
} }
_, err := getToken() _, err := Config.GetToken()
if err == nil { if err == nil {
logs.Info("青龙已连接") logs.Info("青龙已连接")
} }
} }
func getToken() (string, error) { func (ql *QingLong) GetToken() (string, error) {
if token != "" && expiration > time.Now().Unix() { if ql.Token != "" && expiration > time.Now().Unix() {
return token, nil return ql.Token, nil
} }
req := httplib.Get(fmt.Sprintf("%s/open/auth/token?client_id=%s&client_secret=%s", Config.Host, Config.ClientID, Config.ClientSecret)) req := httplib.Get(fmt.Sprintf("%s/open/auth/token?client_id=%s&client_secret=%s", ql.Host, ql.ClientID, ql.ClientSecret))
data, err := req.Bytes() data, err := req.Bytes()
if err != nil { if err != nil {
msg := fmt.Sprintf("青龙连接失败：%v", err) msg := fmt.Sprintf("青龙连接失败：%v", err)
@@ -69,13 +73,13 @@ func getToken() (string, error) {
logs.Warn(msg) logs.Warn(msg)
return "", errors.New(msg) return "", errors.New(msg)
} }
token, _ = jsonparser.GetString(data, "data", "token") ql.Token, _ = jsonparser.GetString(data, "data", "token")
expiration, _ = jsonparser.GetInt(data, "data", "expiration") expiration, _ = jsonparser.GetInt(data, "data", "expiration")
return token, nil return ql.Token, nil
} }
func Req(ps ...interface{}) error { func (ql *QingLong) Req(ps ...interface{}) error {
token, err := getToken() token, err := ql.GetToken()
if err != nil { if err != nil {
return err return err
} }
@@ -116,22 +120,19 @@ func Req(ps ...interface{}) error {
api = strings.Trim(api, " ") api = strings.Trim(api, " ")
switch method { switch method {
case GET: case GET:
req = httplib.Get(Config.Host + "/open/" + api) req = httplib.Get(ql.Host + "/open/" + api)
case POST: case POST:
req = httplib.Post(Config.Host + "/open/" + api) req = httplib.Post(ql.Host + "/open/" + api)
case DELETE: case DELETE:
req = httplib.Delete(Config.Host + "/open/" + api) req = httplib.Delete(ql.Host + "/open/" + api)
case PUT: case PUT:
req = httplib.Put(Config.Host + "/open/" + api) req = httplib.Put(ql.Host + "/open/" + api)
} }
req.Header("Authorization", fmt.Sprintf("Bearer %s", token)) req.Header("Authorization", fmt.Sprintf("Bearer %s", token))
req.Header("Content-Type", "application/json;charset=UTF-8") req.Header("Content-Type", "application/json;charset=UTF-8")
if method != GET { if method != GET {
req.Body(body) req.Body(body)
} }
// fmt.Println(Config.Host + "/open/" + api)
// fmt.Println(method)
// fmt.Println(string(body))
data, err := req.Bytes() data, err := req.Bytes()
if err != nil { if err != nil {
return err return err