dal/pkg/proto/request.go

package proto

import (
	"fmt"
	"reflect"
	"slices"
	"strings"

	"github.com/nesterow/dal/pkg/adapter"
	"github.com/nesterow/dal/pkg/builder"
)

//go:generate msgp

type BuilderMethod struct {
	Method string        `msg:"method"`
	Args   []interface{} `msg:"args"`
}

type Request struct {
	Id       uint32          `msg:"id"`
	Db       string          `msg:"db"`
	Commands []BuilderMethod `msg:"commands"`
}

var allowedMethods = strings.Split(builder.BUILDER_CLIENT_METHODS, "|")

func (q *Request) Parse(dialect adapter.Dialect) (adapter.Query, error) {
	if q.Db == "" {
		return adapter.Query{}, fmt.Errorf("Request format: db url is required")
	}
	if len(q.Commands) == 0 {
		return adapter.Query{}, fmt.Errorf("Request format: commands are required")
	}
	b := builder.New(dialect)
	exec := false
	for _, cmd := range q.Commands {
		if !slices.Contains(allowedMethods, cmd.Method) {
			return adapter.Query{}, fmt.Errorf(
				"method %s is not allowed, available methods are %v",
				cmd.Method,
				allowedMethods,
			)
		}
		method := reflect.ValueOf(b).MethodByName(cmd.Method)
		if !method.IsValid() {
			return adapter.Query{}, fmt.Errorf("method %s not found", cmd.Method)
		}
		if cmd.Method == "Insert" || cmd.Method == "Set" || cmd.Method == "Delete" {
			exec = true
		}
		// check if raw is an exec query
		if cmd.Method == "Raw" {
			qo, ok := cmd.Args[0].(map[string]interface{})
			if ok {
				sq := qo["s"].(string)
				exec = !strings.HasPrefix(sq, "SELECT")
			}
		}
		args := make([]reflect.Value, len(cmd.Args))
		for i, arg := range cmd.Args {
			args[i] = reflect.ValueOf(arg)
		}
		method.Call(args)
	}
	expr, data := b.Sql()
	return adapter.Query{
		Db:          q.Db,
		Expression:  expr,
		Data:        data,
		Transaction: b.Transaction,
		Exec:        exec,
	}, nil
}