package cmds

import (
	"context"
	"fmt"
	"log"
	"os"
	"text/tabwriter"

	"github.com/spf13/cobra"
	"lmika.dev/lmika/weiro/config"
	"lmika.dev/lmika/weiro/models"
	"lmika.dev/lmika/weiro/services"
)

func PubTargetsAdd() *cobra.Command {
	var (
		siteGUID   string
		targetType string
		targetRef  string
		targetKey  string
		baseURL    string
		enabled    bool
	)

	cmd := &cobra.Command{
		Use:   "add",
		Short: "Add a publication target",
		Run: func(cmd *cobra.Command, args []string) {
			cfg, err := config.LoadConfig()
			if err != nil {
				log.Fatal(err)
			}

			svcs, err := services.New(cfg)
			if err != nil {
				log.Fatal(err)
			}
			defer svcs.Close()

			ctx := context.Background()
			site, err := svcs.DB.SelectSiteByGUID(ctx, siteGUID)
			if err != nil {
				log.Fatal(err)
			}

			target := &models.SitePublishTarget{
				SiteID:     site.ID,
				GUID:       models.NewNanoID(),
				Enabled:    enabled,
				BaseURL:    baseURL,
				TargetType: targetType,
				TargetRef:  targetRef,
				TargetKey:  targetKey,
			}

			if err := svcs.DB.SavePublishTarget(ctx, target); err != nil {
				log.Fatal(err)
			}

			fmt.Printf("Added publish target %s\n", target.GUID)
		},
	}

	cmd.Flags().StringVarP(&siteGUID, "site", "s", "", "Site GUID")
	cmd.Flags().StringVarP(&targetType, "type", "t", "", "Target type (localfs, netlify)")
	cmd.Flags().StringVarP(&targetRef, "ref", "r", "", "Target reference")
	cmd.Flags().StringVarP(&targetKey, "key", "k", "", "Target key")
	cmd.Flags().StringVarP(&baseURL, "url", "u", "", "Base URL")
	cmd.Flags().BoolVar(&enabled, "enabled", true, "Enable target")

	cmd.MarkFlagRequired("site")
	cmd.MarkFlagRequired("type")
	cmd.MarkFlagRequired("ref")
	cmd.MarkFlagRequired("url")

	return cmd
}

func PubTargets() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "pubtargets <site-guid>",
		Short: "Manage publication targets",
		Args:  cobra.ExactArgs(1),
		Run: func(cmd *cobra.Command, args []string) {
			cfg, err := config.LoadConfig()
			if err != nil {
				log.Fatal(err)
			}

			svcs, err := services.New(cfg)
			if err != nil {
				log.Fatal(err)
			}
			defer svcs.Close()

			ctx := context.Background()
			site, err := svcs.DB.SelectSiteByGUID(ctx, args[0])
			if err != nil {
				log.Fatal(err)
			}

			targets, err := svcs.DB.SelectPublishTargetsOfSite(ctx, site.ID)
			if err != nil {
				log.Fatal(err)
			}

			w := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0)
			fmt.Fprintln(w, "GUID\tTARGET_TYPE\tENABLED\tTARGET_REF")
			for _, target := range targets {
				fmt.Fprintf(w, "%s\t%s\t%v\t%s\n", target.GUID, target.TargetType, target.Enabled, target.TargetRef)
			}
			w.Flush()
		},
	}
	cmd.AddCommand(PubTargetsAdd())
	return cmd
}