database

package

v0.0.8 Latest Latest Go to latest Published: Jun 20, 2025 License: MIT Imports: 9 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type DB struct {
	*sqlx.DB
}

func Initialize(dbPath string) (*DB, error)

Initialize creates a new database connection and runs migrations

func (db *DB) CreateAgent(req models.CreateAgentRequest) (*models.Agent, error)

func (db *DB) CreateCommand(agentID string, req models.CreateCommandRequest) (*models.Command, error)

func (db *DB) CreateEvent(agentID string, req models.CreateEventRequest) (*models.Event, error)

func (db *DB) CreateTask(req models.CreateTaskRequest) (*models.Task, error)

func (db *DB) CreateTodo(agentID string, req models.CreateTodoRequest) (*models.TodoItem, error)

func (db *DB) DeleteAgent(id string) error

func (db *DB) DeleteTask(id string) error

func (db *DB) DeleteTodo(agentID, todoID string) error

func (db *DB) GetAgent(id string) (*models.Agent, error)

func (db *DB) GetCommand(agentID, commandID string) (*models.Command, error)

func (db *DB) GetFleetStatus() (*models.FleetStatus, error)

func (db *DB) GetRecentUpdates(limit int, since *time.Time) ([]models.Event, error)

func (db *DB) GetTask(id string) (*models.Task, error)

func (db *DB) GetTodo(agentID, todoID string) (*models.TodoItem, error)

func (db *DB) ListAgents(status string, limit, offset int) ([]models.Agent, int, error)

func (db *DB) ListCommands(agentID, status string, limit int) ([]models.Command, error)

func (db *DB) ListEvents(agentID, eventType string, since *time.Time, limit, offset int) ([]models.Event, int, error)

func (db *DB) ListTasks(status, assignedAgentID, priority string, limit, offset int) ([]models.Task, int, error)

func (db *DB) ListTodos(agentID string) ([]models.TodoItem, error)

func (db *DB) UpdateAgent(id string, req models.UpdateAgentRequest) (*models.Agent, error)

func (db *DB) UpdateCommand(agentID, commandID string, req models.UpdateCommandRequest) (*models.Command, error)

func (db *DB) UpdateTask(id string, req models.UpdateTaskRequest) (*models.Task, error)

func (db *DB) UpdateTodo(agentID, todoID string, req models.UpdateTodoRequest) (*models.TodoItem, error)