tvshow-symlinks.pl

#!/usr/bin/perl
use strict;
use warnings;
use Encode;
use File::Basename qw(basename);
use LWP::Simple;
use File::Path;
use File::Spec;
use Cwd;
use Unicode::Normalize;

our $BaseDir = "$ENV{HOME}/Movies/Plex";

mkdir $BaseDir, 0777 unless -e $BaseDir;

if ($ARGV[0] eq '-t') { selftest() }

my $aliases = {};
if (open my $fh, "<:utf8", "$ENV{HOME}/.plexshowaliases") {
    while (<$fh>) {
        chomp;
        my($orig, $alias) = split ',', $_, 2;
        $aliases->{$orig} = $alias;
    }
}

my $current = cwd;
for my $file (@ARGV) {
    $file = File::Spec->file_name_is_absolute($file) ? $file : "$current/$file";
    if (my $info = parse_info($file, $aliases)) {
        my $link = generate_link($info, $file) or next;
        unlink $link;
        symlink $file, $link;
        warn "symlinked $link\n";
    }
}

sub parse_info {
    my $base = NFKC(decode_utf8(basename(shift)));
    my $aliases = shift || {};

    my $ext;
    $base =~ s/\.(\w+)$/$ext = $1; ""/e;

    my %want_ext = map { $_ => 1 } qw( avi mp4 divx m4v mov mkv flv wmv );
    return unless $want_ext{lc $ext};

    $base =~ s/_/ /g;
    $base =~ s/\s+(RAW)$//i;
    trim($base);

    my %pair = ('[' => ']', '(', => ')', "\x{3010}" => "\x{3011}");
    my $tag_re = join "|", map { quotemeta($_) . "([^" . quotemeta($pair{$_}) . "]*?)" . quotemeta($pair{$_}) } keys %pair;

    my @tags;
    while ($base =~ s/^(?:$tag_re)\s*|\s*(?:$tag_re)\.?$//) {
        push @tags, $1 || $2 || $3 || $4 || $5 || $6;
    }

    if ($base =~ s/\.(HR|[HP]DTV|WS|AAC|AC3|DVDRip|PROPER|DVDSCR|720p|1080p|[hx]264(?:-\w+)?|dd51)\.(.*)//i) {
        $base =~ s/\./ /g;
        # ad-hoc: rescue DD.MM.YY(YY)
        $base =~ s/(\d\d) (\d\d) (\d\d(\d\d)?)\b/$1.$2.$3/;
    }

    1 while $base =~ s/\s*(RAW|end|finale|\(\x{7d42}\))\s*$//i;

    # strip episode title
    $base =~ s/\s*[\x{300c}\x{ff62}].*?[\x{300d}\x{ff63}]\s*$//;

    for my $orig (keys %$aliases) {
        $base =~ s/^$orig/$aliases->{$orig}/i
            and last;
    }

    my $date_re = '(\d{2,4})[\.\-\x{5e74}](\d{2})[\.\-\x{6708}](\d{2})[\.\x{65e5}]?';
    
    my $info;
    if ($base =~ s/\s*(?:EP?|\#)(\d+)\s*$//i) {
        $info->{episode} = $1 + 0;
    } elsif ($base =~ s/(?:\s+-)?\s+(\d+)\s*$//) {
        $info->{episode} = $1 + 0;
    } elsif ($base =~ s/(?:\s+-)?\s*\x{7b2c}(\d+)(?:\x{8a71}|\x{591c}|\x{56de})\s*$//) {
        $info->{episode} = $1 + 0;
    } elsif ($base =~ s/(?:\s+-)?\s*$date_re\s*$//) {
        $info->{date} = [ $1, $2, $3 ];
    } else {
        for my $tag (@tags) {
            if ($tag =~ /$date_re/) {
                $info->{date} = [ $1, $2, $3 ];
                last;
            }
        }
    }

    if ($base =~ s/\s+(?:S(?:eason)?)?\s*(\d+)\d*$//i) {
        $info->{season} = $1 + 0;
    }

    $info->{season} ||= 1;
    $info->{series} = trim($base);

    return $info;
}

sub generate_link {
    my($info, $file, $test) = @_;

    my $ext = ($file =~ /\.(\w+)$/)[0];
    $info->{series} = normalize_series($info->{series}) if $info->{series};

    my($path, $link);
    if ($info->{episode}) {
        $path = "$BaseDir/$info->{series}/Season $info->{season}";
        $link = sprintf "%s - S%02dE%02d.%s", $info->{series}, $info->{season}, $info->{episode}, $ext;
    } elsif ($info->{date}) {
        $info->{date}->[0] += 2000 if $info->{date}->[0] < 100;
        $path = "$BaseDir/$info->{series}";
        $link = sprintf "%s - %04d.%02d.%02d.%s", $info->{series}, @{$info->{date}}, $ext;
    } else {
        warn "Didn't match any episode or date: $file";
        return;
    }

    mkpath $path unless $test;
    return "$path/$link";
}

sub normalize_series {
    my $name = shift;
    $name =~ s/^\s*|\s*$//g; # Plex doesn't like in series name apparently
    return $name;
}

sub trim {
    $_[0] =~ s/^\s*|\s*$//g;
    $_[0];
}

sub selftest {
    use Data::Dumper;
    while (<DATA>) {
        my $info = parse_info($_, { "Name Show" => "Show Name" });
        warn Dumper $info;
        warn Dumper generate_link($info, $_, 1);
    }

}

__DATA__