Implement deletion of the current open task
[infodrom.org/service.infodrom.org] / src / InfoCon / stempel / stempel
1 #! /usr/bin/perl
2
3 #  Time Tracker
4 #
5 #  Copyright (c) 2007,8  Martin Schulze <joey@infodrom.org>
6 #
7 #  This program is free software; you can redistribute it and/or modify
8 #  it under the terms of the GNU General Public License as published by
9 #  the Free Software Foundation; either version 2 of the License, or
10 #  (at your option) any later version.
11 #
12 #  This program is distributed in the hope that it will be useful,
13 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
14 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 #  GNU General Public License for more details.
16 #
17 #  You should have received a copy of the GNU General Public License
18 #  along with this program; if not, write to the Free Software
19 #  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
20
21 use strict;
22 use warnings;
23
24 use DBI;
25 use Getopt::Long;
26 use Term::ReadLine;
27
28 my $table = "stempel";
29 my $engine = "dbi:Pg:dbname=infocon";
30
31 my $dbh = DBI->connect($engine);
32 if (!$dbh) {
33     print "Access to database denied!\n";
34     return 1;
35 }
36 $dbh->do("SET DateStyle = 'ISO'");
37
38 sub min2hour
39 {
40     my $minutes = shift;
41
42     return sprintf('%02d:%02d', $minutes/60, $minutes%60);
43 }
44
45 sub is_open
46 {
47     my $query = q{SELECT count(*) AS count FROM stempel WHERE stop IS NULL};
48     my $sth = $dbh->prepare ($query);
49     if ($sth && $sth->execute > 0) {
50         my $row = $sth->fetchrow_hashref;
51         return $row->{count} > 0 ? 1 : 0;
52     }
53     return 0;
54 }
55
56 sub customerlist
57 {
58     my @res;
59
60     my $query = q{SELECT DISTINCT customer FROM stempel ORDER BY customer};
61     my $sth = $dbh->prepare ($query);
62     if ($sth && $sth->execute > 0) {
63         while ((my $row = $sth->fetchrow_hashref)) {
64             push @res, $row->{customer};
65         }
66     }
67     return @res;
68 }
69
70 sub open_task
71 {
72     my $term = new Term::ReadLine '';
73     my @customers = customerlist;
74
75     $term->addhistory($_) foreach (@customers);
76
77     my $attribs = $term->Attribs;
78     $attribs->{completion_entry_function} = $attribs->{list_completion_function};
79     $attribs->{completion_word} = \@customers;
80
81     printf "[%s]\n", join (', ', @customers);
82
83     my $customer = $term->readline ('Kunde: ');
84
85     if (defined $customer && length($customer) > 0) {
86         $customer =~ s/\s*$//;
87         $attribs->{completion_word} = undef;
88         my $task = $term->readline ('Aufgabe: ');
89         if (length($task) > 0) {
90             my $query = q{INSERT INTO stempel (start,customer,task) VALUES('now()',?,?)};
91             my $sth = $dbh->prepare($query);
92             $sth->execute($customer, $task);
93         }
94     }
95 }
96
97 sub quarter
98 {
99     my $min = shift;
100
101     return 15 if $min < 15;
102
103     return $min - $min%15 if ($min%15 < 3);
104
105     return $min + 15-$min%15;
106 }
107
108 sub hdiff
109 {
110     my ($sh, $sm, $eh, $em) = @_;
111
112     if ($em >= $sm) {
113         return ($eh-$sh)*60 + $em-$sm;
114     } else {
115         return ($eh-$sh)*60 - ($sm-$em);
116     }
117 }
118
119 sub close_task
120 {
121     my ($d_sec,$d_min,$d_hour,$d_mday,$d_mon,$d_year,$d_wday,$d_isdst) = localtime();
122
123     my $query = q{SELECT oid,start FROM stempel WHERE stop IS NULL};
124     my $sth = $dbh->prepare ($query);
125     if ($sth && $sth->execute > 0) {
126         while ((my $row = $sth->fetchrow_hashref)) {
127             my @arr = split(/ /, $row->{start});
128             my @d = split(/-/, $arr[0]);
129
130             if ($d[0] != $d_year+1900 || $d[1] != $d_mon+1 || $d[2] != $d_mday) {
131                 printf "Task not started today, aborting.\n";
132                 next;
133             }
134
135             my @t = split(/:/, $arr[1]);
136             my $int = quarter(hdiff($t[0], $t[1], $d_hour, $d_min));
137
138             $query = sprintf('UPDATE stempel SET stop=now(),time=%d WHERE oid = %d',
139                              $int, $row->{oid});
140             $dbh->do ($query);
141             printf "Wrote %s.\n", min2hour($int);
142         }
143     }
144     exit(0);
145 }
146
147 sub delete_task
148 {
149     my $query = q{DELETE FROM stempel WHERE stop IS NULL};
150     $dbh->do ($query);
151
152     exit(0);
153 }
154
155 sub list_open
156 {
157     my ($d_sec,$d_min,$d_hour,$d_mday,$d_mon,$d_year,$d_wday,$d_isdst) = localtime();
158     my $query = q{SELECT customer,start,task FROM stempel WHERE time IS NULL ORDER BY start,customer};
159
160     my $sth = $dbh->prepare ($query);
161     if ($sth && $sth->execute > 0) {
162         while ((my $row = $sth->fetchrow_hashref)) {
163             my @arr = split(/ /, $row->{start});
164             my $day = $arr[0];
165             my @d = split(/-/, $day);
166
167             if ($d[0] != $d_year+1900 || $d[1] != $d_mon+1 || $d[2] != $d_mday) {
168                 printf "Task not started today, aborting.\n";
169                 next;
170             }
171
172             my @t = split(/:/, $arr[1]);
173             my $time = sprintf('%02d:%02d', $t[0], $t[1]);
174             my $int = quarter(hdiff($t[0], $t[1], $d_hour, $d_min));
175
176             printf "%-15s  %s %s (%s)  %s\n", $row->{customer}, $day, $time, min2hour($int), $row->{task};
177         }
178     }
179     exit 0;
180 }
181
182 sub list
183 {
184     my $month = shift;
185     my ($d_sec,$d_min,$d_hour,$d_mday,$d_mon,$d_year,$d_wday,$d_isdst) = localtime();
186     my $query = q{SELECT start,customer,time,task FROM stempel WHERE time IS NOT NULL };
187     
188     if ($month =~ /^(\d{4})-?(\d\d?)$/) {
189         my $pivot = $1 . '-' . $2 . '-01';
190         $query .= "AND start >= '$pivot' AND start <= '$pivot'::date + interval '1 month' ";
191     } elsif ($month =~ /^(\d{4})$/) {
192         my $pivot = $1 . '-01-01';
193         $query .= "AND start >= '$pivot' AND start <= '$pivot'::date + interval '1 year' ";
194     } elsif ($month =~ /^(\d\d?)$/) {
195         my $pivot = $d_year+1900 . '-' . $1 . '-01';
196         $query .= "AND start >= '$pivot' AND start <= '$pivot'::date + interval '1 month' ";
197     } elsif ($month !~ /^all$/) {
198         my $pivot = sprintf('%04d-%02d-01', $d_year+1900, $d_mon+1);
199         $query .= "AND start >= '$pivot' AND start <= '$pivot'::date + interval '1 month' ";
200     }
201     $query .= "ORDER BY customer,start";
202
203     my $sth = $dbh->prepare ($query);
204
205     if ($sth && $sth->execute > 0) {
206         while ((my $row = $sth->fetchrow_hashref)) {
207             if (defined $row->{time}) {
208                 my $day = (split(/ /, $row->{start}))[0];
209                 my $time = min2hour $row->{time};
210                 printf "%-15s  %s  %s  %s\n", $row->{customer}, $day, $time, $row->{task};
211             }
212         }
213     }
214     exit 0;
215 }
216
217 sub toggle_task
218 {
219     if (is_open) {
220         list_open;
221     } else {
222         open_task;
223     }
224 }
225
226 sub late_close_task
227 {
228     my $delta = shift;
229     my ($d_sec,$d_min,$d_hour,$d_mday,$d_mon,$d_year,$d_wday,$d_isdst) = localtime();
230
231     return unless is_open;
232
233     my $query = sprintf("UPDATE stempel SET stop=start + interval '%d minutes',time=%d WHERE stop IS NULL",
234                         $delta, quarter($delta));
235     $dbh->do($query);
236 }
237
238 sub help
239 {
240     print <<"END";
241 stempel  Copyright (c) 2007,8  Martin Schulze <joey\@infodrom.org>
242     --list [month] list month [default=current|all]
243     --open         list open
244     --close time   close open task
245     --help         this text
246     --end|-d       terminate task
247     --delete       delete open task
248 END
249     exit;
250 }
251
252 my $opt_close = undef;
253 my $opt_list = undef;
254 my %options = ('list:s' => \$opt_list,
255                'open' => \&list_open,
256                'help' => \&help,
257                'close=i' => \$opt_close,
258                'delete' => \&delete_task,
259                'terminate|end|d' => \&close_task,
260                );
261 GetOptions %options;
262
263 if ($opt_close) {
264     late_close_task $opt_close;
265 } elsif (defined $opt_list) {
266     list $opt_list;
267 } else {
268     toggle_task;
269 }