Allow reading files/URIs from STDIN

[Fey] / lib / App / Fey.pm

diff --git a/lib/App/Fey.pm b/lib/App/Fey.pm
index e17a22afb535847771927af2e3c804e70584e68d..8b802a0d70e7a1881af2e3bcf1baae2a074a7051 100644 (file)
--- a/lib/App/Fey.pm
+++ b/lib/App/Fey.pm
@@ -8,40 +8,127 @@ our @EXPORT_OK = qw(fey);
 our $version = '0.01';
 
 sub new {
-    my ($class, $args) = @_;
+    my ($class, $options) = @_;
     my $config = do ($ENV{XDG_CONFIG_HOME} // "$ENV{HOME}/.config") . '/fey/config.pl';
 
     my $self = {
-        mime_query => $args->{mime_query} // $config->{mime_query} // sub { `file --brief --mime-type "$_[0]"` },
-        contexts => $args->{contexts} // $config->{contexts} // { default => sub { 1 } },
-        targets => $args->{targets} // $config->{targets} // {}
+        mime_query => $options->{mime_query} // $config->{mime_query} // sub {
+            open my $mime_type, '-|', 'file', '--brief', '--mime-type', $_[0];
+            <$mime_type>;
+        },
+        contexts => $options->{contexts} // $config->{contexts} // { default => sub { 1 } },
+        targets => $options->{targets} // $config->{targets} // {}
     };
 
     bless $self, $class;
 }
 
 sub launch {
-    my ($self, $file_or_uri) = @_;
+    my $self = shift;
+    my $options = ref $_[0] ? shift : {};
 
-    if ($file_or_uri =~ m|^file://(.+)|) {
-        $file_or_uri = $1;
+    if (!(-t STDIN)) {
+        _read_stdin(\@_);
     }
+    if (!@_ || $options->{interactive}) {
+        open STDIN, '<', '/dev/tty'; 
+        _read_stdin(\@_);
+    }
+    die "No files or URIs specified.\n" unless @_;
 
-    my ($mime_or_uri, $targets);
-    if (-e $file_or_uri) {
-        $mime_or_uri = $self->{mime_query}->($file_or_uri)
+    if ($options->{group}) {
+        $self->_launch_group($options, @_);
+    } elsif ($options->{single}) {
+        $self->_launch_single($options, @_);
     } else {
-        $mime_or_uri = $file_or_uri;
+        $self->_launch($options, @_);
     }
+}
+
+sub _launch {
+    my $self = shift;
+    my $options = shift;
+
+    if ($options->{fork}) {
+        for my $file_or_uri (@_) {
+            my $pid = fork;
+            next if $pid;
+
+            my $handler = $self->_get_handler($options, $file_or_uri);
+            $handler->($file_or_uri) if $handler;
+            return;
+        }
+    } else {
+        for my $file_or_uri (@_) {
+            my $handler = $self->_get_handler($options, $file_or_uri);
+            $handler->($file_or_uri) if $handler;
+        }
+    }
+}
+
+sub _launch_group {
+    my $self = shift;
+    my $options = shift;
+
+    my ($groups, $handlers) = ({}, {});
+    for my $file_or_uri (@_) {
+        my $handler = $self->_get_handler($options, $file_or_uri);
+        if ($handler) {
+            $groups->{"$handler"} //= [];
+            push @{ $groups->{"$handler"} }, $file_or_uri;
+            $handlers->{"$handler"} = $handler;
+        }
+    }
+
+    if ($options->{fork}) {
+        for my $group (keys %{ $groups }) {
+            if ($options->{fork}) {
+                my $pid = fork;
+                next if $pid;
+            }
+
+            $handlers->{$group}->(@{ $groups->{$group} });
+            return;
+        }
+    } else {
+        for my $group (keys %{ $groups }) {
+            $handlers->{$group}->(@{ $groups->{$group} });
+        }
+    }
+}
+
+sub _launch_single {
+    my $self = shift;
+    my $options = shift;
+
+    if ($options->{fork}) {
+        my $pid = fork;
+        return if $pid;
+    }
+
+    my $handler = $self->_get_handler($options, $_[0]);
+    $handler->(@_) if $handler;
+}
+
+sub _get_handler {
+    my $self = shift;
+    my $options = shift;
+    
+    my $file_or_uri = $_[0] =~ m|^file://(.+)| ? $1 : $_[0];
+    my $mime_or_uri = -e $file_or_uri ? $self->{mime_query}->($file_or_uri) : $file_or_uri;
 
     for my $target (@{ $self->{targets} }) {
         for my $pattern (@{ $target->{patterns} }) {
             if ($mime_or_uri =~ /$pattern/) {
                 my $associations = $target->{associations};
+
+                my $context = $options->{context};
+                my $handler = $associations->{$context} if $context;
+                return $handler if defined $handler;
+
                 for my $context (keys %{ $associations }) {
                     if ($self->{contexts}->{$context}->()) {
-                        $associations->{$context}->($file_or_uri);
-                        return;
+                        return $associations->{$context};
                     }
                 }
             }
@@ -49,6 +136,13 @@ sub launch {
     }
 }
 
+sub _read_stdin {
+    for my $file_or_uri (<STDIN>) {
+        chomp $file_or_uri;
+        push @{ $_[0] }, $file_or_uri;
+    }
+}
+
 sub fey {
-    App::Fey->new($_[1] // {})->launch($_[0] // die 'Error: No file or URI specified.');
+    App::Fey->new(ref $_[0] ? $_[0] : {})->launch(@_);
 }