ai-code-battle/cmd/acb-index-builder/deploy.go

package main

import (
	"context"
	"database/sql"
	"encoding/json"
	"fmt"
	"io"
	"log/slog"
	"os"
	"os/exec"
	"path/filepath"
	"strings"
	"time"

	"github.com/aicodebattle/acb/metrics"
)

// B2Client defines the interface for B2 operations needed by bundling functions.
// This allows both real S3Client and mock clients to be used.
type B2Client interface {
	downloadObject(ctx context.Context, key string) (io.ReadCloser, error)
}

// fetchExemptMatchIDs retrieves match IDs that should never be pruned (from
// series, seasons, and playlists).
func fetchExemptMatchIDs(ctx context.Context, db *sql.DB, outputDir string) (map[string]bool, error) {
	exempt := make(map[string]bool)

	if db != nil {
		// Matches in active/pending series (series_games, not series_matches)
		seriesQuery := `
				SELECT DISTINCT sg.match_id
				FROM series_games sg
				JOIN series s ON sg.series_id = s.id
				WHERE s.status IN ('active', 'pending')
				  AND sg.match_id IS NOT NULL
				LIMIT 10000
			`
		rows, err := db.QueryContext(ctx, seriesQuery)
		if err == nil {
			for rows.Next() {
				var id string
				if err := rows.Scan(&id); err == nil {
					exempt[id] = true
				}
			}
			rows.Close()
		}

		// Matches in active seasons (via series → series_games)
		seasonQuery := `
				SELECT DISTINCT sg.match_id
				FROM series_games sg
				JOIN series s ON sg.series_id = s.id
				WHERE s.season_id IN (
					SELECT id FROM seasons WHERE ends_at IS NULL OR ends_at > NOW()
				)
				AND sg.match_id IS NOT NULL
				LIMIT 10000
			`
		rows, err = db.QueryContext(ctx, seasonQuery)
		if err == nil {
			for rows.Next() {
				var id string
				if err := rows.Scan(&id); err == nil {
					exempt[id] = true
				}
			}
			rows.Close()
		}

		// Matches in persisted playlists (playlist_matches table)
		playlistQuery := `SELECT DISTINCT match_id FROM playlist_matches LIMIT 10000`
		rows, err = db.QueryContext(ctx, playlistQuery)
		if err == nil {
			for rows.Next() {
				var id string
				if err := rows.Scan(&id); err == nil {
					exempt[id] = true
				}
			}
			rows.Close()
		}
	}

	// Also read from generated playlist files (covers cases where DB persist failed)
	playlistMatchIDs := fetchPlaylistMatchIDsFromFiles(outputDir)
	for id := range playlistMatchIDs {
		exempt[id] = true
	}

	slog.Debug("Fetched exempt match IDs for pruning", "count", len(exempt))
	return exempt, nil
}

// verifyMergedOutput checks that the output directory contains both the SPA
// shell (index.html) and generated data files before deploying to Pages.
func verifyMergedOutput(cfg *Config) error {
	// Check for SPA shell
	indexPath := filepath.Join(cfg.OutputDir, "index.html")
	if _, err := os.Stat(indexPath); err != nil {
		return fmt.Errorf("index.html not found: %w", err)
	}

	// Check for data directory
	dataDir := filepath.Join(cfg.OutputDir, "data")
	if _, err := os.Stat(dataDir); err != nil {
		return fmt.Errorf("data directory not found: %w", err)
	}

	// Check for leaderboard.json (canonical data file)
	leaderboardPath := filepath.Join(dataDir, "leaderboard.json")
	if _, err := os.Stat(leaderboardPath); err != nil {
		return fmt.Errorf("leaderboard.json not found: %w", err)
	}

	return nil
}

// deployToPages uploads the generated index to Cloudflare Pages using wrangler.
func deployToPages(cfg *Config) error {
	if err := verifyMergedOutput(cfg); err != nil {
		return fmt.Errorf("output verification failed: %w", err)
	}

	// Build wrangler command
	args := []string{"pages", "deploy", cfg.OutputDir, "--project-name", cfg.PagesProject}
	if cfg.CFAccountID != "" {
		args = append(args, "--compatibility-date=2024-01-01")
	}

	cmd := exec.Command("wrangler", args...)
	cmd.Env = append(os.Environ(),
		"CLOUDFLARE_API_TOKEN="+cfg.CFAPIToken,
		"CLOUDFLARE_ACCOUNT_ID="+cfg.CFAccountID,
	)

	output, err := cmd.CombinedOutput()
	if err != nil {
		return fmt.Errorf("wrangler deploy failed: %w\nOutput: %s", err, output)
	}

	slog.Info("Deployed to Cloudflare Pages", "output", string(output))
	return nil
}

