#!/bin/bash

[ "${EUID:-$(id -u)}" -eq 0 ] || { >&2 echo "Please run this as root."; exit 1; }

ZSHENV_PATH="/etc/zshenv"

case "$(realpath $ZSHENV_PATH)" in
    *"/nix/"* )
        ZSHENV_PATH="/etc/zshenv.local"
        ;;
esac 

echo """#!/bin/zsh

export ZDOTDIR=\"\${XDG_CONFIG_HOME:-\$HOME/.config}/zsh\"""" > /etc/__zshenv && \
mv -i /etc/__zshenv $ZSHENV_PATH
ln -si $ZSHENV_PATH /etc/zsh/zshenv