From 9b5aa1652ab90be944b60d840b6bd5e1df05b0c3 Mon Sep 17 00:00:00 2001
From: Sr Komodo <komodosr@gmail.com>
Date: Wed, 17 Oct 2018 21:15:46 -0300
Subject: [PATCH] set dark theme in prefs.js

---
 install.go | 41 ++++++++++++++++++++++-------------------
 1 file changed, 22 insertions(+), 19 deletions(-)

diff --git a/install.go b/install.go
index b382ffc..fdd724d 100644
--- a/install.go
+++ b/install.go
@@ -11,9 +11,11 @@ import (
 	"time"
 )
 
-const darkThemeConfig = `
-user_pref("lightweightThemes.selectedThemeID", "firefox-compact-dark@mozilla.org");
-user_pref("devtools.theme", "dark");`
+var darkThemeConfig = map[string]string{
+	"lightweightThemes.selectedThemeID": "\"firefox-compact-dark@mozilla.org\"",
+	"browser.uidensity":                 "1",
+	"devtools.theme":                    "\"dark\"",
+}
 
 func uninstall(profile string) (string, error) {
 	err := os.RemoveAll(filepath.Join(profile, "chrome", "ShadowFox_customization"))
@@ -221,33 +223,34 @@ func install(profilePath string, generateUUIDs bool, setTheme bool) (string, err
 
 	// Set dark theme
 	if setTheme {
-		userJs := filepath.Join(profilePath, "user.js")
-		userJsContent := []byte{}
+		prefs := filepath.Join(profilePath, "prefs.js")
+		prefsContent := []byte{}
 
-		exists, _, err := pathExists(userJs)
+		exists, _, err := pathExists(prefs)
 		if exists {
-			userJsContent, err = ioutil.ReadFile(userJs)
+			prefsContent, err = ioutil.ReadFile(prefs)
 			if err != nil {
-				return "Couldn't read user.js", err
+				return "Couldn't read prefs.js", err
 			}
 		} else {
-			err = createFile(userJs)
+			err = createFile(prefs)
 			if err != nil {
-				return "Couldn't create user.js", err
+				return "Couldn't create prefs.js", err
 			}
 		}
 
-		if !strings.Contains(string(userJsContent), darkThemeConfig) {
-			err = backUp(userJs)
-			if err != nil {
-				return "Couldn't backup user.js", err
+		for key, value := range darkThemeConfig {
+			regex := regexp.MustCompile("user_pref(\"" + key + "\", .+);")
+			replace := []byte("user_pref(\"" + key + "\", " + value + ");")
+			if regex.Match(prefsContent) {
+				prefsContent = regex.ReplaceAll(prefsContent, replace)
+			} else {
+				prefsContent = append(append(prefsContent, replace...), '\n')
 			}
+		}
 
-			userJsContent = append(userJsContent, []byte(darkThemeConfig)...)
-
-			if err := ioutil.WriteFile(userJs, userJsContent, 0644); err != nil {
-				return "Couldn't write user.js", err
-			}
+		if err := ioutil.WriteFile(prefs, prefsContent, 0644); err != nil {
+			return "Couldn't write prefs.js", err
 		}
 	}