// nonce_timestamp.go
package crypt

import (
	"context"

	"github.com/sashabaranov/go-openai"
	"github.com/sashabaranov/go-openai/jsonschema"
)

type NonceTimestampTool struct{}

func (t *NonceTimestampTool) Name() string { return "nonce_timestamp" }

func (t *NonceTimestampTool) Description() string {
	return "时间戳和随机数生成规范。当文档要求携带timestamp、nonce等防重放参数时使用"
}

func (t *NonceTimestampTool) GetFunctionDefinition() openai.FunctionDefinition {
	return openai.FunctionDefinition{
		Name:        t.Name(),
		Description: t.Description(),
		Parameters: jsonschema.Definition{
			Type:       jsonschema.Object,
			Properties: map[string]jsonschema.Definition{},
			Required:   []string{},
		},
	}
}

func (t *NonceTimestampTool) GetDetail(ctx context.Context, params map[string]string) (string, error) {
	return `
### 时间戳和随机数生成规范

**代码模板**：

` + "```go" + `
package crypto

import (
    "crypto/rand"
    "fmt"
    "math/big"
    "time"
)

// GenerateTimestamp 生成秒级时间戳
func GenerateTimestamp() string {
    return fmt.Sprintf("%d", time.Now().Unix())
}

// GenerateTimestampMillis 生成毫秒级时间戳
func GenerateTimestampMillis() string {
    return fmt.Sprintf("%d", time.Now().UnixMilli())
}

// GenerateNonce 生成指定长度的随机字符串（加密安全）
func GenerateNonce(length int) (string, error) {
    const charset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
    b := make([]byte, length)
    for i := range b {
        num, err := rand.Int(rand.Reader, big.NewInt(int64(len(charset))))
        if err != nil {
            return "", err
        }
        b[i] = charset[num.Int64()]
    }
    return string(b), nil
}
` + "```" + `

**注意事项**：
- 确认文档要求的是秒级还是毫秒级时间戳
- 确认nonce的长度要求（通常16-32位）
- 生产环境建议使用加密安全的随机数生成器
`, nil
}

func (t *NonceTimestampTool) Execute(ctx context.Context, params map[string]string) (string, error) {
	return t.GetDetail(ctx, params)
}