Update desligaRunpod.sh

desligaRunpod.sh

@@ -1,21 +1,18 @@
 #!/usr/bin/env bash
-# removeRunpodSafely.sh — Remove o pod atual do RunPod após um atraso controlado.
-# Requisitos: bash, runpodctl, variáveis RUNPOD_POD_ID (injetada pelo RunPod).
-# Comportamento: aguarda SECONDS_ALIVE (default 7200), mostra barra de progresso,
-# permite cancelamento com Ctrl+C e depois executa "runpodctl remove pod $RUNPOD_POD_ID".
+# removeRunpodSafely.sh — Remove o POD atual do RunPod após um atraso.
+# Requisitos: bash, runpodctl; env RUNPOD_POD_ID presente.
+# Comportamento: aguarda SECONDS_ALIVE (default 7200), barra de progresso,
+# Ctrl+C cancela com segurança, janela final de 5s, e então REMOVE o pod.
 
-set -u -o pipefail
+set -Eeuo pipefail
 umask 077
 
-#-----------------------------#
-# Configuração e utilitários  #
-#-----------------------------#
-DEFAULT_SECONDS=7200             # 2 horas por padrão
-MAX_SECONDS=$((14*24*3600))      # limite de segurança: 14 dias
-PROGRESS_WIDTH=50                # largura da barra de progresso
-GRACE_SECONDS=5                  # janela de segurança final para abortar
+DEFAULT_SECONDS=7200          # 2 horas
+MAX_SECONDS=$((14*24*3600))   # limite de segurança: 14 dias
+PROGRESS_WIDTH=50
+GRACE_SECONDS=5
 
