powerline-go: add zsh integration

modules/programs/powerline-go.nix

@@ -106,8 +106,9 @@ in {
     };
   };
 
-  config = mkIf (cfg.enable && config.programs.bash.enable) {
-    programs.bash.initExtra = ''
+  config = {
+    programs.bash.initExtra =
+      mkIf (cfg.enable && config.programs.bash.enable) ''
         function _update_ps1() {
           local old_exit_status=$?
           PS1="$(${pkgs.powerline-go}/bin/powerline-go -error $old_exit_status ${commandLineArguments})"
@@ -119,5 +120,25 @@ in {
           PROMPT_COMMAND="_update_ps1;$PROMPT_COMMAND"
         fi
       '';
+
+    programs.zsh.initExtra = mkIf (cfg.enable && config.programs.zsh.enable) ''
+      function powerline_precmd() {
+        PS1="$(${pkgs.powerline-go}/bin/powerline-go -error $? -shell zsh ${commandLineArguments})"
+        ${cfg.extraUpdatePS1}
+      }
+
+      function install_powerline_precmd() {
+        for s in "$\{precmd_functions[@]}"; do
+          if [ "$s" = "powerline_precmd" ]; then
+            return
+          fi
+        done
+        precmd_functions+=(powerline_precmd)
+      }
+
+      if [ "$TERM" != "linux" ]; then
+        install_powerline_precmd
+      fi
+    '';
   };
 }

tests/modules/programs/powerline-go/default.nix

@@ -1 +1,4 @@
-{ powerline-go-standard = ./standard.nix; }
+{
+  powerline-go-bash = ./bash.nix;
+  powerline-go-zsh = ./zsh.nix;
+}

tests/modules/programs/powerline-go/zsh.nix

@@ -0,0 +1,31 @@
+{ config, lib, pkgs, ... }:
+
+with lib;
+
+{
+  config = {
+    programs = {
+      zsh.enable = true;
+
+      powerline-go = {
+        enable = true;
+        newline = true;
+        modules = [ "nix-shell" ];
+        pathAliases = { "\\~/project/foo" = "prj-foo"; };
+        settings = {
+          ignore-repos = [ "/home/me/project1" "/home/me/project2" ];
+        };
+      };
+    };
+
+    nixpkgs.overlays =
+      [ (self: super: { zsh = pkgs.writeScriptBin "dummy-zsh" ""; }) ];
+
+    nmt.script = ''
+      assertFileExists home-files/.zshrc
+      assertFileContains \
+        home-files/.zshrc \
+        '/bin/powerline-go -error $? -shell zsh -modules nix-shell -newline -path-aliases \~/project/foo=prj-foo -ignore-repos /home/me/project1,/home/me/project2'
+    '';
+  };
+}