#!/bin/bash

# Couleurs pour le terminal
GREEN='\033[0;32m'
BLUE='\033[0;34m'
NC='\033[0m'

echo -e "${BLUE}[*] Initialisation de l'environnement pour Aiandcc M6...${NC}"

# 1. Mise à jour et installation des paquets de base
echo -e "${BLUE}[*] Installation des outils système et de reverse engineering...${NC}"
sudo apt update
sudo apt install -y \
    build-essential \
    cmake \
    git \
    libusb-1.0-0-dev \
    python3 \
    python3-pip \
    python3-pyusb \
    binwalk \
    sigrok \
    pulseview \
    flashrom

# 2. Récupération de l'outil de dump/flash spécifique aux puces Actions (ATJ2127/2157)
if [ ! -d "tools/actions_flash" ]; then
    echo -e "${BLUE}[*] Clonage de l'utilitaire actions_flash (Ilya Kurdyukov)...${NC}"
    mkdir -p tools
    git clone https://github.com/ilyakurdyukov/actions_flash.git tools/actions_flash
    cd tools/actions_flash && make && cd ../..
else
    echo -e "${GREEN}[✓] L'outil actions_flash est déjà présent.${NC}"
fi

# 3. Configuration des règles UDEV pour éviter le mode Sudo sur l'USB
echo -e "${BLUE}[*] Configuration des règles udev pour le lecteur MP3 (ID Actions Semiconductor)...${NC}"
cat << 'EOF' | sudo tee /etc/udev/rules.d/80-actions.rules
# Mode Flash de base / Stockage de masse
SUBSYSTEMS=="usb", ATTRS{idVendor}=="10d6", ATTRS{idProduct}=="1101", MODE="0666", TAG+="uaccess"
# Mode ADFU (Actions Device Firmware Upgrade) pour le flash brut
SUBSYSTEMS=="usb", ATTRS{idVendor}=="10d6", ATTRS{idProduct}=="10d6", MODE="0666", TAG+="uaccess"
EOF

# Recharger les règles udev sans redémarrer
sudo udevadm control --reload-rules
sudo udevadm trigger

echo -e "${GREEN}[✓] Environnement configuré avec succès !${NC}"
echo -e "${GREEN}[!] Tu peux maintenant exécuter PulseView pour l'analyseur logique et utiliser l'outil dans ./tools/actions_flash/${NC}"