package db

import (
	"time"
)

// CmEventRecord represents a persisted cumulative move state transition.
type CmEventRecord struct {
	ID        int64   `json:"id"`
	Coin      string  `json:"coin"`
	PrevState string  `json:"prev_state"`
	NewState  string  `json:"new_state"`
	Direction string  `json:"direction"`
	Score     float64 `json:"score"`
	AvgChange float64 `json:"avg_change"`
	ExAgree   int     `json:"ex_agree"`
	ExTotal   int     `json:"ex_total"`
	BG1m      float64 `json:"bg_1m"`
	HL1m      float64 `json:"hl_1m"`
	BN1m      float64 `json:"bn_1m"`
	OKX1m     float64 `json:"okx_1m"`
	BG5m      float64 `json:"bg_5m"`
	HL5m      float64 `json:"hl_5m"`
	BN5m      float64 `json:"bn_5m"`
	OKX5m     float64 `json:"okx_5m"`
	CreatedAt string  `json:"created_at"`
}

// InsertCmEvent saves a cumulative move event to the database.
func (d *DB) InsertCmEvent(coin, prevState, newState, direction string, score, avgChange float64, exAgree, exTotal int, bg1m, hl1m, bn1m, okx1m, bg5m, hl5m, bn5m, okx5m float64) error {
	_, err := d.Exec(`
		INSERT INTO cm_events (coin, prev_state, new_state, direction, score, avg_change, ex_agree, ex_total, bg_1m, hl_1m, bn_1m, okx_1m, bg_5m, hl_5m, bn_5m, okx_5m, created_at)
		VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
		coin, prevState, newState, direction, score, avgChange, exAgree, exTotal, bg1m, hl1m, bn1m, okx1m, bg5m, hl5m, bn5m, okx5m, Now().Format(time.RFC3339))
	return err
}

// GetCmEvents returns cumulative move events ordered by creation time descending.
func (d *DB) GetCmEvents(limit int) ([]CmEventRecord, error) {
	if limit <= 0 {
		limit = 100
	}
	rows, err := d.Query(`
		SELECT id, coin, prev_state, new_state, direction, score, avg_change, ex_agree, ex_total, bg_1m, hl_1m, bn_1m, okx_1m, bg_5m, hl_5m, bn_5m, okx_5m, created_at
		FROM cm_events
		ORDER BY created_at DESC
		LIMIT ?`, limit)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var result []CmEventRecord
	for rows.Next() {
		var r CmEventRecord
		if err := rows.Scan(&r.ID, &r.Coin, &r.PrevState, &r.NewState, &r.Direction,
			&r.Score, &r.AvgChange, &r.ExAgree, &r.ExTotal,
			&r.BG1m, &r.HL1m, &r.BN1m, &r.OKX1m,
			&r.BG5m, &r.HL5m, &r.BN5m, &r.OKX5m, &r.CreatedAt); err != nil {
			return nil, err
		}
		result = append(result, r)
	}
	return result, rows.Err()
}