package main

import (
	"encoding/json"
	"fmt"
	"log"
	"os"
	"regexp"
	"strings"

	"github.com/aspect-build/bazel-lib/tools/common"
	"golang.org/x/exp/maps"
)

func main() {
	args := os.Args[1:]

	if len(args) == 1 {
		if args[0] == "--version" || args[0] == "-v" {
			fmt.Printf("expand_template %s\n", common.Version())
			return
		}
	}

	if len(args) != 6 {
		fmt.Println("Usage: expand_template <template> <out> <substitutions_json> <volatile_status_file> <stable_status_file> <executable>")
		os.Exit(1)
	}

	executable := args[5] == "true" || args[5] == "1"

	substitutions := map[string]string{}
	substitutionJson, err := os.ReadFile(args[2])
	if err != nil {
		log.Fatal(fmt.Errorf("failed to read substitutions file: %w", err))
	}
	if err := json.Unmarshal(substitutionJson, &substitutions); err != nil {
		log.Fatal(fmt.Errorf("failed to parse substitutions file: %w", err))
	}

	volatileStatus, err := parseStatusFile(args[3])
	if err != nil {
		log.Fatal(fmt.Errorf("failed to parse volatile status file: %w", err))
	}
	stableStatus, err := parseStatusFile(args[4])
	if err != nil {
		log.Fatal(fmt.Errorf("failed to parse stable status file: %w", err))
	}

	statuses := map[string]string{}
	maps.Copy(statuses, volatileStatus)
	maps.Copy(statuses, stableStatus)

	for key, value := range substitutions {
		for token, replacement := range statuses {
			value = strings.ReplaceAll(value, fmt.Sprintf("{{%s}}", token), replacement)
		}
		substitutions[key] = value
	}

	contentBytes, err := os.ReadFile(args[0])
	if err != nil {
		log.Fatal(fmt.Errorf("failed to read the template file: %w", err))
	}
	content := string(contentBytes)

	for key, value := range substitutions {
		content = strings.ReplaceAll(content, key, value)
	}

	var mode os.FileMode = 0x666
	if executable {
		mode = 0o777
	}
	err = os.WriteFile(args[1], []byte(content), mode)
	if err != nil {
		log.Fatal(fmt.Errorf("failed to write output file: %w", err))
	}
}

// captures every `KEY VALUE` line in the status file.
// for explanation see: https://regex101.com/r/B28NN7/1
var STATUS_REGEX = regexp.MustCompile(`(?m)^([^\s]+)\s+([^\n ]*)$`)

func parseStatusFile(statusFilePath string) (map[string]string, error) {
	statusFile, err := os.ReadFile(statusFilePath)
	if err != nil {
		return nil, err
	}

	matches := STATUS_REGEX.FindAllStringSubmatch(string(statusFile), -1)

	results := map[string]string{}

	for _, match := range matches {
		results[match[1]] = match[2]
	}

	return results, nil

}