package prompt

import (
	"fmt"
	"gitee.com/kzangv/gsf-ai-agent/utils"
	"regexp"
	"strings"
)

const StringPromptDelimiterChar = "="

type StringPrompt struct {
	tmp       string
	paramName [][2]string
}

func (t *StringPrompt) Init(tmp string) error {
	// 用来匹配 {{keyword}} 的正则表达式
	re := regexp.MustCompile(`{{(.*?)}}`)

	// 使用正则表达式替换所有的关键词为 %s
	t.tmp = re.ReplaceAllStringFunc(tmp, func(match string) string {
		sv := strings.Split(strings.Trim(match, " {}"), StringPromptDelimiterChar)
		var v, d string
		v = sv[0]
		if len(sv) > 1 {
			d = strings.Join(sv[1:], StringPromptDelimiterChar)
		}
		t.paramName = append(t.paramName, [2]string{v, d})
		return "%s"
	})
	return nil
}

func (t *StringPrompt) Format(params utils.ParamsManager) (string, error) {
	if t.tmp == "" {
		return "", fmt.Errorf("模板字段为空")
	}

	// 格式化
	fParams := make([]interface{}, 0, len(t.paramName))
	for k := range t.paramName {
		v, exist := params.Get(t.paramName[k][0])
		if !exist {
			v = t.paramName[k][1]
		}
		fParams = append(fParams, v)
	}

	return fmt.Sprintf(t.tmp, fParams...), nil
}