package main

import (
	"flag"
	"fmt"
	"log"
	"os"
	"path/filepath"
	"runtime" // for detecting system architecture and platform
	"spitfire-builder/spitfire"
	"time"
	//"errors"
)

var (
	// Define all flags as package-level variables
	all             bool
	buildFlag       bool
	clean           bool
	fullClean       bool
	update          bool
	prePatch        bool
	postPatch       bool
	skipPatchUpdate bool
	run             bool
	compress        bool
	target          string
	version         string
	component       string
	arch            string
	release         string
	platform        string
	upload          bool
	uploadPath      string
	sourceRepo      = "https://hg.mozilla.org/releases/mozilla-release"
	patchesRepo     = "https://weforge.xyz/Spitfire/Patcher.git"
	url             = "https://spitfirebrowser.xyz/"
	licence         = "AGPL-3.0"
	name            = "Spitfire"
	packageName     = "spitfire-browser"
	maintainer      = "Internet Addict"
	initialDir      string
	skipDeps        bool
)

func init() {
	flag.StringVar(&target, "t", "", "Target location format: component-arch-release-platform")
	flag.BoolVar(&compress, "c", false, "Compress the build directory into a tar.gz file before uploading")
	flag.StringVar(&version, "v", "", "Specify version for the package. For nightly, use current date if not specified.")
	flag.StringVar(&component, "component", "browser", "Component name (default: browser)")
	flag.StringVar(&arch, "arch", runtime.GOARCH, "Architecture (default: system architecture)")
	flag.StringVar(&release, "release", "nightly", "Release type (default: nightly)")
	flag.StringVar(&platform, "platform", runtime.GOOS, "Platform (default: system platform)")
	flag.BoolVar(&all, "a", false, "Perform all steps (build, clean, update)")
	flag.BoolVar(&buildFlag, "b", false, "Build Spitfire")
	flag.BoolVar(&clean, "clean", false, "Revert uncommitted changes without removing build artifacts")
	flag.BoolVar(&fullClean, "full-clean", false, "Perform a full clean by removing all build artifacts (clobber)")
	flag.BoolVar(&update, "u", false, "Update Mozilla repository")
	flag.BoolVar(&prePatch, "pre-patch", false, "Apply pre-build patches")
	flag.BoolVar(&postPatch, "post-patch", false, "Apply post-build patches")
	flag.BoolVar(&skipPatchUpdate, "skip-patch-update", false, "Skip updating the patches repository")
	flag.BoolVar(&run, "r", false, "Run the project after build")
	flag.BoolVar(&upload, "upload", false, "Upload the compressed build file to SourceForge")
	flag.StringVar(&uploadPath, "upload-path", "", "Path to the file to upload if no build present")
	flag.BoolVar(&skipDeps, "skip-deps", false, "Skip checking for required system dependencies")
	flag.Bool("h", false, "Display help message")
}

func printHelp() {
	fmt.Println("Usage: go run . -p=<path-to-build> -t=<target> [-c|--compress] [-v|--version=<version>] [-component=<component>] [-arch=<architecture>] [-release=<release>] [-platform=<platform>]")
	flag.PrintDefaults()
	fmt.Println("Example: go run . --upload -c -a")
	os.Exit(0)
}

func main() {
	flag.Parse()

	if flag.Lookup("h").Value.(flag.Getter).Get().(bool) {
		printHelp()
	}

	if !skipDeps {
		if err := spitfire.CheckDependencies(); err != nil {
			log.Fatalf("System check failed: %v", err)
		}
	}

	if version == "" && release == "nightly" {
		version = time.Now().Format("2006.01.02")
	}

	// Save the initial directory
	var err error
	initialDir, err = os.Getwd()
	if err != nil {
		log.Fatalf("Failed to get current working directory: %v", err)
	}

	var buildDir string // Store the resolved build directory here

	// Perform build if necessary
	if all || buildFlag || prePatch || postPatch || clean || update || run {
		buildDir = BuildProcess()
		if buildDir == "" {
			log.Fatalf("Build process completed, but no build directory was found.")
		}
		fmt.Printf("Build directory from process: %s\n", buildDir)
	}

	// Resolve build directory for compression/upload
	if compress || upload {
		if buildDir == "" { // Resolve dynamically if no build was performed
			fmt.Println("No build directory detected during build process. Resolving dynamically...")
			sourcePath, err := spitfire.ResolvePath("./mozilla-release")
			if err != nil {
				log.Fatalf("Error resolving source path: %v", err)
			}
			fmt.Printf("Resolved source path: %s\n", sourcePath)

			buildDir, err = spitfire.ResolveBuildDir(sourcePath)
			if err != nil {
				log.Fatalf("Error resolving build directory dynamically: %v", err)
			}
		}

		if buildDir == "" {
			log.Fatalf("No build directory found for compression or upload.")
		}
		fmt.Printf("Resolved build directory for compress/upload: %s\n", buildDir)

		PackageAndUploadProcess(buildDir)
	}

	spitfire.PrintErrors()
}

