Add cleanup config (#23)

portapps · Jun 9, 2020 · d2bff65 · d2bff65
1 parent 18bdad6
commit d2bff65
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changelog
 
+## 2020.05-15 (2020/06/09)
+
+* Add `cleanup` config (#23)
+
 ## 2020.05-14 (2020/05/29)
 
 * Fix addons startup (portapps/phyrox-portable#3) 

diff --git a/assets/assets.go b/assets/assets.go
diff --git a/build.properties b/build.properties
@@ -6,7 +6,7 @@ app = waterfox
 app.name = Waterfox
 app.type = archive
 app.version = 2020.05
-app.release = 14
+app.release = 15
 app.homepage = https://www.waterfox.net/
 
 # Portable app

diff --git a/main.go b/main.go
@@ -25,6 +25,7 @@ type config struct {
 	Profile           string `yaml:"profile" mapstructure:"profile"`
 	MultipleInstances bool   `yaml:"multiple_instances" mapstructure:"multiple_instances"`
 	DisableTelemetry  bool   `yaml:"disable_telemetry" mapstructure:"disable_telemetry"`
+	Cleanup           bool   `yaml:"cleanup" mapstructure:"cleanup"`
 }
 
 var (
@@ -40,6 +41,7 @@ func init() {
 		Profile:           "default",
 		MultipleInstances: false,
 		DisableTelemetry:  false,
+		Cleanup:           false,
 	}
 
 	// Init app
@@ -88,6 +90,16 @@ func main() {
 		}
 	}
 
+	// Cleanup on exit
+	if cfg.Cleanup {
+		defer func() {
+			utl.Cleanup([]string{
+				path.Join(os.Getenv("APPDATA"), "Waterfox"),
+				path.Join(os.Getenv("LOCALAPPDATA"), "Waterfox"),
+			})
+		}()
+	}
+
 	// Multiple instances
 	if cfg.MultipleInstances {
 		log.Info().Msg("Multiple instances enabled")