X-Git-Url: https://git.sev.monster/~sev/dotfiles.git/blobdiff_plain/b31a3fb1d2a23158add217e47783cce2f887e0e7..dd6dcbd457b22b4d7398329c1d2457781b3d8de4:/etc/zsh/.zshenv

diff --git a/etc/zsh/.zshenv b/etc/zsh/.zshenv
index f14590e..69ac9bc 100644
--- a/etc/zsh/.zshenv
+++ b/etc/zsh/.zshenv
@@ -1,38 +1,78 @@
-### critical exports, the rest are in zprofile
-export CHARSET=${CHARSET:-UTF-8}
-export LANG=${LANG:-en_US.UTF-8}
-## alternative home for pulling in bin & config, used for zsu
-[[ -v _sev_home ]] || export _sev_home=$HOME
+### functions
+## cleanup
+# XXX: only call after relevant vars have been set up, defined early so that
+#      below code can utilize it after they do so
+function _sev_zcleanup {
+    local x p y
 
-## fix broken term
-# NOTE: we do this here instead of .zshrc since we might echo stuff
-if [[ -t 1 ]] { # only if stdout is tty
-    [[ ! -v TERM ]] && export TERM=xterm-256color >/dev/null 2>&1
-    if [[ $#terminfo -eq 0 ]] {
-        _oldterm=$TERM
-        export TERM=xterm >/dev/null 2>&1
-        [[ -o interactive ]] &&
-          print -P "%F{red}!!! Can't find terminfo for $_oldterm, using $TERM%f"
-        unset _oldterm
+    # gpg forwarding
+    if [[ -d $_sev_gpg_forward_dir && ( -z $1 || $1 == 'gpg-forward' ) ]] {
+        # clean up forward dirs if its session is dead or we ask for it
+        find $_sev_gpg_forward_dir -mindepth 1 -maxdepth 1 -type d |
+          while {read -r x} {
+            # NOTE: the only way we can get here is if we have not been
+            #       forwarded before, if the user asks for it, or during
+            #       logout. if our own pid already has a dir, it is most likely
+            #       stale, the user wants it removed, or something is very
+            #       broken—in all 3 of these cases the best choice is remove it.
+            p=$(basename $x)
+            if {[[ -v _sev_gpg_forward_clean || $$ == $p ]] ||
+                    ! kill -0 $p 2>/dev/null} {
+                find $x -mindepth 1 -maxdepth 1 | while {read -r y} {
+                    # XXX: real dirs will stop unlink, consider it a feature
+                    unlink $y
+                }
+                # don't force in case something important is still there
+                rmdir $x
+            }
+        }
+        # reset GNUPGHOME if we removed our own dir
+        if [[ $GNUPGHOME =~ '/.ssh_forward/\d+/*$' && ! -d $GNUPGHOME ]]
+            GNUPGHOME=${GNUPGHOME%$MATCH}
+    }
+
+    # custom tmp
+    # NOTE: _sev_tmp is not unset so session dirs will not be recreated if
+    #       called during runtime; unset _sev_tmp and re-source to fix
+    # NOTE: XDG dirs that use our tmp are not unset here, they are in zlogout
+    #       after this function is called
+    if [[ -d $_sev_tmp && ( -z $1 || $1 == 'tmp' ) ]] {
+        # clean up tmp dirs if its session is dead or we ask for it
+        find $_sev_tmp -mindepth 1 -maxdepth 1 -name '.session.*' -type d |
+          while {read -r x} {
+            # NOTE: same rationale as above
+            p=${$(basename $x)#.session.}
+            if {[[ -v _sev_tmp_clean || $$ == $p ]] ||
+                    ! kill -0 $p 2>/dev/null} {
+                rm -rf $x
+            }
+        }
     }
 }
 
-## path
-# NOTE: we utilize the fact that unique arrays keep the first occurrence and
-#       remove any further occurences to capture elements from the old PATH
-#       that we did not anticipate and shift them to the front, since they are
-#       probably important to the system
-if [[ ! -v _sev_setup_path || -o login ]] {
-    typeset -U path fpath
+function _sev_setpath {
     # add as many generic paths as possible to keep the order we want
-    # NOTE: /usr/{local,pkg,games} are unix/bsdisms
-    # XXX: PREFIX not validated, non-posix but Termux uses it, maybe others
+    # NOTE: tied arrays path and fpath already exist, but are not unique (-U);
+    #       we utilize the fact that unique arrays keep the first occurrence
+    #       and remove any further occurences to check for elements from the
+    #       old PATH that we did not anticipate and shift them to the front,
+    #       since they are probably important to the system
+    typeset -gU path fpath
+    local -a syspath=("$path[@]")
+    # NOTE: /usr/{pkg,games} are unix/bsdisms
+    # NOTE: some systems (esp. research machines) may have multiple versions of
+    #       packages installed in /opt/[pkg]/[ver]/bin or other dirs, managed
+    #       with something like Environment Modules. this code does not account
+    #       for this type of usage and will add all valid paths. any undesired
+    #       paths can be removed using .zshenv.local.
+    # XXX: PREFIX not validated, non-POSIX but Termux uses it, maybe others
     # XXX: XDG specifies ~/.local/bin as the only user-writable dir for
     #      executables, but we specify more; technically this is against spec
-    syspath=("$path[@]")
-    path=({{${_sev_home:-~},~}{/.local,},{$PREFIX,}{,/opt,/usr{,/local,/pkg}}}/{s,}bin
-          /usr/{X11R{7,6}/bin,games})
-    ((len=$#path))
+    path=({{${_sev_home:-~},~}{/.local,},{$PREFIX,}{,/usr{,/local,/pkg},/opt{,/*{/*,}}}}/{s,}bin(N)
+          {$PREFIX,}/usr/{X11R{7,6}/bin,games}
+          # emulate Arch Linux flatpak-bindir.sh for use on other systems
+          {${XDG_DATA_HOME:-~/.local/share},{$PREFIX,}/var/lib}/flatpak/exports/bin)
+    local -i i len=$#path
     path=("$path[@]" "$syspath[@]")
     # remove bad paths
     for (( i = 1; i <= $#path; i++ )) {
@@ -45,20 +85,220 @@ if [[ ! -v _sev_setup_path || -o login ]] {
     }
     # shift valid system paths to the front if there are any left
     ((len > 0 && len < $#path)) && path=("${(@)path[len + 1, -1]}" "${(@)path[1, len]}")
-    unset syspath len i
     # include our zsh dir in fpath. unlike above, we always prefer our paths
     fpath=({${ZDOTDIR:-~/.zsh},{${_sev_home:-~},~}/.zsh}/functions/**/*(/N) "$fpath[@]")
+    # remove bad paths
+    for (( i = 1; i <= $#fpath; i++ )) {
+        if [[ ! -d $fpath[$i] ]] {
+            fpath[$i]=()
+            ((i--))
+            continue
+        }
+    }
     # FPATH is not exported by default
     export FPATH
+    # un-unique system arrays for consistency
     typeset +U path fpath
-    export _sev_setup_path=
+}
+
+### common exports
+export CHARSET=${CHARSET:-UTF-8}
+export LANG=${LANG:-en_US.UTF-8}
+
+## alternative home for pulling in bin & config, used for zsu
+[[ -v _sev_home ]] || export _sev_home=$HOME
+
+## fix broken term
+# NOTE: we do this here instead of .zshrc since we might print stuff
+if [[ -t 1 ]] { # only if stdout is tty
+    [[ ! -v TERM ]] && export TERM=xterm-256color >/dev/null 2>&1
+    if [[ $#terminfo -eq 0 ]] {
+        _oldterm=$TERM
+        export TERM=xterm >/dev/null 2>&1
+        [[ -o interactive ]] &&
+          print -P "%F{red}!!! Can't find terminfo for $_oldterm, using $TERM%f"
+        unset _oldterm
+    }
+}
+
+## path
+if [[ ! -v _sev_setup_path || -o login ]] {
+    _sev_setpath
+    # NOTE: do not set _sev_setup_path, it is set in zprofile
+}
+
+### home dir setup & exports
+# XXX: traditionally, zshenv should just contain exports, and not touch the
+#      filesystem. however, some system profile scripts that are sourced in the
+#      system zprofile may attempt to do things that rely on some of these
+#      vars. for example, `flatpak-bindir.sh` in the Arch Linux flatpak package
+#      references $XDG_DATA_HOME with no fallback. since we do special handling
+#      for these vars before we export them, we're forced to do it all here
+#      instead of zprofile.
+
+## xdg local dir
+# NOTE: need this for tmp, so confirm it exists.
+# XXX: perms are not specified for XDG dirs except runtime, but I think 760
+#      makes the most sense. shouldn't break anything since no one else should
+#      be poking around in our dir.
+[[ -e ~/.local ]] || mkdir -m760 ~/.local
+
+## tmp
+# NOTE: specs say that POSIX tmp and XDG runtime directories should exist
+#       until the last session is logged out (POSIX can exist for longer).
+#       since we can't reliably keep track of sessions in a cross-platform
+#       manner, the current implementation should use a separate directory per
+#       toplevel session (i.e. SHLVL=1). this should placate most applications,
+#       though it is not expressly spec compliant. this may also cause problems
+#       with disowned applications that still try to use the directories after
+#       the toplevel shell has already logged out and the dirs removed, but the
+#       chances of that are slim.
+if [[ ! -v _sev_tmp ]] {
+    _sev_tmp=~/.local/tmp
+    # create personal TMPDIR under system tmp
+    # NOTE: under proot with uid remapping, we can reuse old tmp, without
+    #       worrying about permission issues; intended for proot under termux.
+    # XXX: _sev_proot_old_user only works if /tmp is shared!
+    t=${TMPDIR:-${TEMPDIR:-${TEMP:-${TMP:-${${TMPPREFIX%/zsh}:-/tmp}}}}}/.home-${_sev_proot_old_username:-$LOGNAME}
+    [[ -e $t ]] || mkdir -m700 $t 2>/dev/null
+    if [[ ! -d $t ]] {
+        # fallback TMPDIR to bare local directory or existing softlink
+        [[ -o interactive ]] &&
+          print -P "%F{orange}*** Can't create tmp dir $t, using $_sev_tmp%f"
+        [[ -h $_sev_tmp && ! -d _sev_tmp ]] && unlink $_sev_tmp 2>/dev/null
+        [[ ! -e $_sev_tmp ]] && mkdir -m700 $_sev_tmp 2>/dev/null
+        if [[ ! -d $_sev_tmp ]] {
+            _sev_tmp=${$(mktemp 2>/dev/null):/tmp}
+            [[ -o interactive ]] &&
+              print -P "%F{red}!!! Can't create tmp dir, using $_sev_tmp%f"
+        }
+    } elif [[ -e $_sev_tmp && ! -h $_sev_tmp ]] {
+        # non-softlink node is on our local dir
+        [[ -o interactive ]] &&
+          print -P "%F{orange}*** $_sev_tmp exists, can't link to tmp dir $t, ignoring it%f"
+        _sev_tmp=$t
+    } else {
+        if [[ -h $_sev_tmp ]] {
+            [[ -o interactive && $_sev_tmp:P != $t:P ]] &&
+              print -P "%F{orange}*** $_sev_tmp links to ${_sev_tmp:P} and not ${t:P}, unlinking it%f"
+            # NOTE: ln -f doesn't seem to work reliably with softlink
+            #       directories, so explicitly remove the target if it exists
+            # XXX: potential race condition
+            # TODO: handle cleanup of old dir if it doesn't match?
+            unlink $_sev_tmp 2>/dev/null
+        }
+        # link local dir to tmp dir
+        ln -s $t $_sev_tmp 2>/dev/null
+    }
+    # ensure dir is clean
+    _sev_zcleanup tmp
+    # finally create our subdir for this session
+    t=$_sev_tmp/.session.$$
+    if ! mkdir -m700 $t 2>/dev/null; then
+        [[ -o interactive ]] &&
+          print -P "%F{red}!!! Can't create session tmp subdir $t, using $_sev_tmp%f"
+        t=$_sev_tmp
+    fi
+    export _sev_tmp TMPDIR=$t TEMPDIR=$t TEMP=$t TMP=$t TMPPREFIX=$t/zsh
+    unset t
+}
+
+## xdg
+if [[ ! -v _sev_setup_xdg ]] {
+    ## merge with any existing dirs and remove duplicates using unique arrays
+    # NOTE: we are accepting whatever value might be set for CONFIG and DATA;
+    #       if it wasn't set, we just use default and leave it unset
+    # NOTE: include and then remove CONFIG_HOME and DATA_HOME to ensure they
+    #       are not present in the array if it was added before we got to it
+
+    # source user dirs before other vars; technically it is against spec to
+    # include any of the below dirs there, but you never know what crazy shit
+    # people will do. I rather handle them sanely with our own code than let
+    # them override after the fact.
+    [[ -f $XDG_CONFIG_HOME/user-dirs.dirs ]] &&
+      emulate sh -c "source $XDG_CONFIG_HOME/user-dirs.dirs"
+
+    typeset -UT XDG_DATA_DIRS xdg_data_dirs
+    [[ -v XDG_DATA_HOME ]] && export XDG_DATA_HOME
+    [[ -e ${XDG_DATA_HOME:-~/.local/share} ]] ||
+      mkdir -m760 ${XDG_DATA_HOME:-~/.local/share}
+    xdg_data_dirs=($XDG_DATA_HOME ${XDG_DATA_DIRS:+${xdg_data_dirs%%/}}
+      /{usr{,/local,/pkg},opt{,/*{/*,}}}/share(N))
+    xdg_data_dirs=($xdg_data_dirs(/N))
+    export XDG_DATA_DIRS
+
+    typeset -UT XDG_CONFIG_DIRS xdg_config_dirs
+    [[ -v XDG_CONFIG_HOME ]] && export XDG_CONFIG_HOME
+    [[ -e ${XDG_CONFIG_HOME:-~/.config} ]] ||
+      mkdir -m760 ${XDG_CONFIG_HOME:-~/.config}
+    # I am of the belief .local should follow FHS /usr/local...
+    [[ -e ~/.local/etc ]] || ln -s ${XDG_CONFIG_HOME:-~/.config} ~/.local/etc
+    xdg_config_dirs=($XDG_CONFIG_HOME ${XDG_CONFIG_DIRS:+${xdg_config_dirs%%/}}
+      {,/usr{,/local,/pkg},opt{,/*{/*,}}}/etc/xdg(N))
+    xdg_config_dirs=($xdg_config_dirs(/N))
+    export XDG_CONFIG_DIRS
+
+    [[ -v XDG_STATE_HOME ]] && export XDG_STATE_HOME
+    [[ -e ${XDG_STATE_HOME:-~/.local/state} ]] ||
+      mkdir -m760 ${XDG_STATE_HOME:-~/.local/state}
+
+    if [[ -v XDG_CACHE_HOME ]] {
+        export XDG_CACHE_HOME
+    } else {
+        export XDG_CACHE_HOME=$_sev_tmp/.xdg.cache
+    }
+    [[ -e ${XDG_CACHE_HOME:-~/.cache} ]] ||
+      mkdir -m700 ${XDG_CACHE_HOME:-~/.cache}
+
+    # NOTE: this can be set by systemd or other pre-shell supervisor, and if
+    #       any services were started such as pipewire, we need to use the
+    #       existing runtime dir to preserve runtime state
+    if [[ -v XDG_RUNTIME_DIR ]] {
+        export XDG_RUNTIME_DIR
+    } else {
+        # make runtime dir in our session-specific tmpdir
+        export XDG_RUNTIME_DIR=$TMPDIR/.xdg.runtime
+        # same as in tmpdir creation, ensure dir doesn't exist
+        if [[ -h $XDG_RUNTIME_DIR ]] {
+            unlink $XDG_RUNTIME_DIR 2>/dev/null
+        } elif [[ -e $XDG_RUNTIME_DIR ]] {
+            rm -rf $XDG_RUNTIME_DIR 2>/dev/null
+        }
+    }
+    [[ ! -e $XDG_RUNTIME_DIR ]] &&
+      mkdir -m700 $XDG_RUNTIME_DIR 2>/dev/null
+
+    export _sev_setup_xdg=
+}
+
+### app setup & exports
+# NOTE: we set these up here since some scripts might need them
+## gpg home
+if [[ ! -v GNUPGHOME ]] {
+    export GNUPGHOME=${XDG_CONFIG_HOME:-~/.config}/gnupg
+    # move existing gnupg dir to our new home
+    if [[ -d ~/.gnupg && ! -d $GNUPGHOME ]] {
+        mv ~/.gnupg $GNUPGHOME
+    }
+}
+
+## perl local lib
+[[ ! -v PERL_LOCAL_LIB_ROOT && -v commands[perl] &&
+   -d $XDG_DATA_HOME/perl5/lib/perl5 ]] &&
+  eval $(perl -I$XDG_DATA_HOME/perl5/lib/perl5 \
+              -Mlocal::lib=$XDG_DATA_HOME/perl5 2>/dev/null)
+
+## go
+if [[ -v commands[go] ]] {
+    [[ ! -v GOPATH ]] && export GOPATH=${XDG_DATA_HOME:-~/.local/share}/go:~/go
+    [[ ! -v GOBIN  ]] && export GOBIN=~/.local/bin
 }
 
 ### load zshenv site-specific
 autoload -Uz load-site-dotfile
 load-site-dotfile zshenv
 
-### source .zprofile early for non-login shells to fix issues
+### source .zprofile early for non-login shells that should be
 if [[ ! -v _sev_first_display && ( -v DISPLAY || -v WAYLAND_DISPLAY ) ]] {
     # most graphical login/session managers will spawn the user's shell as a
     # parent of all child processes for that session. however, if the parent shell
@@ -82,6 +322,6 @@ if [[ ! -v _sev_first_display && ( -v DISPLAY || -v WAYLAND_DISPLAY ) ]] {
     export _sev_first_display=
     [[ ! -o login ]] && source ${ZDOTDIR:-~}/.zprofile
 } elif [[ ${+TERMUX_VERSION} -eq 0 && ! -o login && $SHLVL -eq 1 ]] {
-    ## Termux first process isn't login shell, so source early
+    # Termux first process isn't login shell, so source early
     source ${ZDOTDIR:-~}/.zprofile
 }