// Update the BuildProcess function to pass patchesRepo
func BuildProcess() string {
	sourcePath, err := spitfire.ResolvePath("./mozilla-release")
	if err != nil {
		log.Fatalf("Error resolving source path: %v", err)
	}

	var buildDir string

	if all {
		spitfire.DownloadSource(sourcePath, sourceRepo)
		spitfire.DiscardChanges(sourcePath)
		spitfire.CleanBuild(sourcePath, fullClean)
		spitfire.UpdateRepo(sourcePath)
		if err := spitfire.ApplyPatches(sourcePath, patchesRepo, "pre-compile-patches", filepath.Join(sourcePath, "patcher"), skipPatchUpdate); err != nil {
			log.Fatalf("Error during patch application: %v", err)
		}
		spitfire.Configure(sourcePath)
		spitfire.Build(sourcePath)
		// Detect the build directory dynamically
		buildDir, err = spitfire.ResolveBuildDir(sourcePath)
		if err != nil {
			log.Fatalf("Error resolving build directory: %v", err)
		}
		if err := spitfire.ApplyPatches(buildDir, patchesRepo, "post-compile-patches", filepath.Join(sourcePath, "patcher"), skipPatchUpdate); err != nil {
			log.Fatalf("Error during patch application: %v", err)
		}
		if run {
			spitfire.RunProject(sourcePath)
		}
		fmt.Println("Spitfire build completed successfully.")
	} else if buildFlag {
		if prePatch {
			if err := spitfire.ApplyPatches(sourcePath, patchesRepo, "pre-compile-patches", filepath.Join(sourcePath, "patcher"), skipPatchUpdate); err != nil {
				log.Fatalf("Error during patch application: %v", err)
			}
		}
		spitfire.Configure(sourcePath)
		spitfire.Build(sourcePath)
		// Detect the build directory dynamically
		buildDir, err = spitfire.ResolveBuildDir(sourcePath)
		if err != nil {
			log.Fatalf("Error resolving build directory: %v", err)
		}
		if postPatch {
			if err := spitfire.ApplyPatches(buildDir, patchesRepo, "post-compile-patches", filepath.Join(sourcePath, "patcher"), skipPatchUpdate); err != nil {
				log.Fatalf("Error during patch application: %v", err)
			}
		}
		if run {
			spitfire.RunProject(sourcePath)
		}
		fmt.Println("Spitfire build completed successfully.")
	} else if clean {
		spitfire.CleanBuild(sourcePath, fullClean)
		fmt.Println("Cleaned Firefox build.")
	} else if update {
		spitfire.DownloadSource(sourcePath, sourceRepo)
		spitfire.UpdateRepo(sourcePath)
		fmt.Println("Mozilla repository updated.")
	} else if prePatch {
		spitfire.DownloadSource(sourcePath, sourceRepo)
		if err := spitfire.ApplyPatches(sourcePath, patchesRepo, "pre-compile-patches", filepath.Join(sourcePath, "patcher"), skipPatchUpdate); err != nil {
			log.Fatalf("Error during patch application: %v", err)
		}
		fmt.Println("Patches updated.")
	} else if postPatch {
		// Detect the build directory dynamically
		buildDir, err = spitfire.ResolveBuildDir(sourcePath)
		if err != nil {
			log.Fatalf("Error resolving build directory: %v", err)
		}
		if _, err := os.Stat(buildDir); err == nil {
			if err := spitfire.ApplyPatches(buildDir, patchesRepo, "post-compile-patches", filepath.Join(sourcePath, "patcher"), skipPatchUpdate); err != nil {
				log.Fatalf("Error during patch application: %v", err)
			}
		}
		if run {
			spitfire.RunProject(sourcePath)
		}
	} else if run {
		spitfire.RunProject(sourcePath)
	}

	return buildDir
}

