#!/bin/bash

function folder_dialog()
{
  local result=$(osascript << EOT
    tell application "Finder"
      activate
      set fpath to POSIX path of (choose folder)
      set thePassword to display dialog "Enter your password:" default answer "" with hidden answer buttons {"Cancel", "OK"} default button "OK" with icon 1
      set thePasswordText to text returned of thePassword
      return fpath
    end tell
  EOT)

  echo "$result"
}

# Ask for folder
lib=$(folder_dialog)

if [ ! -d "$lib" ]; then
  echo "🥁 Invalid Kontakt library path"
  read -p "🎹 Press Enter to continue..."
  exit 1
fi

# Path must begin with "/Volumes"
if [[ "$lib" != /Volumes/* ]]; then
  vol=$(ls -l /Volumes | grep ' -> /' | awk '{$1=""; $2=""; $3=""; $4=""; $5=""; $6=""; $7=""; $8=""; print $0}' | awk '{$1=$1};1')
  vol=${vol/ -> \//}
  lib=/Volumes/$vol$lib
fi

xml=/var/tmp/kontaktLibraryHints.xml

find "$lib" -iname "*.nicnt" -o -iname "*_info.nkx" -type f | while read file
do
  # Extract library version (`.nicnt` only)
  cver=

  if [[ "$file" = *.nicnt ]]; then
    cver=$(dd skip=66 count=10 bs=1 if="$file" 2> /dev/null | sed 's/\x00//g')
    echo $cver | grep '\d\.\d\.\d' > /dev/null

    if [ $? -ne 0 ]; then
      cver=$(dd skip=66 count=6 bs=1 if="$file" 2> /dev/null | sed 's/\x00//g')
      echo $cver | grep '\d\.\d' > /dev/null
    fi

    [ $? -ne 0 ] && cver=
  else
    # Skip `_info.nkx` if `.nicnt` is present
    ldir=$(dirname "$file")
    hasnicnt=$(ls "$ldir" | grep -i '.nicnt' | wc -l)
    if [ $hasnicnt -ne 0 ]; then
      continue
    fi
  fi

  # Extract library installation hints XML tree
  # and remove old `.plist` and `.xml` (if present) while getting its name
  awk '/<ProductHints[ >]/, $NF ~ /<\/ProductHints>/' "$file" | LC_ALL=C sed 's/<\/ProductHints>.*/<\/ProductHints>/' | xmllint --format --recover --encode "UTF-8" - > "$xml"
  name=$(xmllint --xpath "string(//Name)" "$xml")
  regkey=$(xmllint --xpath "string(//RegKey)" "$xml")
  plist="/Library/Preferences/com.native-instruments.$regkey.plist"
  xmldist="/Library/Application Support/Native Instruments/Service Center/$name.xml"

# Check for bad `.nicnt` (improperly hand-made, unofficial)
if grep -iq '<HU>' "$xml"; then
  nohu=0
else
  nohu=1
fi
if grep -iq '<JDX>' "$xml"; then
  nojdx=0
else
  nojdx=1
fi
if grep -iq '<ProductSpecific>' "$xml"; then
  nops=0
else
  nops=1
fi

# These encryption keys fool Kontakt into believing
# that library is legit
if (( nohu != 0 )) && (( nojdx != 0 )); then
  cp "$xml" "$xml.tmp"
  if (( nops != 0 )); then
    sed 's/<\/SNPID>/<\/SNPID>|    <ProductSpecific>|      <HU>6C70AC13E02414D1A552685A1301D859<\/HU>|      <JDX>023733942B73318EAEAD914E3981EC68BE72519A2F5738F828A6A028C4E1DBAC<\/JDX>|      <Visibility type="Number">3<\/Visibility>|    <\/ProductSpecific>/' "$xml.tmp" | tr '|' '\n' > "$xml"
  else
    sed 's/      <Visibility type="Number">/      <HU>6C70AC13E02414D1A552685A1301D859<\/HU>|      <JDX>023733942B73318EAEAD914E3981EC68BE72519A2F5738F828A6A028C4E1DBAC<\/JDX>|      <Visibility type="Number">/' "$xml.tmp" | tr '|' '\n' > "$xml"
  fi
  rm -f "$xml.tmp"
fi

# Integrate into Service Center
echo "$thePassword" | sudo -S mkdir -p "/Library/Application Support/Native Instruments/Service Center"
echo "$thePassword" | sudo -S chmod 755 "/Library/Application Support/Native Instruments/Service Center"
echo "$thePassword" | sudo -S cp "$xml" "$xmldist"
echo "$thePassword" | sudo -S chmod 755 "$xmldist"

# Set `ContentDir`
echo "$thePassword" | sudo -S rm -f "$plist"
ContentDir=$(dirname "$file" | tr / :)
ContentDir=${ContentDir//::/:}
ContentDir=${ContentDir//::/:}
echo "$thePassword" | sudo -S defaults write "$plist" ContentDir "${ContentDir#:*:}:"

# Obtain rest of parameters from extracted XML
for key in RegKey SNPID Name HU JDX UPID AuthSystem ; do
  val=$(xmllint --xpath "string(//$key)" "$xml")
  if [[ -n "$val" ]]
  then
    if ! echo "$thePassword" | sudo -S defaults write "$plist" "$key" "$val"; then
      echo "Error: Failed to write $key to $plist" >&2
      exit 1
    fi
  fi
done

# Write `ContentVersion`
if [ -n "$cver" ]; then
  if ! echo "$thePassword" | sudo -S defaults write "$plist" ContentVersion "$cver"; then
    echo "Error: Failed to write ContentVersion to $plist" >&2
    exit 1
  fi
fi

# Get `Visibility`
vis=$(xmllint --xpath "string(//ProductSpecific/Visibility)" "$xml")
if ! echo "$thePassword" | sudo -S defaults write "$plist" Visibility -int "$vis"; then
  echo "Error: Failed to write Visibility to $plist" >&2
  exit 1
fi

# Review
cat "$xml"
defaults read "$plist"
rm -f "$xml"

# display dialog box
osascript -e 'tell app "System Events" to display dialog "🎸 Have fun! 🎻" buttons {"OK"} default button 1'