package tools

import (
	"context"
	"fmt"
	"time"

	"eino-test/internal/rag"
	"eino-test/internal/store"

	"github.com/cloudwego/eino/components/tool"
	"github.com/cloudwego/eino/components/tool/utils"
)

type CreateNoteInput struct {
	Title   string   `json:"title" jsonschema:"description=Note title,required"`
	Content string   `json:"content" jsonschema:"description=Markdown content of the note,required"`
	Tags    []string `json:"tags" jsonschema:"description=Tags for categorization"`
}

func NewCreateNoteTool(pipeline *rag.RAGPipeline) tool.InvokableTool {
	t, _ := utils.InferTool("create_note", "Create a new note in the knowledge base.", func(ctx context.Context, input *CreateNoteInput) (string, error) {
		note := &store.Note{
			ID:        input.Title,
			Title:     input.Title,
			Content:   input.Content,
			Tags:      input.Tags,
			CreatedAt: time.Now(),
			UpdatedAt: time.Now(),
		}
		if err := pipeline.NoteStore().Create(ctx, note); err != nil {
			return "", fmt.Errorf("create note: %w", err)
		}
		if err := pipeline.IndexNote(ctx, note); err != nil {
			return "", fmt.Errorf("index note: %w", err)
		}
		return fmt.Sprintf("Note created: %s", note.ID), nil
	})
	return t
}