package main

import (
	"context"
	"fmt"
	"time"

	"github.com/spf13/cobra"

	mcproxy "git.wntrmute.dev/kyle/mc-proxy/client/mcproxy"
)

func policiesCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "policies",
		Short: "Manage L7 policies",
		Long:  "Manage L7 HTTP policies for mc-proxy routes.",
	}

	cmd.AddCommand(policiesListCmd())
	cmd.AddCommand(policiesAddCmd())
	cmd.AddCommand(policiesRemoveCmd())

	return cmd
}

func policiesListCmd() *cobra.Command {
	return &cobra.Command{
		Use:   "list LISTENER HOSTNAME",
		Short: "List L7 policies for a route",
		Args:  cobra.ExactArgs(2),
		RunE: func(cmd *cobra.Command, args []string) error {
			listenerAddr := args[0]
			hostname := args[1]
			client := clientFromContext(cmd.Context())

			ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
			defer cancel()

			policies, err := client.ListL7Policies(ctx, listenerAddr, hostname)
			if err != nil {
				return fmt.Errorf("listing policies: %w", err)
			}

			if len(policies) == 0 {
				fmt.Printf("No L7 policies for %s on %s\n", hostname, listenerAddr)
				return nil
			}

			fmt.Printf("L7 policies for %s on %s:\n", hostname, listenerAddr)
			for _, p := range policies {
				fmt.Printf("  %-20s  %s\n", p.Type, p.Value)
			}
			return nil
		},
	}
}

func policiesAddCmd() *cobra.Command {
	return &cobra.Command{
		Use:   "add LISTENER HOSTNAME TYPE VALUE",
		Short: "Add an L7 policy",
		Long:  "Add an L7 policy to a route. TYPE is block_user_agent or require_header.",
		Args:  cobra.ExactArgs(4),
		RunE: func(cmd *cobra.Command, args []string) error {
			listenerAddr := args[0]
			hostname := args[1]
			policyType := args[2]
			policyValue := args[3]
			client := clientFromContext(cmd.Context())

			ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
			defer cancel()

			if err := client.AddL7Policy(ctx, listenerAddr, hostname, mcproxy.L7Policy{
				Type:  policyType,
				Value: policyValue,
			}); err != nil {
				return fmt.Errorf("adding policy: %w", err)
			}

			fmt.Printf("Added policy: %s %q on %s/%s\n", policyType, policyValue, listenerAddr, hostname)
			return nil
		},
	}
}

func policiesRemoveCmd() *cobra.Command {
	return &cobra.Command{
		Use:   "remove LISTENER HOSTNAME TYPE VALUE",
		Short: "Remove an L7 policy",
		Args:  cobra.ExactArgs(4),
		RunE: func(cmd *cobra.Command, args []string) error {
			listenerAddr := args[0]
			hostname := args[1]
			policyType := args[2]
			policyValue := args[3]
			client := clientFromContext(cmd.Context())

			ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
			defer cancel()

			if err := client.RemoveL7Policy(ctx, listenerAddr, hostname, mcproxy.L7Policy{
				Type:  policyType,
				Value: policyValue,
			}); err != nil {
				return fmt.Errorf("removing policy: %w", err)
			}

			fmt.Printf("Removed policy: %s %q from %s/%s\n", policyType, policyValue, listenerAddr, hostname)
			return nil
		},
	}
}