#!/usr/bin/env perl

use lib 'lib';
use History::SQLite;
use Environment::SQLite;
use Term::ReadLine;
use Cwd;
use File::HomeDir;
use CustomCommands;
$ENV{TERM} = 'xterm-256color' unless exists $ENV{TERM};


# Initialize command history and environment
my $history = History::SQLite->new(
    db_path => 'pshell_history.db'
);
my $env = Environment::SQLite->new(
    db_path => 'pshell_env.db'
);

# Signal handling setup
$SIG{INT} = sub { print "\n"; };
$SIG{CHLD} = 'IGNORE';

print "Perl Shell (pshell) - Type 'exit' to quit\n";

my $term = Term::ReadLine->new('pshell');
$term->ornaments(0);

# Workaround for Ctrl-R binding with Term::ReadLine::Gnu
$term->add_defun('history-search', sub {
    my ($count, $key) = @_;
    my $search_term = $term->readline('Search history: ');
    return unless defined $search_term && length $search_term;
    
    my @matches = grep { $_ =~ /\Q$search_term\E/i } $history->get_all;
    if (@matches) {
        print "\nMatching commands:\n";
        for my $i (0..$#matches) {
            print sprintf("%2d: %s\n", $i+1, $matches[$i]);
        }
        
        my $choice = $term->readline('\nSelect command to run (number or Enter to cancel): ');
        if ($choice =~ /^\d+$/ && $choice > 0 && $choice <= @matches) {
            $term->addhistory($matches[$choice-1]);
            $term->replace_line($matches[$choice-1]);
            $term->redisplay;
        }
    } else {
        print "\nNo matching commands found\n";
    }
    
    return 0;
});
$term->parse_and_bind('"\C-r": history-search');

# Load previous history
my @history = $history->get_all;
$term->addhistory($_) for @history;

# Load environment variables
my %env_vars = $env->get_all;
$ENV{$_} = $env_vars{$_} for keys %env_vars;

# Load and execute .pshellrc if it exists
my $rc_file = File::HomeDir->my_home . '/.pshellrc';
if (-e $rc_file && -r $rc_file) {
    open my $fh, '<', $rc_file or warn "Could not open $rc_file: $!";
    while (my $line = <$fh>) {
        chomp $line;
        next unless length $line;
        system($line);
    }
    close $fh;
}

while (1) {
    my $cwd = getcwd();
    my $prompt = "pshell:$cwd> ";
    my $command = $term->readline($prompt);
    
    last unless defined $command;
    chomp $command;
    last if $command =~ /^exit$/i;
    
    # Skip empty commands
    next unless length $command;
    
    # Handle command
    my $cmd_status = CustomCommands::handle($command, $env);
    if ($cmd_status >= 0) {  # 1=handled, 2=system command
        # Save valid commands to history
        $history->add($command);
        $term->addhistory($command);
        next;
    }
    
    # Command not found
    print "Command not found: $command\n";
    next;
}