#!/usr/bin/perl

=head1 NAME

test-smart-collection.pl - Simple SMART data collection test

=head1 DESCRIPTION

Simplified SMART data collection test for autoSMART deployment verification.

=cut

use strict;

use warnings;

use FindBin qw($Bin);

use lib "$Bin/../lib";

use SmartCollector;

use DBI;

use JSON::XS;

# Configuration from environment

my $config = {

    db_host => $ENV{AUTOSMART_DB_HOST} || '192.168.2.102',

    db_port => $ENV{AUTOSMART_DB_PORT} || '5432',

    db_name => $ENV{AUTOSMART_DB_NAME} || 'autosmart',

    db_user => $ENV{AUTOSMART_DB_USER} || 'autosmart',

    db_pass => $ENV{AUTOSMART_DB_PASS} || 'autoSMART2025!',

    node_id => $ENV{AUTOSMART_NODE_ID} || 'ebony',

    debug => $ENV{AUTOSMART_DEBUG} || 2,

};

print "=== autoSMART SMART Collection Test ===\n\n";

# Test database connection

print "Testing database connection...\n";

my $dsn = "DBI:Pg:dbname=$config->{db_name};host=$config->{db_host};port=$config->{db_port}";

my $dbh = DBI->connect($dsn, $config->{db_user}, $config->{db_pass}, {

    RaiseError => 1,

    AutoCommit => 1,

    PrintError => 0

}) or die "Failed to connect to database: $DBI::errstr\n";

print "✓ Database connection successful\n\n";

# Initialize collector

print "Initializing SMART collector...\n";

my $collector = SmartCollector->new($config);

print "✓ Collector initialized\n\n";

# Discover available drives

print "Discovering storage devices...\n";

my @devices = glob('/dev/sd[a-z]');

for my $device (@devices) {

    print "Found device: $device\n";

    # Test SMART data collection

    print "  Collecting SMART data...\n";

    my $smart_data = $collector->collect_smart_data($device);

    if ($smart_data) {

        print "  ✓ SMART data collected successfully\n";

        print "    Serial: $smart_data->{serial_number}\n";

        print "    Model: $smart_data->{model_name}\n";

        print "    Temperature: $smart_data->{temperature}°C\n";

        print "    Parameters: " . scalar(keys %{$smart_data->{parameters}}) . "\n";

        # Create drive info structure

        my $drive_info = {

            device_path => $device,

            serial_number => $smart_data->{serial_number},

            model_name => $smart_data->{model_name}

        };

        # Store in database

        print "  Storing in database...\n";

        if ($collector->store_smart_data($drive_info, $smart_data)) {

            print "  ✓ Data stored successfully\n";

        } else {

            print "  ✗ Failed to store data\n";

        }

    } else {

        print "  ✗ Failed to collect SMART data\n";

    }

    print "\n";

}

# Check database contents

print "Checking database contents:\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";

$sth = $dbh->prepare("SELECT COUNT(*) FROM hdd_migrations");

$sth->execute();

my ($migration_count) = $sth->fetchrow_array();

print "  HDD Migrations: $migration_count migrations\n";

# Show recent readings

if ($reading_count > 0) {

    print "\nRecent SMART readings:\n";

    $sth = $dbh->prepare(q{

        SELECT hi.serial_number, hi.model_name, sr.timestamp, sr.temperature, sr.reading_type

        FROM smart_readings sr

        JOIN hdd_inventory hi ON sr.hdd_id = hi.id

        ORDER BY sr.timestamp DESC

        LIMIT 5

    });

    $sth->execute();

    while (my $row = $sth->fetchrow_hashref()) {

        printf "  %s (%s) - %s - %d°C - %s\n",

               $row->{serial_number},

               $row->{model_name},

               $row->{timestamp},

               $row->{temperature} || 0,

               $row->{reading_type};

    }

}

$dbh->disconnect();

print "\n=== Test Complete ===\n";