dotfiles/.config/shell/functions.zsh

set-cred ()
{
  export BW_SESSION="$(bw unlock $(secret-tool lookup drive bitwarden) --raw)"
  bw sync
  printf "$(bw get password cli-gitlab)" | secret-tool store --label='Gitlab Token for CLI' 'token' 'GITLAB_TOKEN' 
  printf "$(bw get password cli-ollama-token)" | secret-tool store --label='Ollama Token for CLI' 'token' 'OLLAMA_TOKEN' 
  printf "$(bw get password cli-litellm-token)" | secret-tool store --label='LiteLLM Token for CLI' 'token' 'LITELLM_TOKEN' 

  unset BW_SESSION
}

set-env ()
{
  export GITLAB_TOKEN="$(secret-tool lookup token GITLAB_TOKEN)"
  export OLLAMA_TOKEN="$(secret-tool lookup token OLLAMA_TOKEN)"
  export LITELLM_TOKEN="$(secret-tool lookup token LITELLM_TOKEN)"
}

clear-env ()
{
  unset GITLAB_TOKEN
  unset OLLAMA_TOKEN
  unset LITELLM_TOKEN
}

tmux-new () {
  local dir=${1:-}

  [ -z "${dir}" ] && local dir=$(find $HOME/Documents/gitlab  -mindepth 1 -maxdepth 1 -type d| awk -F/ '{print $(NF-1)"/"$NF " " $0}' | fzf --with-nth=1 | awk '{print $2}')
  [ -z "$dir" ] && return 1


  local name=$(basename "$dir")

  if [[ $name == .* ]]; then
    name=${name#.}
  fi

  tmux has-session -t ${name} > /dev/null && { tmux switch-client -t ${name}; return; }
  tmux new-session -d -s "$name" -c "$dir"

  RIGHT_PANE_WIDTH=60
  BOTTOM_PANE_WIDTH=15

  tmux split-window -h -t "$name:1" -c "$dir"
  tmux resize-pane -t "$name:1.2" -x $RIGHT_PANE_WIDTH

  tmux split-window -v -t "$name:1.1" -c "$dir"
  tmux resize-pane -t "$name:1.1" -y $(($(tmux display -p '#{window_height}') - $BOTTOM_PANE_WIDTH))

  tmux new-window "$name" -n lazygit -c "$dir"

  sleep 1
  tmux send-keys -t "$name:1.1" 'nvim -c "Neotree filesystem reveal left"' C-m

  sleep 0.1
  tmux send-keys -t "$name:1.3" 'set-env; nvim +CodeCompanionChat -c only' C-m

  sleep 1
  tmux send-keys -t "$name:2" 'lg' C-m

  tmux select-window -t "$name:1"
  tmux select-pane -t "$name:1.1"

  tmux switch-client -t "$name"
}

load-profile () {
  ansible-playbook /home/user/.dotfiles/ansible/.config/ansible/local.yml -K
}

decode_base64_url() {
  local len=$((${#1} % 4))
  local result="$1"
  if [ $len -eq 2 ]; then result="$1"'=='
  elif [ $len -eq 3 ]; then result="$1"'=' 
  fi
  echo "$result" | tr '_-' '/+' | openssl enc -d -base64
}

decode-jwtpayload(){
   decode_base64_url $(echo -n $1 | cut -d "." -f '2') | jq .
}
decode-jwtheader(){
   decode_base64_url $(echo -n $1 | cut -d "." -f '1') | jq .
}