package storage

import (
	"context"
	"fmt"
)

// HistoryMsg — одна запись истории диалога.
type HistoryMsg struct {
	Role    string `json:"role"`    // "user" | "assistant"
	Content string `json:"content"`
}

// AppendHistory добавляет сообщение в историю задачи.
func (s *Storage) AppendHistory(ctx context.Context, taskID int64, role, content string) error {
	now := Now()
	_, err := s.db.ExecContext(ctx, `
		INSERT INTO task_history (task_id, role, content, created_at)
		VALUES (?, ?, ?, ?)`, taskID, role, content, now,
	)
	if err != nil {
		return fmt.Errorf("%w: append history: %w", ErrDB, err)
	}
	return nil
}

// GetHistory возвращает историю задачи, отсортированную по времени.
func (s *Storage) GetHistory(ctx context.Context, taskID int64) ([]HistoryMsg, error) {
	rows, err := s.db.QueryContext(ctx, `
		SELECT role, content FROM task_history
		WHERE task_id = ? ORDER BY id ASC`, taskID)
	if err != nil {
		return nil, fmt.Errorf("%w: get history: %w", ErrDB, err)
	}
	defer rows.Close()

	var msgs []HistoryMsg
	for rows.Next() {
		var m HistoryMsg
		if err := rows.Scan(&m.Role, &m.Content); err != nil {
			return nil, fmt.Errorf("%w: scan history: %w", ErrDB, err)
		}
		msgs = append(msgs, m)
	}
	return msgs, rows.Err()
}

// ClearHistory удаляет всю историю задачи (при /start).
func (s *Storage) ClearHistory(ctx context.Context, taskID int64) error {
	_, err := s.db.ExecContext(ctx, `DELETE FROM task_history WHERE task_id = ?`, taskID)
	if err != nil {
		return fmt.Errorf("%w: clear history: %w", ErrDB, err)
	}
	return nil
}