Merge branch 'master' of github.com:kivitendo/kivitendo-erp
[kivitendo-erp.git] / SL / DBUpgrade2.pm
1 package SL::DBUpgrade2;
2
3 use IO::File;
4 use List::MoreUtils qw(any);
5
6 use SL::Common;
7 use SL::DBUpgrade2::Base;
8 use SL::DBUtils;
9 use SL::Iconv;
10
11 use strict;
12
13 sub new {
14   my $package = shift;
15
16   return bless({}, $package)->init(@_);
17 }
18
19 sub init {
20   my ($self, %params) = @_;
21
22   if ($params{auth}) {
23     $params{path_suffix} = "-auth";
24     $params{schema}      = "auth.";
25   }
26
27   $params{path_suffix} ||= '';
28   $params{schema}      ||= '';
29   $params{path}          = "sql/" . $params{dbdriver} . "-upgrade2" . $params{path_suffix};
30
31   map { $self->{$_} = $params{$_} } keys %params;
32
33   return $self;
34 }
35
36 sub path {
37   $_[0]{path};
38 }
39
40 sub parse_dbupdate_controls {
41   $::lxdebug->enter_sub();
42
43   my ($self) = @_;
44
45   my $form   = $self->{form};
46   my $locale = $::locale;
47
48   local *IN;
49   my %all_controls;
50
51   my $path = $self->path;
52
53   foreach my $file_name (<$path/*.sql>, <$path/*.pl>) {
54     next unless (open(IN, $file_name));
55
56     my $file = $file_name;
57     $file =~ s|.*/||;
58
59     my $control = {
60       "priority" => 1000,
61       "depends"  => [],
62     };
63
64     while (<IN>) {
65       chomp();
66       next unless (/^(--|\#)\s*\@/);
67       s/^(--|\#)\s*\@//;
68       s/\s*$//;
69       next if ($_ eq "");
70
71       my @fields = split(/\s*:\s*/, $_, 2);
72       next unless (scalar(@fields) == 2);
73
74       if ($fields[0] eq "depends") {
75         push(@{$control->{"depends"}}, split(/\s+/, $fields[1]));
76       } else {
77         $control->{$fields[0]} = $fields[1];
78       }
79     }
80
81     next if ($control->{ignore});
82
83     $control->{charset} = 'UTF-8' if $file =~ m/\.pl$/;
84     $control->{charset} = $control->{charset} || $control->{encoding} || Common::DEFAULT_CHARSET;
85
86     if (!$control->{"tag"}) {
87       _control_error($form, $file_name, $locale->text("Missing 'tag' field.")) ;
88     }
89
90     if ($control->{"tag"} =~ /[^a-zA-Z0-9_\(\)\-]/) {
91       _control_error($form, $file_name, $locale->text("The 'tag' field must only consist of alphanumeric characters or the carachters - _ ( )"))
92     }
93
94     if (defined($all_controls{$control->{"tag"}})) {
95       _control_error($form, $file_name, sprintf($locale->text("More than one control file with the tag '%s' exist."), $control->{"tag"}))
96     }
97
98     if (!$control->{"description"}) {
99       _control_error($form, $file_name, sprintf($locale->text("Missing 'description' field."))) ;
100     }
101
102     $control->{"priority"}  *= 1;
103     $control->{"priority"} ||= 1000;
104     $control->{"file"}       = $file;
105
106     delete @{$control}{qw(depth applied)};
107
108     $all_controls{$control->{"tag"}} = $control;
109
110     close(IN);
111   }
112
113   foreach my $control (values(%all_controls)) {
114     foreach my $dependency (@{$control->{"depends"}}) {
115       _control_error($form, $control->{"file"}, sprintf($locale->text("Unknown dependency '%s'."), $dependency)) if (!defined($all_controls{$dependency}));
116     }
117
118     map({ $_->{"loop"} = 0; } values(%all_controls));
119     _check_for_loops($form, $control->{"file"}, \%all_controls, $control->{"tag"});
120   }
121
122   map({ _dbupdate2_calculate_depth(\%all_controls, $_->{"tag"}) }
123       values(%all_controls));
124
125   $self->{all_controls} = \%all_controls;
126
127   $::lxdebug->leave_sub();
128
129   return $self;
130 }
131
132 sub process_query {
133   $::lxdebug->enter_sub();
134
135   my ($self, $dbh, $filename, $version_or_control, $db_charset) = @_;
136
137   my $form  = $self->{form};
138   my $fh    = IO::File->new($filename, "r") or $form->error("$filename : $!\n");
139   my $query = "";
140   my $sth;
141   my @quote_chars;
142
143   my $file_charset = Common::DEFAULT_CHARSET;
144   while (<$fh>) {
145     last if !/^--/;
146     next if !/^--\s*\@(?:charset|encoding):\s*(.+)/;
147     $file_charset = $1;
148     last;
149   }
150   $fh->seek(0, SEEK_SET);
151
152   $db_charset ||= Common::DEFAULT_CHARSET;
153
154   $dbh->begin_work();
155
156   while (<$fh>) {
157     $_ = SL::Iconv::convert($file_charset, $db_charset, $_);
158
159     # Remove DOS and Unix style line endings.
160     chomp;
161
162     # remove comments
163     s/--.*$//;
164
165     for (my $i = 0; $i < length($_); $i++) {
166       my $char = substr($_, $i, 1);
167
168       # Are we inside a string?
169       if (@quote_chars) {
170         if ($char eq $quote_chars[-1]) {
171           pop(@quote_chars);
172         } elsif (length $quote_chars[-1] > 1
173              &&  substr($_, $i, length $quote_chars[-1]) eq $quote_chars[-1]) {
174           $i   += length($quote_chars[-1]) - 1;
175           $char = $quote_chars[-1];
176           pop(@quote_chars);
177         }
178         $query .= $char;
179
180       } else {
181         my ($tag, $tag_end);
182         if (($char eq "'") || ($char eq "\"")) {
183           push(@quote_chars, $char);
184
185         } elsif ($char eq '$'                                            # start of dollar quoting
186              && ($tag_end  = index($_, '$', $i + 1)) > -1                # ends on same line
187              && (do { $tag = substr($_, $i + 1, $tag_end - $i - 1); 1 }) # extract tag
188              &&  $tag      =~ /^ (?= [A-Za-z_] [A-Za-z0-9_]* | ) $/x) {  # tag is identifier
189           push @quote_chars, $char = '$' . $tag . '$';
190           $i = $tag_end;
191         } elsif ($char eq ";") {
192
193           # Query is complete. Send it.
194
195           $sth = $dbh->prepare($query);
196           if (!$sth->execute()) {
197             my $errstr = $dbh->errstr;
198             $sth->finish();
199             $dbh->rollback();
200             $form->dberror("The database update/creation did not succeed. " .
201                            "The file ${filename} containing the following " .
202                            "query failed:<br>${query}<br>" .
203                            "The error message was: ${errstr}<br>" .
204                            "All changes in that file have been reverted.");
205           }
206           $sth->finish();
207
208           $char  = "";
209           $query = "";
210         }
211
212         $query .= $char;
213       }
214     }
215
216     # Insert a space at the end of each line so that queries split
217     # over multiple lines work properly.
218     if ($query ne '') {
219       $query .= @quote_chars ? "\n" : ' ';
220     }
221   }
222
223   if (ref($version_or_control) eq "HASH") {
224     $dbh->do("INSERT INTO " . $self->{schema} . "schema_info (tag, login) VALUES (" . $dbh->quote($version_or_control->{"tag"}) . ", " . $dbh->quote($form->{"login"}) . ")");
225   } elsif ($version_or_control) {
226     $dbh->do("UPDATE defaults SET version = " . $dbh->quote($version_or_control));
227   }
228   $dbh->commit();
229
230   $fh->close();
231
232   $::lxdebug->leave_sub();
233 }
234
235 # Process a Perl script which updates the database.
236 # If the script returns 1 then the update was successful.
237 # Return code "2" means "needs more interaction; remove
238 # users/nologin and end current request".
239 # All other return codes are fatal errors.
240 sub process_perl_script {
241   $::lxdebug->enter_sub();
242
243   my ($self, $dbh, $filename, $version_or_control, $db_charset) = @_;
244
245   my %form_values = map { $_ => $::form->{$_} } qw(dbconnect dbdefault dbdriver dbhost dbmbkiviunstable dbname dboptions dbpasswd dbport dbupdate dbuser login template_object version);
246
247   $dbh->begin_work;
248
249   # setup dbup_ export vars & run script
250   my %dbup_myconfig = map { ($_ => $::form->{$_}) } qw(dbname dbuser dbpasswd dbhost dbport dbconnect);
251   my $result        = SL::DBUpgrade2::Base::execute_script(
252     file_name => $filename,
253     tag       => $version_or_control->{tag},
254     dbh       => $dbh,
255     myconfig  => \%dbup_myconfig,
256   );
257
258   if (1 != ($result // 1)) {
259     $dbh->rollback();
260   }
261
262   if (!defined($result)) {
263     print $::form->parse_html_template("dbupgrade/error", { file  => $filename, error => $@ });
264     ::end_of_request();
265   } elsif (1 != $result) {
266     unlink("users/nologin") if (2 == $result);
267     ::end_of_request();
268   }
269
270   if (ref($version_or_control) eq "HASH") {
271     $dbh->do("INSERT INTO " . $self->{schema} . "schema_info (tag, login) VALUES (" . $dbh->quote($version_or_control->{tag}) . ", " . $dbh->quote($::form->{login}) . ")");
272   } elsif ($version_or_control) {
273     $dbh->do("UPDATE defaults SET version = " . $dbh->quote($version_or_control));
274   }
275   $dbh->commit();
276
277   # Clear $::form of values that may have been set so that following
278   # Perl upgrade scripts won't have to work with old data (think of
279   # the usual 'continued' mechanism that's used for determining
280   # whether or not the upgrade form must be displayed).
281   delete @{ $::form }{ keys %{ $::form } };
282   $::form->{$_} = $form_values{$_} for keys %form_values;
283
284   $::lxdebug->leave_sub();
285 }
286
287 sub process_file {
288   my ($self, $dbh, $filename, $version_or_control, $db_charset) = @_;
289
290   if ($filename =~ m/sql$/) {
291     $self->process_query($dbh, $filename, $version_or_control, $db_charset);
292   } else {
293     $self->process_perl_script($dbh, $filename, $version_or_control, $db_charset);
294   }
295 }
296
297 sub update_available {
298   my ($self, $cur_version) = @_;
299
300   local *SQLDIR;
301
302   my $dbdriver = $self->{dbdriver};
303   opendir SQLDIR, "sql/${dbdriver}-upgrade" || error("", "sql/${dbdriver}-upgrade: $!");
304   my @upgradescripts = grep /${dbdriver}-upgrade-\Q$cur_version\E.*\.(sql|pl)$/, readdir SQLDIR;
305   closedir SQLDIR;
306
307   return ($#upgradescripts > -1);
308 }
309
310 sub update2_available {
311   $::lxdebug->enter_sub();
312
313   my ($self, $dbh) = @_;
314
315   map { $_->{applied} = 0; } values %{ $self->{all_controls} };
316
317   my $sth = $dbh->prepare(qq|SELECT tag FROM | . $self->{schema} . qq|schema_info|);
318   if ($sth->execute) {
319     while (my ($tag) = $sth->fetchrow_array) {
320       $self->{all_controls}->{$tag}->{applied} = 1 if defined $self->{all_controls}->{$tag};
321     }
322   }
323   $sth->finish();
324
325   my $needs_update = any { !$_->{applied} } values %{ $self->{all_controls} };
326
327   $::lxdebug->leave_sub();
328
329   return $needs_update;
330 }
331
332 sub unapplied_upgrade_scripts {
333   my ($self, $dbh) = @_;
334
335   my @all_scripts = map { $_->{applied} = 0; $_ } $self->sort_dbupdate_controls;
336
337   my $query = qq|SELECT tag FROM | . $self->{schema} . qq|schema_info|;
338   my $sth   = $dbh->prepare($query);
339   $sth->execute || $self->{form}->dberror($query);
340   while (my ($tag) = $sth->fetchrow_array()) {
341     $self->{all_controls}->{$tag}->{applied} = 1 if defined $self->{all_controls}->{$tag};
342   }
343   $sth->finish;
344
345   return grep { !$_->{applied} } @all_scripts;
346 }
347
348 sub apply_admin_dbupgrade_scripts {
349   my ($self, $called_from_admin) = @_;
350
351   return 0 if !$self->{auth};
352
353   my $dbh               = $::auth->dbconnect;
354   my @unapplied_scripts = $self->unapplied_upgrade_scripts($dbh);
355
356   return 0 if !@unapplied_scripts;
357
358   my $db_charset           = $::lx_office_conf{system}->{dbcharset} || Common::DEFAULT_CHARSET;
359   $self->{form}->{login} ||= 'admin';
360
361   map { $_->{description} = SL::Iconv::convert($_->{charset}, $db_charset, $_->{description}) } values %{ $self->{all_controls} };
362
363   if ($called_from_admin) {
364     $self->{form}->{title} = $::locale->text('Dataset upgrade');
365     $self->{form}->header;
366   }
367
368   print $self->{form}->parse_html_template("dbupgrade/header", { dbname => $::auth->{DB_config}->{db} });
369
370   foreach my $control (@unapplied_scripts) {
371     $::lxdebug->message(LXDebug->DEBUG2(), "Applying Update $control->{file}");
372     print $self->{form}->parse_html_template("dbupgrade/upgrade_message2", $control);
373
374     $self->process_file($dbh, "sql/$self->{dbdriver}-upgrade2-auth/$control->{file}", $control, $db_charset);
375   }
376
377   print $self->{form}->parse_html_template("dbupgrade/footer", { is_admin => 1 }) if $called_from_admin;
378
379   return 1;
380 }
381
382 sub _check_for_loops {
383   my ($form, $file_name, $controls, $tag, @path) = @_;
384
385   push(@path, $tag);
386
387   my $ctrl = $controls->{$tag};
388
389   if ($ctrl->{"loop"} == 1) {
390     # Not done yet.
391     _control_error($form, $file_name, $::locale->text("Dependency loop detected:") . " " . join(" -> ", @path))
392
393   } elsif ($ctrl->{"loop"} == 0) {
394     # Not checked yet.
395     $ctrl->{"loop"} = 1;
396     map({ _check_for_loops($form, $file_name, $controls, $_, @path); } @{ $ctrl->{"depends"} });
397     $ctrl->{"loop"} = 2;
398   }
399 }
400
401 sub _control_error {
402   my ($form, $file_name, $message) = @_;
403
404   $form = $::form;
405   my $locale = $::locale;
406
407   $form->error(sprintf($locale->text("Error in database control file '%s': %s"), $file_name, $message));
408 }
409
410 sub _dbupdate2_calculate_depth {
411   $::lxdebug->enter_sub(2);
412
413   my ($tree, $tag) = @_;
414
415   my $node = $tree->{$tag};
416
417   return $::lxdebug->leave_sub(2) if (defined($node->{"depth"}));
418
419   my $max_depth = 0;
420
421   foreach $tag (@{$node->{"depends"}}) {
422     _dbupdate2_calculate_depth($tree, $tag);
423     my $value = $tree->{$tag}->{"depth"};
424     $max_depth = $value if ($value > $max_depth);
425   }
426
427   $node->{"depth"} = $max_depth + 1;
428
429   $::lxdebug->leave_sub(2);
430 }
431
432 sub sort_dbupdate_controls {
433   my $self = shift;
434
435   $self->parse_dbupdate_controls unless $self->{all_controls};
436
437   return sort { ($a->{depth} <=> $b->{depth}) || ($a->{priority} <=> $b->{priority}) || ($a->{tag} cmp $b->{tag}) } values %{ $self->{all_controls} };
438 }
439
440 1;
441 __END__
442
443 =pod
444
445 =encoding utf8
446
447 =head1 NAME
448
449 SL::DBUpgrade2 - Parse database upgrade files stored in
450 C<sql/Pg-upgrade2> and C<sql/Pg-upgrade2-auth> (and also in
451 C<SQL/Pg-upgrade>)
452
453 =head1 SYNOPSIS
454
455   use SL::User;
456   use SL::DBUpgrade2;
457
458   # Apply outstanding updates to the authentication database
459   my $scripts = SL::DBUpgrade2->new(
460     form     => $::form,
461     dbdriver => 'Pg',
462     auth     => 1
463   );
464   $scripts->apply_admin_dbupgrade_scripts(1);
465
466   # Apply updates to a user database
467   my $scripts = SL::DBUpgrade2->new(
468     form     => $::form,
469     dbdriver => $::form->{dbdriver},
470     auth     => 1
471   );
472   User->dbupdate2($form, $scripts->parse_dbupdate_controls);
473
474 =head1 OVERVIEW
475
476 Database upgrade files are used to upgrade the database structure and
477 content of both the authentication database and the user
478 databases. They're applied when a user logs in. As long as the
479 authentication database is not up to date users cannot log in in
480 general, and the admin has to log in first in order to get his
481 database updated.
482
483 Database scripts form a tree by specifying which upgrade file depends
484 on which other upgrade file. This means that such files are always
485 applied in a well-defined order.
486
487 Each script is run in a separate transaction. If a script fails the
488 current transaction is rolled back and the whole upgrade process is
489 stopped. The user/admin is required to fix the issue manually.
490
491 A list of applied upgrade scripts is maintained in a table called
492 C<schema_info> for the user database and C<auth.schema_info>) for the
493 authentication database. They contain the tags, the login name of the
494 user having applied the script and the timestamp when the script was
495 applied.
496
497 Database upgrade files come in two flavours: SQL files and Perl
498 files. For both there are control fields that determine the order in
499 which they're executed, what charset the scripts are written in
500 etc. The control fields are tag/value pairs contained in comments.
501
502 =head1 OLD UPGRADE FILES
503
504 The files in C<sql/Pg-upgrade> are so old that I don't bother
505 documenting them. They're handled by this class, too, but new files
506 are only created as C<Pg-upgrade2> files.
507
508 =head1 CONTROL FIELDS
509
510 =head2 SYNTAX
511
512 Control fields for Perl files:
513
514   # @tag1: value1
515   # @tag2: some more values
516   sub do_stuff {
517   }
518   1;
519
520 Control fields for SQL files:
521
522   -- @tag1: value1
523   -- @tag2: some more values
524   ALTER TABLE ...;
525
526 =head2 TAGS AND THEIR MEANING
527
528 The following tags are recognized:
529
530 =over 4
531
532 =item tag
533
534 The name for this file. The C<tag> is also used for dependency
535 resolution (see C<depends>).
536
537 This is mandatory.
538
539 =item description
540
541 A description presented to the user when the update is applied.
542
543 This is mandatory.
544
545 =item depends
546
547 A space-separated list of tags of scripts this particular script
548 depends on. All other upgrades listed in C<depends> will be applied
549 before the current one is applied.
550
551 =item charset
552 =item encoding
553
554 The charset this file uses. Defaults to C<ISO-8859-15> if
555 missing. Both terms are recognized.
556
557 =item priority
558
559 Ordering the scripts by their dependencies alone produces a lot of
560 groups of scripts that could be applied at the same time (e.g. if both
561 B and C depend only on A then B could be applied before C or the other
562 way around). This field determines the order inside such a
563 group. Scripts with lower priority fields are executed before scripts
564 with higher priority fields.
565
566 If two scripts have equal priorities then their tag name decides.
567
568 The priority defaults to 1000.
569
570 =back
571
572 =head1 FUNCTIONS
573
574 =over 4
575
576 =item C<apply_admin_dbupgrade_scripts $called_from_admin>
577
578 Applies all unapplied upgrade files to the authentication/admin
579 database. The parameter C<$called_from_admin> should be truish if the
580 function is called from the web interface and falsish if it's called
581 from e.g. a command line script like C<scripts/dbupgrade2_tool.pl>.
582
583 =item C<init %params>
584
585 Initializes the object. Is called directly from L<new> and should not
586 be called again.
587
588 =item C<new %params>
589
590 Creates a new object. Possible parameters are:
591
592 =over 4
593
594 =item path
595
596 Path to the upgrade files to parse. Required.
597
598 =item form
599
600 C<SL::Form> object to use. Required.
601
602 =item dbdriver
603
604 Name of the database driver. Currently only C<Pg> for PostgreSQL is
605 supported.
606
607 =item auth
608
609 Optional parameter defaulting to 0. If trueish then the scripts read
610 are the ones applying to the authentication database.
611
612 =back
613
614 =item C<parse_dbupdate_controls>
615
616 Parses all files located in C<path> (see L<new>), ananlyzes their
617 control fields, builds the tree, and signals errors if control fields
618 are missing/wrong (e.g. a tag name listed in C<depends> is not
619 found). Sets C<$Self-&gt;{all_controls}> to the list of database
620 scripts.
621
622 =item C<process_file $dbh, $filename, $version_or_control, $db_charset>
623
624 Applies a single database upgrade file. Calls L<process_perl_script>
625 for Perl update files and C<process_query> for SQL update
626 files. Requires an open database handle(C<$dbh>), the file name
627 (C<$filename>), a hash structure of the file's control fields as
628 produced by L<parse_dbupdate_controls> (C<$version_or_control>) and
629 the database charset (for on-the-fly charset recoding of the script if
630 required, C<$db_charset>).
631
632 Returns the result of the actual function called.
633
634 =item C<process_perl_script $dbh, $filename, $version_or_control, $db_charset>
635
636 Applies a single Perl database upgrade file. Requires an open database
637 handle(C<$dbh>), the file name (C<$filename>), a hash structure of the
638 file's control fields as produced by L<parse_dbupdate_controls>
639 (C<$version_or_control>) and the database charset (for on-the-fly
640 charset recoding of the script if required, C<$db_charset>).
641
642 Perl scripts are executed via L<eval>. If L<eval> returns falsish then
643 an error is expected. There are two special return values: If the
644 script returns C<1> then the update was successful. Return code C<2>
645 means "needs more interaction from the user; remove users/nologin and
646 end current upgrade process". All other return codes are fatal errors.
647
648 Inside the Perl script several local variables exist that can be used:
649
650 =over 4
651
652 =item $dbup_locale
653
654 A locale object for translating messages
655
656 =item $dbh
657
658 The database handle (inside a transaction).
659
660 =item $::form
661
662 The global C<SL::Form> object.
663
664 =back
665
666 A Perl script can actually implement queries that fail while
667 continueing the process by handling the transaction itself, e.g. with
668 the following function:
669
670   sub do_query {
671     my ($query, $may_fail) = @_;
672
673     if (!$dbh->do($query)) {
674       die($dbup_locale->text("Database update error:") . "<br>$msg<br>" . $DBI::errstr) unless $may_fail;
675       $dbh->rollback();
676       $dbh->begin_work();
677     }
678   }
679
680 =item C<process_query $dbh, $filename, $version_or_control, $db_charset>
681
682 Applies a single SQL database upgrade file. Requires an open database
683 handle(C<$dbh>), the file name (C<$filename>), a hash structure of the
684 ofile's control fields as produced by L<parse_dbupdate_controls>
685 (C<$version_or_control>) and the database charset (for on-the-fly
686 charset recoding of the script if required, C<$db_charset>).
687
688 =item C<sort_dbupdate_controls>
689
690 Sorts the database upgrade scripts according to their C<tag> and
691 C<priority> control fields. Returns a list of their hash
692 representations that can be applied in order.
693
694 =item C<unapplied_upgrade_scripts $dbh>
695
696 Returns a list if upgrade scripts (their internal hash representation)
697 that haven't been applied to a database yet. C<$dbh> is an open handle
698 to the database that is checked.
699
700 Requires that the scripts have been parsed.
701
702 =item C<update2_available $dbh>
703
704 Returns trueish if at least one upgrade script hasn't been applied to
705 a database yet. C<$dbh> is an open handle to the database that is
706 checked.
707
708 Requires that the scripts have been parsed.
709
710 =back
711
712 =head1 BUGS
713
714 Nothing here yet.
715
716 =head1 AUTHOR
717
718 Moritz Bunkus E<lt>m.bunkus@linet-services.deE<gt>
719
720 =cut