1. 1
2. 2
3. 3
4. 4
5. 5
6. 6
7. 7
8. 8
9. 9
10. 10
11. 11
12. 12
13. 13
14. 14
15. 15
16. 16
17. 17
18. 18
19. 19
20. 20
21. 21
22. 22
23. 23
24. 24
25. 25
26. 26
27. 27
28. 28
29. 29
30. 30
31. 31
32. 32
33. 33
34. 34
35. 35
36. 36
37. 37
38. 38
39. 39
40. 40
41. 41
42. 42
43. 43
44. 44
45. 45
46. 46
47. 47
48. 48
49. 49
50. 50
51. 51
52. 52
53. 53
54. 54
55. 55
56. 56
57. 57
58. 58
59. 59
60. 60
61. 61
62. 62
63. 63
64. 64
65. 65
66. 66
67. 67
68. 68
69. 69
70. 70
71. 71
72. 72
73. 73
74. 74
75. 75
76. 76
77. 77
78. 78
79. 79
80. 80
81. 81
82. 82
83. 83
84. 84
85. 85
86. 86
87. 87
88. 88
89. 89
90. 90
91. 91
92. 92
93. 93
94. 94
95. 95
96. 96
97. 97
98. 98
99. 99
100. 100
101. 101
102. 102
103. 103
104. 104
105. 105
106. 106
107. 107
108. 108
109. 109
110. 110
111. 111
112. 112
113. 113
114. 114
115. 115
116. 116
117. 117
118. 118
119. 119
120. 120
121. 121
122. 122
123. 123
124. 124
125. 125
126. 126
127. 127
128. 128
129. 129
130. 130
131. 131
132. 132
133. 133
134. 134
135. 135
136. 136
137. 137
138. 138
139. 139
140. 140
141. 141
142. 142
143. 143
144. 144
145. 145
146. 146
147. 147
148. 148
149. 149
150. 150
151. 151
152. 152
153. 153
154. 154
155. 155
156. 156
157. 157
158. 158
159. 159
160. 160
161. 161
162. 162
163. 163

# Copyright 2023 Shota FUJI <pockawoooh@gmail.com>
#
# Permission to use, copy, modify, and/or distribute this software for any
# purpose with or without fee is hereby granted.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
# REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
# AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
# LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
# OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
# PERFORMANCE OF THIS SOFTWARE.
#
# SPDX-License-Identifier: 0BSD

{ config
, lib
, pkgs
, ...
}:
let
  cfg = config.features.basics.zsh;
in
{
  options = {
    features.basics.zsh.theme = {
      text = lib.mkOption {
        type = lib.types.nonEmptyStr;
        default = "%f";
      };

      vi.insert = lib.mkOption {
        type = lib.types.nonEmptyStr;
        default = "%{$fg[blue]%}";
      };

      vi.normal = lib.mkOption {
        type = lib.types.nonEmptyStr;
        default = "%{$fg[green]%}";
      };

      vcs = {
        info = lib.mkOption {
          type = lib.types.nonEmptyStr;
          default = "%{$fg[white]%}";
        };

        staged = lib.mkOption {
          type = lib.types.nonEmptyStr;
          default = "%{$fg[green]%}";
        };

        unstaged = lib.mkOption {
          type = lib.types.nonEmptyStr;
          default = "%{$fg[red]%}";
        };
      };

      symbol = lib.mkOption {
        type = lib.types.nonEmptyStr;
        default = "%F{8}";
      };
    };
  };

  config = {
    programs = {
      zsh = {
        enable = true;

        # The default base keymap to use.
        defaultKeymap = "viins";

        # Enable zsh completion. Don’t forget to add
        enableCompletion = true;

        # Options related to commands history configuration.
        history = {
          # Do not enter command lines into the history list if they are duplicates of the previous event.
          ignoreDups = true;

          # Save timestamp into the history file.
          extended = true;

          # Number of history lines to keep.
          size = 1000;

          # Share command history between zsh sessions.
          share = false;
        };

        sessionVariables = {
          # Insert space between completed string and ampersand or pipe
          ZLE_SPACE_SUFFIX_CHARS = "&|";
        };

        initExtra = with cfg.theme; ''
          # Activate colors module in order to colourise prompt
          autoload -Uz colors
          colors

          # Branch character (for readability)
          CH_BRANCH=$'\ue0a0'

          function custom-prompt() {
            PROMPT_PREFIX=""

            if [[ -n $SSH_CLIENT ]] || [[ -n $SSH_TTY ]]; then
              PROMPT_PREFIX="''${PROMPT_PREFIX}[SSH]"
            fi

            if [[ -n $IN_NIX_SHELL ]]; then
              PROMPT_PREFIX="''${PROMPT_PREFIX}[Nix]"
            fi

            if [[ -n $PROMPT_PREFIX ]]; then
              PROMPT_PREFIX="''${PROMPT_PREFIX} "
            fi

            echo -e "

          %f%k%b%F{cyan}''${PROMPT_PREFIX}%f''${1}%1d ''${vcs_info_msg_0_}%k%f%b
          ${symbol}%# ${text}"
          }

          # VCS
          autoload -Uz vcs_info

          precmd () { vcs_info }

          zstyle ":vcs_info:git:*" check-for-changes true
          zstyle ":vcs_info:git:*" stagedstr "${vcs.staged}*"
          zstyle ":vcs_info:git:*" unstagedstr "${vcs.unstaged}*"
          zstyle ":vcs_info:*" formats "${vcs.info}''${CH_BRANCH} %b%c%u${text}"
          zstyle ":vcs_info:*" actionformats "[%b|%a]"

          zstyle ":completion:*" ignored-patterns "dpipe|exiv2"

          function zle-line-init zle-keymap-select {
            case $KEYMAP in
              vicmd)
                PROMPT=$(custom-prompt "${vi.normal}")
                ;;
              main|viins)
                PROMPT=$(custom-prompt "${vi.insert}")
                ;;
            esac
            zle reset-prompt
          }

          zle -N zle-line-init
          zle -N zle-keymap-select

          # Aliases nix commands to use Zsh rather than bash
          if command -v nix-your-shell > /dev/null; then
            nix-your-shell zsh | source /dev/stdin
          fi
        '';
      };
    };
  };
}