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 project 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_project
 123 {
 124         return undef if ( ! -f $trk_dir . "/last" );
 125         open ( CUR, "<" .  $trk_dir . "/last" ) or die ("Unable to read last project file");
 126         my $id = <CUR>;
 127         chomp($id);
 128         close(CUR);
 129         return $id;
 130 }
 131
 132 sub get_current_project
 133 {
 134         return undef if ( ! -f $trk_dir . "/current" );
 135         open ( CUR, "<" .  $trk_dir . "/current" ) or die ("Unable to read current project file");
 136         my $id = <CUR>;
 137         chomp($id);
 138         close(CUR);
 139         return $id;
 140 }
 141
 142 sub set_current_project ($)
 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 project file");
 147         printf(CUR "%s\n", $id );
 148         close(CUR);
 149
 150         open ( LAST, ">" .  $trk_dir . "/last" ) or die ("Unable to write last project file");
 151         printf(LAST "%s\n", $id );
 152         close(LAST);
 153 }
 154
 155 sub current_task
 156 {
 157         my $project = get_current_project();
 158         return undef if not $project;
 159
 160         open ( CUR, "<" .  $trk_dir . "/current" ) or die ("Unable to read current project file");
 161         <CUR>;
 162         my $id = <CUR>;
 163         chomp($id);
 164         close(CUR);
 165         return $id;
 166
 167 }
 168
 169 sub get_projects
 170 {
 171         my %projects;
 172
 173         foreach my $d ( <$trk_dir/*> )
 174         {
 175                 next if not -d $d;
 176                 next if not -f $d . "/info";
 177
 178                 my $id = basename($d);
 179                 my $title = get_project_name( $id );
 180
 181                 $projects{$id} = $title unless not defined $title;
 182         }
 183
 184         return \%projects;
 185
 186 }
 187
 188 sub get_project_id ($)
 189 {
 190         my $title = shift;
 191
 192         # Get hash of project-id's and -names from get_projects
 193         my $projects = get_projects();
 194
 195         # Look up name in list
 196         foreach my $id ( keys $projects )
 197         {
 198                 # Return ID for name
 199                 return $id if ( $projects->{$id} eq $title )
 200         }
 201
 202         # If no match, return undef.
 203         return undef;
 204 }
 205
 206 sub get_project_name ($)
 207 {
 208         my $id = shift;
 209         open(PRO, "<" . $trk_dir . "/" . $id . "/info" ) or die ("Unable to read project medatata file!");
 210         my $title = undef;
 211         while (<PRO>)
 212         {
 213                 next if not $_ =~ /^title:(.*)/;
 214                 $title = $1;
 215         }
 216         close(PRO);
 217         return $title;
 218 }
 219
 220 sub create_project ($)
 221 {
 222         my $title = shift;
 223
 224         my $id;
 225         do
 226         {
 227                 $id = gen_puuid(8);
 228
 229         } while ( -d $trk_dir . "/" . $id );
 230         mkdir ( $trk_dir . "/" . $id );
 231
 232         open(PRO, ">" . $trk_dir . "/" . $id . "/info" ) or die ("Unable to create project medatata file!");
 233         printf(PRO "title:%s\n", $title);
 234         close(PRO);
 235
 236         return $id;
 237 }
 238
 239 ############################################################
 240
 241 if ( ! -d $trk_dir )
 242 {
 243         mkdir $trk_dir or die("Unable to create data directory");
 244 }
 245
 246 if ( $#ARGV < 0 )
 247 {
 248         help();
 249 }
 250
 251 my $command = $ARGV[0];
 252
 253 if ( ( $command eq "start") || ($command eq "on" ) )
 254 {
 255         if ( $#ARGV < 1)
 256         {
 257                 help(START);
 258         }
 259
 260         my ( $start_time, $title ) = parse_arguments(START);
 261
 262         my $current = get_current_project();
 263         if ( not $current )
 264         {
 265                 $current = get_project_id( $title );
 266                 if ( not $current )
 267                 {
 268                         printf("No project by that name! Creating a new one.\n");
 269                         $current = create_project($title);
 270                 }
 271                 else
 272                 {
 273                         printf("Continuing tracking for existing project.\n");
 274                 }
 275                 set_current_project($current);
 276         }
 277         else
 278         {
 279                 printf("A project is being tracked: %s\n", get_project_name( $current ) );
 280                 printf("Stop current tracking before starting a new one\n");
 281                 exit(0);
 282         }
 283
 284         # First iteration is VERY naive: simply add the start time to the bottom of the tracking file
 285         # Will have to do more logic: if the start point is before one of the times already in the track,
 286         # the file will have to be manipulated to get coherent tracking!
 287         open (TRACK, ">>" . $trk_dir . "/" . $current . "/tracking" ) or die ("Unable to open file, $!");
 288         printf(TRACK "[%s]", time2str($start_time));
 289         close (TRACK);
 290
 291         printf("Started tracking of '%s' at %s\n\n", $title, scalar localtime $start_time);
 292 }
 293 elsif ( ( $command eq "stop") || ($command eq "off" ) )
 294 {
 295         if ( $#ARGV < 0)
 296         {
 297                 help(STOP);
 298         }
 299
 300
 301         my $stop_time = parse_arguments(STOP);
 302
 303         my $current = get_current_project();
 304         if ( not $current )
 305         {
 306                 printf("No project is currently tracked. To stop, please start first\n");
 307                 exit(0);
 308         }
 309         my $title = get_project_name( $current );
 310
 311         die ("Project exists, but tracking file does not!") if ( not -f  $trk_dir . "/" . $current . "/tracking" );
 312
 313         # First iteration is VERY naive: simply add the stop time to the bottom line of the tracking file
 314         # Will have to do more logic: if the start point is before one of the times already in the track,
 315         # the file will have to be manipulated to get coherent tracking!
 316         # In addtion to this: actually do some file sanity checking!
 317         open (TRACK, ">>" . $trk_dir . "/" . $current . "/tracking" ) or die ("Unable to open file, $!");
 318         printf(TRACK " to [%s]\n", time2str($stop_time));
 319         close (TRACK);
 320
 321         unlink ( $trk_dir . "/current" );
 322
 323         printf("Stopped tracking of '%s' at %s\n\n", $title, scalar localtime $stop_time);
 324 }
 325 elsif ( ( $command eq "projects" ) || ( $command eq "list" )  )
 326 {
 327         # Todo/future extensions:
 328         # TODO: Sort list of names alphabetically
 329         # TODO: Get total-hours for projects
 330         # TODO:
 331         my $projects = get_projects();
 332         printf("Currently tracked project names:\n\n");
 333         my $current = get_current_project();
 334
 335         foreach my $id ( keys $projects )
 336         {
 337                 printf(" %s %s\n", ($id eq $current ? ">" : " " ),$projects->{$id} );
 338         }
 339         print("\n");
 340 }
 341 elsif ( $command eq "edit" )
 342 {
 343
 344         my ( undef, $title ) = parse_arguments(EDIT);
 345         my $id = get_last_project();
 346
 347         if ( $title )
 348         {
 349                 $id = get_project_id($title);
 350                 if ( not $id )
 351                 {
 352                         printf("No project by that name. Try 'list'\n");
 353                         exit(0);
 354                 }
 355         }
 356
 357         system( "/usr/bin/editor " . $trk_dir . "/" . $id . "/tracking" );
 358 }
 359 else
 360 {
 361         help();
 362 }