package cli

import (
	"encoding/json"
	"fmt"
	"os"

	"github.com/spf13/cobra"
)

// campaignListEntry is the JSON shape for one row of `campaign list --json`.
// Kept minimal and separate from any internal campaign type so this command
// can grow real data (fetched from the API/DB, per the TODO below) without
// dragging engine internals into the CLI's JSON contract.
type campaignListEntry struct {
	ID       string `json:"id"`
	Status   string `json:"status"`
	Target   string `json:"target"`
	Findings int    `json:"findings"`
}

var campaignCmd = &cobra.Command{
	Use:   "campaign",
	Short: "Manage penetration testing campaigns",
}

var campaignListCmd = &cobra.Command{
	Use:     "list",
	Short:   "List all campaigns",
	Example: "  pentestswarm campaign list\n  pentestswarm campaign list --json",
	RunE: func(cmd *cobra.Command, args []string) error {
		// TODO: fetch from API/DB
		entries := []campaignListEntry{}

		if OutputIsJSON() {
			enc := json.NewEncoder(os.Stdout)
			enc.SetIndent("", "  ")
			return enc.Encode(entries)
		}

		fmt.Println(colorBold("ID                                   STATUS       TARGET              FINDINGS"))
		fmt.Println(colorDim("───────────────────────────────────────────────────────────────────────────────"))
		if len(entries) == 0 {
			fmt.Println(colorDim("  (no campaigns yet — run: pentestswarm scan <target> --scope <scope>)"))
		}
		return nil
	},
}

var campaignStatusCmd = &cobra.Command{
	Use:     "status <id>",
	Short:   "Show detailed status of a campaign",
	Args:    cobra.ExactArgs(1),
	Example: "  pentestswarm campaign status abc-123",
	RunE: func(cmd *cobra.Command, args []string) error {
		id := args[0]
		fmt.Printf("%s Campaign: %s\n", colorBold("*"), id)
		fmt.Printf("  Status:  %s\n", colorYellow("unknown"))
		fmt.Printf("  Target:  %s\n", colorDim("fetch from API"))
		// TODO: fetch from API
		return nil
	},
}

var campaignStopCmd = &cobra.Command{
	Use:     "stop <id>",
	Short:   "Emergency stop a running campaign",
	Args:    cobra.ExactArgs(1),
	Example: "  pentestswarm campaign stop abc-123",
	RunE: func(cmd *cobra.Command, args []string) error {
		fmt.Printf("%s Stopping campaign %s...\n", colorRed("*"), args[0])
		// TODO: call API
		fmt.Printf("%s Campaign stopped. Cleanup actions executed.\n", colorGreen("*"))
		return nil
	},
}

func init() {
	campaignCmd.AddCommand(campaignListCmd)
	campaignCmd.AddCommand(campaignStatusCmd)
	campaignCmd.AddCommand(campaignStopCmd)

	rootCmd.AddCommand(campaignCmd)
}