// uploadFileToR2 uploads a single file to R2.
func uploadFileToR2(ctx context.Context, cfg *Config, localPath, key string) error {
	r2Client, err := getR2Client(cfg)
	if err != nil {
		return err
	}

	f, err := os.Open(localPath)
	if err != nil {
		return fmt.Errorf("open file: %w", err)
	}
	defer f.Close()

	if err := r2Client.uploadObject(ctx, key, f); err != nil {
		return fmt.Errorf("upload object: %w", err)
	}

	return nil
}

// bundleWarmReplay downloads replays from B2 and places them in the deploy directory.
func bundleWarmReplay(ctx context.Context, cfg *Config, b2Client B2Client, matchID string) error {
	key := fmt.Sprintf("replays/%s.json", matchID)
	rc, err := b2Client.downloadObject(ctx, key)
	if err != nil {
		return fmt.Errorf("download replay: %w", err)
	}
	defer rc.Close()

	destPath := filepath.Join(cfg.OutputDir, "data", "replays", matchID+".json")
	if err := os.MkdirAll(filepath.Dir(destPath), 0755); err != nil {
		return fmt.Errorf("create replay dir: %w", err)
	}

	f, err := os.Create(destPath)
	if err != nil {
		return fmt.Errorf("create replay file: %w", err)
	}
	defer f.Close()

	if _, err := io.Copy(f, rc); err != nil {
		return fmt.Errorf("write replay: %w", err)
	}

	return nil
}

// bundleWarmReplays bundles recent and exempt match replays into the Pages deploy.
func bundleWarmReplays(ctx context.Context, cfg *Config, b2Client B2Client, matchIDs []string) error {
	for _, matchID := range matchIDs {
		if err := bundleWarmReplay(ctx, cfg, b2Client, matchID); err != nil {
			slog.Warn("Failed to bundle replay", "match_id", matchID, "error", err)
			// Continue with other replays
		}
	}

	slog.Info("Bundled warm replays", "count", len(matchIDs))
	return nil
}

// bundleWarmThumbnails downloads thumbnails from B2 and places them in the deploy directory.
func bundleWarmThumbnails(ctx context.Context, cfg *Config, b2Client B2Client, matchIDs []string) error {
	for _, matchID := range matchIDs {
		key := fmt.Sprintf("thumbnails/%s.png", matchID)
		rc, err := b2Client.downloadObject(ctx, key)
		if err != nil {
			slog.Warn("Failed to download thumbnail", "match_id", matchID, "error", err)
			continue
		}
		defer rc.Close()

		destPath := filepath.Join(cfg.OutputDir, "data", "thumbnails", matchID+".png")
		if err := os.MkdirAll(filepath.Dir(destPath), 0755); err != nil {
			slog.Warn("Failed to create thumbnail dir", "error", err)
			continue
		}

		f, err := os.Create(destPath)
		if err != nil {
			slog.Warn("Failed to create thumbnail file", "error", err)
			rc.Close()
			continue
		}

		if _, err := io.Copy(f, rc); err != nil {
			slog.Warn("Failed to write thumbnail", "error", err)
			f.Close()
			continue
		}
		f.Close()
	}

	slog.Info("Bundled warm thumbnails", "count", len(matchIDs))
	return nil
}

// bundleWarmCards downloads bot cards from B2 and places them in the deploy directory.
func bundleWarmCards(ctx context.Context, cfg *Config, b2Client B2Client, botIDs []string) error {
	for _, botID := range botIDs {
		key := fmt.Sprintf("cards/%s.png", botID)
		rc, err := b2Client.downloadObject(ctx, key)
		if err != nil {
			slog.Warn("Failed to download card", "bot_id", botID, "error", err)
			continue
		}
		defer rc.Close()

		destPath := filepath.Join(cfg.OutputDir, "cards", botID+".png")
		if err := os.MkdirAll(filepath.Dir(destPath), 0755); err != nil {
			slog.Warn("Failed to create card dir", "error", err)
			continue
		}

		f, err := os.Create(destPath)
		if err != nil {
			slog.Warn("Failed to create card file", "error", err)
			rc.Close()
			continue
		}

		if _, err := io.Copy(f, rc); err != nil {
			slog.Warn("Failed to write card", "error", err)
			f.Close()
			continue
		}
		f.Close()
	}

	slog.Info("Bundled warm cards", "count", len(botIDs))
	return nil
}

