Bug 23382: (RM follow-up) Revert "hours_between should match..."
[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     if ( $unit eq 'hours' ) {
148         # Fixed for legacy support. Should be set as a branch parameter
149         my $return_by_hour = 10;
150
151         $dt = $self->addHours($startdate, $add_duration, $return_by_hour);
152     } else {
153         # days
154         $dt = $self->addDays($startdate, $add_duration);
155     }
156     return $dt;
157 }
158
159 sub addHours {
160     my ( $self, $startdate, $hours_duration, $return_by_hour ) = @_;
161     my $base_date = $startdate->clone();
162
163     $base_date->add_duration($hours_duration);
164
165     # If we are using the calendar behave for now as if Datedue
166     # was the chosen option (current intended behaviour)
167
168     if ( $self->{days_mode} ne 'Days' &&
169           $self->is_holiday($base_date) ) {
170
171         if ( $hours_duration->is_negative() ) {
172             $base_date = $self->prev_open_days($base_date, 1);
173         } else {
174             $base_date = $self->next_open_days($base_date, 1);
175         }
176
177         $base_date->set_hour($return_by_hour);
178
179     }
180
181     return $base_date;
182 }
183
184 sub addDays {
185     my ( $self, $startdate, $days_duration ) = @_;
186     my $base_date = $startdate->clone();
187
188     $self->{days_mode} ||= q{};
189
190     if ( $self->{days_mode} eq 'Calendar' ) {
191         # use the calendar to skip all days the library is closed
192         # when adding
193         my $days = abs $days_duration->in_units('days');
194
195         if ( $days_duration->is_negative() ) {
196             while ($days) {
197                 $base_date = $self->prev_open_days($base_date, 1);
198                 --$days;
199             }
200         } else {
201             while ($days) {
202                 $base_date = $self->next_open_days($base_date, 1);
203                 --$days;
204             }
205         }
206
207     } else { # Days, Datedue or Dayweek
208         # use straight days, then use calendar to push
209         # the date to the next open day as appropriate
210         # if Datedue or Dayweek
211         $base_date->add_duration($days_duration);
212
213         if ( $self->{days_mode} eq 'Datedue' ||
214             $self->{days_mode} eq 'Dayweek') {
215             # Datedue or Dayweek, then use the calendar to push
216             # the date to the next open day if holiday
217             if ( $self->is_holiday($base_date) ) {
218                 my $dow = $base_date->day_of_week;
219                 my $days = $days_duration->in_units('days');
220                 # Is it a period based on weeks
221                 my $push_amt = $days % 7 == 0 ?
222                     $self->get_push_amt($base_date) : 1;
223                 if ( $days_duration->is_negative() ) {
224                     $base_date = $self->prev_open_days($base_date, $push_amt);
225                 } else {
226                     $base_date = $self->next_open_days($base_date, $push_amt);
227                 }
228             }
229         }
230     }
231
232     return $base_date;
233 }
234
235 sub get_push_amt {
236     my ( $self, $base_date) = @_;
237
238     my $dow = $base_date->day_of_week;
239     return (
240         # We're using Dayweek useDaysMode option
241         $self->{days_mode} eq 'Dayweek' &&
242         # It's not a permanently closed day
243         !$self->{weekly_closed_days}->[$dow] == 1
244     ) ? 7 : 1;
245 }
246
247 sub is_holiday {
248     my ( $self, $dt ) = @_;
249
250     my $localdt = $dt->clone();
251     my $day   = $localdt->day;
252     my $month = $localdt->month;
253
254     #Change timezone to "floating" before doing any calculations or comparisons
255     $localdt->set_time_zone("floating");
256     $localdt->truncate( to => 'day' );
257
258
259     if ( $self->exception_holidays->contains($localdt) ) {
260         # exceptions are not holidays
261         return 0;
262     }
263
264     my $dow = $localdt->day_of_week;
265     # Representation fix
266     # DateTime object dow (1-7) where Monday is 1
267     # Arrays are 0-based where 0 = Sunday, not 7.
268     if ( $dow == 7 ) {
269         $dow = 0;
270     }
271
272     if ( $self->{weekly_closed_days}->[$dow] == 1 ) {
273         return 1;
274     }
275
276     if ( exists $self->{day_month_closed_days}->{$month}->{$day} ) {
277         return 1;
278     }
279
280     my $ymd   = $localdt->ymd('')  ;
281     if ($self->single_holidays(  $ymd  ) == 1 ) {
282         return 1;
283     }
284
285     # damn have to go to work after all
286     return 0;
287 }
288
289 sub next_open_days {
290     my ( $self, $dt, $to_add ) = @_;
291     my $base_date = $dt->clone();
292
293     $base_date->add(days => $to_add);
294     while ($self->is_holiday($base_date)) {
295         my $add_next = $self->get_push_amt($base_date);
296         $base_date->add(days => $add_next);
297     }
298     return $base_date;
299 }
300
301 sub prev_open_days {
302     my ( $self, $dt, $to_sub ) = @_;
303     my $base_date = $dt->clone();
304
305     # It feels logical to be passed a positive number, though we're
306     # subtracting, so do the right thing
307     $to_sub = $to_sub > 0 ? 0 - $to_sub : $to_sub;
308
309     $base_date->add(days => $to_sub);
310
311     while ($self->is_holiday($base_date)) {
312         my $sub_next = $self->get_push_amt($base_date);
313         # Ensure we're subtracting when we need to be
314         $sub_next = $sub_next > 0 ? 0 - $sub_next : $sub_next;
315         $base_date->add(days => $sub_next);
316     }
317
318     return $base_date;
319 }
320
321 sub days_forward {
322     my $self     = shift;
323     my $start_dt = shift;
324     my $num_days = shift;
325
326     return $start_dt unless $num_days > 0;
327
328     my $base_dt = $start_dt->clone();
329
330     while ($num_days--) {
331         $base_dt = $self->next_open_days($base_dt, 1);
332     }
333
334     return $base_dt;
335 }
336
337 sub days_between {
338     my $self     = shift;
339     my $start_dt = shift;
340     my $end_dt   = shift;
341
342     # Change time zone for date math and swap if needed
343     $start_dt = $start_dt->clone->set_time_zone('floating');
344     $end_dt = $end_dt->clone->set_time_zone('floating');
345     if( $start_dt->compare($end_dt) > 0 ) {
346         ( $start_dt, $end_dt ) = ( $end_dt, $start_dt );
347     }
348
349     # start and end should not be closed days
350     my $days = $start_dt->delta_days($end_dt)->delta_days;
351     while( $start_dt->compare($end_dt) < 1 ) {
352         $days-- if $self->is_holiday($start_dt);
353         $start_dt->add( days => 1 );
354     }
355     return DateTime::Duration->new( days => $days );
356 }
357
358 sub hours_between {
359     my ($self, $start_date, $end_date) = @_;
360     my $start_dt = $start_date->clone()->set_time_zone('floating');
361     my $end_dt = $end_date->clone()->set_time_zone('floating');
362     my $duration = $end_dt->delta_ms($start_dt);
363     $start_dt->truncate( to => 'day' );
364     $end_dt->truncate( to => 'day' );
365     # NB this is a kludge in that it assumes all days are 24 hours
366     # However for hourly loans the logic should be expanded to
367     # take into account open/close times then it would be a duration
368     # of library open hours
369     my $skipped_days = 0;
370     for (my $dt = $start_dt->clone();
371         $dt <= $end_dt;
372         $dt->add(days => 1)
373     ) {
374         if ($self->is_holiday($dt)) {
375             ++$skipped_days;
376         }
377     }
378     if ($skipped_days) {
379         $duration->subtract_duration(DateTime::Duration->new( hours => 24 * $skipped_days));
380     }
381
382     return $duration;
383
384 }
385
386 sub set_daysmode {
387     my ( $self, $mode ) = @_;
388
389     # if not testing this is a no op
390     if ( $self->{test} ) {
391         $self->{days_mode} = $mode;
392     }
393
394     return;
395 }
396
397 sub clear_weekly_closed_days {
398     my $self = shift;
399     $self->{weekly_closed_days} = [ 0, 0, 0, 0, 0, 0, 0 ];    # Sunday only
400     return;
401 }
402
403 1;
404 __END__
405
406 =head1 NAME
407
408 Koha::Calendar - Object containing a branches calendar
409
410 =head1 SYNOPSIS
411
412   use Koha::Calendar
413
414   my $c = Koha::Calendar->new( branchcode => 'MAIN' );
415   my $dt = DateTime->now();
416
417   # are we open
418   $open = $c->is_holiday($dt);
419   # when will item be due if loan period = $dur (a DateTime::Duration object)
420   $duedate = $c->addDate($dt,$dur,'days');
421
422
423 =head1 DESCRIPTION
424
425   Implements those features of C4::Calendar needed for Staffs Rolling Loans
426
427 =head1 METHODS
428
429 =head2 new : Create a calendar object
430
431 my $calendar = Koha::Calendar->new( branchcode => 'MAIN' );
432
433 The option branchcode is required
434
435
436 =head2 addDate
437
438     my $dt = $calendar->addDate($date, $dur, $unit)
439
440 C<$date> is a DateTime object representing the starting date of the interval.
441
442 C<$offset> is a DateTime::Duration to add to it
443
444 C<$unit> is a string value 'days' or 'hours' toflag granularity of duration
445
446 Currently unit is only used to invoke Staffs return Monday at 10 am rule this
447 parameter will be removed when issuingrules properly cope with that
448
449
450 =head2 addHours
451
452     my $dt = $calendar->addHours($date, $dur, $return_by_hour )
453
454 C<$date> is a DateTime object representing the starting date of the interval.
455
456 C<$offset> is a DateTime::Duration to add to it
457
458 C<$return_by_hour> is an integer value representing the opening hour for the branch
459
460 =head2 get_push_amt
461
462     my $amt = $calendar->get_push_amt($date)
463
464 C<$date> is a DateTime object representing a closed return date
465
466 Using the days_mode syspref value and the nature of the closed return
467 date, return how many days we should jump forward to find another return date
468
469 =head2 addDays
470
471     my $dt = $calendar->addDays($date, $dur)
472
473 C<$date> is a DateTime object representing the starting date of the interval.
474
475 C<$offset> is a DateTime::Duration to add to it
476
477 C<$unit> is a string value 'days' or 'hours' toflag granularity of duration
478
479 Currently unit is only used to invoke Staffs return Monday at 10 am rule this
480 parameter will be removed when issuingrules properly cope with that
481
482
483 =head2 single_holidays
484
485 my $rc = $self->single_holidays(  $ymd  );
486
487 Passed a $date in Ymd (yyyymmdd) format -  returns 1 if date is a single_holiday, or 0 if not.
488
489
490 =head2 is_holiday
491
492 $yesno = $calendar->is_holiday($dt);
493
494 passed a DateTime object returns 1 if it is a closed day
495 0 if not according to the calendar
496
497 =head2 days_between
498
499 $duration = $calendar->days_between($start_dt, $end_dt);
500
501 Passed two dates returns a DateTime::Duration object measuring the length between them
502 ignoring closed days. Always returns a positive number irrespective of the
503 relative order of the parameters
504
505 =head2 next_open_days
506
507 $datetime = $calendar->next_open_days($duedate_dt, $to_add)
508
509 Passed a Datetime and number of days,  returns another Datetime representing
510 the next open day after adding the passed number of days. It is intended for
511 use to calculate the due date when useDaysMode syspref is set to either
512 'Datedue', 'Calendar' or 'Dayweek'.
513
514 =head2 prev_open_days
515
516 $datetime = $calendar->prev_open_days($duedate_dt, $to_sub)
517
518 Passed a Datetime and a number of days, returns another Datetime
519 representing the previous open day after subtracting the number of passed
520 days. It is intended for use to calculate the due date when useDaysMode
521 syspref is set to either 'Datedue', 'Calendar' or 'Dayweek'.
522
523 =head2 set_daysmode
524
525 For testing only allows the calling script to change days mode
526
527 =head2 clear_weekly_closed_days
528
529 In test mode changes the testing set of closed days to a new set with
530 no closed days. TODO passing an array of closed days to this would
531 allow testing of more configurations
532
533 =head2 add_holiday
534
535 Passed a datetime object this will add it to the calendar's list of
536 closed days. This is for testing so that we can alter the Calenfar object's
537 list of specified dates
538
539 =head1 DIAGNOSTICS
540
541 Will croak if not passed a branchcode in new
542
543 =head1 BUGS AND LIMITATIONS
544
545 This only contains a limited subset of the functionality in C4::Calendar
546 Only enough to support Staffs Rolling loans
547
548 =head1 AUTHOR
549
550 Colin Campbell colin.campbell@ptfs-europe.com
551
552 =head1 LICENSE AND COPYRIGHT
553
554 Copyright (c) 2011 PTFS-Europe Ltd All rights reserved
555
556 This program is free software: you can redistribute it and/or modify
557 it under the terms of the GNU General Public License as published by
558 the Free Software Foundation, either version 2 of the License, or
559 (at your option) any later version.
560
561 This program is distributed in the hope that it will be useful,
562 but WITHOUT ANY WARRANTY; without even the implied warranty of
563 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
564 GNU General Public License for more details.
565
566 You should have received a copy of the GNU General Public License
567 along with this program.  If not, see <http://www.gnu.org/licenses/>.