// PackageAndUploadProcess handles compressing, packaging, and uploading the build to SourceForge.
func PackageAndUploadProcess(buildDir string) {

	// Restore working directory before performing SourceForge operations
	restoreWorkingDirectory()

	pathToUse := buildDir
	if upload && uploadPath != "" {
		pathToUse = uploadPath
	}

	if pathToUse == "" {
		log.Fatalf("Error: no valid build or upload path provided.")
	}

	// Calculate & display uncompressed size
	uncompressedSize, err := spitfire.GetDirectorySize(pathToUse)
	if err != nil {
		log.Fatalf("Failed to calculate uncompressed size: %v", err)
	}
	fmt.Printf("Uncompressed directory size: %s\n", spitfire.BytesToHumanReadable(uncompressedSize))

	// Create the tar.gz name, e.g. "browser-amd64-nightly-linux.tar.gz"
	outputCompressedFile := filepath.Join(".", fmt.Sprintf(
		"%s-%s-%s-%s.tar.gz",
		component, arch, release, platform,
	))

	// Compress if requested
	if compress {
		err := spitfire.CompressDirectory(pathToUse, outputCompressedFile)
		if err != nil {
			log.Fatalf("Failed to compress build directory: %v", err)
		}
		fmt.Printf("Build directory compressed to: %s\n", outputCompressedFile)
	}

	// Compressed size
	compressedSize, err := spitfire.GetFileSize(outputCompressedFile)
	if err != nil {
		log.Fatalf("Failed to get compressed file size: %v", err)
	}
	fmt.Printf("Compressed file size: %s\n", spitfire.BytesToHumanReadable(compressedSize))

	// Show compression ratio & efficiency
	compressionRatio, efficiency := spitfire.CalculateCompressionEfficiency(uncompressedSize, compressedSize)
	fmt.Printf("Compression ratio: %.2f:1\n", compressionRatio)
	fmt.Printf("Compression efficiency: %.2f%%\n", efficiency)

	// Display compressed directory path
	fmt.Printf("Compressed dir: %s\n", pathToUse)

	// If not uploading, we're done
	if !upload {
		return
	}

	// Load SourceForge config
	config, err := spitfire.LoadConfig()
	if err != nil {
		log.Fatalf("Failed to load SourceForge config: %v", err)
	}

	// Check tarball existence
	if _, err := os.Stat(outputCompressedFile); err != nil {
		log.Fatalf("No compressed file found to upload: %v", err)
	}

	// The subdirectory path in the SF project
	// e.g. "browser/amd64/nightly/2024.12.23"
	uploadDir := fmt.Sprintf("%s/%s/%s/%s", component, arch, release, version)

	// 1) Upload the file to SourceForge once
	err = spitfire.Upload(config, outputCompressedFile,
		"/home/frs/project/spitfire-browser/"+uploadDir+"/")
	if err != nil {
		log.Fatalf("Failed to upload compressed file: %v", err)
	}
	fmt.Println("Compressed file uploaded successfully.")

	// 2) Download existing APPINDEX or create new
	err = spitfire.DownloadAPPINDEX(config, "/home/frs/project/spitfire-browser/")
	if err != nil {
		fmt.Println("Failed to download APPINDEX. A new one will be created and uploaded.")
	}

	// 3) Update the APPINDEX
	err = spitfire.PackageAPPINDEX(
		packageName, // e.g. "spitfire-browser"
		release,     // e.g. "nightly"
		version,     // e.g. "2024.12.23"
		arch,
		fmt.Sprintf("%d", compressedSize),
		fmt.Sprintf("%d", uncompressedSize),
		name, // e.g. "Spitfire"
		url,
		licence,
		component,
		maintainer,
		"", // dependencies
		platform,
		uploadDir,
	)
	if err != nil {
		log.Fatalf("Failed to update APPINDEX: %v", err)
	}
	fmt.Println("APPINDEX updated successfully.")

	// 4) Clean
	if err := spitfire.CleanAppIndex(); err != nil {
		log.Fatalf("Failed to clean APPINDEX: %v", err)
	}

	// 5) Upload the updated APPINDEX
	err = spitfire.UploadAPPINDEX(config, "/home/frs/project/spitfire-browser/")
	if err != nil {
		log.Fatalf("Failed to upload updated APPINDEX: %v", err)
	}
	fmt.Println("APPINDEX uploaded successfully.")
}

// restoreWorkingDirectory restores the initial working directory after any operation that might change it.
func restoreWorkingDirectory() {
	err := os.Chdir(initialDir)
	if err != nil {
		log.Fatalf("Failed to restore the working directory: %v", err)
	}
	fmt.Printf("Restored working directory to: %s\n", initialDir)
}