diff options
author | Tom Feist <shabble@metavore.org> | 2011-02-19 15:06:28 +0000 |
---|---|---|
committer | Tom Feist <shabble@metavore.org> | 2011-02-19 15:06:28 +0000 |
commit | ca08f1290f9798c3c65c1c6c924bd49546edd863 (patch) | |
tree | 971982c8943ea0cbe1ea1bf398ba577d2eedb77a /feature-tests | |
parent | initial commit, probably about to get blasted. (diff) | |
download | irssi-scripts-ca08f1290f9798c3c65c1c6c924bd49546edd863.tar.gz irssi-scripts-ca08f1290f9798c3c65c1c6c924bd49546edd863.zip |
semi-working wrapper for irssi allowing command/keystroke injection. Still need
to work on the output format though.
Diffstat (limited to 'feature-tests')
-rwxr-xr-x | feature-tests/auto-testing.pl | 184 |
1 files changed, 132 insertions, 52 deletions
diff --git a/feature-tests/auto-testing.pl b/feature-tests/auto-testing.pl index db45226..5ed9d70 100755 --- a/feature-tests/auto-testing.pl +++ b/feature-tests/auto-testing.pl @@ -1,74 +1,154 @@ #!/usr/bin/env perl -use strict; use warnings; -use feature qw/say/; - -package Manager; -use Moose; -with qw(MooseX::Workers); +use strict; -use POE qw(Filter::Reference Filter::Stream); +sub PROGRAM () { "/opt/stow/repo/irssi-debug/bin/irssi" } +use POSIX; +use POE qw( Wheel::ReadWrite Wheel::Run Filter::Stream ); use Term::VT102; +use Term::TermInfo; + +my $ti = Term::Terminfo->new(); + +my $vt = Term::VT102->new(rows => 24, cols => 80); +$vt->callback_set('OUTPUT', \&vt_output, undef); +$vt->callback_set('ROWCHANGE', \&vt_rowchange, undef); + +sub vt_rowchange { + my ($vt, $cb_name, $arg1, $arg2, $priv_data) = @_; + #print STDERR "Row $arg1 changing: $arg2\n"; + print $ti->getstr("clear"); + print vt_dump(); + +} + +sub vt_output { + my ($vt, $cb_name, $cb_data, $priv_data) = @_; + #print "X:" . $cb_data; +} -has 'vt' - => ( - is => 'ro', - isa => 'Term::VT102', - required => 1, - lazy => 1, - builder => '_build_vt' - ); +sub vt_dump { + my $str = ''; + for my $y (1..24) { + $str .= $vt->row_sgrtext($y) . "\n"; + } + return $str; +} + +### Handle the _start event. This sets things in motion. +sub handle_start { + my ($kernel, $heap) = @_[KERNEL, HEAP]; + + save_term_settings($heap); + + # Set a signal handler. + $kernel->sig(CHLD => "got_sigchld"); + + make_raw(); -sub _build_vt { - my $vt = Term::VT102->new(cols => 80, rows => 24); - $vt->callback_set('OUTPUT', \&vt_output, undef); + # Start the terminal reader/writer. + $heap->{stdio} = POE::Wheel::ReadWrite->new( + InputHandle => \*STDIN, + OutputHandle => \*STDOUT, + InputEvent => "got_terminal_stdin", + Filter => POE::Filter::Stream->new(), + ); - return $vt; + # Start the asynchronous child process. + $heap->{program} = POE::Wheel::Run->new( + Program => PROGRAM, + Conduit => "pty", + StdoutEvent => "got_child_stdout", + StdioFilter => POE::Filter::Stream->new(), + ); } -sub vt_output { - my ($vt, $cb_name, $data, $priv) = @_; - say "Data is: $data" +### Handle the _stop event. This restores the original terminal +### settings when we're done. That's very important. +sub handle_stop { + my $heap = $_[HEAP]; + $heap->{stdin_tio}->setattr(0, TCSANOW); + $heap->{stdout_tio}->setattr(1, TCSANOW); + $heap->{stderr_tio}->setattr(2, TCSANOW); } -sub run { - my $self = shift; - my $job = MooseX::Workers::Job->new - ( - name => "Irssi", - command => "/opt/stow/repo/irssi-debug/bin/irssi", - args => [ "--home=/tmp" ], - ); - - $self->spawn($job); - - POE::Kernel->run(); + +### Handle terminal STDIN. Send it to the background program's STDIN. +### If the user presses ^C, then also go berserk a little. + +sub handle_terminal_stdin { + my ($heap, $input) = @_[HEAP, ARG0]; + while ($input =~ m/\003/g) { + $input = "/echo I like cakes\n"; + } + $heap->{program}->put($input); } -# Implement our Interface -# These two are both optional; if defined (as here), they -# should return a subclass of POE::Filter. -sub stdout_filter { new POE::Filter::Stream } -sub stderr_filter { new POE::Filter::Line } +### Handle STDOUT from the child program. +sub handle_child_stdout { + my ($heap, $input) = @_[HEAP, ARG0]; + # process via vt + $vt->process($input); + # send to terminal +# $heap->{stdio}->put($input); +} -sub worker_stdout { - my ( $self, $data ) = @_; +### Handle SIGCHLD. Shut down if the exiting child process was the +### one we've been managing. - $self->vt->process($data); +sub handle_sigchld { + my ($heap, $child_pid) = @_[HEAP, ARG1]; + if ($child_pid == $heap->{program}->PID) { + delete $heap->{program}; + delete $heap->{stdio}; + } + return 0; } -sub worker_manager_start { warn 'started worker manager' } -sub worker_manager_stop { warn 'stopped worker manager' } -sub max_workers_reached { warn 'maximum worker count reached' } -sub worker_error { shift; warn join ' ', @_; } -sub worker_done { shift; warn join ' ', @_; } -sub worker_started { shift; warn join ' ', @_; } -sub sig_child { shift; warn join ' ', @_; } -sub sig_TERM { shift; warn 'Handled TERM' } +### Start a session to encapsulate the previous features. +POE::Session->create( + inline_states => { + _start => \&handle_start, + _stop => \&handle_stop, + got_terminal_stdin => \&handle_terminal_stdin, + got_child_stdout => \&handle_child_stdout, + got_sigchld => \&handle_sigchld, + }, +); + + +sub make_raw { + + # Put the terminal into raw input mode. Otherwise discrete + # keystrokes will not be read immediately. + my $tio = POSIX::Termios->new(); + $tio->getattr(0); + my $lflag = $tio->getlflag; + $lflag &= ~(ECHO | ECHOE | ECHOK | ECHONL | ICANON | IEXTEN | ISIG); + $tio->setlflag($lflag); + my $iflag = $tio->getiflag; + $iflag &= ~(BRKINT | INPCK | ISTRIP | IXON); + $tio->setiflag($iflag); + my $cflag = $tio->getcflag; + $cflag &= ~(CSIZE | PARENB); + $tio->setcflag($cflag); + $tio->setattr(0, TCSANOW); -no Moose; +} -Manager->new->run(); +sub save_term_settings { + my ($heap) = @_; + # Save the original terminal settings so they can be restored later. + $heap->{stdin_tio} = POSIX::Termios->new(); + $heap->{stdin_tio}->getattr(0); + $heap->{stdout_tio} = POSIX::Termios->new(); + $heap->{stdout_tio}->getattr(1); + $heap->{stderr_tio} = POSIX::Termios->new(); + $heap->{stderr_tio}->getattr(2); +} +### Start POE's main loop, which runs the session until it's done. +$poe_kernel->run(); +exit 0; |