From bf3b69bedec589c939c335f3d19dd6f61455af47 Mon Sep 17 00:00:00 2001
From: Ryan Orendorff <12442942+ryanorendorff@users.noreply.github.com>
Date: Thu, 22 Dec 2022 21:15:24 -0700
Subject: [PATCH 1/3] math-symbols-input: Add package

---
 .../math-symbols-input/commands-to-plist.nix  |  29 +++++
 .../math-symbols-input/custom-commands.nix    |  14 +++
 .../services/math-symbols-input/default.nix   |  45 ++++++++
 .../generate-commands-plist/default.nix       |   6 +
 .../generate_commands_plist.py                | 106 ++++++++++++++++++
 .../math-symbols-input/package/default.nix    |  34 ++++++
 6 files changed, 234 insertions(+)
 create mode 100644 modules/services/math-symbols-input/commands-to-plist.nix
 create mode 100644 modules/services/math-symbols-input/custom-commands.nix
 create mode 100644 modules/services/math-symbols-input/default.nix
 create mode 100644 modules/services/math-symbols-input/generate-commands-plist/default.nix
 create mode 100644 modules/services/math-symbols-input/generate-commands-plist/generate_commands_plist.py
 create mode 100644 modules/services/math-symbols-input/package/default.nix

diff --git a/modules/services/math-symbols-input/commands-to-plist.nix b/modules/services/math-symbols-input/commands-to-plist.nix
new file mode 100644
index 00000000..3b9f8603
--- /dev/null
+++ b/modules/services/math-symbols-input/commands-to-plist.nix
@@ -0,0 +1,29 @@
+{ stdenv, callPackage, python3Packages, writers }:
+commands:
+
+let
+
+  custom-commmands-file = callPackage ./custom-commands.nix { };
+
+  generate-commands-plist =
+    python3Packages.callPackage ./generate-commands-plist {
+      inherit (writers) writePython3Bin;
+    };
+  math-symbols-input = callPackage ./package { };
+
+in stdenv.mkDerivation {
+  name = "commands-plist";
+
+  phases = [ "installPhase" ];
+
+  installPhase = ''
+    mkdir -p $out
+
+    # Call the generate_commands_plist python cli to convert the default
+    # list of replacements and a custom list of replacements to a plist file.
+    ${generate-commands-plist}/bin/generate_commands_plist \
+    ${math-symbols-input}/Math\ Symbols\ Input.app/Contents/Resources/commands.txt \
+    ${custom-commmands-file commands} \
+    $out/com.mathsymbolsinput.inputmethod.MathSymbolsInput.plist
+  '';
+}
diff --git a/modules/services/math-symbols-input/custom-commands.nix b/modules/services/math-symbols-input/custom-commands.nix
new file mode 100644
index 00000000..3a3afd99
--- /dev/null
+++ b/modules/services/math-symbols-input/custom-commands.nix
@@ -0,0 +1,14 @@
+# Converts an attr list to the text file format used by Math Symbols Input.
+# for example, the following command attribute set
+#
+#    {"alpha" = "α"; "beta" = "β";}
+#
+# would be converted to
+#
+#    \alpha α
+#    \beta β
+#
+{ writeText, lib }:
+commands:
+writeText "custom-commands.txt" (builtins.concatStringsSep "\n"
+  (lib.attrsets.mapAttrsToList (name: value: "\\${name} ${value}") commands))
diff --git a/modules/services/math-symbols-input/default.nix b/modules/services/math-symbols-input/default.nix
new file mode 100644
index 00000000..229f5c4b
--- /dev/null
+++ b/modules/services/math-symbols-input/default.nix
@@ -0,0 +1,45 @@
+{ config, lib, pkgs, ... }:
+
+with lib;
+
+let
+
+  cfg = config.services.math-symbols-input;
+  package = pkgs.callPackage ./package { };
+  commands-to-plist = pkgs.callPackage ./commands-to-plist.nix { };
+  # module = types.submoduleWith { description = "Math Symbols Input"; };
+
+in {
+
+  options.services.math-symbols-input = {
+    enable = mkEnableOption ''
+      LaTeX-style mathematical symbols input method for macOS
+    '';
+
+    symbols = mkOption {
+      type = types.attrs;
+      default = { };
+      example = literalExample ''
+        {
+          "xd" = "😆";
+        }
+      '';
+      description = "Custom symbols to add to Math Symbols Input";
+    };
+  };
+
+  # Note that this only makes sense for homebrew
+  config = mkIf cfg.enable {
+    system.activationScripts.preActivation.text = ''
+      echo "Setting up Math Symbols Input"
+
+      rm /Library/Input\ Methods/Math\ Symbols\ Input.app
+      rm ~/Library/Preferences/com.mathsymbolsinput.inputmethod.MathSymbolsInput.plist
+
+      ln -s ${package}/Math\ Symbols\ Input.app /Library/Input\ Methods/
+      ln -s ${
+        commands-to-plist cfg.symbols
+      }/com.mathsymbolsinput.inputmethod.MathSymbolsInput.plist ~/Library/Preferences/
+    '';
+  };
+}
diff --git a/modules/services/math-symbols-input/generate-commands-plist/default.nix b/modules/services/math-symbols-input/generate-commands-plist/default.nix
new file mode 100644
index 00000000..d977d282
--- /dev/null
+++ b/modules/services/math-symbols-input/generate-commands-plist/default.nix
@@ -0,0 +1,6 @@
+{ lib, writePython3Bin, cytoolz }:
+
+writePython3Bin "generate_commands_plist" {
+  libraries = [ cytoolz ];
+  flakeIgnore = [ "E501" ];
+} (builtins.readFile ./generate_commands_plist.py)
diff --git a/modules/services/math-symbols-input/generate-commands-plist/generate_commands_plist.py b/modules/services/math-symbols-input/generate-commands-plist/generate_commands_plist.py
new file mode 100644
index 00000000..e0973b5b
--- /dev/null
+++ b/modules/services/math-symbols-input/generate-commands-plist/generate_commands_plist.py
@@ -0,0 +1,106 @@
+from functools import reduce
+from pathlib import Path
+from collections.abc import Iterable
+from typing import TypedDict
+
+from operator import ior
+from cytoolz import compose_left
+from cytoolz.curried import filter, map
+
+###############################################################################
+#                                    Types                                    #
+###############################################################################
+
+Commands = dict[str, str]
+
+
+# This is the structure in the Math Symbols Input plist file
+class MathSymbolsPlist(TypedDict):
+    CustomCommands: Commands
+    DefaultCommands: Commands
+    PreferencesTab: str
+
+
+###############################################################################
+#                                Pure Commands                                #
+###############################################################################
+
+
+def valid_line(line: str) -> bool:
+    """Is this line a command line (true) or a comment (false)"""
+    return not line.startswith("#") and not line.isspace()
+
+
+def clean_line(line: str) -> str:
+    """Remove whitespace so commands don't have random whitespace in them"""
+    return line.strip()
+
+
+def command_line_to_commands(line: str) -> Commands:
+    """Convert a single command string line to a Commands"""
+    command, symbol = line.split(" ")
+    return {command: symbol}
+
+
+def combine_commands(commands: Iterable[Commands]) -> Commands:
+    """Combine a bunch of Commands dicts into one
+
+    Equivalent to combining a bunch of [d1, d2, ..., dn] as
+
+    d = {}
+    d |= d1
+    d |= d2
+    ...
+    d |= dn
+    """
+    return reduce(ior, commands, dict())
+
+
+# Operates on a generator, so only reads one line at a time
+def lines_to_commands(lines: Iterable[str]) -> Commands:
+    """Convert an iterable of command strings onto one Commands dict"""
+    return compose_left(
+        filter(valid_line),
+        map(compose_left(clean_line, command_line_to_commands)),
+        combine_commands,
+    )(lines)
+
+
+def commands_to_plist(custom: Commands, default: Commands) -> MathSymbolsPlist:
+    return MathSymbolsPlist(
+        CustomCommands=custom, DefaultCommands=default, PreferencesTab="custom-commands"
+    )
+
+
+###############################################################################
+#                          Stateful/Impure functions                          #
+###############################################################################
+
+
+def path_to_commands(p: Path) -> Commands:
+    with open(p, "r") as f:
+        return lines_to_commands(f)
+
+
+def command_paths_to_plist(custom_path: Path, default_path: Path) -> MathSymbolsPlist:
+    custom = path_to_commands(custom_path)
+    default = path_to_commands(default_path)
+
+    return commands_to_plist(custom, default)
+
+
+###############################################################################
+#                                     Main                                    #
+###############################################################################
+
+if __name__ == "__main__":
+    import sys
+    import plistlib
+
+    locations = sys.argv[1:4]
+    default_path, custom_path, output_path = map(Path, locations)
+
+    commands = command_paths_to_plist(custom_path, default_path)
+
+    with open(output_path, "wb") as f:
+        plistlib.dump(commands, f, fmt=plistlib.FMT_BINARY)
diff --git a/modules/services/math-symbols-input/package/default.nix b/modules/services/math-symbols-input/package/default.nix
new file mode 100644
index 00000000..4576144f
--- /dev/null
+++ b/modules/services/math-symbols-input/package/default.nix
@@ -0,0 +1,34 @@
+{ stdenv, fetchurl, xar, zip, cpio }:
+
+let
+  patch = fetchurl {
+    url =
+      "https://github.com/knrafto/MathSymbolsInput/commit/19352edd3e181d7ff3030460b3a87bbbaef7bccc.patch";
+    sha256 = "sha256-0q0PJE3ZxaIHof5o6epgncMtgrVBWBIm+CZbkfl1Z2E=";
+  };
+
+in stdenv.mkDerivation rec {
+  pname = "MathSymbolsInput";
+  version = "v1.2";
+
+  src = fetchurl {
+    url =
+      "https://github.com/knrafto/${pname}/releases/download/${version}/${pname}.pkg";
+    sha256 = "sha256-/zQ9PEP9qO6fc77PcWeHMJa7gICyJL6UmuoZPNezDVc=";
+  };
+
+  phases = [ "unpackPhase" "buildPhase" "installPhase" ];
+  unpackPhase = "xar -xf $src";
+
+  buildInputs = [ xar zip cpio ];
+
+  buildPhase = ''
+    cat MathSymbolsInput.pkg/Payload  | gunzip -dc | cpio -i
+    patch -u Math\ Symbols\ Input.app/Contents/Resources/commands.txt -i ${patch}
+  '';
+
+  installPhase = ''
+    mkdir -p $out
+    cp -r "Math Symbols Input.app" $out
+  '';
+}

