1 files changed, 58 insertions, 0 deletions
diff --git a/lib/App/REPL/Plugin/Colors.pm b/lib/App/REPL/Plugin/Colors.pm
new file mode 100644
index 0000000..b06f857
--- /dev/null
+++ b/lib/App/REPL/Plugin/Colors.pm
@@ -0,0 +1,58 @@
+package App::REPL::Plugin::Colors;
+use strict;
+use warnings;
+
+use base 'App::REPL::Plugin';
+
+use Term::ANSIColor;
+
+sub new {
+    my $class = shift;
+
+    my $self = $class->SUPER::new(@_);
+    $self->{error} = 'red';
+    $self->{warning} = 'yellow';
+    $self->{result} = 'green';
+
+    return $self;
+}
+
+sub evaluate {
+    my $self = shift;
+    my ($next, @args) = @_;
+
+    local $SIG{__WARN__} = sub { $self->print_warn(@_) };
+    $next->(@args);
+}
+
+sub print_error {
+    my $self = shift;
+    my ($next, $error) = @_;
+
+    print color($self->{error});
+    $next->($error);
+    local $| = 1;
+    print color('reset');
+}
+
+sub print_result {
+    my $self = shift;
+    my ($next, @result) = @_;
+
+    print color($self->{result});
+    $next->(@result);
+    local $| = 1;
+    print color('reset');
+}
+
+sub print_warn {
+    my $self = shift;
+    my ($warning) = @_;
+
+    print color($self->{warning});
+    print $warning;
+    local $| = 1;
+    print color('reset');
+}
+
+1;