package store

import (
	"context"
	"database/sql"
	"time"
)

// ControlPlane is one control plane agents may dial.
type ControlPlane struct {
	// URL is what an agent dials: https://10.0.0.150:7443.
	URL     string
	Name    string
	AddedAt time.Time
}

// ControlPlanes lists them, oldest first: the first is the one agents try
// first.
func (s *DB) ControlPlanes(ctx context.Context) ([]ControlPlane, error) {
	return queryAll(ctx, s.db, `SELECT url, name, added_at FROM control_planes ORDER BY added_at, url`, nil,
		func(rows *sql.Rows) (ControlPlane, error) {
			var c ControlPlane
			var at int64
			err := rows.Scan(&c.URL, &c.Name, &at)
			c.AddedAt = fromMillis(at)
			return c, err
		})
}

// PutControlPlane records one, keeping when it was first added.
func (s *DB) PutControlPlane(ctx context.Context, c ControlPlane) error {
	if c.AddedAt.IsZero() {
		c.AddedAt = time.Now()
	}
	_, err := s.db.ExecContext(ctx, `INSERT INTO control_planes (url, name, added_at) VALUES (?, ?, ?)
		ON CONFLICT (url) DO UPDATE SET name = excluded.name`, c.URL, c.Name, millis(c.AddedAt))
	return err
}

// RemoveControlPlane forgets one.
func (s *DB) RemoveControlPlane(ctx context.Context, url string) error {
	_, err := s.db.ExecContext(ctx, `DELETE FROM control_planes WHERE url = ?`, url)
	return err
}