package main

import (
	"fmt"
	"log"
	"os"
	"path/filepath"
	"strings"
)

func generateEnv(yamlFile string, wantDiff bool, allowWrite bool, useColor bool) {
	config := loadConfig(yamlFile)

	// Check that all local directories exist
	for _, logEntry := range config.Logs {
		if _, err := os.Stat(logEntry.LocalDirectory); os.IsNotExist(err) {
			log.Fatalf("User is required to create %s", logEntry.LocalDirectory)
		}
	}

	// Generate .env file for each log
	for _, logEntry := range config.Logs {
		envPath := filepath.Join(logEntry.LocalDirectory, ".env")

		// Create combined roots.pem file
		rootsPemPath := filepath.Join(logEntry.LocalDirectory, "roots.pem")
		err := createCombinedRootsPemWithStatus(config.Roots, logEntry.ExtraRoots, rootsPemPath, wantDiff, allowWrite, useColor)
		if err != nil {
			log.Fatalf("Failed to create %s: %v", rootsPemPath, err)
		}

		// Build TESSERACT_ARGS string
		args := []string{
			fmt.Sprintf("--private_key=%s", logEntry.Secret),
			fmt.Sprintf("--origin=%s", logEntry.Origin),
			fmt.Sprintf("--storage_dir=%s", logEntry.LocalDirectory),
			fmt.Sprintf("--roots_pem_file=%s", rootsPemPath),
			"--batch_max_size=512",
			"--batch_max_age=500ms",
			"--pushback_max_outstanding=4096",
			"--pushback_max_dedupe_in_flight=250",
		}

		// Add http_endpoint if Listen is specified
		if logEntry.Listen != "" {
			args = append(args, fmt.Sprintf("--http_endpoint=%s", logEntry.Listen))
		}

		// Add not_after flags if specified
		if !logEntry.NotAfterStart.IsZero() {
			args = append(args, fmt.Sprintf("--not_after_start=%s", logEntry.NotAfterStart.Format("2006-01-02T15:04:05Z")))
		}
		if !logEntry.NotAfterLimit.IsZero() {
			args = append(args, fmt.Sprintf("--not_after_limit=%s", logEntry.NotAfterLimit.Format("2006-01-02T15:04:05Z")))
		}

		tesseractArgs := strings.Join(args, " ")
		envContent := fmt.Sprintf("TESSERACT_ARGS=\"%s\"\nOTEL_EXPORTER_OTLP_ENDPOINT=http://localhost:4318\n", tesseractArgs)

		err = writeFileWithStatus(envPath, []byte(envContent), wantDiff, allowWrite, useColor)
		if err != nil {
			log.Fatalf("Failed to write %s: %v", envPath, err)
		}
	}
}

func createCombinedRootsPemWithStatus(rootsFile, extraRootsFile, outputPath string, wantDiff bool, allowWrite bool, useColor bool) error {
	// Read main roots file
	var combinedContent []byte
	if rootsFile != "" {
		rootsData, err := os.ReadFile(rootsFile)
		if err != nil {
			return fmt.Errorf("failed to read roots file %s: %v", rootsFile, err)
		}
		combinedContent = append(combinedContent, rootsData...)
	}

	// Append extra roots file if it exists
	if extraRootsFile != "" {
		extraRootsData, err := os.ReadFile(extraRootsFile)
		if err != nil {
			return fmt.Errorf("failed to read extra roots file %s: %v", extraRootsFile, err)
		}
		combinedContent = append(combinedContent, extraRootsData...)
	}

	return writeFileWithStatus(outputPath, combinedContent, wantDiff, allowWrite, useColor)
}