api.go

package main

import (
	"context"
	"errors"
	"fmt"
	"github.com/dpwgc/easierweb"
	"reflect"
	"regexp"
	"strings"
)

type API struct {
	route map[string]APIOptions
}

func NewAPI() (*API, error) {
	route := make(map[string]APIOptions)
	for _, db := range Config().DBs {
		s, err := NewStore(db)
		if err != nil {
			return nil, err
		}
		for i := 0; i < len(db.Groups); i++ {
			for j := 0; j < len(db.Groups[i].APIs); j++ {
				db.Groups[i].APIs[j].Store = s
				route[fmt.Sprintf("%s.%s.%s", db.Name, db.Groups[i].Name, db.Groups[i].APIs[j].Name)] = db.Groups[i].APIs[j]
			}
		}
	}
	return &API{
		route: route,
	}, nil
}

type CommandReply struct {
	Result map[string]int64 `json:"result,omitempty"`
	Error  string           `json:"error,omitempty"`
}

type QueryReply struct {
	Result []map[string]any `json:"result,omitempty"`
	Error  string           `json:"error,omitempty"`
}

type InfoReply struct {
	Result []map[string]any `json:"result,omitempty"`
}

type ErrorReply struct {
	Error string `json:"error,omitempty"`
}

func (a *API) Query(ctx *easierweb.Context, request map[string]any) (*QueryReply, error) {
	api := a.route[fmt.Sprintf("%s.%s.%s", ctx.Path.Get("db"), ctx.Path.Get("group"), ctx.Path.Get("api"))]
	if len(api.Name) == 0 || len(api.Sql) == 0 {
		return nil, errors.New("not found")
	}
	hide := ctx.Query.Get("hide")
	if len(hide) > 0 {
		api.Hide = strings.Split(hide, ",")
	}
	show := ctx.Query.Get("show")
	if len(show) > 0 {
		api.Show = strings.Split(show, ",")
	}
	format := ctx.Query.Get("format")
	if len(format) > 0 {
		api.Format = format
	}
	sql := api.Sql
	sql, args := a.handleSql(sql, request, api.Params)
	rows, err := api.Store.Query(context.Background(), api, sql, args...)
	return &QueryReply{
		Error:  errorToString(err),
		Result: rows,
	}, err
}

func (a *API) Command(ctx *easierweb.Context, request map[string]any) (*CommandReply, error) {
	api := a.route[fmt.Sprintf("%s.%s.%s", ctx.Path.Get("db"), ctx.Path.Get("group"), ctx.Path.Get("api"))]
	if len(api.Name) == 0 || len(api.Sql) == 0 {
		return nil, errors.New("not found")
	}
	sql := api.Sql
	sql, args := a.handleSql(sql, request, api.Params)
	res, err := api.Store.Command(context.Background(), sql, args...)
	return &CommandReply{
		Error:  errorToString(err),
		Result: res,
	}, err
}

func (a *API) Info(ctx *easierweb.Context) *InfoReply {
	reply := &InfoReply{}
	for _, db := range Config().DBs {
		for _, group := range db.Groups {
			for _, api := range group.APIs {
				reply.Result = append(reply.Result, map[string]any{
					"db":     db.Name,
					"group":  group.Name,
					"api":    api.Name,
					"params": api.Params,
				})
			}
		}
	}
	return reply
}

func (a *API) Health(ctx *easierweb.Context) {

}

func (a *API) handleSql(sql string, request map[string]any, params []string) (string, []any) {
	var args []any
	for _, field := range params {
		value := request[field]
		if value == nil {
			sql = a.delStr(sql, fmt.Sprintf("{#%s}", field), fmt.Sprintf("{/%s}", field))
		}
	}
	for _, field := range params {
		value := request[field]
		if value == nil {
			continue
		}
		f := fmt.Sprintf("{%s}", field)
		if strings.Contains(sql, f) {
			if reflect.TypeOf(value).Kind() == reflect.Slice {
				var in []string
				arr := value.([]any)
				for _, item := range arr {
					in = append(in, "?")
					args = append(args, item)
				}
				sql = strings.ReplaceAll(sql, f, fmt.Sprintf("(%s)", strings.Join(in, ",")))
			} else {
				sql = strings.ReplaceAll(sql, f, "?")
				args = append(args, value)
			}
		}
		sql = strings.ReplaceAll(sql, fmt.Sprintf("{#%s}", field), "")
		sql = strings.ReplaceAll(sql, fmt.Sprintf("{/%s}", field), "")
	}
	if Config().Server.Debug {
		fmt.Println("[DEBUG]", "REQ:", request, "/", "SQL:", sql, "/", "ARGS:", args)
	}
	return sql, args
}

func (a *API) delStr(s string, start string, end string) string {
	// 查找需要删除的部分
	startIndex := strings.Index(s, start)
	endIndex := strings.Index(s, end) + len(end)
	if startIndex == -1 || endIndex == -1 {
		return s
	}

	// 删除指定部分（包括起始和结束字符串）
	re := regexp.MustCompile(start + "(.*?)" + end)
	return re.ReplaceAllLiteralString(s, "")
}

func errorToString(err error) string {
	if err == nil {
		return ""
	}
	return err.Error()
}