-# Imprime duração como HH:MM:SS
+#--- utils ---#
 format_duration() {
   local sec=${1:-0}
   local h=$(( sec / 3600 ))
@@ -23,8 +20,6 @@ format_duration() {
   local s=$(( sec % 60 ))
   printf "%02d:%02d:%02d" "$h" "$m" "$s"
 }
-
-# Repete um caractere N vezes sem depender de 'seq'
 repeat_char() {
   local n=${1:-0} ch="${2:-#}" buf=
   (( n < 0 )) && n=0
@@ -32,137 +27,88 @@ repeat_char() {
   echo "${buf// /$ch}"
 }
 
-#-----------------------------#
-# Validações de ambiente      #
-#-----------------------------#
-# Carrega SECONDS_ALIVE ou usa padrão.
+#--- parâmetros ---#
 SECONDS_TOTAL="${SECONDS_ALIVE:-$DEFAULT_SECONDS}"
-
-# Verifica se é inteiro não-negativo
 if [[ ! "$SECONDS_TOTAL" =~ ^[0-9]+$ ]]; then
   echo "ERRO: SECONDS_ALIVE precisa ser um inteiro >= 0 (em segundos)." >&2
   exit 2
 fi
-
-# Limite máximo de segurança
 if (( SECONDS_TOTAL > MAX_SECONDS )); then
-  echo "Aviso: SECONDS_ALIVE=$SECONDS_TOTAL excede o limite de segurança."
-  echo "Limitando para $MAX_SECONDS segundos."
+  echo "Aviso: SECONDS_ALIVE=$SECONDS_TOTAL excede o limite de segurança; limitando a $MAX_SECONDS."
   SECONDS_TOTAL="$MAX_SECONDS"
 fi
 
-# runpodctl disponível?
+#--- pré-checagens mínimas ---#
 if ! command -v runpodctl >/dev/null 2>&1; then
   echo "ERRO: 'runpodctl' não encontrado no PATH. Execute dentro de um pod do RunPod." >&2
   exit 3
 fi
-
-# RUNPOD_POD_ID definido?
 POD_ID="${RUNPOD_POD_ID:-}"
 if [[ -z "$POD_ID" ]]; then
   echo "ERRO: RUNPOD_POD_ID não definido. Este script deve rodar dentro de um pod do RunPod." >&2
   exit 4
 fi
 
-# Pod existe/é acessível?
-if ! runpodctl get pod "$POD_ID" >/dev/null 2>&1; then
-  echo "ERRO: Não consegui localizar/ler o pod '$POD_ID' via runpodctl." >&2
-  exit 5
-fi
-
-#-----------------------------#
-# Cabeçalho / Avisos          #
-#-----------------------------#
+#--- header ---#
 echo "==================== REMOVER POD RUNPOD (SEGURO) ===================="
-echo "Pod atual: $POD_ID"
-echo "Ação: REMOVE (apagar/terminar o pod)."
+echo "Pod: $POD_ID"
+echo "Ação: REMOVE (apagar/terminar)"
 echo "Tempo até a remoção: $(format_duration "$SECONDS_TOTAL")"
-echo "Pressione Ctrl+C a QUALQUER momento para cancelar com segurança (nada será removido)."
+echo "Pressione Ctrl+C a qualquer momento para CANCELAR sem remover."
 echo "======================================================================"
 
-#-----------------------------#
-# Barra de progresso + cancel #
-#-----------------------------#
+#--- progresso + cancelamento ---#
 CANCELLED=0
-on_cancel() {
-  CANCELLED=1
-}
+on_cancel(){ CANCELLED=1; }
 trap on_cancel INT TERM
 
 START_TS=$(date +%s)
 END_TS=$(( START_TS + SECONDS_TOTAL ))
 
-# Atualiza a barra de progresso uma vez por segundo
 while :; do
-  # Se cancelado, saia sem remover
-  if (( CANCELLED )); then
-    echo -e "\nOperação cancelada pelo usuário. Nenhuma alteração foi feita."
-    exit 130
-  fi
-
+  (( CANCELLED )) && { echo -e "\nCancelado pelo usuário. Nada foi removido."; exit 130; }
   NOW=$(date +%s)
-  # Fim da espera?
-  if (( NOW >= END_TS )); then
-    break
-  fi
+  (( NOW >= END_TS )) && break
 
   ELAPSED=$(( NOW - START_TS ))
   REMAIN=$(( END_TS - NOW ))
 
-  # Evita divisão por zero
   if (( SECONDS_TOTAL > 0 )); then
     PERCENT=$(( (ELAPSED * 100) / SECONDS_TOTAL ))
     FILLED=$(( (ELAPSED * PROGRESS_WIDTH) / SECONDS_TOTAL ))
   else
-    PERCENT=100
-    FILLED=$PROGRESS_WIDTH
+    PERCENT=100; FILLED=$PROGRESS_WIDTH
   fi
   (( FILLED > PROGRESS_WIDTH )) && FILLED=$PROGRESS_WIDTH
   EMPTY=$(( PROGRESS_WIDTH - FILLED ))
 
   BAR_FILLED=$(repeat_char "$FILLED" "#")
   BAR_EMPTY=$(repeat_char "$EMPTY" ".")
-
   printf "\r[%s%s] %3d%% | faltam %s " \
     "$BAR_FILLED" "$BAR_EMPTY" "$PERCENT" "$(format_duration "$REMAIN")"
 
-  # O sleep pode ser interrompido por Ctrl+C; não é erro aqui
   sleep 1 || true
 done
-
-# Linha nova após a barra
 echo
 
-# Se cancelado exatamente no fim
-if (( CANCELLED )); then
-  echo "Operação cancelada pelo usuário. Nenhuma alteração foi feita."
-  exit 130
-fi
+(( CANCELLED )) && { echo "Cancelado pelo usuário. Nada foi removido."; exit 130; }
 
-#-----------------------------#
-# Janela de segurança final   #
-#-----------------------------#
+#--- janela de segurança final ---#
 echo "Tempo esgotado."
 echo "Janela de segurança: removendo o pod em ${GRACE_SECONDS}s (Ctrl+C para abortar)."
 for (( i=GRACE_SECONDS; i>0; i-- )); do
-  if (( CANCELLED )); then
-    echo -e "\nCancelado durante a janela de segurança. Nenhuma alteração foi feita."
-    exit 130
-  fi
+  (( CANCELLED )) && { echo -e "\nCancelado durante a janela de segurança. Nada foi removido."; exit 130; }
   printf "\rRemovendo em %ds... " "$i"
   sleep 1 || true
 done
 echo
 
-#-----------------------------#
-# Execução da remoção         #
-#-----------------------------#
+#--- remoção ---#
 echo "Executando: runpodctl remove pod \"$POD_ID\""
-# A remoção deve encerrar este processo (seu terminal pode fechar).
 if runpodctl remove pod "$POD_ID"; then
-  echo "Comando de remoção enviado. (Este pod será encerrado.)"
+  echo "Comando enviado. Este pod será encerrado."
 else
-  RC=$?
-  echo "ERRO: falha ao executar 'runpodctl remove pod \"$POD_ID\"' (código $RC)" >&2
-  exit "$RC"
+  rc=$?
+  echo "ERRO: falha ao executar 'runpodctl remove pod \"$POD_ID\"' (código $rc)" >&2
+  exit "$rc"
 fi