From 55b3754e21169b24c3b2f7c85183df93b209b0db Mon Sep 17 00:00:00 2001
From: Ryan Orendorff <12442942+ryanorendorff@users.noreply.github.com>
Date: Sat, 24 Dec 2022 12:08:55 -0700
Subject: [PATCH 2/3] [math-symbols-input] Symbols change after switch

Previously if you modified the `services.math-symbols.input.symbols`
dictionary, then the changes were not seen because the updates were done
without using the `UserDefaults` toolchain in macOS. Now we use
`defaults` to write the preference file to disk, which causes the cache
used by Math Symbols Input to be refreshed.
---
 .../services/math-symbols-input/default.nix   | 28 ++++++++++++++-----
 1 file changed, 21 insertions(+), 7 deletions(-)

diff --git a/modules/services/math-symbols-input/default.nix b/modules/services/math-symbols-input/default.nix
index 229f5c4b..fe9e4084 100644
--- a/modules/services/math-symbols-input/default.nix
+++ b/modules/services/math-symbols-input/default.nix
@@ -6,9 +6,16 @@ let
 
   cfg = config.services.math-symbols-input;
   package = pkgs.callPackage ./package { };
-  commands-to-plist = pkgs.callPackage ./commands-to-plist.nix { };
-  # module = types.submoduleWith { description = "Math Symbols Input"; };
 
