main.go

package main

import (
	"flag"
	"fmt"
	"math"
	"os"
	"path/filepath"

	tea "github.com/charmbracelet/bubbletea"
)

func main() {
	// CLI flags
	ralphDir := flag.String("ralph-dir", "", "Path to ralph directory containing prd.json and prompt.md (default: ./scripts/ralph)")
	prdFile := flag.String("prd", "", "Path to prd.json (overrides --ralph-dir)")
	promptFile := flag.String("prompt", "", "Path to prompt.md (overrides --ralph-dir)")
	flag.Parse()

	// Get current working directory (project root)
	cwd, err := os.Getwd()
	if err != nil {
		fmt.Fprintf(os.Stderr, "Error getting current directory: %v\n", err)
		os.Exit(1)
	}

	// Determine paths
	var prdPath, promptPath string

	if *prdFile != "" && *promptFile != "" {
		// Explicit file paths provided
		prdPath = *prdFile
		promptPath = *promptFile
		if !filepath.IsAbs(prdPath) {
			prdPath = filepath.Join(cwd, prdPath)
		}
		if !filepath.IsAbs(promptPath) {
			promptPath = filepath.Join(cwd, promptPath)
		}
	} else if *ralphDir != "" {
		// Custom ralph directory
		dir := *ralphDir
		if !filepath.IsAbs(dir) {
			dir = filepath.Join(cwd, dir)
		}
		prdPath = filepath.Join(dir, "prd.json")
		promptPath = filepath.Join(dir, "prompt.md")
	} else {
		// Default: scripts/ralph in current directory
		prdPath = filepath.Join(cwd, "scripts", "ralph", "prd.json")
		promptPath = filepath.Join(cwd, "scripts", "ralph", "prompt.md")
	}

	projectRoot := cwd

	// Validate prd.json exists
	if _, err := os.Stat(prdPath); os.IsNotExist(err) {
		fmt.Fprintf(os.Stderr, "Error: prd.json not found at %s\n", prdPath)
		fmt.Fprintf(os.Stderr, "\nUsage:\n")
		fmt.Fprintf(os.Stderr, "  ralphy                           # Uses ./scripts/ralph/\n")
		fmt.Fprintf(os.Stderr, "  ralphy --ralph-dir ./custom/path # Custom ralph directory\n")
		fmt.Fprintf(os.Stderr, "  ralphy --prd ./prd.json --prompt ./prompt.md\n")
		os.Exit(1)
	}

	// Validate prompt.md exists
	if _, err := os.Stat(promptPath); os.IsNotExist(err) {
		fmt.Fprintf(os.Stderr, "Error: prompt.md not found at %s\n", promptPath)
		os.Exit(1)
	}

	// Load PRD
	prd, err := LoadPRD(prdPath)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Error loading prd.json: %v\n", err)
		os.Exit(1)
	}

	// Calculate max iterations
	pendingCount := CountPending(prd.UserStories)
	maxIterations := int(math.Ceil(float64(pendingCount) * 1.3))
	if maxIterations < 1 {
		maxIterations = 1
	}

	// Create model and run
	model := NewModel(prdPath, promptPath, projectRoot, maxIterations)

	p := tea.NewProgram(
		model,
		tea.WithAltScreen(),
		tea.WithMouseCellMotion(),
	)

	if _, err := p.Run(); err != nil {
		fmt.Fprintf(os.Stderr, "Error running program: %v\n", err)
		os.Exit(1)
	}
}