From f2c91727517cdd23530a8359b795b35bf7bcbb02 Mon Sep 17 00:00:00 2001
From: Colin Powell <colin@unbl.ink>
Date: Sat, 13 Jan 2024 13:13:02 -0500
Subject: [PATCH] [zsh] Add sure commands and atuin

---
 zsh/.zshrc | 44 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/zsh/.zshrc b/zsh/.zshrc
index aac670b..bc68348 100644
--- a/zsh/.zshrc
+++ b/zsh/.zshrc
@@ -17,5 +17,49 @@ alias fsync="find . -type f -name '*.sync-conflict*'"
 alias atlas="ssh atlas.local"
 alias pandora="ssh pandora.unbl.ink"
 alias bastion="ssh bastion.unbl.ink"
+alias pull="git pull --rebase --autostash"
+alias asso="aws sso login --profile $argv"
+
+repogo() {
+  export PRIVATE_REPO_TOKEN="$(aws codeartifact get-authorization-token --domain sure --query authorizationToken --output text --profile registries-read)"
+  export CODEARTIFACT_AUTH_TOKEN="$PRIVATE_REPO_TOKEN"
+  poetry config http-basic.codeartifact aws $PRIVATE_REPO_TOKEN
+  aws codeartifact login --namespace sureapp --tool npm --repository npm --domain sure --domain-owner 348777858795 --profile registries-read
+}
+
+kush() {
+  if [[ $1 == "sbx" ]]
+  then
+    aws eks update-kubeconfig --name surepreme-sandbox-main-cluster --profile sure-inc
+  fi
+
+  if [[ $1 == "qa" ]]
+  then
+    aws eks update-kubeconfig --name farmers-qa-main-cluster --profile np-farmers
+  fi 
+
+  if [[ $1 == "prod" ]] 
+  then
+    echo "from platform_core.apps.core_protections.models import *" | pbcopy
+    aws eks update-kubeconfig --name surepreme-production-main-cluster --profile sure-inc
+    kubectl exec -it -n bastion $(kubectl get po -n bastion | grep "surepreme.*bastion" | head -n 1 | awk '{print $1}') -- python manage.py shell
+  fi
+
+  if (( $2 ));
+  then
+    if [[ $2 = "-p" ]]
+    then
+      echo "Import command is on your clipboard"
+      echo "from platform_core.apps.core_protections.models import *" | pbcopy
+      kubectl exec -it $(kubectl get pods | fzf | awk '{print $1}'|head -n 1) -- bash -c "python manage.py shell"
+    else
+      kubectl exec -it $(kubectl get pods | fzf | awk '{print $1}'|head -n 1) -- bash
+    fi
+  else
+    kubectl exec -it $(kubectl get pods | fzf | awk '{print $1}'|head -n 1) -- bash
+  fi
+}
 
 [ -f ~/.fzf.zsh ] && source ~/.fzf.zsh
+
+eval "$(atuin init zsh)"