feat: revert and expose updater functions

spicetify · Dec 7, 2023 · 24a3f7e · 24a3f7e
1 parent 28c33e3
commit 24a3f7e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/jsHelper/spicetifyWrapper.js b/jsHelper/spicetifyWrapper.js
@@ -311,7 +311,8 @@ window.Spicetify = {
 	},
 	ReactComponent: {},
 	ReactHook: {},
-	ReactFlipToolkit: {}
+	ReactFlipToolkit: {},
+	Snackbar: {}
 };
 
 (async function hotloadWebpackModules() {
@@ -449,7 +450,6 @@ window.Spicetify = {
 		// Snackbar notifications
 		// https://github.com/iamhosseindhv/notistack
 		Snackbar: {
-			...Spicetify.Snackbar,
 			SnackbarProvider: functionModules.find(m => m.toString().includes("enqueueSnackbar called with invalid argument")),
 			useSnackbar: functionModules.find(m => m.toString().match(/\{return\(0,\w+\.useContext\)\(\w+\)\}/))
 		},

diff --git a/src/preprocess/preprocess.go b/src/preprocess/preprocess.go
@@ -465,8 +465,18 @@ if (${1}.popper?.firstChild?.id === "context-menu") {
 	// Snackbar https://github.com/iamhosseindhv/notistack
 	utils.Replace(
 		&input,
-		`\w+\s*=\s*\w\.call\(this,[^)]+\)\s*\|\|\s*this\)\.enqueueSnackbar`,
-		`Spicetify.Snackbar=${0}`)
+		`\(\w+\s*=\s*\w\.call\(this,[^)]+\)\s*\|\|\s*this\)\.enqueueSnackbar`,
+		` Spicetify.Snackbar.enqueueSnackbar=${0}`)
+
+	utils.Replace(
+		&input,
+		`\w+.closeSnackbar=function`,
+		`Spicetify.Snackbar.closeSnackbar=${0}`)
+
+	utils.Replace(
+		&input,
+		`(this\.updater)\s*=\s*(.*)`,
+		`${1}=Spicetify.Snackbar.updater=${2}`)
 
 	return input
 }