#!/usr/bin/perl

=head1 NAME

simple-smart-test.pl - Very simple SMART data test

=head1 DESCRIPTION

Direct SMART data collection and database storage test.

=cut

use strict;
use warnings;
use DBI;
use JSON::XS;

print "=== Simple SMART Test ===\n\n";

# Database connection
my $dsn = "DBI:Pg:dbname=autosmart;host=192.168.2.102;port=5432";
my $dbh = DBI->connect($dsn, "autosmart", "autoSMART2025!", {
    RaiseError => 1,
    AutoCommit => 1,
    PrintError => 0
}) or die "Failed to connect to database: $DBI::errstr\n";

print "✓ Database connected\n";

# Test SMART data collection manually
my @devices = glob('/dev/sd[a-z]');

for my $device (@devices) {
    print "\nTesting device: $device\n";
    
    # Get basic device info
    my $smartctl_output = `smartctl -i $device 2>/dev/null`;
    if ($? != 0) {
        print "  ✗ SMART not available\n";
        next;
    }
    
    # Parse basic info
    my ($model) = $smartctl_output =~ /Device Model:\s+(.+)/;
    my ($serial) = $smartctl_output =~ /Serial Number:\s+(.+)/;
    
    if (!$model || !$serial) {
        print "  ✗ Could not parse model/serial\n";
        next;
    }
    
    print "  Model: $model\n";
    print "  Serial: $serial\n";
    
    # Get SMART attributes
    my $smart_output = `smartctl -A $device 2>/dev/null`;
    my %parameters;
    
    # Parse SMART attributes
    for my $line (split /\n/, $smart_output) {
        if ($line =~ /^\s*(\d+)\s+(\w+)\s+0x[\da-f]+\s+(\d+)\s+(\d+)\s+(\d+)\s+\S+\s+\S+\s+\S+\s+(\d+)/) {
            my ($id, $name, $current, $worst, $threshold, $raw) = ($1, $2, $3, $4, $5, $6);
            $parameters{$name} = $raw;
        }
    }
    
    # Get temperature
    my $temp = $parameters{'Temperature_Celsius'} || 0;
    print "  Temperature: ${temp}°C\n";
    print "  Parameters: " . scalar(keys %parameters) . "\n";
    
    # Check if HDD exists in inventory
    my $sth = $dbh->prepare("SELECT id FROM hdd_inventory WHERE serial_number = ? AND model_name = ?");
    $sth->execute($serial, $model);
    my ($hdd_id) = $sth->fetchrow_array();
    
    if (!$hdd_id) {
        # Create new HDD
        print "  Creating new HDD entry...\n";
        $sth = $dbh->prepare(q{
            INSERT INTO hdd_inventory 
            (serial_number, model_name, current_device_path, current_node_id, status)
            VALUES (?, ?, ?, 'ebony', 'active')
            RETURNING id
        });
        $sth->execute($serial, $model, $device);
        ($hdd_id) = $sth->fetchrow_array();
        print "  ✓ HDD created with ID: $hdd_id\n";
    } else {
        print "  ✓ HDD exists with ID: $hdd_id\n";
        # Update location
        $sth = $dbh->prepare("UPDATE hdd_inventory SET current_device_path = ?, last_seen = NOW() WHERE id = ?");
        $sth->execute($device, $hdd_id);
    }
    
    # Store SMART reading
    print "  Storing SMART reading...\n";
    my $parameters_json = encode_json(\%parameters);
    
    $sth = $dbh->prepare(q{
        INSERT INTO smart_readings
        (hdd_id, serial_number, device_path, node_id, timestamp, 
         collection_ok, temperature, parameters_json, reading_type)
        VALUES (?, ?, ?, 'ebony', NOW(), true, ?, ?, 'full')
    });
    
    $sth->execute($hdd_id, $serial, $device, $temp, $parameters_json);
    print "  ✓ SMART reading stored\n";
}

# Show results
print "\n=== Database Summary ===\n";

my $sth = $dbh->prepare("SELECT COUNT(*) FROM hdd_inventory");
$sth->execute();
my ($hdd_count) = $sth->fetchrow_array();
print "HDD Inventory: $hdd_count drives\n";

$sth = $dbh->prepare("SELECT COUNT(*) FROM smart_readings");
$sth->execute();
my ($reading_count) = $sth->fetchrow_array();
print "SMART Readings: $reading_count readings\n";

# Show latest readings
$sth = $dbh->prepare(q{
    SELECT hi.serial_number, hi.model_name, sr.timestamp, sr.temperature
    FROM smart_readings sr
    JOIN hdd_inventory hi ON sr.hdd_id = hi.id
    ORDER BY sr.timestamp DESC
    LIMIT 5
});
$sth->execute();

print "\nLatest readings:\n";
while (my $row = $sth->fetchrow_hashref()) {
    printf "  %s (%s) - %s - %d°C\n",
           substr($row->{serial_number}, 0, 12),
           substr($row->{model_name}, 0, 20),
           $row->{timestamp},
           $row->{temperature} || 0;
}

$dbh->disconnect();
print "\n=== Test Complete ===\n";