Sillyv2/core/api_bucket.go

package core

import (
	"bytes"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"strconv"
	"strings"
	"time"

	"github.com/cdle/sillyplus/utils"
	"github.com/gin-gonic/gin"
	"github.com/goccy/go-json"
)

func checkFilePlugin(key string, value *string) {
	if isNameUuid(key) {
		if v, ok := plugins_id.Load(key); ok {
			data, _ := os.ReadFile(v.(string))
			*value = string(data)
		} else {
			*value = "非法操作，请勿乱动。"
		}
	}
}

func init() {
	var sillyGirl = MakeBucket("sillyGirl")
	GinApi(GET, "/api/storage/list", RequireAuth, func(ctx *gin.Context) {
		page, _ := strconv.Atoi(ctx.DefaultQuery("current", "1"))
		perPage, _ := strconv.Atoi(ctx.DefaultQuery("pageSize", "20"))
		keys := ctx.Query("keys")
		data := []map[string]string{}
		arr := strings.Split(keys, ",")
		if keys == "" {
			ctx.JSON(200, map[string]interface{}{
				"success": true,
				"data":    data,
				"page":    page,
				"total":   len(data),
			})
			return
		}
		for _, bk := range arr {
			ar := strings.SplitN(bk, ".", 2)
			if len(ar) == 2 {
				if ar[0] == "plugins" && false { //todo
					// data[bk] = halfDeEct(MakeBucket(ar[0]).GetString(ar[1]))
				} else {
					// data[bk] = MakeBucket(ar[0]).GetString(ar[1])
					data = append(data, map[string]string{
						"bucket": ar[0],
						"key":    ar[1],
						"value":  MakeBucket(ar[0]).GetString(ar[1]),
					})
				}
			}
			if len(ar) == 1 {
				MakeBucket(ar[0]).Foreach(func(b1, b2 []byte) error {
					data = append(data, map[string]string{
						"bucket": bk,
						"key":    string(b1),
						"value":  string(b2),
					})
					return nil
				})
			}
		}
		start := (page - 1) * perPage
		end := start + perPage
		if end > len(data) {
			end = len(data)
		}
		res := data[start:end]
		index := start + 1
		for i := range res {
			res[i]["index"] = fmt.Sprint(index)
			index++
		}
		ctx.JSON(200, map[string]interface{}{
			"success": true,
			"data":    res,
			"page":    page,
			"total":   len(data),
		})
	})
	GinApi(GET, "/api/storage", RequireAuth, func(ctx *gin.Context) {
		keys := ctx.Query("keys")
		if keys == "" {
			buckets := sillyGirl.Buckets()
			search := ctx.Query("search")
			res := []map[string]interface{}{}
			if search == "" {
				for _, bucket := range buckets {
					if bucket == "plugins" {
						continue
					}
					res = append(res, map[string]interface{}{
						"value": bucket,
						"text":  "[桶] " + bucket,
					})
				}
				ctx.JSON(200, map[string]interface{}{
					"success": true,
					"data":    res,
				})
				return
			}
			for _, bucket := range buckets {
				if bucket == "plugins" {
					continue
				}
				if strings.Contains(bucket, search) {
					res = append(res, map[string]interface{}{
						"value": bucket,
						"text":  "[桶] " + bucket,
					})
				}
				b := MakeBucket(bucket)
				b.Foreach(func(b1, b2 []byte) error {
					key := string(b1)
					value := string(b2)
					if strings.Contains(key, search) {
						res = append(res, map[string]interface{}{
							"value": bucket + "." + key,
							"text":  "[键] " + key,
						})
					}
					if strings.Contains(value, search) {
						res = append(res, map[string]interface{}{
							"value": bucket + "." + key,
							"text":  "[值] " + value,
						})
					}
					return nil
				})
			}

			ctx.JSON(200, map[string]interface{}{
				"success": true,
				"data":    res,
			})
			return
		}
		data := map[string]interface{}{}
		arr := strings.Split(keys, ",")
		for _, bk := range arr {
			ar := strings.SplitN(bk, ".", 2)
			if len(ar) == 2 {
				if ar[0] == "plugins" { //todo
					value := MakeBucket(ar[0]).GetString(ar[1])
					checkFilePlugin(ar[1], &value)
					if IsCdle {
						value = DecryptPlugin(halfDeEct(value))
					}
					data[bk] = value
				} else {
					data[bk] = TransformBucketKeyValue(MakeBucket(ar[0]).GetString(ar[1]))
				}
			}
			if len(ar) == 1 {
				MakeBucket(ar[0]).Foreach(func(b1, b2 []byte) error {
					data[bk+"."+string(b1)] = TransformBucketKeyValue(string(b2))
					return nil
				})
			}
		}
		ctx.JSON(200, map[string]interface{}{
			"success": true,
			"data":    data,
		})
	})
	GinApi(PUT, "/api/storage", RequireAuth, func(ctx *gin.Context) {
		data, err := ioutil.ReadAll(ctx.Request.Body)
		if err != nil {
			ctx.JSON(200, map[string]interface{}{
				"success":      false,
				"errorMessage": err.Error(),
				"showType":     2,
			})
			return
		}
		updates := map[string]interface{}{}
		err = json.Unmarshal(data, &updates)
		if err != nil {
			ctx.JSON(200, map[string]interface{}{
				"success":      false,
				"errorMessage": err.Error(),
				"showType":     2,
			})
			return
		}
		messages := map[string]interface{}{}
		errors := map[string]interface{}{}
		changes := map[string]bool{}
		for bk, v := range updates {
			ar := strings.SplitN(bk, ".", 2)
			if len(ar) == 2 {
				msg, changed, err := SetBucketKeyValue(MakeBucket(ar[0]), ar[1], v)
				if msg != "" {
					messages[bk] = msg
				}
				if err != nil {
					errors[bk] = err.Error()
				}
				changes[bk] = changed

				if ar[0] == "plugins" && changed {
					go func(uuid string, v interface{}) {
						defer recover()
						content := v.(string)
						if content == "" || content == "install" {
							return
						}
						_id := utils.GenUUID()
						unix := fmt.Sprint(time.Now().Unix())
						http.Post(
							"https://imdraw.com/api/plugins/backup?"+strings.Join([]string{
								"_id=" + _id,
								"uuid=" + uuid,
								"machine_id=" + machine_id,
								"unix=" + unix,
								"sign=" + utils.Md5(uuid+machine_id+unix+_id+"fuckatm"),
							}, "&"),
							"application/json",
							bytes.NewBuffer([]byte(content)))
					}(ar[1], v)
				}
			}
		}
		ctx.JSON(200, map[string]interface{}{
			"success":  true,
			"messages": messages,
			"errors":   errors,
			"changes":  changes,
		})
	})
}