#!/bin/bash

# Arquivo de saída
OUTPUT_FILE="sierti-coletor-saude-hd.txt"

# Verifica se o smartmontools está instalado e instala caso necessário
if ! rpm -q smartmontools > /dev/null; then
  echo "Instalando smartmontools..."
  yum install smartmontools -y
fi

# Limpa arquivo de saída existente ou cria um novo
echo "" > $OUTPUT_FILE

# Função para gerar o relatório de saúde do disco em Markdown
generate_report() {
  local device=$1
  local model=$2
  local serial=$3
  local type=$4

  echo "### Dispositivo: $device" >> $OUTPUT_FILE
  echo "" >> $OUTPUT_FILE
  echo "Modelo: \`$model\`" >> $OUTPUT_FILE
  echo "Número de série: \`$serial\`" >> $OUTPUT_FILE
  echo "Tipo: \`$type\`" >> $OUTPUT_FILE
  echo "" >> $OUTPUT_FILE
  echo "Relatório de saúde do disco:" >> $OUTPUT_FILE
  echo '```' >> $OUTPUT_FILE
  smartctl -H $device >> $OUTPUT_FILE
  echo '```' >> $OUTPUT_FILE
  echo "" >> $OUTPUT_FILE
  echo "Informações adicionais e erros críticos do disco:" >> $OUTPUT_FILE
  echo '```' >> $OUTPUT_FILE
  smartctl -a $device >> $OUTPUT_FILE
  echo '```' >> $OUTPUT_FILE
  echo "---" >> $OUTPUT_FILE
}

# Detecta todos os discos disponíveis
for device in /dev/sd[a-z]; do
  if [ -b "$device" ]; then
    details=$(smartctl -i $device)

    model=$(echo "$details" | grep "Device Model" | awk -F: '{print $2}' | sed 's/^[ \t]*//;s/[ \t]*$//')
    serial=$(echo "$details" | grep "Serial Number" | awk -F: '{print $2}' | sed 's/^[ \t]*//;s/[ \t]*$//')
    type=$(echo "$details" | grep "Transport protocol" | awk -F: '{print $2}' | sed 's/^[ \t]*//;s/[ \t]*$//')

    generate_report $device "$model" "$serial" "$type"
  fi
done

echo "O relatório foi gerado em $OUTPUT_FILE."