package main

import (
	"log"
	"os"

	"gitea.d3m0k1d.ru/d3m0k1d/HellreigN/agent/internal/config"
	"gitea.d3m0k1d.ru/d3m0k1d/HellreigN/agent/internal/registration"
)

func main() {
	cfgPath := os.Getenv("CONFIG_FILE")
	if cfgPath == "" {
		cfgPath = "/etc/hellreign-agent/config.yml"
	}

	cfg, err := config.Load(cfgPath)
	if err != nil {
		log.Fatalf("Failed to load config: %v", err)
	}

	log.Printf("Agent label: %s", cfg.Label)

	if cfg.RegistrationToken == "" {
		log.Fatal("No registration token provided")
	}

	// Generate key and CSR
	key, csrPEM, err := registration.GenerateKeyAndCSR(cfg.Label)
	if err != nil {
		log.Fatalf("Failed to generate key and CSR: %v", err)
	}
	log.Println("Generated ECDSA key pair and CSR")

	// Register with backend
	certs, err := registration.Register(cfg.BackendURL, cfg.RegistrationToken, csrPEM)
	if err != nil {
		log.Fatalf("Failed to register: %v", err)
	}
	log.Println("Successfully registered, received certificates")

	// Save certificates
	if err := registration.SaveCerts(cfg.CertDir, certs, key); err != nil {
		log.Fatalf("Failed to save certificates: %v", err)
	}
	log.Printf("Certificates saved to %s", cfg.CertDir)

	log.Println("Agent registration complete")
}