package cli

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"

	"github.com/spf13/cobra"
)

func newCreateCmd() *cobra.Command {
	var (
		serverURL    string
		instructions string
		workingDir   string
		model        string
		agentType    string
		parentID     string
		budget       float64
		timeout      string
		priority     string
		autoStart    bool
	)

	cmd := &cobra.Command{
		Use:   "create <name>",
		Short: "Create a task via the running server",
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			return createTask(serverURL, args[0], instructions, workingDir, model, agentType, parentID, budget, timeout, priority, autoStart)
		},
	}

	cmd.Flags().StringVar(&serverURL, "server", defaultServerURL, "claudomator server URL")
	cmd.Flags().StringVarP(&instructions, "instructions", "i", "", "task instructions (required)")
	cmd.Flags().StringVarP(&workingDir, "working-dir", "d", "", "working directory for the task")
	cmd.Flags().StringVarP(&model, "model", "m", "", "model to use")
	cmd.Flags().StringVar(&agentType, "agent-type", "claude", "agent type: claude, gemini")
	cmd.Flags().StringVar(&parentID, "parent-id", "", "parent task ID (makes this a subtask)")
	cmd.Flags().Float64Var(&budget, "budget", 1.0, "max budget in USD")
	cmd.Flags().StringVar(&timeout, "timeout", "15m", "task timeout")
	cmd.Flags().StringVar(&priority, "priority", "normal", "priority: high, normal, low")
	cmd.Flags().BoolVar(&autoStart, "start", false, "queue for execution immediately after creating")
	_ = cmd.MarkFlagRequired("instructions")

	return cmd
}

func createTask(serverURL, name, instructions, workingDir, model, agentType, parentID string, budget float64, timeout, priority string, autoStart bool) error {
	body := map[string]interface{}{
		"name":     name,
		"timeout":  timeout,
		"priority": priority,
		"agent": map[string]interface{}{
			"type":           agentType,
			"instructions":   instructions,
			"project_dir":    workingDir,
			"model":          model,
			"max_budget_usd": budget,
		},
	}
	if parentID != "" {
		body["parent_task_id"] = parentID
	}

	data, _ := json.Marshal(body)
	resp, err := httpClient.Post(serverURL+"/api/tasks", "application/json", bytes.NewReader(data)) //nolint:noctx
	if err != nil {
		return fmt.Errorf("POST /api/tasks: %w", err)
	}
	defer resp.Body.Close()

	raw, _ := io.ReadAll(resp.Body)
	var result map[string]interface{}
	if err := json.Unmarshal(raw, &result); err != nil {
		return fmt.Errorf("server returned invalid JSON (status %d): %s", resp.StatusCode, string(raw))
	}

	if resp.StatusCode >= 300 {
		return fmt.Errorf("server returned %d: %v", resp.StatusCode, result["error"])
	}

	id, _ := result["id"].(string)
	if id == "" {
		return fmt.Errorf("server returned task without id field")
	}
	fmt.Printf("Created task %s\n", id)

	if autoStart {
		return startTask(serverURL, id, agentType)
	}
	return nil
}