git.defcon.no Git - trk/blob - trk

   1 #!/usr/bin/perl
   2 use Time::Local;
   3 use Digest::MD5  qw(md5_hex);
   4 use File::Basename;
   5 use POSIX;
   6
   7
   8 my $trk_dir = "$ENV{HOME}/.trk";
   9
  10 use constant {
  11         START => 1,
  12         TIMEFORMAT => 2,
  13         STOP => 3,
  14         EDIT => 4,
  15 };
  16
  17 sub help
  18 {
  19         my $code = shift;
  20         printf("It seems you require assistance\n");
  21
  22         if ( $code )
  23         {
  24                 printf("How to start\n") if $code == START;
  25                 printf("How to time\n")  if $code == TIMEFORMAT;
  26         }
  27         exit(-1);
  28 }
  29
  30
  31 sub gen_puuid (;$)
  32 {
  33         my $id_length = shift;
  34         $id_length = 32 if not defined $id_length;
  35
  36         my $w = time;
  37
  38         for(my $i=0 ; $i<128;)
  39         {
  40                 my $tc = chr(int(rand(127)));
  41                 if($tc =~ /[a-zA-Z0-9]/)
  42                 {
  43                         $w .=$tc;
  44                         $i++;
  45                 }
  46         }
  47         $w = md5_hex( $w );
  48
  49         while ( length($w) < $id_length )
  50         {
  51                 $w .= gen_puuid( $id_length - length( $w ) );
  52         }
  53
  54         $w = substr( $w, 0, $id_length );
  55         return $w;
  56 }
  57
  58 # Input to parse_time is:
  59 #   * date -> date-string in the form YYYY-MM-DD
  60 #   * time -> time-string in the form HH:MM
  61 # Return value is a unix timestamp, as returned by time()
  62 sub parse_time ($$)
  63 {
  64         my ( $Y, $M, $D ) = split ("-", shift );
  65         my ( $h, $m ) = split(":", shift );
  66         return timelocal(0, $m, $h, $D, ($M-1), $Y);
  67 }
  68
  69 sub str2time ($)
  70 {
  71         my $i = shift;
  72         printf("%s\n", $i);
  73         return 0 if not $i =~ m/(\d\d\d\d-\d\d-\d\d) (\d\d:\d\d)/;
  74         return parse_time($1, $2);
  75 }
  76
  77 sub time2str ($)
  78 {
  79         my $t = shift;
  80         return strftime("%Y-%m-%d %H:%M", localtime($t));
  81 }
  82
  83 sub parse_arguments ($)
  84 {
  85
  86         my $step = shift;
  87
  88         my $start_time = time;
  89         my $title = undef;
  90
  91         if (( $#ARGV >= 1) && ( $ARGV[1] eq "at" ))
  92         {
  93                 # Start and Activity require a title to be present.
  94                 # All other (stop, main...) do not ^^.
  95                 if ( ($step == START) || ($step == TASK) )
  96                 {
  97                         # TODO: Allow no title!
  98                         # If no title is given, read ID of previously used track in stead :)
  99                         help($step) unless $#ARGV > 3;
 100                         $title = join(" ", @ARGV[4..$#ARGV]);
 101                 }
 102                 help(TIMEFORMAT) unless ( $ARGV[2] =~ m/\d\d\d\d-\d\d-\d\d/ && $ARGV[3] =~ m/\d\d:\d\d/);
 103
 104                 $start_time = parse_time( $ARGV[2], $ARGV[3] );
 105         }
 106         elsif ( ($step == START) || ($step == TASK) || ($step == EDIT))
 107         {
 108                 shift(@ARGV);
 109                 $title = join(" ", @ARGV);
 110         }
 111
 112         if ( not defined $title )
 113         {
 114                 return $start_time;
 115         }
 116         else
 117         {
 118                 return ( $start_time, $title );
 119         }
 120 }
 121
 122 sub get_last_id
 123 {
 124         return undef if ( ! -f $trk_dir . "/last" );
 125         open ( CUR, "<" .  $trk_dir . "/last" ) or die ("Unable to read last track file");
 126         my $id = <CUR>;
 127         chomp($id);
 128         close(CUR);
 129         return $id;
 130 }
 131
 132 sub get_current_id
 133 {
 134         return undef if ( ! -f $trk_dir . "/current" );
 135         open ( CUR, "<" .  $trk_dir . "/current" ) or die ("Unable to read current track file");
 136         my $id = <CUR>;
 137         chomp($id);
 138         close(CUR);
 139         return $id;
 140 }
 141
 142 sub set_current_id ($)
 143 {
 144         my $id = shift;
 145         return undef if ( -f $trk_dir . "/current" );
 146         open ( CUR, ">" .  $trk_dir . "/current" ) or die ("Unable to write current track file");
 147         printf(CUR "%s\n", $id );
 148         close(CUR);
 149
 150         open ( LAST, ">" .  $trk_dir . "/last" ) or die ("Unable to write last track file");
 151         printf(LAST "%s\n", $id );
 152         close(LAST);
 153 }
 154
 155 sub get_tracks
 156 {
 157         my %tracks;
 158
 159         foreach my $d ( <$trk_dir/*> )
 160         {
 161                 next if not -d $d;
 162                 next if not -f $d . "/info";
 163
 164                 my $id = basename($d);
 165                 my $title = get_track_name( $id );
 166
 167                 $tracks{$id} = $title unless not defined $title;
 168         }
 169
 170         return \%tracks;
 171
 172 }
 173
 174 sub get_track_id ($)
 175 {
 176         my $title = shift;
 177
 178         # Get hash of track-id's and -names from get_tracks
 179         my $tracks = get_tracks();
 180
 181         # Look up name in list
 182         foreach my $id ( keys $tracks )
 183         {
 184                 # Return ID for name
 185                 return $id if ( $tracks->{$id} eq $title )
 186         }
 187
 188         # If no match, return undef.
 189         return undef;
 190 }
 191
 192 sub get_track_name ($)
 193 {
 194         my $id = shift;
 195         open(PRO, "<" . $trk_dir . "/" . $id . "/info" ) or die ("Unable to read track medatata file!");
 196         my $title = undef;
 197         while (<PRO>)
 198         {
 199                 next if not $_ =~ /^title:(.*)/;
 200                 $title = $1;
 201         }
 202         close(PRO);
 203         return $title;
 204 }
 205
 206 sub create_track ($)
 207 {
 208         my $title = shift;
 209
 210         my $id;
 211         do
 212         {
 213                 $id = gen_puuid(8);
 214
 215         } while ( -d $trk_dir . "/" . $id );
 216         mkdir ( $trk_dir . "/" . $id );
 217
 218         open(PRO, ">" . $trk_dir . "/" . $id . "/info" ) or die ("Unable to create track medatata file!");
 219         printf(PRO "title:%s\n", $title);
 220         close(PRO);
 221
 222         return $id;
 223 }
 224
 225 sub start_track ($$)
 226 {
 227         my $start_time = shift;
 228         my $title = shift;
 229
 230
 231         my $current = get_current_id();
 232         if ( not $current )
 233         {
 234                 if ( not $title )
 235                 {
 236                         $current = get_last( );
 237                 }
 238                 else
 239                 {
 240                         $current = get_track_id( $title );
 241                         if ( not $current )
 242                         {
 243                                 printf("No track by that name! Creating a new one.\n");
 244                                 $current = create_track($title);
 245                         }
 246                 }
 247
 248                 # Break off here if we haven't gotten an ID yet.
 249                 return undef if not $current;
 250
 251                 set_current_id($current);
 252
 253                 # First iteration is VERY naive: simply add the start time to the bottom of the tracking file
 254                 # Will have to do more logic: if the start point is before one of the times already in the track,
 255                 # the file will have to be manipulated to get coherent tracking!
 256                 open (TRACK, ">>" . $trk_dir . "/" . $current . "/tracking" ) or die ("Unable to open file, $!");
 257                 printf(TRACK "[%s]", time2str($start_time));
 258                 close (TRACK);
 259
 260                 return $current;
 261         }
 262
 263         return undef;
 264 }
 265
 266 sub close_track ($)
 267 {
 268
 269         my $stop_time = shift;
 270
 271         my $current = get_current_id();
 272
 273         die ("Project exists, but tracking file does not!") if ( not -f  $trk_dir . "/" . $current . "/tracking" );
 274
 275         # First iteration is VERY naive: simply add the stop time to the bottom line of the tracking file
 276         # Will have to do more logic: if the start point is before one of the times already in the track,
 277         # the file will have to be manipulated to get coherent tracking!
 278         # In addtion to this: actually do some file sanity checking!
 279         open (TRACK, ">>" . $trk_dir . "/" . $current . "/tracking" ) or die ("Unable to open file, $!");
 280         printf(TRACK " to [%s]\n", time2str($stop_time));
 281         close (TRACK);
 282
 283         unlink ( $trk_dir . "/current" );
 284 }
 285
 286 ############################################################
 287
 288 if ( ! -d $trk_dir )
 289 {
 290         mkdir $trk_dir or die("Unable to create data directory");
 291 }
 292
 293 if ( $#ARGV < 0 )
 294 {
 295         help();
 296 }
 297
 298 my $command = $ARGV[0];
 299
 300 if ( ( $command eq "start") || ($command eq "on" ) )
 301 {
 302         if ( $#ARGV < 1)
 303         {
 304                 help(START);
 305         }
 306
 307         my ( $start_time, $title ) = parse_arguments(START);
 308
 309         my $current = get_current_id();
 310         if ( not $current )
 311         {
 312                 $current = start_track( $start_time, $title );
 313
 314                 if ( not $current )
 315                 {
 316                         printf("Something weird happened.\n");
 317                         exit(1);
 318                 }
 319         }
 320         else
 321         {
 322                 printf("A project is being tracked: %s\n", get_track_name( $current ) );
 323                 close_track($start_time);
 324                 $current = start_track( $start_time, $title );
 325         }
 326
 327         printf("Started tracking of '%s' at %s\n\n", $title, scalar localtime $start_time);
 328 }
 329 elsif ( ( $command eq "stop") || ($command eq "off" ) )
 330 {
 331         if ( $#ARGV < 0)
 332         {
 333                 help(STOP);
 334         }
 335
 336
 337         my $stop_time = parse_arguments(STOP);
 338
 339         my $current = get_current_id();
 340         if ( not $current )
 341         {
 342                 printf("No project is currently tracked. To stop, please start first\n");
 343                 exit(0);
 344         }
 345         my $title = get_track_name( $current );
 346
 347         close_track($stop_time);
 348
 349         printf("Stopped tracking of '%s' at %s\n\n", $title, scalar localtime $stop_time);
 350 }
 351 elsif ( ( $command eq "projects" ) || ( $command eq "list" )  )
 352 {
 353         # Todo/future extensions:
 354         # TODO: Sort list of names alphabetically
 355         # TODO: Get total-hours for projects
 356         # TODO:
 357         my $tracks = get_tracks();
 358         printf("Currently tracked project names:\n\n");
 359         my $current = get_current_id();
 360
 361         foreach my $id ( keys $tracks )
 362         {
 363                 printf(" %s %s\n", ($id eq $current ? ">" : " " ),$tracks->{$id} );
 364         }
 365         print("\n");
 366 }
 367 elsif ( $command eq "edit" )
 368 {
 369
 370         my ( undef, $title ) = parse_arguments(EDIT);
 371         my $id = get_last_id();
 372
 373         if ( $title )
 374         {
 375                 $id = get_track_id($title);
 376                 if ( not $id )
 377                 {
 378                         printf("No project by that name. Try 'list'\n");
 379                         exit(0);
 380                 }
 381         }
 382
 383         system( "/usr/bin/editor " . $trk_dir . "/" . $id . "/tracking" );
 384 }
 385 else
 386 {
 387         help();
 388 }