#!/bin/bash

# Define the ASCII art maps
large1=''' 
┌─────────────────────────────────────────────────────────────┐
│                                                             │
│                           large1                            │
│                                                             │
│ ┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐ │
│ │                 │ │                 │ │                 │ │
│ │       1         │ │        2        │ │        3        │ │
│ │                 │ │                 │ │                 │ │
│ └─────────────────┘ └─────────────────┘ └─────────────────┘ │
│ ┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐ │
│ │                 │ │                 │ │                 │ │
│ │       4         │ │        5        │ │        6        │ │
│ │                 │ │                 │ │                 │ │
│ └─────────────────┘ └─────────────────┘ └─────────────────┘ │
│ ┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐ │
│ │                 │ │                 │ │                 │ │
│ │       7         │ │        8        │ │        9        │ │
│ │                 │ │                 │ │                 │ │
│ └─────────────────┘ └─────────────────┘ └─────────────────┘ │
└─────────────────────────────────────────────────────────────┘
'''

medium1=''' 
┌─────────────────────────────────────────────────────────────┐
│                                                             │
│ ┌────────────┐ ┌────────────┐ ┌────────────┐ ┌────────────┐ │
│ │     1      │ │     2      │ │      3     │ │     4      │ │
│ └────────────┘ └────────────┘ └────────────┘ └────────────┘ │
│                                                             │
│ ┌────────────┐ ┌────────────┐ ┌────────────┐ ┌────────────┐ │
│ │     5      │ │     6      │ │      7     │ │     8      │ │
│ └────────────┘ └────────────┘ └────────────┘ └────────────┘ │
│                                                             │
│                                                             │
│                                                             │
│                                                 ┌─────────┐ │
│                           medium1               │    9    │ │
│                                                 └─────────┘ │
│                                                 ┌─────────┐ │
│                                                 │   10    │ │
│                                                 └─────────┘ │
│                                                             │
└─────────────────────────────────────────────────────────────┘
'''

medium2=''' 
┌──────────────────────────────────────────────────────────────┐
│                                                              │
│                                                              │
│                                                              │
│                                                              │
│                            medium2                           │
│                                                              │
│                                                              │
│                                                              │
│ ┌────┐┌────┐┌────┐┌────┐┌────┐┌────┐┌────┐┌────┐┌────┐┌────┐ │
│ │    ││    ││    ││    ││    ││    ││    ││    ││    ││    │ │
│ │    ││    ││    ││    ││    ││    ││    ││    ││    ││    │ │
│ │    ││    ││    ││    ││    ││    ││    ││    ││    ││    │ │
│ │ 1  ││ 2  ││ 3  ││ 4  ││ 5  ││ 6  ││ 7  ││ 8  ││ 9  ││ 10 │ │
│ │    ││    ││    ││    ││    ││    ││    ││    ││    ││    │ │
│ │    ││    ││    ││    ││    ││    ││    ││    ││    ││    │ │
│ │    ││    ││    ││    ││    ││    ││    ││    ││    ││    │ │
│ └────┘└────┘└────┘└────┘└────┘└────┘└────┘└────┘└────┘└────┘ │
│                                                              │
└──────────────────────────────────────────────────────────────┘
'''

microGeneric=''' 
 ┌─┐  ┌─┐ 
┌└─┘──└─┘┐
│ 1    2 │
│        │
│        │
│        │
│        │
│        │
│        │
│        │
└────────┘
'''

# Get the hostname
HOSTNAME=$(hostname)

# ASCII art based on hostname
case "$HOSTNAME" in
    "large1")
        echo -e "$large1"
        ;;
    "medium1")
        echo -e "$medium1"
        ;;
    "medium2")
        echo -e "$medium2"
        ;;
    "micro1" | "micro2")
        echo -e "$microGeneric"
        ;;
    *)
        echo -e "No ASCII map defined for this hostname."
        ;;
esac

# Show NVMe Drives
echo -e "\n=== NVMe Drives ==="
sudo nvme list

# Show SATA Drives
echo -e "\n=== SATA Drives ==="
lsblk -o NAME,SIZE,TYPE,MOUNTPOINT,MODEL | grep disk

# Show PCI BDF for NVMe
echo -e "\n=== PCI BDF for NVMe ==="
lspci | grep -i nvme

# Show PCI BDF for SATA
echo -e "\n=== PCI BDF for SATA ==="
lspci | grep -i sata

# Show Drives by ID
echo -e "\n=== Drives by ID ==="
ls -l /dev/disk/by-id