package main

import (
	"context"
	"errors"
	"log"
	"os"
	"os/signal"
	"path/filepath"
	"syscall"
	"time"

	"git.ipao.vip/rogee/mohomo-docker/internal/bootstrap"
)

const (
	defaultRoot         = "/opt/clash"
	defaultSSClashTemp  = "/tmp/ssclash"
	defaultCoreSource   = "/usr/local/lib/ssclash/clash"
	defaultConfigSource = "/usr/local/share/ssclash/config.yaml"
	ssclashBinary       = "/usr/local/bin/ssclash"
	defaultRuntimeDir   = "/dev/shm/mohomo"
	defaultSecretPath   = "/run/secrets/subscription"
	defaultDataDir      = "/data"
)

func main() {
	log.SetFlags(log.Ldate | log.Ltime | log.LUTC)
	if len(os.Args) == 2 && os.Args[1] == "candidate" {
		ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
		defer stop()
		if err := bootstrap.PublishCandidate(ctx, bootstrap.CandidateConfig{
			SecretPath:   defaultSecretPath,
			DataDir:      defaultDataDir,
			TemplatePath: defaultConfigSource,
			MihomoBinary: defaultCoreSource,
		}); err != nil {
			log.Fatalf("bootstrap: candidate update failed: %v", err)
		}
		log.Print("bootstrap: candidate configuration published")
		return
	}
	if len(os.Args) != 1 {
		log.Fatal("bootstrap: usage: bootstrap [candidate]")
	}
	root := envOrDefault("SSCLASH_ROOT", defaultRoot)
	log.Printf("bootstrap: preparing persistent runtime root=%s", root)

	result, err := bootstrap.Prepare(bootstrap.Config{
		Root:         root,
		SSClashTemp:  envOrDefault("SSCLASH_TMP", defaultSSClashTemp),
		CoreSource:   defaultCoreSource,
		ConfigSource: defaultConfigSource,
	})
	if err != nil {
		log.Fatalf("bootstrap: runtime preparation failed: %v", err)
	}
	adminPasswordInitialized, err := bootstrap.EnsureAdminPassword(root, ssclashBinary, os.Getenv("SSCLASH_PASSWORD"))
	if err != nil {
		log.Fatalf("bootstrap: admin authentication setup failed: %v", err)
	}
	log.Printf(
		"bootstrap: ready root=%s core_initialized=%t config_initialized=%t config_migrated=%t server_settings_changed=%t admin_password_initialized=%t",
		root,
		result.CoreInitialized,
		result.ConfigInitialized,
		result.ConfigMigrated,
		result.ServerSettingsChanged,
		adminPasswordInitialized,
	)

	subscriptionURL := os.Getenv("SUBSCRIPTION_URL")
	if subscriptionURL == "" {
		log.Fatal("bootstrap: SUBSCRIPTION_URL is required")
	}

	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
	defer stop()
	err = bootstrap.Run(ctx, bootstrap.RuntimeConfig{
		Root:            root,
		CoreBinary:      filepath.Join(root, "bin", "clash"),
		SSClashBinary:   ssclashBinary,
		ConfigSource:    filepath.Join(root, "config.yaml"),
		RuntimeDir:      defaultRuntimeDir,
		SubscriptionURL: subscriptionURL,
		UpdateInterval:  time.Hour,
	})
	if err != nil && !errors.Is(err, context.Canceled) {
		log.Fatalf("bootstrap: service failed: %v", err)
	}
}

func envOrDefault(key, fallback string) string {
	if value := os.Getenv(key); value != "" {
		return value
	}
	return fallback
}