#!/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
: > $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 -e "### Dispositivo: $device\n" >> $OUTPUT_FILE
  echo -e "Modelo: \`$model\`" >> $OUTPUT_FILE
  echo -e "Número de série: \`$serial\`" >> $OUTPUT_FILE
  echo -e "Tipo: \`$type\`\n" >> $OUTPUT_FILE
  echo -e "Relatório de saúde do disco:\n" >> $OUTPUT_FILE
  echo -e '```\n' >> $OUTPUT_FILE
  LC_ALL=C.UTF-8 smartctl -H $device >> $OUTPUT_FILE
  echo -e '```\n' >> $OUTPUT_FILE
  echo -e "Informações adicionais e erros críticos do disco:\n" >> $OUTPUT_FILE
  echo -e '```\n' >> $OUTPUT_FILE
  LC_ALL=C.UTF-8 smartctl -a $device >> $OUTPUT_FILE
  echo -e '```\n' >> $OUTPUT_FILE
  echo -e "---\n" >> $OUTPUT_FILE
}

# Detecta todos os discos disponíveis
for device in /dev/sd[a-z]; do
  if [ -b "$device" ]; then
    details=$(LC_ALL=C.UTF-8 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."