Infrastructure cleanups

[aoc.git] / lib / Y / AoC.pm

diff --git a/lib/Y/AoC.pm b/lib/Y/AoC.pm
index 8fbb606a6486cba86f744dd9af0332441b61e6de..31b4532694b97a840f63e4fe9ce9a07e1847d8e2 100644 (file)
--- a/lib/Y/AoC.pm
+++ b/lib/Y/AoC.pm
@@ -7,6 +7,7 @@ our @EXPORT_OK = qw(red grey white yellow day year);
 
 use FindBin qw($Bin);
 use Term::ANSIColor;
+use Cwd;
 
 sub red    { colored(['bright_red    on_black'], @_); }
 sub white  { colored(['bright_white  on_black'], @_); }
@@ -20,6 +21,9 @@ sub day {
 
 sub year {
        my ($num) = $Bin =~ /.*(\d{4})/;
+       if (!defined $num) {
+               ($num) = getcwd =~ /.*(\d{4})/;
+       }
        $num;
 }