Update file association scripts.

Author: f32y

README.md

@@ -1,5 +1,8 @@
 # macOS Initialization Scripts
 
-> Modify some system preferences
->
-> Change default file associations
+My preferences.
+
+1. Modify system preferences
+2. Change default file associations
+
+Get more from https://github.com/kevinSuttle/macOS-Defaults

associate_changer_file_extensions.zsh

@@ -0,0 +1,49 @@
+#!/bin/zsh
+
+export LS_PLIST="$HOME/Library/Preferences/com.apple.LaunchServices/com.apple.launchservices.secure.plist"
+
+function associate_extension {
+  EXT=$1
+  APP=$2
+
+  i=0
+  while true; do
+    # Check if we reach the end of LSHandlers array.
+    CHECK_EXIST_CMD='Print :LSHandlers:'$i
+    EXIST=$(/usr/libexec/PlistBuddy -c "$CHECK_EXIST_CMD" $LS_PLIST 2>/dev/null)
+    if [ $? -ne 0 ]; then
+      break
+    fi
+
+    CHECK_TAG_CMD='Print :LSHandlers:'$i':LSHandlerContentTag'
+    # Check if there is a LSHandlerContentTag and it is the one we need to modify.
+    PREVIOUS=$(/usr/libexec/PlistBuddy -c "$CHECK_TAG_CMD" $LS_PLIST 2>/dev/null)
+    if [[ "$PREVIOUS" = "$EXT" ]]; then
+      # Found. Modify it!
+      echo "Modified existing association: *.$EXT -> $APP"
+      MODIFY_CMD='Set :LSHandlers:'$i':LSHandlerRoleAll $APP'
+      /usr/libexec/PlistBuddy -c "$MODIFY_CMD" $LS_PLIST
+      return
+    fi
+
+    i=$((i+1))
+  done
+
+  # Exist entry not found. Just create a new one.
+  CONTENT="<dict>
+             <key>LSHandlerContentTag</key>
+             <string>$EXT</string>
+             <key>LSHandlerContentTagClass</key>
+             <string>public.filename-extension</string>
+             <key>LSHandlerPreferredVersions</key>
+             <dict>
+				       <key>LSHandlerRoleAll</key>
+				         <string>-</string>
+			       </dict>
+             <key>LSHandlerRoleAll</key>
+             <string>$APP</string>
+           </dict>"
+  defaults write com.apple.LaunchServices/com.apple.launchservices.secure LSHandlers -array-add $CONTENT
+
+  echo "Created new association: *.$EXT -> $APP"
+}

association_changer.swift association_changer_public_uti.swift

@@ -1,5 +1,7 @@
 #!/usr/bin/xcrun swift
 
+// For public.* UTIs.
+
 import Foundation
 
 func main() -> Void {

change_file_associations.sh

@@ -0,0 +1,44 @@
+#!/bin/zsh
+
+source $PWD/associate_changer_file_extensions.zsh
+
+set -xe
+
+VSCODE='com.microsoft.vscode'
+
+OPEN_WITH_VSCODE=(
+  'net.daringfireball.markdown'
+  'public.c-header'
+  'public.objective-c-source'
+  'public.plain-text'
+  'public.patch-file'
+  'public.python-script'
+  'public.c-plus-plus-source'
+  'com.netscape.javascript-source'
+  'public.xml'
+  'public.yaml'
+  'public.protobuf-source'
+  'public.html'
+)
+
+OPEN_WITH_DUMB_BROWSER=(
+  'public.url'
+)
+
+# PDF
+$PWD/association_changer_public_uti.swift com.adobe.pdf com.readdle.pdfexpert-mac
+
+# Open with VSCode
+for ty in $OPEN_WITH_VSCODE do
+  $PWD/association_changer_public_uti.swift $ty $VSCODE
+done
+
+# For some custom exension names
+associate_extension "gn" $VSCODE
+associate_extension "gni" $VSCODE
+associate_extension "meta" $VSCODE
+associate_extension "toml" $VSCODE
+associate_extension "props" $VSCODE
+
+# Restart LaunchServices
+/System/Library/Frameworks/CoreServices.framework/Versions/A/Frameworks/LaunchServices.framework/Versions/A/Support/lsregister -kill -r -domain local -domain system -domain user