From d95d46585099cfda82dd971351ff3177d66884cb Mon Sep 17 00:00:00 2001
From: Ole Tange <ole@tange.dk>
Date: Thu, 13 Dec 2018 21:03:25 +0100
Subject: [PATCH] ytv: Read files/urls from STDIN if not given on the command
 line.

---
 ytv/ytv | 72 +++++++++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 57 insertions(+), 15 deletions(-)

diff --git a/ytv/ytv b/ytv/ytv
index 2e8e1e3..34f6dbc 100755
--- a/ytv/ytv
+++ b/ytv/ytv
@@ -74,30 +74,58 @@ my @tor = $opt::tor ? qw(--proxy socks4a://127.0.0.1:9050/) : ();
 `mkdir -p ~/tmp/Videos`;
 chdir($ENV{'HOME'}."/tmp/Videos");
 map { $before{$_} = -M $_ } <*>;
+my @youtubeid;
+if(not @ARGV) {
+    chomp(@ARGV = <>);
+}
+
 for my $url (@ARGV) {
+    print "Playing $url\n";
     if(not fork()) {
 	# Download in the background
 	system(qw(youtube-dl --all-subs --skip-download), @tor,$url);
 	system(qw(youtube-dl -f best), @tor,$url);
 	exit;
     }
-}
 
-$SIG{'CHLD'} = sub { exit; };
+    do {
+	# Sleep until there are matching files
+	@new = ((grep { -e $_ } $url),
+		(grep { !/Frag\d/ } grep { -s $_ > 1000000 }
+		 grep { not $before{$_} or $before{$_} > -M $_ } <*>),
+		map { my $id = $_; grep { /$id/ } <*> }
+		grep { s/^.*watch.v=([a-z0-9]+).*$/$1/i
+			   or
+			   s/^.*youtu.be.([a-z0-9]+).*$/$1/i } $url);
+	usleep(10);
+    } until @new;
 
-do {
-    # Sleep until there are matching files
-    sleep(1);
-    @new = (grep { !/Frag\d/ } grep { -s $_ > 1000000 }
-	    grep { not $before{$_} or $before{$_} > -M $_ } <*>);
-} until @new;
-
-while(@new) {
-    # Mark as seen
-    map { $before{$_} = -M $_ } @new;
-    # Run VLC on the matching files
-    system("vlc", map { $a=$b=$_; $b=~s/.part//; s/.temp//; $a,$b,$_ } @new);
-    @new = grep { not $before{$_} or $before{$_} > -M $_ } <*>;
+    while(@new) {
+	# Mark as seen
+	map { $before{$_} = -M $_ } @new;
+	# Run VLC on the matching files
+	if($opt::kodi) {
+	    for(uniq(map { $a=$b=$_; $b=~s/.part//; s/.temp//; $a,$b,$_ } @new)) {
+		my $answer;
+		do {
+		    print "KODI playing $_\n";
+		    if(not fork()) {
+			exec("idok", $_);
+		    }
+		    print "Press ENTER\n";
+		    open(my $tty_fh, "<", "/dev/tty") || ::die_bug("interactive-tty");
+		    $answer = <$tty_fh>;
+		    close $tty_fh;
+		    if($answer =~ /d/i) {
+			unlink $_;
+		    }
+		} while($answer =~ /r/i);
+	    }
+	} else {
+	    system("vlc", map { $a=$b=$_; $b=~s/.part//; s/.temp//; $a,$b,$_ } @new);
+	}
+	@new = grep { not $before{$_} or $before{$_} > -M $_ } <*>;
+    }
 }
 
 sub get_options_from_array {
@@ -118,6 +146,7 @@ sub get_options_from_array {
     my @retval = GetOptions
 	("debug|D" => \$opt::debug,
 	 "tor" => \$opt::tor,
+	 "kodi|k" => \$opt::kodi,
 	 "version|V" => \$opt::version,
 	);
     if(not $this_is_ARGV) {
@@ -126,3 +155,16 @@ sub get_options_from_array {
     }
     return @retval;
 }
+
+sub usleep {
+    # Sleep this many milliseconds.
+    # Input:
+    #   $ms = milliseconds to sleep
+    my $ms = shift;
+    select(undef, undef, undef, $ms/1000);
+}
+
+sub uniq {
+    # Remove duplicates and return unique values
+    return keys %{{ map { $_ => 1 } @_ }};
+}