Upload desligaRunpod.sh

desligaRunpod.sh

@@ -0,0 +1,168 @@
+#!/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".
+
+set -u -o 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
+
+# Imprime duração como HH:MM:SS
+format_duration() {
+  local sec=${1:-0}
+  local h=$(( sec / 3600 ))
+  local m=$(( (sec % 3600) / 60 ))
+  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
+  printf -v buf '%*s' "$n" ''
+  echo "${buf// /$ch}"
+}
+
+#-----------------------------#
+# Validações de ambiente      #
+#-----------------------------#
+# Carrega SECONDS_ALIVE ou usa padrão.
+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."
+  SECONDS_TOTAL="$MAX_SECONDS"
+fi
+
+# runpodctl disponível?
+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          #
+#-----------------------------#
+echo "==================== REMOVER POD RUNPOD (SEGURO) ===================="
+echo "Pod atual: $POD_ID"
+echo "Ação: REMOVE (apagar/terminar o pod)."
+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 "======================================================================"
+
+#-----------------------------#
+# Barra de progresso + cancel #
+#-----------------------------#
+CANCELLED=0
+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
+
+  NOW=$(date +%s)
+  # Fim da espera?
+  if (( NOW >= END_TS )); then
+    break
+  fi
+
+  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
+  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
+
+#-----------------------------#
+# 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
+  printf "\rRemovendo em %ds... " "$i"
+  sleep 1 || true
+done
+echo
+
+#-----------------------------#
+# Execução da 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.)"
+else
+  RC=$?
+  echo "ERRO: falha ao executar 'runpodctl remove pod \"$POD_ID\"' (código $RC)" >&2
+  exit "$RC"
+fi