dotfiles/emacs/.emacs.d/config/init-aws.el
Jeremy Dormitzer 0973393714 Add vterm AWS profile switch command and keybinding
Added a new function `vterm-aws-switch-profile` to switch AWS profiles in the current vterm buffer and associated it with the keybinding `C-c a` for quick access. This facilitates easier management of AWS profiles within vterm sessions.
2024-07-24 12:08:14 -04:00

91 lines
3.2 KiB
EmacsLisp

;; -*- lexical-binding: t; -*-
;; Manage AWS profiles
(defvar aws-profiles '("default")
"AWS profile names")
(defvar aws-current-profile nil
"Currently active AWS profile")
(defun aws-local-profile ()
(make-local-variable 'aws-current-profile))
(add-hook 'eshell-mode-hook #'aws-local-profile)
(add-hook 'vterm-mode-hook #'aws-local-profile)
(add-hook 'term-mode-hook #'aws-local-profile)
(defun aws-list-profiles ()
(when (not (executable-find "aws"))
(user-error "Unable to find executable \"aws\"!"))
(let ((profile-string (shell-command-to-string "aws configure list-profiles")))
(s-split "\n" profile-string)))
(defun aws-switch-profile (profile)
(interactive (list (completing-read "Profile: " (aws-list-profiles))))
(setenv "AWS_PROFILE" profile)
(when (eq major-mode 'vterm-mode)
(vterm-send-C-e)
(vterm-send-C-u)
(vterm-send-string (format "export AWS_PROFILE=%s\n" profile)))
(setq aws-current-profile profile))
(defun vterm-aws-switch-profile (profile)
"Switch the AWS profile for the current vterm command."
(interactive (list (completing-read "Profile: " (aws-list-profiles))))
(vterm-send-C-a)
(vterm-send-string (format "AWS_PROFILE=%s " profile)))
(with-eval-after-load 'vterm
(define-key vterm-mode-map (kbd "C-c a") #'vterm-aws-switch-profile))
(defun aws-sso-login ()
(interactive)
(let ((buffer "*aws-sso-login*"))
(async-shell-command "AWS_PROFILE=default aws sso login" buffer)
(with-current-buffer buffer
(let ((proc (get-buffer-process (current-buffer))))
(set-process-sentinel
proc
(lambda (proc sentinel)
(when (string-match-p "finished" sentinel)
(with-selected-window (get-buffer-window (process-buffer proc) t)
(quit-window)))))))))
(add-hook 'emacs-startup-hook
(lambda ()
(aws-switch-profile "default")))
;; A command to MFA to AWS
(defun aws-mfa (mfa-token)
(interactive (list
(let ((prompt (if aws-current-profile
(format "MFA code for %s: " aws-current-profile)
"MFA code: ")))
(read-from-minibuffer prompt))))
(let ((proc (start-process "aws-mfa"
"*aws-mfa*"
"aws-mfa"
"--force")))
(set-process-sentinel
proc
(make-success-err-msg-sentinel "*aws-mfa*"
"AWS MFA succeeded"
"AWS MFA failed, check *aws-mfa* buffer for details"))
(process-send-string proc (concat mfa-token "\n"))))
(define-minor-mode aws-modeline-mode
"Displays the current AWS profile in the modeline." :global t
(if aws-modeline-mode
(add-to-list 'mode-line-misc-info
'(:eval (format " AWS:%s " (or aws-current-profile "default"))))
(setq mode-line-misc-info
(delete '(:eval (format " AWS:%s " (or aws-current-profile "default")))
mode-line-misc-info))))
(add-hook 'after-init-hook #'aws-modeline-mode)
;; A dired interface to S3
(use-package s3ed
:commands (s3ed-find-file
s3ed-save-file))
(provide 'init-aws)