package memory

import (
	"time"

	"git.db123.ir/db123/divacode/internal/storage"
)

type SelfMemory struct {
	db *storage.DB
}

type Promise struct {
	ID        string
	Promise   string
	Context   string
	Fulfilled bool
	CreatedAt time.Time
	Deadline  *time.Time
}

type Strategy struct {
	ID        string
	Situation string
	Strategy  string
	Outcome   string
	CreatedAt time.Time
}

func NewSelfMemory(db *storage.DB) *SelfMemory {
	return &SelfMemory{db: db}
}

func (sm *SelfMemory) AddPromise(promise, context string, deadline *time.Time) error {
	id := time.Now().Format("150405.000")
	return sm.db.Exec(
		"INSERT INTO agent_promises (id, promise, context, deadline) VALUES (?, ?, ?, ?)",
		id, promise, context, deadline,
	)
}

func (sm *SelfMemory) FulfillPromise(id string) error {
	return sm.db.Exec("UPDATE agent_promises SET fulfilled = 1 WHERE id = ?", id)
}

func (sm *SelfMemory) ActivePromises() ([]Promise, error) {
	rows, err := sm.db.Query(
		"SELECT id, promise, context, fulfilled, created_at FROM agent_promises WHERE fulfilled = 0 ORDER BY created_at DESC",
	)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var promises []Promise
	for rows.Next() {
		var p Promise
		var created string
		if err := rows.Scan(&p.ID, &p.Promise, &p.Context, &p.Fulfilled, &created); err != nil {
			continue
		}
		promises = append(promises, p)
	}
	return promises, nil
}

func (sm *SelfMemory) AddStrategy(situation, strategy, outcome string) error {
	id := time.Now().Format("150405.000")
	return sm.db.Exec(
		"INSERT INTO agent_strategies (id, situation, strategy, outcome) VALUES (?, ?, ?, ?)",
		id, situation, strategy, outcome,
	)
}

func (sm *SelfMemory) StrategiesByOutcome(outcome string) ([]Strategy, error) {
	rows, err := sm.db.Query(
		"SELECT id, situation, strategy, outcome FROM agent_strategies WHERE outcome = ? ORDER BY created_at DESC",
		outcome,
	)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var strategies []Strategy
	for rows.Next() {
		var s Strategy
		if err := rows.Scan(&s.ID, &s.Situation, &s.Strategy, &s.Outcome); err != nil {
			continue
		}
		strategies = append(strategies, s)
	}
	return strategies, nil
}