;;; core/cli/env.el -*- lexical-binding: t; -*-

(dispatcher! env
  (let ((env-file (abbreviate-file-name doom-env-file)))
    (pcase (car args)
      ("refresh"
       (doom-reload-env-file 'force))
      ("enable"
       (setenv "DOOMENV" "1")
       (print! (green "Enabling auto-reload of %S" env-file))
       (doom-reload-env-file 'force)
       (print! (green "Done! `doom reload' will now refresh your envvar file.")))
      ("clear"
       (setenv "DOOMENV" nil)
       (unless (file-exists-p env-file)
         (user-error "%S does not exist to be cleared" env-file))
       (delete-file env-file)
       (print! (green "Disabled envvar file by deleting %S" env-file)))
      (_
       (message "No valid subcommand provided. See `doom help env`."))))
  "Manages your envvars file.

  env [SUBCOMMAND]

Available subcommands:

  refresh  Create or regenerate your envvar file
  enable   enable auto-reloading of your envvars file (on `doom refresh`)
  clear    deletes your envvar file (if it exists) and disables auto-reloading

An envvars file (its location is controlled by the `doom-env-file' variable)
will contain a list of environment variables scraped from your shell environment
and loaded when Doom starts (if it exists). This is necessary when Emacs can't
be launched from your shell environment (e.g. on MacOS or certain app launchers
on Linux).

To generate a file, run `doom env refresh`. If you'd like this file to be
auto-reloaded when running `doom refresh`, run `doom env enable` instead (only
needs to be run once).")


;;
;; Helpers

(defvar doom-ignored-env-vars
  '("DBUS_SESSION_BUS_ADDRESS"
    "GPG_AGENT_INFO"
    "SSH_AGENT_PID"
    "SSH_AUTH_SOCK")
  "Environment variables to not save in `doom-env-file'.")

;; Borrows heavily from Spacemacs'`spacemacs//init-spacemacs-env'.
(defun doom-reload-env-file (&optional force-p)
  "Generates `doom-env-file', if it doesn't exist (or FORCE-P is non-nil)."
  (when (or force-p (not (file-exists-p doom-env-file)))
    (with-temp-file doom-env-file
      (message "%s envvars file at %S"
               (if (file-exists-p doom-env-file)
                   "Regenerating"
                 "Generating")
               (abbreviate-file-name doom-env-file))
      (insert
       (concat
        "# -*- mode: dotenv -*-\n"
        "# ---------------------------------------------------------------------------\n"
        "# This file was auto-generated by Doom. It contains all environment variables\n"
        "# scraped from your default shell (excluding variables blacklisted in\n"
        "# doom-ignored-env-vars).\n"
        "#\n"
        "# It is NOT safe to edit this file. Changes will be overwritten next time\n"
        "# that `doom env reload` is executed. Alternatively, create your own env file\n"
        "# in your DOOMDIR and load that with `(load-env-vars FILE)`.\n"
        "#\n"
        "# To auto-regenerate this file when `doom reload` is run, use `doom env enable'\n"
        "# or set DOOMENV=1 in your shell environment/config.\n"
        "# ---------------------------------------------------------------------------\n\n"))
      (let ((env-point (point))
            (switches
             (cond (IS-WINDOWS '("-c"))
                   ;; Execute twice, once in a non-interactive login shell and
                   ;; once in an interactive shell in order to capture all the
                   ;; init files possible.
                   ((or IS-MAC IS-LINUX) '("-lc" "-ic"))))
            (executable (if IS-WINDOWS
                            "set"
                          (executable-find "env"))))
        (dolist (shell-command-switch switches)
          (insert (shell-command-to-string executable)))
        ;; sort the environment variables
        (sort-lines nil env-point (point-max))
        ;; remove adjacent duplicated lines
        (delete-duplicate-lines env-point (point-max) nil t)
        ;; remove ignored environment variables
        (dolist (var doom-ignored-env-vars)
          (flush-lines (concat "^" var "=") env-point (point-max)))))))