package main

import (
	"database/sql"
	"fmt"
	"io"
	"log"
	"os"

	_ "modernc.org/sqlite"
)

func main() {
	fmt.Println("==================================================")
	fmt.Println("  CHALLENGER POS - DB INTEGRITY & CLEANUP TOOL")
	fmt.Println("==================================================")

	dbPath := "C:/D3/PRM/CHALLENGER.db"
	backupPath := "C:/D3/PRM/CHALLENGER_PRE_CLEAN.db"

	fmt.Println("[*] Step 1: Creating safety backup...")
	if err := copyFile(dbPath, backupPath); err != nil {
		log.Fatalf("[!] Backup failed! Aborting to be safe: %v", err)
	}
	fmt.Println("[+] Backup secured at:", backupPath)

	db, err := sql.Open("sqlite", dbPath+"?_pragma=busy_timeout(5000)")
	if err != nil {
		log.Fatalf("FATAL: Could not open database. %v", err)
	}
	defer db.Close()

	fmt.Println("\n[*] Step 2: Running SQLite Integrity Check...")
	var integrityResult string
	err = db.QueryRow("PRAGMA integrity_check;").Scan(&integrityResult)
	if err != nil {
		fmt.Printf("[!] Integrity check error: %v\n", err)
	} else {
		fmt.Printf("[+] Integrity Check Result: %s\n", integrityResult)
	}

	fmt.Println("\n[*] Step 3: Executing FULL VACUUM...")
	fmt.Println("    (Physically rebuilding the file to destroy ghost data)")
	_, err = db.Exec("VACUUM;")
	if err != nil {
		fmt.Printf("[!] VACUUM failed: %v\n", err)
	} else {
		fmt.Println("[+] VACUUM Complete! Database defragmented and rebuilt.")
	}

	fmt.Println("\n[*] Step 4: Optimizing indexes & clearing memory logs...")
	db.Exec("PRAGMA optimize;")
	db.Exec("PRAGMA wal_checkpoint(TRUNCATE);")
	fmt.Println("[+] Optimization complete.")

	fmt.Println("\n==================================================")
	fmt.Println("  CLEANUP FINISHED. You may now restart facet.exe.")
	fmt.Println("==================================================")
}

func copyFile(src, dst string) error {
	in, err := os.Open(src)
	if err != nil { return err }
	defer in.Close()
	out, err := os.Create(dst)
	if err != nil { return err }
	defer out.Close()
	_, err = io.Copy(out, in)
	if err != nil { return err }
	return out.Sync()
}