Containr/app/backend/sqlc/queries/agents.sql

-- name: GetAgentByHostAndIP :one
SELECT * FROM node_agents
WHERE hostname = $1 AND ip_address = $2
LIMIT 1;

-- name: GetAgent :one
SELECT * FROM node_agents WHERE id = $1;

-- name: ListAgents :many
SELECT * FROM node_agents ORDER BY created_at ASC;

-- name: CreateAgent :one
INSERT INTO node_agents (
    id, name, hostname, ip_address, port, status, version,
    capabilities, resources, last_heartbeat, metadata
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11)
RETURNING *;

-- name: UpdateAgent :one
UPDATE node_agents SET
    name = $2,
    hostname = $3,
    ip_address = $4,
    port = $5,
    status = $6,
    version = $7,
    capabilities = $8,
    resources = $9,
    last_heartbeat = $10,
    metadata = $11,
    auto_prune = $12,
    schedulable = $13,
    updated_at = NOW()
WHERE id = $1
RETURNING *;

-- name: UpdateAgentHeartbeat :exec
UPDATE node_agents SET
    status = $2,
    resources = $3,
    last_heartbeat = $4,
    updated_at = NOW()
WHERE id = $1;

-- name: DeleteAgent :exec
DELETE FROM node_agents WHERE id = $1;

-- name: InsertAgentHeartbeat :exec
INSERT INTO agent_heartbeats (
    id, node_agent_id, timestamp, status, resources,
    container_count, system_load, uptime, version
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9);

-- name: ListAgentHeartbeatsSince :many
SELECT * FROM agent_heartbeats
WHERE node_agent_id = $1 AND timestamp >= $2
ORDER BY timestamp ASC;

-- name: ListPendingCommands :many
SELECT * FROM agent_commands
WHERE node_agent_id = $1 AND status = 'pending'
ORDER BY created_at ASC
LIMIT 25;

-- name: ListCommandsForAgent :many
SELECT * FROM agent_commands
WHERE node_agent_id = $1
ORDER BY created_at DESC;

-- name: GetCommandForAgent :one
SELECT * FROM agent_commands
WHERE id = $1 AND node_agent_id = $2;

-- name: CreateCommand :one
INSERT INTO agent_commands (
    id, type, node_agent_id, container_id, payload, status
) VALUES ($1, $2, $3, $4, $5, 'pending')
RETURNING *;

-- name: CompleteCommand :one
UPDATE agent_commands SET
    status = $3,
    result = $4,
    error = $5,
    completed_at = NOW(),
    updated_at = NOW()
WHERE id = $1 AND node_agent_id = $2
RETURNING *;

-- name: ListContainersForAgent :many
SELECT * FROM container_instances
WHERE node_agent_id = $1;

-- name: GetContainerForAgent :one
SELECT * FROM container_instances
WHERE id = $1 AND node_agent_id = $2;

-- name: GetContainer :one
SELECT * FROM container_instances WHERE id = $1;

-- name: CreateContainer :one
INSERT INTO container_instances (
    id, name, image, project_id, service_id, node_agent_id,
    status, resources, ports, environment, volumes, networks,
    restart_policy, health_check
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14)
RETURNING *;

-- name: UpdateContainerStatus :exec
UPDATE container_instances SET status = $2, updated_at = NOW()
WHERE id = $1;

-- name: GetLastAgentCommandByType :one
SELECT * FROM agent_commands
WHERE node_agent_id = $1 AND type = $2
ORDER BY created_at DESC LIMIT 1;

-- name: PickLeastLoadedAgent :one
SELECT a.id FROM node_agents a
LEFT JOIN LATERAL (
    SELECT h.container_count FROM agent_heartbeats h
    WHERE h.node_agent_id = a.id
    ORDER BY h.timestamp DESC LIMIT 1
) h ON true
WHERE a.status = 'online' AND a.schedulable
ORDER BY COALESCE(h.container_count, 0) ASC, a.created_at ASC
LIMIT 1;

-- name: SetAgentSchedulable :exec
UPDATE node_agents SET schedulable = $2, updated_at = NOW() WHERE id = $1;

-- name: ClearServiceNodePins :execrows
UPDATE services SET node_id = NULL, updated_at = NOW() WHERE node_id = $1;

-- name: UpsertServiceContainer :exec
INSERT INTO container_instances (
    id, name, image, project_id, service_id, node_agent_id,
    status, resources, ports, environment, volumes, networks,
    restart_policy, health_check
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14)
ON CONFLICT (id) DO UPDATE SET
    image = EXCLUDED.image,
    status = EXCLUDED.status,
    updated_at = NOW();

-- name: DeleteServiceContainersOnAgent :exec
DELETE FROM container_instances WHERE node_agent_id = $1 AND service_id = $2;

-- name: ScrubCommandPayload :exec
UPDATE agent_commands SET payload = $2, updated_at = NOW() WHERE id = $1;