Allow closing tasks not started today with -D
[infodrom.org/service.infodrom.org] / src / InfoCon / stempel / stempel
index be9b66c..7bb7bbc 100755 (executable)
@@ -1,7 +1,7 @@
 #! /usr/bin/perl
 
-#  Time Tracker
-#  Copyright (c) 2007,8,13,14  Martin Schulze <joey@infodrom.org>
+#  InfoCon Time Tracker
+#  Copyright (c) 2007,8,13,14,15  Martin Schulze <joey@infodrom.org>
 #
 #  This program is free software; you can redistribute it and/or modify
 #  it under the terms of the GNU General Public License as published by
@@ -131,6 +131,7 @@ sub open_task
                next;
            }
 
+           $task =~ s/\s*$//;
            my $query = q{INSERT INTO stempel (start,customer,task) VALUES('now()',?,?)};
            my $sth = $dbh->prepare($query);
            $sth->execute($customer, $task);
@@ -143,7 +144,7 @@ sub quarter
 {
     my $min = shift;
 
-    return 15 if $min < 15;
+    return 15 if $min > 0 && $min < 15;
 
     return $min - $min%15 if ($min%15 < 3);
 
@@ -163,7 +164,9 @@ sub hdiff
 
 sub close_task
 {
+    my $really = shift;
     my ($d_sec,$d_min,$d_hour,$d_mday,$d_mon,$d_year,$d_wday,$d_isdst) = localtime();
+    my $today = 1;
 
     my $query = q{SELECT oid,start FROM stempel WHERE stop IS NULL};
     my $sth = $dbh->prepare ($query);
@@ -173,12 +176,18 @@ sub close_task
            my @d = split(/-/, $arr[0]);
 
            if ($d[0] != $d_year+1900 || $d[1] != $d_mon+1 || $d[2] != $d_mday) {
-               printf "Task not started today, aborting.\n";
-               next;
+               if ($really) {
+                   printf "Task not started today.\n";
+                   $today = 0;
+               } else {
+                   printf "Task not started today, aborting, use -D if this is intentional.\n";
+                   next;
+               }
            }
 
            my @t = split(/:/, $arr[1]);
            my $int = quarter(hdiff($t[0], $t[1], $d_hour, $d_min));
+           $int += 24 * 60 unless $today;
 
            $query = sprintf('UPDATE stempel SET stop=now(),time=%d WHERE oid = %d',
                             $int, $row->{oid});
@@ -197,11 +206,25 @@ sub delete_task
     exit(0);
 }
 
+sub reopen_task
+{
+    my $sql = q{SELECT max(id) AS max_id FROM stempel};
+    my $sth = $dbh->prepare($sql);
+    if ($sth && $sth->execute > 0) {
+       my $row = $sth->fetchrow_hashref;
+       $sql = sprintf("UPDATE stempel SET stop = NULL, time = NULL WHERE id = %d", $row->{max_id});
+       $dbh->do($sql);
+    }
+
+    exit(0);
+}
+
 sub list_open
 {
     my $exit = shift;
     my ($d_sec,$d_min,$d_hour,$d_mday,$d_mon,$d_year,$d_wday,$d_isdst) = localtime();
     my $query = q{SELECT customer,start,task FROM stempel WHERE time IS NULL ORDER BY start,customer};
+    my $today = 1;
 
     my $sth = $dbh->prepare ($query);
     if ($sth && $sth->execute > 0) {
@@ -212,12 +235,14 @@ sub list_open
 
            if ($d[0] != $d_year+1900 || $d[1] != $d_mon+1 || $d[2] != $d_mday) {
                printf "Task not started today.\n";
-               next if $exit eq 1;
+               $today = 0;
+               next if defined $exit && $exit eq 1;
            }
 
            my @t = split(/:/, $arr[1]);
            my $time = sprintf('%02d:%02d', $t[0], $t[1]);
            my $int = quarter(hdiff($t[0], $t[1], $d_hour, $d_min));
+           $int += 24 * 60 unless $today;
 
            printf "%-15s  %s %s (%s)  %s\n", $row->{customer}, $day, $time, min2hour($int), $row->{task};
        }
@@ -299,6 +324,18 @@ sub alter_task
 
     my $term = new Term::ReadLine '';
 
+    my $sql = q{SELECT customer,task FROM stempel WHERE stop IS NULL};
+    my $sth = $dbh->prepare($sql);
+    $sth->execute;
+    my $row = $sth->fetchrow_hashref;
+
+    my $tasks = tasklist $row->{'customer'};
+    $term->addhistory($_) foreach (@$tasks);
+
+    my $attribs = $term->Attribs;
+    $attribs->{completion_word} = $tasks;
+    $term->{completion_function} = sub {return complete_customer $row->{'customer'}, @_};
+
     my $task = $term->readline ('Aufgabe: ');
 
     if (length($task) > 0) {
@@ -319,7 +356,9 @@ stempel  Copyright (c) 2007,8  Martin Schulze <joey\@infodrom.org>
     --close time   close open task
     --help         this text
     --end|-d       terminate task
+    -D             terminate task not started today
     --delete       delete open task
+    --reopen       re-open last task
     --task         alter task content
 END
     exit;
@@ -334,8 +373,10 @@ my %options = ('list:s' => \$opt_list,
               'help' => \&help,
               'close=i' => \$opt_close,
               'delete' => \&delete_task,
+              'reopen' => \&reopen_task,
               'task|t' => \&alter_task,
               'terminate|end|d' => \&close_task,
+              'D' => sub {close_task(1);},
               );
 GetOptions %options;