multica/server/pkg/db/generated/agent_invocation_target.sql.go

// Code generated by sqlc. DO NOT EDIT.
// versions:
//   sqlc v1.31.1
// source: agent_invocation_target.sql

package db

import (
	"context"

	"github.com/jackc/pgx/v5/pgtype"
)

const createAgentInvocationTarget = `-- name: CreateAgentInvocationTarget :exec
INSERT INTO agent_invocation_target (agent_id, target_type, target_id, created_by)
VALUES ($1, $2, $3, $4)
ON CONFLICT (agent_id, target_type, target_id) DO UPDATE SET
    created_by = EXCLUDED.created_by,
    created_at = now()
`

type CreateAgentInvocationTargetParams struct {
	AgentID    pgtype.UUID `json:"agent_id"`
	TargetType string      `json:"target_type"`
	TargetID   pgtype.UUID `json:"target_id"`
	CreatedBy  pgtype.UUID `json:"created_by"`
}

// Idempotent upsert: re-adding an existing (agent, target_type, target)
// refreshes created_by/created_at rather than erroring. Callers replace the
// whole set via DeleteAgentInvocationTargets + a series of these, so the
// ON CONFLICT is belt-and-suspenders against races.
func (q *Queries) CreateAgentInvocationTarget(ctx context.Context, arg CreateAgentInvocationTargetParams) error {
	_, err := q.db.Exec(ctx, createAgentInvocationTarget,
		arg.AgentID,
		arg.TargetType,
		arg.TargetID,
		arg.CreatedBy,
	)
	return err
}

const deleteAgentInvocationTargets = `-- name: DeleteAgentInvocationTargets :exec
DELETE FROM agent_invocation_target
WHERE agent_id = $1
`

// Clears every target for an agent. Used before re-writing the allow-list so
// the update is a wholesale replace, matching the composio_toolkit_allowlist
// write model.
func (q *Queries) DeleteAgentInvocationTargets(ctx context.Context, agentID pgtype.UUID) error {
	_, err := q.db.Exec(ctx, deleteAgentInvocationTargets, agentID)
	return err
}

const deleteAgentInvocationTargetsByArchivedRuntimeAgents = `-- name: DeleteAgentInvocationTargetsByArchivedRuntimeAgents :exec
DELETE FROM agent_invocation_target
WHERE agent_id IN (
    SELECT id FROM agent WHERE runtime_id = $1 AND archived_at IS NOT NULL
)
`

// Application-layer replacement for the (deliberately absent) agent_id ON
// DELETE CASCADE: removes invocation targets for the archived agents a runtime
// delete is about to hard-delete. MUST run in the same tx as, and BEFORE,
// DeleteArchivedAgentsByRuntime so no orphan target rows survive the agent
// rows they belonged to. Mirrors the agent hard-delete predicate exactly.
func (q *Queries) DeleteAgentInvocationTargetsByArchivedRuntimeAgents(ctx context.Context, runtimeID pgtype.UUID) error {
	_, err := q.db.Exec(ctx, deleteAgentInvocationTargetsByArchivedRuntimeAgents, runtimeID)
	return err
}

const deleteAgentInvocationTargetsByMember = `-- name: DeleteAgentInvocationTargetsByMember :exec
DELETE FROM agent_invocation_target ait
USING agent a
WHERE ait.agent_id = a.id
  AND a.workspace_id = $1
  AND ait.target_type = 'member'
  AND ait.target_id = $2
`

type DeleteAgentInvocationTargetsByMemberParams struct {
	WorkspaceID pgtype.UUID `json:"workspace_id"`
	TargetID    pgtype.UUID `json:"target_id"`
}

// Removes member-target grants pointing at a leaving user, SCOPED to a single
// workspace. A user may belong to multiple workspaces; removing them from one
// must NOT touch their invocation grants on agents in another workspace. Joins
// through agent (agent_invocation_target has no workspace_id column and no FK)
// to bound the delete to @workspace_id.
func (q *Queries) DeleteAgentInvocationTargetsByMember(ctx context.Context, arg DeleteAgentInvocationTargetsByMemberParams) error {
	_, err := q.db.Exec(ctx, deleteAgentInvocationTargetsByMember, arg.WorkspaceID, arg.TargetID)
	return err
}

const listAgentInvocationTargets = `-- name: ListAgentInvocationTargets :many

SELECT id, agent_id, target_type, target_id, created_by, created_at FROM agent_invocation_target
WHERE agent_id = $1
ORDER BY target_type ASC, created_at ASC
`

// Agent invocation permission targets (MUL-3963). Rows are the allow-list for
// agents whose permission_mode = 'public_to'. See migration 130.
func (q *Queries) ListAgentInvocationTargets(ctx context.Context, agentID pgtype.UUID) ([]AgentInvocationTarget, error) {
	rows, err := q.db.Query(ctx, listAgentInvocationTargets, agentID)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	items := []AgentInvocationTarget{}
	for rows.Next() {
		var i AgentInvocationTarget
		if err := rows.Scan(
			&i.ID,
			&i.AgentID,
			&i.TargetType,
			&i.TargetID,
			&i.CreatedBy,
			&i.CreatedAt,
		); err != nil {
			return nil, err
		}
		items = append(items, i)
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}
	return items, nil
}

const listAgentInvocationTargetsByAgentIDs = `-- name: ListAgentInvocationTargetsByAgentIDs :many
SELECT id, agent_id, target_type, target_id, created_by, created_at FROM agent_invocation_target
WHERE agent_id = ANY($1::uuid[])
ORDER BY agent_id, target_type ASC, created_at ASC
`

// Batch load for the agent list endpoint so we don't N+1 per agent.
func (q *Queries) ListAgentInvocationTargetsByAgentIDs(ctx context.Context, agentIds []pgtype.UUID) ([]AgentInvocationTarget, error) {
	rows, err := q.db.Query(ctx, listAgentInvocationTargetsByAgentIDs, agentIds)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	items := []AgentInvocationTarget{}
	for rows.Next() {
		var i AgentInvocationTarget
		if err := rows.Scan(
			&i.ID,
			&i.AgentID,
			&i.TargetType,
			&i.TargetID,
			&i.CreatedBy,
			&i.CreatedAt,
		); err != nil {
			return nil, err
		}
		items = append(items, i)
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}
	return items, nil
}