+  plist-base-path = "com.mathsymbolsinput.inputmethod.MathSymbolsInput.plist ";
+  plist-full-path = "~/Library/Preferences/${plist-base-path}";
+
+  commands-to-plist = pkgs.callPackage ./commands-to-plist.nix { };
+  custom-commands-plist = "${commands-to-plist cfg.symbols}/${plist-base-path}";
+
+  command-to-run = pkgs.writeScriptBin "write_defaults" ''
+    defaults write ${plist-full-path} "$(defaults read ${custom-commands-plist} | sed 's|\\\\|\\|g')"
+  '';
 in {
 
   options.services.math-symbols-input = {
@@ -33,13 +40,20 @@ in {
     system.activationScripts.preActivation.text = ''
       echo "Setting up Math Symbols Input"
 
-      rm /Library/Input\ Methods/Math\ Symbols\ Input.app
-      rm ~/Library/Preferences/com.mathsymbolsinput.inputmethod.MathSymbolsInput.plist
+      if [ -d /Library/Input\ Methods/Math\ Symbols\ Input.app ]; then
+        rm /Library/Input\ Methods/Math\ Symbols\ Input.app
+      fi
 
       ln -s ${package}/Math\ Symbols\ Input.app /Library/Input\ Methods/
-      ln -s ${
-        commands-to-plist cfg.symbols
-      }/com.mathsymbolsinput.inputmethod.MathSymbolsInput.plist ~/Library/Preferences/
+
+      if [ ! -f ${plist-full-path} ]; then
+        defaults write ${plist-full-path} CustomCommands ""
+        chmod 600 ${plist-full-path}
+        chown $SUDO_USER:staff ${plist-full-path}
+      fi
+
+      # Force a cache reload by writing
+      su - $SUDO_USER -c "${command-to-run}/bin/write_defaults"
     '';
   };
 }

From 8fbdc6464ef5fcb47bb05be8386cf4b81e0098ea Mon Sep 17 00:00:00 2001
From: Ryan Orendorff <12442942+ryanorendorff@users.noreply.github.com>
Date: Sun, 8 Jan 2023 21:40:44 -0700
Subject: [PATCH 3/3] [MathSymbolsInput] v1.2 -> v1.3

---
 .../math-symbols-input/package/default.nix         | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/modules/services/math-symbols-input/package/default.nix b/modules/services/math-symbols-input/package/default.nix
index 4576144f..6183398b 100644
--- a/modules/services/math-symbols-input/package/default.nix
+++ b/modules/services/math-symbols-input/package/default.nix
@@ -1,20 +1,13 @@
 { stdenv, fetchurl, xar, zip, cpio }:
 
-let
-  patch = fetchurl {
-    url =
-      "https://github.com/knrafto/MathSymbolsInput/commit/19352edd3e181d7ff3030460b3a87bbbaef7bccc.patch";
-    sha256 = "sha256-0q0PJE3ZxaIHof5o6epgncMtgrVBWBIm+CZbkfl1Z2E=";
-  };
-
-in stdenv.mkDerivation rec {
+stdenv.mkDerivation rec {
   pname = "MathSymbolsInput";
-  version = "v1.2";
+  version = "v1.3";
 
   src = fetchurl {
     url =
       "https://github.com/knrafto/${pname}/releases/download/${version}/${pname}.pkg";
-    sha256 = "sha256-/zQ9PEP9qO6fc77PcWeHMJa7gICyJL6UmuoZPNezDVc=";
+    sha256 = "sha256-+s+TFoAv2rWl3gOHp8LLUZm+OTMDxgZtK0vmHQDJzFU=";
   };
 
   phases = [ "unpackPhase" "buildPhase" "installPhase" ];
@@ -24,7 +17,6 @@ in stdenv.mkDerivation rec {
 
   buildPhase = ''
     cat MathSymbolsInput.pkg/Payload  | gunzip -dc | cpio -i
-    patch -u Math\ Symbols\ Input.app/Contents/Resources/commands.txt -i ${patch}
   '';
 
   installPhase = ''