A bit of refactoring in report.php for timesheet assignment.
[timetracker.git] / WEB-INF / lib / ttTimesheetHelper.class.php
1 <?php
2 // +----------------------------------------------------------------------+
3 // | Anuko Time Tracker
4 // +----------------------------------------------------------------------+
5 // | Copyright (c) Anuko International Ltd. (https://www.anuko.com)
6 // +----------------------------------------------------------------------+
7 // | LIBERAL FREEWARE LICENSE: This source code document may be used
8 // | by anyone for any purpose, and freely redistributed alone or in
9 // | combination with other software, provided that the license is obeyed.
10 // |
11 // | There are only two ways to violate the license:
12 // |
13 // | 1. To redistribute this code in source form, with the copyright
14 // |    notice or license removed or altered. (Distributing in compiled
15 // |    forms without embedded copyright notices is permitted).
16 // |
17 // | 2. To redistribute modified versions of this code in *any* form
18 // |    that bears insufficient indications that the modifications are
19 // |    not the work of the original author(s).
20 // |
21 // | This license applies to this document only, not any other software
22 // | that it may be combined with.
23 // |
24 // +----------------------------------------------------------------------+
25 // | Contributors:
26 // | https://www.anuko.com/time_tracker/credits.htm
27 // +----------------------------------------------------------------------+
28
29 import('ttUserHelper');
30
31 // Class ttTimesheetHelper is used to help with project related tasks.
32 class ttTimesheetHelper {
33
34   // The getTimesheetByName looks up a project by name.
35   static function getTimesheetByName($name) {
36     global $user;
37     $mdb2 = getConnection();
38
39     $user_id = $user->getUser();
40     $group_id = $user->getGroup();
41     $org_id = $user->org_id;
42
43     $sql = "select id from tt_timesheets".
44       " where group_id = $group_id and org_id = $org_id and user_id = $user_id and name = ".$mdb2->quote($name).
45       " and status is not null";
46     $res = $mdb2->query($sql);
47     if (!is_a($res, 'PEAR_Error')) {
48       $val = $res->fetchRow();
49       if ($val && $val['id'])
50         return $val;
51     }
52     return false;
53   }
54
55   // createTimesheet function creates a new timesheet.
56   static function createTimesheet($fields)
57   {
58     // Create a new timesheet entry.
59     global $user;
60     $mdb2 = getConnection();
61
62     $user_id = $user->getUser();
63     $group_id = $user->getGroup();
64     $org_id = $user->org_id;
65
66     $client_id = $fields['client_id'];
67     $project_id = $fields['project_id'];
68     $name = $fields['name'];
69     $comment = $fields['comment'];
70
71     $start_date = new DateAndTime($user->date_format, $fields['start_date']);
72     $start = $start_date->toString(DB_DATEFORMAT);
73
74     $end_date = new DateAndTime($user->date_format, $fields['end_date']);
75     $end = $end_date->toString(DB_DATEFORMAT);
76
77     $sql = "insert into tt_timesheets (user_id, group_id, org_id, client_id, project_id, name, comment, start_date, end_date)".
78       " values ($user_id, $group_id, $org_id, ".$mdb2->quote($client_id).", ".$mdb2->quote($project_id).", ".$mdb2->quote($name).
79       ", ".$mdb2->quote($comment).", ".$mdb2->quote($start).", ".$mdb2->quote($end).")";
80     $affected = $mdb2->exec($sql);
81     if (is_a($affected, 'PEAR_Error'))
82       return false;
83
84     $last_id = $mdb2->lastInsertID('tt_timesheets', 'id');
85
86     // Associate tt_log items with timesheet.
87     if (isset($fields['client'])) $client_id = (int) $fields['client_id'];
88     if (isset($fields['project_id'])) $project_id = (int) $fields['project_id'];
89     // sql parts.
90     if ($client_id) $client_part = " and client_id = $client_id";
91     if ($project_id) $project_part = " and project_id = $project_id";
92
93     $sql = "update tt_log set timesheet_id = $last_id".
94       " where status = 1 $client_part $project_part and timesheet_id is null".
95       " and date >= ".$mdb2->quote($start)." and date <= ".$mdb2->quote($end).
96       " and user_id = $user_id and group_id = $group_id and org_id = $org_id";
97     $affected = $mdb2->exec($sql);
98     if (is_a($affected, 'PEAR_Error'))
99       return false;
100
101     return $last_id;
102   }
103
104   // The getActiveTimesheets obtains active timesheets for a user.
105   static function getActiveTimesheets()
106   {
107     global $user;
108     $mdb2 = getConnection();
109
110     $user_id = $user->getUser();
111     $group_id = $user->getGroup();
112     $org_id = $user->org_id;
113
114     $result = array();
115     $sql = "select ts.id, ts.name, ts.client_id, c.name as client_name,".
116       " ts.submit_status, ts.approve_status from tt_timesheets ts".
117       " left join tt_clients c on (c.id = ts.client_id)".
118       " where ts.status = 1 and ts.group_id = $group_id and ts.org_id = $org_id and ts.user_id = $user_id".
119       " order by ts.name";
120     $res = $mdb2->query($sql);
121     $result = array();
122     if (!is_a($res, 'PEAR_Error')) {
123       while ($val = $res->fetchRow()) {
124         $result[] = $val;
125       }
126     }
127     return $result;
128   }
129
130   // The getInactiveTimesheets obtains inactive timesheets for a user.
131   static function getInactiveTimesheets()
132   {
133     global $user;
134     $mdb2 = getConnection();
135
136     $user_id = $user->getUser();
137     $group_id = $user->getGroup();
138     $org_id = $user->org_id;
139
140     $result = array();
141     $sql = "select ts.id, ts.name, ts.client_id, c.name as client_name,".
142       " ts.submit_status, ts.approve_status from tt_timesheets ts".
143       " left join tt_clients c on (c.id = ts.client_id)".
144       " where ts.status = 0 and ts.group_id = $group_id and ts.org_id = $org_id and ts.user_id = $user_id".
145       " order by ts.name";
146     $res = $mdb2->query($sql);
147     $result = array();
148     if (!is_a($res, 'PEAR_Error')) {
149       while ($val = $res->fetchRow()) {
150         $result[] = $val;
151       }
152     }
153     return $result;
154   }
155
156   // getTimesheet - obtains timesheet data from the database.
157   static function getTimesheet($timesheet_id) {
158     global $user;
159     $mdb2 = getConnection();
160
161     $user_id = $user->getUser();
162     $group_id = $user->getGroup();
163     $org_id = $user->org_id;
164
165     $sql = "select ts.*, u.name as user_name, c.name as client_name,".
166       " p.name as project_name from tt_timesheets ts".
167       " left join tt_users u on (ts.user_id = u.id)".
168       " left join tt_clients c on (ts.client_id = c.id)".
169       " left join tt_projects p on (ts.project_id = p.id)".
170       " where ts.id = $timesheet_id and ts.user_id = $user_id and ts.group_id = $group_id and ts.org_id = $org_id and ts.status is not null";
171     $res = $mdb2->query($sql);
172     if (!is_a($res, 'PEAR_Error')) {
173       if ($val = $res->fetchRow())
174         return $val;
175     }
176     return false;
177   }
178
179   // delete - deletes timesheet from the database.
180   static function delete($timesheet_id) {
181     global $user;
182     $mdb2 = getConnection();
183
184     $user_id = $user->getUser();
185     $group_id = $user->getGroup();
186     $org_id = $user->org_id;
187
188     // Handle tt_log records.
189     $sql = "update tt_log set timesheet_id = null".
190       " where timesheet_id = $timesheet_id and user_id = $user_id and group_id = $group_id and org_id = $org_id";
191     $affected = $mdb2->exec($sql);
192     if (is_a($affected, 'PEAR_Error')) return false;
193
194     // Delete timesheet.
195     $sql = "update tt_timesheets set status = null".
196       " where id = $timesheet_id and user_id = $user_id and group_id = $group_id and org_id = $org_id";
197     $affected = $mdb2->exec($sql);
198     return (!is_a($affected, 'PEAR_Error'));
199   }
200
201   // update function - updates the timesheet in database.
202   static function update($fields) {
203     global $user;
204     $mdb2 = getConnection();
205
206     $user_id = $user->getUser();
207     $group_id = $user->getGroup();
208     $org_id = $user->org_id;
209
210     $timesheet_id = $fields['id']; // Timesheet we are updating.
211     $name = $fields['name']; // Timesheet name.
212     $comment = $fields['comment'];
213     $status = $fields['status']; // Timesheet status.
214
215     $sql = "update tt_timesheets set name = ".$mdb2->quote($name).", comment = ".$mdb2->quote($comment).
216       ", status = ".$mdb2->quote($status).
217       " where id = $timesheet_id and user_id = $user_id and group_id = $group_id and org_id = $org_id";
218     $affected = $mdb2->exec($sql);
219     return (!is_a($affected, 'PEAR_Error'));
220   }
221
222   // getReportOptions prepares $options array to be used with ttReportHelper
223   // to obtain items for timesheet view.
224   static function getReportOptions($timesheet) {
225     global $user;
226     $group_by_client = $user->isPluginEnabled('cl') && !$timesheet['client_id'];
227     $trackingMode = $user->getTrackingMode();
228     $group_by_project = MODE_PROJECTS == $trackingMode || MODE_PROJECTS_AND_TASKS == $trackingMode;
229
230     $options['timesheet_id'] = $timesheet['id'];
231     $options['group_by1'] = 'date';
232     if ($group_by_client || $group_by_project) {
233       $options['group_by2'] = $group_by_client ? 'client' : 'project';
234     }
235     if ($options['group_by2'] && $options['group_by2'] != 'project' && $group_by_project) {
236       $options['group_by3'] = 'project';
237     }
238     return $options;
239   }
240
241   // getApprovers obtains a list of users who can approve a timesheet for a given user
242   // and also have an email to receive a notification about it.
243   static function getApprovers() {
244     global $user;
245     $mdb2 = getConnection();
246
247     $user_id = $user->getUser();
248     $group_id = $user->getGroup();
249     $org_id = $user->org_id;
250
251     $approvers = array();
252     $rank = ttUserHelper::getUserRank($user_id);
253     $sql = "select u.id, u.name, u.email".
254       " from tt_users u".
255       " left join tt_roles r on (r.id = u.role_id)".
256       " where u.status = 1 and u.email is not null and u.group_id = $group_id and u.org_id = $org_id".
257       " and (r.rank > $rank and r.rights like '%approve_timesheets%')";
258     $res = $mdb2->query($sql);
259     if (!is_a($res, 'PEAR_Error')) {
260       while ($val = $res->fetchRow()) {
261         $approvers[] = $val;
262       }
263     }
264     return $approvers;
265   }
266
267   // submitTimesheet marks a timesheet as submitted and also sends an email
268   // to a selected approver.
269   static function submitTimesheet($fields) {
270     global $user;
271     $mdb2 = getConnection();
272
273     $user_id = $user->getUser();
274     $group_id = $user->getGroup();
275     $org_id = $user->org_id;
276
277     // First, mark timesheet as submitted.
278     // Even if mail part below does not work, this will get us a functioning workflow
279     // without email notification.
280     $timesheet_id = $fields['timesheet_id'];
281     $sql = "update tt_timesheets set submit_status = 1".
282       " where id = $timesheet_id and user_id = $user_id and group_id = $group_id and org_id = $org_id";
283     $affected = $mdb2->exec($sql);
284     if (is_a($affected, 'PEAR_Error')) return false;
285
286     // TODO: send email to approver here...
287     // $approver_id = $fields['approver_id'];
288
289     return true;
290   }
291
292   // approveTimesheet marks a timesheet as approved and sends an email to submitter.
293   static function approveTimesheet($fields) {
294     global $user;
295     $mdb2 = getConnection();
296
297     $user_id = $user->getUser();
298     $group_id = $user->getGroup();
299     $org_id = $user->org_id;
300
301     // First, mark timesheet as approved.
302     // Even if mail part below does not work, this will get us a functioning workflow
303     // without email notification.
304     $timesheet_id = $fields['timesheet_id'];
305     $comment = $fields['comment'];
306
307     $sql = "update tt_timesheets set approve_status = 1, approve_comment = ".$mdb2->quote($comment).
308       " where id = $timesheet_id and submit_status = 1 and user_id = $user_id and group_id = $group_id and org_id = $org_id";
309     $affected = $mdb2->exec($sql);
310     if (is_a($affected, 'PEAR_Error')) return false;
311
312     // TODO: send email to submitter here...
313     return true;
314   }
315
316   // disapproveTimesheet marks a timesheet as approved and sends an email to submitter.
317   static function disapproveTimesheet($fields) {
318     global $user;
319     $mdb2 = getConnection();
320
321     $user_id = $user->getUser();
322     $group_id = $user->getGroup();
323     $org_id = $user->org_id;
324
325     // First, mark timesheet as disapproved.
326     // Even if mail part below does not work, this will get us a functioning workflow
327     // without email notification.
328     $timesheet_id = $fields['timesheet_id'];
329     $comment = $fields['comment'];
330
331     $sql = "update tt_timesheets set approve_status = 0, approve_comment = ".$mdb2->quote($comment).
332       " where id = $timesheet_id and submit_status = 1 and user_id = $user_id and group_id = $group_id and org_id = $org_id";
333     $affected = $mdb2->exec($sql);
334     if (is_a($affected, 'PEAR_Error')) return false;
335
336     // TODO: send email to submitter here...
337     return true;
338   }
339
340   // The timesheetItemsExist determines whether tt_log records exist in the specified period
341   // for inclusion in a new timesheet.
342   static function timesheetItemsExist($fields) {
343     global $user;
344     $mdb2 = getConnection();
345
346     $user_id = $user->getUser();
347     $group_id = $user->getGroup();
348     $org_id = $user->org_id;
349
350     if (isset($fields['client_id'])) $client_id = (int) $fields['client_id'];
351     if (isset($fields['project_id'])) $project_id = (int) $fields['project_id'];
352
353     $start_date = new DateAndTime($user->date_format, $fields['start_date']);
354     $start = $start_date->toString(DB_DATEFORMAT);
355
356     $end_date = new DateAndTime($user->date_format, $fields['end_date']);
357     $end = $end_date->toString(DB_DATEFORMAT);
358
359     // sql parts.
360     if ($client_id) $client_part = " and client_id = $client_id";
361     if ($project_id) $project_part = " and project_id = $project_id";
362
363     $sql = "select count(*) as num from tt_log".
364       " where status = 1 $client_part $project_part and timesheet_id is null".
365       " and date >= ".$mdb2->quote($start)." and date <= ".$mdb2->quote($end).
366       " and user_id = $user_id and group_id = $group_id and org_id = $org_id";
367     $res = $mdb2->query($sql);
368     if (!is_a($res, 'PEAR_Error')) {
369       $val = $res->fetchRow();
370       if ($val['num']) {
371         return true;
372       }
373     }
374
375     return false;
376   }
377
378   // The overlaps function determines if a new timesheet overlaps with
379   // an already existing timesheet.
380   static function overlaps($fields) {
381     global $user;
382     $mdb2 = getConnection();
383
384     $user_id = $user->getUser();
385     $group_id = $user->getGroup();
386     $org_id = $user->org_id;
387
388     if (isset($fields['client_id'])) $client_id = (int) $fields['client_id'];
389     if (isset($fields['project_id'])) $project_id = (int) $fields['project_id'];
390
391     $start_date = new DateAndTime($user->date_format, $fields['start_date']);
392     $start = $start_date->toString(DB_DATEFORMAT);
393     $quoted_start = $mdb2->quote($start);
394
395     $end_date = new DateAndTime($user->date_format, $fields['end_date']);
396     $end = $end_date->toString(DB_DATEFORMAT);
397     $quoted_end = $mdb2->quote($end);
398
399     // sql parts.
400     if ($client_id) $client_part = " and client_id = $client_id";
401     if ($project_id) $project_part = " and project_id = $project_id";
402
403     $sql = "select id from tt_timesheets".
404       " where status is not null $client_part $project_part".
405       " and (($quoted_start >= start_date and $quoted_start <= end_date)".
406       "   or ($quoted_end >= start_date and $quoted_end <= end_date))".
407       " and user_id = $user_id and group_id = $group_id and org_id = $org_id";
408     $res = $mdb2->query($sql);
409     if (!is_a($res, 'PEAR_Error')) {
410       $val = $res->fetchRow();
411       if ($val['id']) {
412         return true;
413       }
414     }
415     return false;
416   }
417
418   // The canAssign function determines if we can show controls on a report page
419   // for timesheet assignment.
420   //
421   // Conditions:
422   // - Report date range, client_id, and project_id match an existing timesheet
423   //   with approved_status null.
424   static function canAssign($options) {
425     return false;
426   }
427 }