Bug 12063: Change date calculation for reserve expiration to skip all holiday
[koha.git] / Koha / Calendar.pm
1 package Koha::Calendar;
2 use strict;
3 use warnings;
4 use 5.010;
5
6 use DateTime;
7 use DateTime::Set;
8 use DateTime::Duration;
9 use C4::Context;
10 use Koha::Caches;
11 use Carp;
12
13 sub new {
14     my ( $classname, %options ) = @_;
15     my $self = {};
16     bless $self, $classname;
17     for my $o_name ( keys %options ) {
18         my $o = lc $o_name;
19         $self->{$o} = $options{$o_name};
20     }
21     if ( !defined $self->{branchcode} ) {
22         croak 'No branchcode argument passed to Koha::Calendar->new';
23     }
24     $self->_init();
25     return $self;
26 }
27
28 sub _init {
29     my $self       = shift;
30     my $branch     = $self->{branchcode};
31     my $dbh        = C4::Context->dbh();
32     my $weekly_closed_days_sth = $dbh->prepare(
33 'SELECT weekday FROM repeatable_holidays WHERE branchcode = ? AND weekday IS NOT NULL'
34     );
35     $weekly_closed_days_sth->execute( $branch );
36     $self->{weekly_closed_days} = [ 0, 0, 0, 0, 0, 0, 0 ];
37     while ( my $tuple = $weekly_closed_days_sth->fetchrow_hashref ) {
38         $self->{weekly_closed_days}->[ $tuple->{weekday} ] = 1;
39     }
40     my $day_month_closed_days_sth = $dbh->prepare(
41 'SELECT day, month FROM repeatable_holidays WHERE branchcode = ? AND weekday IS NULL'
42     );
43     $day_month_closed_days_sth->execute( $branch );
44     $self->{day_month_closed_days} = {};
45     while ( my $tuple = $day_month_closed_days_sth->fetchrow_hashref ) {
46         $self->{day_month_closed_days}->{ $tuple->{month} }->{ $tuple->{day} } =
47           1;
48     }
49
50     $self->{days_mode}       = C4::Context->preference('useDaysMode');
51     $self->{test}            = 0;
52     return;
53 }
54
55 sub exception_holidays {
56     my ( $self ) = @_;
57
58     my $cache  = Koha::Caches->get_instance();
59     my $cached = $cache->get_from_cache('exception_holidays');
60     return $cached if $cached;
61
62     my $dbh = C4::Context->dbh;
63     my $branch = $self->{branchcode};
64     my $exception_holidays_sth = $dbh->prepare(
65 'SELECT day, month, year FROM special_holidays WHERE branchcode = ? AND isexception = 1'
66     );
67     $exception_holidays_sth->execute( $branch );
68     my $dates = [];
69     while ( my ( $day, $month, $year ) = $exception_holidays_sth->fetchrow ) {
70         push @{$dates},
71           DateTime->new(
72             day       => $day,
73             month     => $month,
74             year      => $year,
75             time_zone => "floating",
76           )->truncate( to => 'day' );
77     }
78     $self->{exception_holidays} =
79       DateTime::Set->from_datetimes( dates => $dates );
80     $cache->set_in_cache( 'exception_holidays', $self->{exception_holidays} );
81     return $self->{exception_holidays};
82 }
83
84 sub single_holidays {
85     my ( $self, $date ) = @_;
86     my $branchcode = $self->{branchcode};
87     my $cache           = Koha::Caches->get_instance();
88     my $single_holidays = $cache->get_from_cache('single_holidays');
89
90     # $single_holidays looks like:
91     # {
92     #   CPL =>  [
93     #        [0] 20131122,
94     #         ...
95     #    ],
96     #   ...
97     # }
98
99     unless ($single_holidays) {
100         my $dbh = C4::Context->dbh;
101         $single_holidays = {};
102
103         # push holidays for each branch
104         my $branches_sth =
105           $dbh->prepare('SELECT distinct(branchcode) FROM special_holidays');
106         $branches_sth->execute();
107         while ( my $br = $branches_sth->fetchrow ) {
108             my $single_holidays_sth = $dbh->prepare(
109 'SELECT day, month, year FROM special_holidays WHERE branchcode = ? AND isexception = 0'
110             );
111             $single_holidays_sth->execute($br);
112
113             my @ymd_arr;
114             while ( my ( $day, $month, $year ) =
115                 $single_holidays_sth->fetchrow )
116             {
117                 my $dt = DateTime->new(
118                     day       => $day,
119                     month     => $month,
120                     year      => $year,
121                     time_zone => 'floating',
122                 )->truncate( to => 'day' );
123                 push @ymd_arr, $dt->ymd('');
124             }
125             $single_holidays->{$br} = \@ymd_arr;
126         }    # br
127         $cache->set_in_cache( 'single_holidays', $single_holidays,
128             { expiry => 76800 } )    #24 hrs ;
129     }
130     my $holidays  = ( $single_holidays->{$branchcode} );
131     for my $hols  (@$holidays ) {
132             return 1 if ( $date == $hols )   #match ymds;
133     }
134     return 0;
135 }
136
137 sub addDate {
138     my ( $self, $startdate, $add_duration, $unit ) = @_;
139
140     # Default to days duration (legacy support I guess)
141     if ( ref $add_duration ne 'DateTime::Duration' ) {
142         $add_duration = DateTime::Duration->new( days => $add_duration );
143     }
144
145     $unit ||= 'days'; # default days ?
146     my $dt;
147
148     if ( $unit eq 'hours' ) {
149         # Fixed for legacy support. Should be set as a branch parameter
150         my $return_by_hour = 10;
151
152         $dt = $self->addHours($startdate, $add_duration, $return_by_hour);
153     } else {
154         # days
155         $dt = $self->addDays($startdate, $add_duration);
156     }
157
158     return $dt;
159 }
160
161 sub addHours {
162     my ( $self, $startdate, $hours_duration, $return_by_hour ) = @_;
163     my $base_date = $startdate->clone();
164
165     $base_date->add_duration($hours_duration);
166
167     # If we are using the calendar behave for now as if Datedue
168     # was the chosen option (current intended behaviour)
169
170     if ( $self->{days_mode} ne 'Days' &&
171           $self->is_holiday($base_date) ) {
172
173         if ( $hours_duration->is_negative() ) {
174             $base_date = $self->prev_open_day($base_date);
175         } else {
176             $base_date = $self->next_open_day($base_date);
177         }
178
179         $base_date->set_hour($return_by_hour);
180
181     }
182
183     return $base_date;
184 }
185
186 sub addDays {
187     my ( $self, $startdate, $days_duration ) = @_;
188     my $base_date = $startdate->clone();
189
190     $self->{days_mode} ||= q{};
191
192     if ( $self->{days_mode} eq 'Calendar' ) {
193         # use the calendar to skip all days the library is closed
194         # when adding
195         my $days = abs $days_duration->in_units('days');
196
197         if ( $days_duration->is_negative() ) {
198             while ($days) {
199                 $base_date = $self->prev_open_day($base_date);
200                 --$days;
201             }
202         } else {
203             while ($days) {
204                 $base_date = $self->next_open_day($base_date);
205                 --$days;
206             }
207         }
208
209     } else { # Days or Datedue
210         # use straight days, then use calendar to push
211         # the date to the next open day if Datedue
212         $base_date->add_duration($days_duration);
213
214         if ( $self->{days_mode} eq 'Datedue' ) {
215             # Datedue, then use the calendar to push
216             # the date to the next open day if holiday
217             if ( $self->is_holiday($base_date) ) {
218
219                 if ( $days_duration->is_negative() ) {
220                     $base_date = $self->prev_open_day($base_date);
221                 } else {
222                     $base_date = $self->next_open_day($base_date);
223                 }
224             }
225         }
226     }
227
228     return $base_date;
229 }
230
231 sub is_holiday {
232     my ( $self, $dt ) = @_;
233
234     my $localdt = $dt->clone();
235     my $day   = $localdt->day;
236     my $month = $localdt->month;
237
238     #Change timezone to "floating" before doing any calculations or comparisons
239     $localdt->set_time_zone("floating");
240     $localdt->truncate( to => 'day' );
241
242
243     if ( $self->exception_holidays->contains($localdt) ) {
244         # exceptions are not holidays
245         return 0;
246     }
247
248     my $dow = $localdt->day_of_week;
249     # Representation fix
250     # DateTime object dow (1-7) where Monday is 1
251     # Arrays are 0-based where 0 = Sunday, not 7.
252     if ( $dow == 7 ) {
253         $dow = 0;
254     }
255
256     if ( $self->{weekly_closed_days}->[$dow] == 1 ) {
257         return 1;
258     }
259
260     if ( exists $self->{day_month_closed_days}->{$month}->{$day} ) {
261         return 1;
262     }
263
264     my $ymd   = $localdt->ymd('')  ;
265     if ($self->single_holidays(  $ymd  ) == 1 ) {
266         return 1;
267     }
268
269     # damn have to go to work after all
270     return 0;
271 }
272
273 sub next_open_day {
274     my ( $self, $dt ) = @_;
275     my $base_date = $dt->clone();
276
277     $base_date->add(days => 1);
278
279     while ($self->is_holiday($base_date)) {
280         $base_date->add(days => 1);
281     }
282
283     return $base_date;
284 }
285
286 sub prev_open_day {
287     my ( $self, $dt ) = @_;
288     my $base_date = $dt->clone();
289
290     $base_date->add(days => -1);
291
292     while ($self->is_holiday($base_date)) {
293         $base_date->add(days => -1);
294     }
295
296     return $base_date;
297 }
298
299 sub days_forward {
300     my $self     = shift;
301     my $start_dt = shift;
302     my $num_days = shift;
303
304     my $base_dt = $start_dt->clone();
305
306     while ($num_days--) {
307         $base_dt = $self->next_open_day($base_dt);
308     }
309
310     return $base_dt;
311 }
312
313 sub days_between {
314     my $self     = shift;
315     my $start_dt = shift;
316     my $end_dt   = shift;
317
318     if ( $start_dt->compare($end_dt) > 0 ) {
319         # swap dates
320         my $int_dt = $end_dt;
321         $end_dt = $start_dt;
322         $start_dt = $int_dt;
323     }
324
325
326     # start and end should not be closed days
327     my $days = $start_dt->delta_days($end_dt)->delta_days;
328     for (my $dt = $start_dt->clone();
329         $dt <= $end_dt;
330         $dt->add(days => 1)
331     ) {
332         if ($self->is_holiday($dt)) {
333             $days--;
334         }
335     }
336     return DateTime::Duration->new( days => $days );
337
338 }
339
340 sub hours_between {
341     my ($self, $start_date, $end_date) = @_;
342     my $start_dt = $start_date->clone();
343     my $end_dt = $end_date->clone();
344     my $duration = $end_dt->delta_ms($start_dt);
345     $start_dt->truncate( to => 'day' );
346     $end_dt->truncate( to => 'day' );
347     # NB this is a kludge in that it assumes all days are 24 hours
348     # However for hourly loans the logic should be expanded to
349     # take into account open/close times then it would be a duration
350     # of library open hours
351     my $skipped_days = 0;
352     for (my $dt = $start_dt->clone();
353         $dt <= $end_dt;
354         $dt->add(days => 1)
355     ) {
356         if ($self->is_holiday($dt)) {
357             ++$skipped_days;
358         }
359     }
360     if ($skipped_days) {
361         $duration->subtract_duration(DateTime::Duration->new( hours => 24 * $skipped_days));
362     }
363
364     return $duration;
365
366 }
367
368 sub set_daysmode {
369     my ( $self, $mode ) = @_;
370
371     # if not testing this is a no op
372     if ( $self->{test} ) {
373         $self->{days_mode} = $mode;
374     }
375
376     return;
377 }
378
379 sub clear_weekly_closed_days {
380     my $self = shift;
381     $self->{weekly_closed_days} = [ 0, 0, 0, 0, 0, 0, 0 ];    # Sunday only
382     return;
383 }
384
385 1;
386 __END__
387
388 =head1 NAME
389
390 Koha::Calendar - Object containing a branches calendar
391
392 =head1 SYNOPSIS
393
394   use Koha::Calendar
395
396   my $c = Koha::Calendar->new( branchcode => 'MAIN' );
397   my $dt = DateTime->now();
398
399   # are we open
400   $open = $c->is_holiday($dt);
401   # when will item be due if loan period = $dur (a DateTime::Duration object)
402   $duedate = $c->addDate($dt,$dur,'days');
403
404
405 =head1 DESCRIPTION
406
407   Implements those features of C4::Calendar needed for Staffs Rolling Loans
408
409 =head1 METHODS
410
411 =head2 new : Create a calendar object
412
413 my $calendar = Koha::Calendar->new( branchcode => 'MAIN' );
414
415 The option branchcode is required
416
417
418 =head2 addDate
419
420     my $dt = $calendar->addDate($date, $dur, $unit)
421
422 C<$date> is a DateTime object representing the starting date of the interval.
423
424 C<$offset> is a DateTime::Duration to add to it
425
426 C<$unit> is a string value 'days' or 'hours' toflag granularity of duration
427
428 Currently unit is only used to invoke Staffs return Monday at 10 am rule this
429 parameter will be removed when issuingrules properly cope with that
430
431
432 =head2 addHours
433
434     my $dt = $calendar->addHours($date, $dur, $return_by_hour )
435
436 C<$date> is a DateTime object representing the starting date of the interval.
437
438 C<$offset> is a DateTime::Duration to add to it
439
440 C<$return_by_hour> is an integer value representing the opening hour for the branch
441
442
443 =head2 addDays
444
445     my $dt = $calendar->addDays($date, $dur)
446
447 C<$date> is a DateTime object representing the starting date of the interval.
448
449 C<$offset> is a DateTime::Duration to add to it
450
451 C<$unit> is a string value 'days' or 'hours' toflag granularity of duration
452
453 Currently unit is only used to invoke Staffs return Monday at 10 am rule this
454 parameter will be removed when issuingrules properly cope with that
455
456
457 =head2 single_holidays
458
459 my $rc = $self->single_holidays(  $ymd  );
460
461 Passed a $date in Ymd (yyyymmdd) format -  returns 1 if date is a single_holiday, or 0 if not.
462
463
464 =head2 is_holiday
465
466 $yesno = $calendar->is_holiday($dt);
467
468 passed a DateTime object returns 1 if it is a closed day
469 0 if not according to the calendar
470
471 =head2 days_between
472
473 $duration = $calendar->days_between($start_dt, $end_dt);
474
475 Passed two dates returns a DateTime::Duration object measuring the length between them
476 ignoring closed days. Always returns a positive number irrespective of the
477 relative order of the parameters
478
479 =head2 next_open_day
480
481 $datetime = $calendar->next_open_day($duedate_dt)
482
483 Passed a Datetime returns another Datetime representing the next open day. It is
484 intended for use to calculate the due date when useDaysMode syspref is set to either
485 'Datedue' or 'Calendar'.
486
487 =head2 prev_open_day
488
489 $datetime = $calendar->prev_open_day($duedate_dt)
490
491 Passed a Datetime returns another Datetime representing the previous open day. It is
492 intended for use to calculate the due date when useDaysMode syspref is set to either
493 'Datedue' or 'Calendar'.
494
495 =head2 set_daysmode
496
497 For testing only allows the calling script to change days mode
498
499 =head2 clear_weekly_closed_days
500
501 In test mode changes the testing set of closed days to a new set with
502 no closed days. TODO passing an array of closed days to this would
503 allow testing of more configurations
504
505 =head2 add_holiday
506
507 Passed a datetime object this will add it to the calendar's list of
508 closed days. This is for testing so that we can alter the Calenfar object's
509 list of specified dates
510
511 =head1 DIAGNOSTICS
512
513 Will croak if not passed a branchcode in new
514
515 =head1 BUGS AND LIMITATIONS
516
517 This only contains a limited subset of the functionality in C4::Calendar
518 Only enough to support Staffs Rolling loans
519
520 =head1 AUTHOR
521
522 Colin Campbell colin.campbell@ptfs-europe.com
523
524 =head1 LICENSE AND COPYRIGHT
525
526 Copyright (c) 2011 PTFS-Europe Ltd All rights reserved
527
528 This program is free software: you can redistribute it and/or modify
529 it under the terms of the GNU General Public License as published by
530 the Free Software Foundation, either version 2 of the License, or
531 (at your option) any later version.
532
533 This program is distributed in the hope that it will be useful,
534 but WITHOUT ANY WARRANTY; without even the implied warranty of
535 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
536 GNU General Public License for more details.
537
538 You should have received a copy of the GNU General Public License
539 along with this program.  If not, see <http://www.gnu.org/licenses/>.