package actions

import (
	"fmt"
	"net/http"
	"net/url"
	"os"
	"time"

	"github.com/pinchtab/pinchtab/internal/cli"
	"github.com/pinchtab/pinchtab/internal/cli/apiclient"
	"github.com/spf13/cobra"
)

func PDF(client *http.Client, base, token string, cmd *cobra.Command) {
	params := url.Values{}
	params.Set("raw", "true")

	outFile, _ := cmd.Flags().GetString("output")
	tabID, _ := cmd.Flags().GetString("tab")

	if v, _ := cmd.Flags().GetBool("landscape"); v {
		params.Set("landscape", "true")
	}
	if v, _ := cmd.Flags().GetString("scale"); v != "" {
		params.Set("scale", v)
	}
	if v, _ := cmd.Flags().GetString("paper-width"); v != "" {
		params.Set("paperWidth", v)
	}
	if v, _ := cmd.Flags().GetString("paper-height"); v != "" {
		params.Set("paperHeight", v)
	}
	if v, _ := cmd.Flags().GetString("margin-top"); v != "" {
		params.Set("marginTop", v)
	}
	if v, _ := cmd.Flags().GetString("margin-bottom"); v != "" {
		params.Set("marginBottom", v)
	}
	if v, _ := cmd.Flags().GetString("margin-left"); v != "" {
		params.Set("marginLeft", v)
	}
	if v, _ := cmd.Flags().GetString("margin-right"); v != "" {
		params.Set("marginRight", v)
	}
	if v, _ := cmd.Flags().GetString("page-ranges"); v != "" {
		params.Set("pageRanges", v)
	}
	if v, _ := cmd.Flags().GetBool("prefer-css-page-size"); v {
		params.Set("preferCSSPageSize", "true")
	}
	if v, _ := cmd.Flags().GetBool("display-header-footer"); v {
		params.Set("displayHeaderFooter", "true")
	}
	if v, _ := cmd.Flags().GetString("header-template"); v != "" {
		params.Set("headerTemplate", v)
	}
	if v, _ := cmd.Flags().GetString("footer-template"); v != "" {
		params.Set("footerTemplate", v)
	}
	if v, _ := cmd.Flags().GetBool("generate-tagged-pdf"); v {
		params.Set("generateTaggedPDF", "true")
	}
	if v, _ := cmd.Flags().GetBool("generate-document-outline"); v {
		params.Set("generateDocumentOutline", "true")
	}
	if v, _ := cmd.Flags().GetBool("file-output"); v {
		params.Del("raw")
		params.Set("output", "file")
	}
	if v, _ := cmd.Flags().GetString("path"); v != "" {
		params.Set("path", v)
	}

	if outFile == "" {
		outFile = fmt.Sprintf("page-%s.pdf", time.Now().Format("20060102-150405"))
	}

	var data []byte
	if tabID != "" {
		data = apiclient.DoGetRaw(client, base, token, fmt.Sprintf("/tabs/%s/pdf", tabID), params)
	} else {
		data = apiclient.DoGetRaw(client, base, token, "/pdf", params)
	}
	if data == nil {
		return
	}
	if err := os.WriteFile(outFile, data, 0600); err != nil {
		cli.Fatal("Write failed: %v", err)
	}
	fmt.Println(cli.StyleStdout(cli.SuccessStyle, fmt.Sprintf("Saved %s (%d bytes)", outFile, len(data))))
}