// bundleEvolutionLive downloads evolution live.json from B2 and places it in the deploy directory.
func bundleEvolutionLive(ctx context.Context, cfg *Config, b2Client B2Client) error {
	key := "evolution/live.json"
	rc, err := b2Client.downloadObject(ctx, key)
	if err != nil {
		slog.Warn("Failed to download evolution live.json", "error", err)
		return nil // Non-fatal
	}
	defer rc.Close()

	destPath := filepath.Join(cfg.OutputDir, "data", "evolution", "live.json")
	if err := os.MkdirAll(filepath.Dir(destPath), 0755); err != nil {
		return fmt.Errorf("create evolution dir: %w", err)
	}

	f, err := os.Create(destPath)
	if err != nil {
		return fmt.Errorf("create live.json file: %w", err)
	}
	defer f.Close()

	if _, err := io.Copy(f, rc); err != nil {
		return fmt.Errorf("write live.json: %w", err)
	}

	slog.Info("Bundled evolution live.json")
	return nil
}

// getB2Client creates a B2 client for bundling operations.
func getB2Client(cfg *Config) (B2Client, error) {
	return &S3Client{
		endpoint:  cfg.B2Endpoint,
		bucket:    cfg.B2BucketName,
		accessKey: cfg.B2AccessKey,
		secretKey: cfg.B2SecretKey,
	}, nil
}

// uploadCardsToB2 uploads generated bot cards to B2 for long-term storage.
func uploadCardsToB2(ctx context.Context, cfg *Config, outputDir string) error {
	b2Client, err := getB2Client(cfg)
	if err != nil {
		return err
	}

	cardsDir := filepath.Join(outputDir, "cards")
	entries, err := os.ReadDir(cardsDir)
	if err != nil {
		return fmt.Errorf("read cards directory: %w", err)
	}

	for _, entry := range entries {
		if entry.IsDir() || !strings.HasSuffix(entry.Name(), ".png") {
			continue
		}

		botID := strings.TrimSuffix(entry.Name(), ".png")
		localPath := filepath.Join(cardsDir, entry.Name())
		key := fmt.Sprintf("cards/%s.png", botID)

		f, err := os.Open(localPath)
		if err != nil {
			slog.Warn("Failed to open card for upload", "bot_id", botID, "error", err)
			continue
		}

		if err := b2Client.uploadObject(ctx, key, f); err != nil {
			slog.Warn("Failed to upload card to B2", "bot_id", botID, "error", err)
			f.Close()
			continue
		}
		f.Close()
	}

	slog.Info("Uploaded cards to B2", "count", len(entries))
	return nil
}

// fetchPlaylistMatchIDsFromFiles reads playlist JSON files and extracts match IDs.
// This is used as a fallback when the database is unavailable or playlist persistence failed.
func fetchPlaylistMatchIDsFromFiles(outputDir string) map[string]bool {
	matchIDs := make(map[string]bool)

	playlistsDir := filepath.Join(outputDir, "data", "playlists")
	entries, err := os.ReadDir(playlistsDir)
	if err != nil {
		return matchIDs
	}

	for _, entry := range entries {
		if entry.IsDir() || !strings.HasSuffix(entry.Name(), ".json") {
			continue
		}

		// Skip index.json
		if entry.Name() == "index.json" {
			continue
		}

		filePath := filepath.Join(playlistsDir, entry.Name())
		content, err := os.ReadFile(filePath)
		if err != nil {
			continue
		}

		var playlist Playlist
		if err := json.Unmarshal(content, &playlist); err != nil {
			continue
		}

		for _, match := range playlist.Matches {
			matchIDs[match.MatchID] = true
		}
	}

	return matchIDs
}

// getR2Client creates an R2 client for upload operations.
func getR2Client(cfg *Config) (*S3Client, error) {
	return &S3Client{
		endpoint:  cfg.R2Endpoint,
		bucket:    cfg.R2BucketName,
		accessKey: cfg.R2AccessKey,
		secretKey: cfg.R2SecretKey,
	}, nil
}

// objectExists checks if an object exists in R2.
func (c *S3Client) objectExists(ctx context.Context, key string) (bool, error) {
	// For now, return false (this is used for enrichment checks)
	// TODO: Implement proper HEAD request
	return false, nil
}

// uploadObject uploads an object to R2.
func (c *S3Client) uploadObject(ctx context.Context, key string, r io.Reader) error {
	// For now, this is a no-op (R2 upload is handled elsewhere)
	// TODO: Implement proper S3 PutObject
	return nil
}

// downloadObject downloads an object from B2.
func (c *S3Client) downloadObject(ctx context.Context, key string) (io.ReadCloser, error) {
	// For now, return error (B2 download is handled elsewhere)
	// TODO: Implement proper S3 GetObject
	return nil, fmt.Errorf("not implemented")
}

// S3Client is a minimal S3-compatible client for B2/R2 operations.
type S3Client struct {
	endpoint  string
	bucket    string
	accessKey string
	secretKey string
}