Add cleanup config

portapps · Dec 15, 2019 · 25b2fe1 · 25b2fe1
1 parent e028648
commit 25b2fe1
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -16,6 +16,7 @@ import (
 )
 
 type config struct {
+	Cleanup      bool `yaml:"cleanup" mapstructure:"cleanup"`
 	DisableDelay bool `yaml:"disable_delay" mapstructure:"disable_delay"`
 }
 
@@ -29,6 +30,7 @@ func init() {
 
 	// Default config
 	cfg = &config{
+		Cleanup:      false,
 		DisableDelay: false,
 	}
 
@@ -45,6 +47,15 @@ func main() {
 		"--user-data-dir=" + app.DataPath,
 	}
 
+	// Cleanup on exit
+	if cfg.Cleanup {
+		defer func() {
+			utl.Cleanup([]string{
+				path.Join(os.Getenv("APPDATA"), "Franz"),
+			})
+		}()
+	}
+
 	// Copy default shortcut
 	shortcutPath := path.Join(os.Getenv("APPDATA"), "Microsoft", "Windows", "Start Menu", "Programs", "Franz Portable.lnk")
 	defaultShortcut, err := assets.Asset("Franz.lnk")