goclaw/internal/bootstrap/backfill_capabilities.go

package bootstrap

import (
	"context"
	"database/sql"
	"log/slog"
)

// BackfillCapabilities seeds CAPABILITIES.md template for all agents that don't have it.
// Runs once at startup, idempotent. Uses a single INSERT ... WHERE NOT EXISTS query
// so it's O(1) regardless of agent count. Returns number of agents backfilled.
func BackfillCapabilities(ctx context.Context, db *sql.DB) (int64, error) {
	if db == nil {
		return 0, nil
	}

	tpl, err := templateFS.ReadFile(templatePath(CapabilitiesFile))
	if err != nil {
		return 0, err
	}

	// Single query: insert CAPABILITIES.md for all agents missing it.
	// Pulls tenant_id from the agents table to maintain tenant isolation.
	// file_name is a constant, only content is parameterized to avoid PG type inference issues.
	res, err := db.ExecContext(ctx, `
		INSERT INTO agent_context_files (id, agent_id, file_name, content, created_at, updated_at, tenant_id)
		SELECT uuid_generate_v7(), a.id, 'CAPABILITIES.md', $1, NOW(), NOW(), a.tenant_id
		FROM agents a
		WHERE NOT EXISTS (
			SELECT 1 FROM agent_context_files acf
			WHERE acf.agent_id = a.id AND acf.file_name = 'CAPABILITIES.md'
		)`,
		string(tpl),
	)
	if err != nil {
		return 0, err
	}

	count, _ := res.RowsAffected()
	if count > 0 {
		slog.Info("bootstrap: backfilled CAPABILITIES.md", "agents", count)
	}
	return count, nil
}