From cb3ec42b92c4586cf3cc870c11765655c33ce1ef Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Mon, 27 Sep 2021 18:41:34 -0600
Subject: [PATCH] Add functions to disable Chrome notifications

---
 scripts/auto_setup.py   |  2 +-
 scripts/wk/setup/win.py | 48 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 49 insertions(+), 1 deletion(-)

diff --git a/scripts/auto_setup.py b/scripts/auto_setup.py
index 9db1d1c4..2f4311ea 100644
--- a/scripts/auto_setup.py
+++ b/scripts/auto_setup.py
@@ -120,7 +120,7 @@ BASE_MENUS = {
       MenuEntry('Software Bundle',          'auto_install_software_bundle'),
       ),
     'Configure System': (
-      MenuEntry('Chrome Notifications',     no_op),
+      MenuEntry('Chrome Notifications',     'auto_disable_chrome_notifications'),
       #MenuEntry('O&O ShutUp 10',            no_op),
       MenuEntry('Open Shell',               'auto_config_open_shell'),
       MenuEntry('uBlock Origin',            'auto_enable_ublock_origin'),
diff --git a/scripts/wk/setup/win.py b/scripts/wk/setup/win.py
index 5b54da1d..bcfcddfb 100644
--- a/scripts/wk/setup/win.py
+++ b/scripts/wk/setup/win.py
@@ -409,6 +409,11 @@ def auto_config_open_shell():
   TRY_PRINT.run('Open Shell...', reg_write_settings, REG_OPEN_SHELL_SETTINGS)
 
 
+def auto_disable_chrome_notifications():
+  """Disable notifications in Google Chrome."""
+  TRY_PRINT.run('Chrome Notifications...', disable_chrome_notifications)
+
+
 def auto_enable_ublock_origin():
   """Enable uBlock Origin in supported browsers."""
   prompt = '    Press Enter to continue...'
@@ -452,6 +457,49 @@ def auto_install_vcredists():
 
 
 # Configure Functions
+def disable_chrome_notifications():
+  """Disable notifications in Google Chrome."""
+  defaults_key = 'default_content_setting_values'
+  profiles = []
+  search_path = case_insensitive_path(
+    f'{os.environ.get("LOCALAPPDATA")}/Google/Chrome/User Data',
+    )
+
+  # Bail early
+  if not search_path:
+    raise GenericWarning('No profiles detected.')
+
+  # Close any running instances of Chrome
+  kill_procs('chrome.exe', force=True)
+
+  # Build list of profiles
+  for item in search_path.iterdir():
+    if not item.is_dir():
+      continue
+
+    if re.match(r'^(Default|Profile).*', item.name, re.IGNORECASE):
+      profiles.append(item)
+
+  # Bail if no profiles were detected
+  if not profiles:
+    raise GenericWarning('No profiles detected.')
+
+  # Set notifications preference
+  for profile in profiles:
+    pref_file = profile.joinpath('Preferences')
+    if not pref_file.exists():
+      continue
+
+    # Update config
+    pref_data = json.loads(pref_file.read_text())
+    if defaults_key not in pref_data['profile']:
+      pref_data['profile'][defaults_key] = {}
+    pref_data['profile'][defaults_key]['notifications'] = 2
+
+    # Save file
+    pref_file.write_text(json.dumps(pref_data, separators=(',', ':')))
+
+
 def enable_ublock_origin():
   """Enable uBlock Origin in supported browsers."""
   base_paths = [