Bug 14423: XSS issues in marc_subfields_structure
[koha.git] / C4 / Context.pm
blob660a489f6b2345a792c80bc3305a7c4699912987
1 package C4::Context;
2 # Copyright 2002 Katipo Communications
4 # This file is part of Koha.
6 # Koha is free software; you can redistribute it and/or modify it
7 # under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 3 of the License, or
9 # (at your option) any later version.
11 # Koha is distributed in the hope that it will be useful, but
12 # WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 # GNU General Public License for more details.
16 # You should have received a copy of the GNU General Public License
17 # along with Koha; if not, see <http://www.gnu.org/licenses>.
19 use strict;
20 use warnings;
21 use vars qw($VERSION $AUTOLOAD $context @context_stack $servers $memcached $ismemcached);
22 BEGIN {
23 if ($ENV{'HTTP_USER_AGENT'}) {
24 require CGI::Carp;
25 # FIXME for future reference, CGI::Carp doc says
26 # "Note that fatalsToBrowser does not work with mod_perl version 2.0 and higher."
27 import CGI::Carp qw(fatalsToBrowser);
28 sub handle_errors {
29 my $msg = shift;
30 my $debug_level;
31 eval {C4::Context->dbh();};
32 if ($@){
33 $debug_level = 1;
35 else {
36 $debug_level = C4::Context->preference("DebugLevel");
39 print q(<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
40 "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
41 <html lang="en" xml:lang="en" xmlns="http://www.w3.org/1999/xhtml">
42 <head><title>Koha Error</title></head>
43 <body>
45 if ($debug_level eq "2"){
46 # debug 2 , print extra info too.
47 my %versions = get_versions();
49 # a little example table with various version info";
50 print "
51 <h1>Koha error</h1>
52 <p>The following fatal error has occurred:</p>
53 <pre><code>$msg</code></pre>
54 <table>
55 <tr><th>Apache</th><td> $versions{apacheVersion}</td></tr>
56 <tr><th>Koha</th><td> $versions{kohaVersion}</td></tr>
57 <tr><th>Koha DB</th><td> $versions{kohaDbVersion}</td></tr>
58 <tr><th>MySQL</th><td> $versions{mysqlVersion}</td></tr>
59 <tr><th>OS</th><td> $versions{osVersion}</td></tr>
60 <tr><th>Perl</th><td> $versions{perlVersion}</td></tr>
61 </table>";
63 } elsif ($debug_level eq "1"){
64 print "
65 <h1>Koha error</h1>
66 <p>The following fatal error has occurred:</p>
67 <pre><code>$msg</code></pre>";
68 } else {
69 print "<p>production mode - trapped fatal error</p>";
71 print "</body></html>";
73 #CGI::Carp::set_message(\&handle_errors);
74 ## give a stack backtrace if KOHA_BACKTRACES is set
75 ## can't rely on DebugLevel for this, as we're not yet connected
76 if ($ENV{KOHA_BACKTRACES}) {
77 $main::SIG{__DIE__} = \&CGI::Carp::confess;
79 } # else there is no browser to send fatals to!
81 # Check if there are memcached servers set
82 $servers = $ENV{'MEMCACHED_SERVERS'};
83 if ($servers) {
84 # Load required libraries and create the memcached object
85 require Cache::Memcached;
86 $memcached = Cache::Memcached->new({
87 servers => [ $servers ],
88 debug => 0,
89 compress_threshold => 10_000,
90 expire_time => 600,
91 namespace => $ENV{'MEMCACHED_NAMESPACE'} || 'koha'
92 });
93 # Verify memcached available (set a variable and test the output)
94 $ismemcached = $memcached->set('ismemcached','1');
97 $VERSION = '3.07.00.049';
100 use DBIx::Connector;
101 use Encode;
102 use ZOOM;
103 use XML::Simple;
104 use POSIX ();
105 use DateTime::TimeZone;
106 use Module::Load::Conditional qw(can_load);
107 use Carp;
109 use C4::Boolean;
110 use C4::Debug;
111 use Koha;
112 use Koha::Config::SysPrefs;
114 =head1 NAME
116 C4::Context - Maintain and manipulate the context of a Koha script
118 =head1 SYNOPSIS
120 use C4::Context;
122 use C4::Context("/path/to/koha-conf.xml");
124 $config_value = C4::Context->config("config_variable");
126 $koha_preference = C4::Context->preference("preference");
128 $db_handle = C4::Context->dbh;
130 $Zconn = C4::Context->Zconn;
132 $stopwordhash = C4::Context->stopwords;
134 =head1 DESCRIPTION
136 When a Koha script runs, it makes use of a certain number of things:
137 configuration settings in F</etc/koha/koha-conf.xml>, a connection to the Koha
138 databases, and so forth. These things make up the I<context> in which
139 the script runs.
141 This module takes care of setting up the context for a script:
142 figuring out which configuration file to load, and loading it, opening
143 a connection to the right database, and so forth.
145 Most scripts will only use one context. They can simply have
147 use C4::Context;
149 at the top.
151 Other scripts may need to use several contexts. For instance, if a
152 library has two databases, one for a certain collection, and the other
153 for everything else, it might be necessary for a script to use two
154 different contexts to search both databases. Such scripts should use
155 the C<&set_context> and C<&restore_context> functions, below.
157 By default, C4::Context reads the configuration from
158 F</etc/koha/koha-conf.xml>. This may be overridden by setting the C<$KOHA_CONF>
159 environment variable to the pathname of a configuration file to use.
161 =head1 METHODS
163 =cut
166 # In addition to what is said in the POD above, a Context object is a
167 # reference-to-hash with the following fields:
169 # config
170 # A reference-to-hash whose keys and values are the
171 # configuration variables and values specified in the config
172 # file (/etc/koha/koha-conf.xml).
173 # dbh
174 # A handle to the appropriate database for this context.
175 # dbh_stack
176 # Used by &set_dbh and &restore_dbh to hold other database
177 # handles for this context.
178 # Zconn
179 # A connection object for the Zebra server
181 # Koha's main configuration file koha-conf.xml
182 # is searched for according to this priority list:
184 # 1. Path supplied via use C4::Context '/path/to/koha-conf.xml'
185 # 2. Path supplied in KOHA_CONF environment variable.
186 # 3. Path supplied in INSTALLED_CONFIG_FNAME, as long
187 # as value has changed from its default of
188 # '__KOHA_CONF_DIR__/koha-conf.xml', as happens
189 # when Koha is installed in 'standard' or 'single'
190 # mode.
191 # 4. Path supplied in CONFIG_FNAME.
193 # The first entry that refers to a readable file is used.
195 use constant CONFIG_FNAME => "/etc/koha/koha-conf.xml";
196 # Default config file, if none is specified
198 my $INSTALLED_CONFIG_FNAME = '__KOHA_CONF_DIR__/koha-conf.xml';
199 # path to config file set by installer
200 # __KOHA_CONF_DIR__ is set by rewrite-confg.PL
201 # when Koha is installed in 'standard' or 'single'
202 # mode. If Koha was installed in 'dev' mode,
203 # __KOHA_CONF_DIR__ is *not* rewritten; instead
204 # developers should set the KOHA_CONF environment variable
206 $context = undef; # Initially, no context is set
207 @context_stack = (); # Initially, no saved contexts
210 =head2 read_config_file
212 Reads the specified Koha config file.
214 Returns an object containing the configuration variables. The object's
215 structure is a bit complex to the uninitiated ... take a look at the
216 koha-conf.xml file as well as the XML::Simple documentation for details. Or,
217 here are a few examples that may give you what you need:
219 The simple elements nested within the <config> element:
221 my $pass = $koha->{'config'}->{'pass'};
223 The <listen> elements:
225 my $listen = $koha->{'listen'}->{'biblioserver'}->{'content'};
227 The elements nested within the <server> element:
229 my $ccl2rpn = $koha->{'server'}->{'biblioserver'}->{'cql2rpn'};
231 Returns undef in case of error.
233 =cut
235 sub read_config_file { # Pass argument naming config file to read
236 my $koha = XMLin(shift, keyattr => ['id'], forcearray => ['listen', 'server', 'serverinfo'], suppressempty => '');
238 if ($ismemcached) {
239 $memcached->set('kohaconf',$koha);
242 return $koha; # Return value: ref-to-hash holding the configuration
245 =head2 ismemcached
247 Returns the value of the $ismemcached variable (0/1)
249 =cut
251 sub ismemcached {
252 return $ismemcached;
255 =head2 memcached
257 If $ismemcached is true, returns the $memcache variable.
258 Returns undef otherwise
260 =cut
262 sub memcached {
263 if ($ismemcached) {
264 return $memcached;
265 } else {
266 return;
270 =head2 db_scheme2dbi
272 my $dbd_driver_name = C4::Context::db_schema2dbi($scheme);
274 This routines translates a database type to part of the name
275 of the appropriate DBD driver to use when establishing a new
276 database connection. It recognizes 'mysql' and 'Pg'; if any
277 other scheme is supplied it defaults to 'mysql'.
279 =cut
281 sub db_scheme2dbi {
282 my $scheme = shift // '';
283 return $scheme eq 'Pg' ? $scheme : 'mysql';
286 sub import {
287 # Create the default context ($C4::Context::Context)
288 # the first time the module is called
289 # (a config file can be optionaly passed)
291 # default context already exists?
292 return if $context;
294 # no ? so load it!
295 my ($pkg,$config_file) = @_ ;
296 my $new_ctx = __PACKAGE__->new($config_file);
297 return unless $new_ctx;
299 # if successfully loaded, use it by default
300 $new_ctx->set_context;
304 =head2 new
306 $context = new C4::Context;
307 $context = new C4::Context("/path/to/koha-conf.xml");
309 Allocates a new context. Initializes the context from the specified
310 file, which defaults to either the file given by the C<$KOHA_CONF>
311 environment variable, or F</etc/koha/koha-conf.xml>.
313 It saves the koha-conf.xml values in the declared memcached server(s)
314 if currently available and uses those values until them expire and
315 re-reads them.
317 C<&new> does not set this context as the new default context; for
318 that, use C<&set_context>.
320 =cut
323 # Revision History:
324 # 2004-08-10 A. Tarallo: Added check if the conf file is not empty
325 sub new {
326 my $class = shift;
327 my $conf_fname = shift; # Config file to load
328 my $self = {};
330 # check that the specified config file exists and is not empty
331 undef $conf_fname unless
332 (defined $conf_fname && -s $conf_fname);
333 # Figure out a good config file to load if none was specified.
334 if (!defined($conf_fname))
336 # If the $KOHA_CONF environment variable is set, use
337 # that. Otherwise, use the built-in default.
338 if (exists $ENV{"KOHA_CONF"} and $ENV{'KOHA_CONF'} and -s $ENV{"KOHA_CONF"}) {
339 $conf_fname = $ENV{"KOHA_CONF"};
340 } elsif ($INSTALLED_CONFIG_FNAME !~ /__KOHA_CONF_DIR/ and -s $INSTALLED_CONFIG_FNAME) {
341 # NOTE: be careful -- don't change __KOHA_CONF_DIR in the above
342 # regex to anything else -- don't want installer to rewrite it
343 $conf_fname = $INSTALLED_CONFIG_FNAME;
344 } elsif (-s CONFIG_FNAME) {
345 $conf_fname = CONFIG_FNAME;
346 } else {
347 warn "unable to locate Koha configuration file koha-conf.xml";
348 return;
352 if ($ismemcached) {
353 # retrieve from memcached
354 $self = $memcached->get('kohaconf');
355 if (not defined $self) {
356 # not in memcached yet
357 $self = read_config_file($conf_fname);
359 } else {
360 # non-memcached env, read from file
361 $self = read_config_file($conf_fname);
364 $self->{"config_file"} = $conf_fname;
365 warn "read_config_file($conf_fname) returned undef" if !defined($self->{"config"});
366 return if !defined($self->{"config"});
368 $self->{"dbh"} = undef; # Database handle
369 $self->{"Zconn"} = undef; # Zebra Connections
370 $self->{"stopwords"} = undef; # stopwords list
371 $self->{"marcfromkohafield"} = undef; # the hash with relations between koha table fields and MARC field/subfield
372 $self->{"userenv"} = undef; # User env
373 $self->{"activeuser"} = undef; # current active user
374 $self->{"shelves"} = undef;
375 $self->{tz} = undef; # local timezone object
377 bless $self, $class;
378 $self->{db_driver} = db_scheme2dbi($self->config('db_scheme')); # cache database driver
379 return $self;
382 =head2 set_context
384 $context = new C4::Context;
385 $context->set_context();
387 set_context C4::Context $context;
390 restore_context C4::Context;
392 In some cases, it might be necessary for a script to use multiple
393 contexts. C<&set_context> saves the current context on a stack, then
394 sets the context to C<$context>, which will be used in future
395 operations. To restore the previous context, use C<&restore_context>.
397 =cut
400 sub set_context
402 my $self = shift;
403 my $new_context; # The context to set
405 # Figure out whether this is a class or instance method call.
407 # We're going to make the assumption that control got here
408 # through valid means, i.e., that the caller used an instance
409 # or class method call, and that control got here through the
410 # usual inheritance mechanisms. The caller can, of course,
411 # break this assumption by playing silly buggers, but that's
412 # harder to do than doing it properly, and harder to check
413 # for.
414 if (ref($self) eq "")
416 # Class method. The new context is the next argument.
417 $new_context = shift;
418 } else {
419 # Instance method. The new context is $self.
420 $new_context = $self;
423 # Save the old context, if any, on the stack
424 push @context_stack, $context if defined($context);
426 # Set the new context
427 $context = $new_context;
430 =head2 restore_context
432 &restore_context;
434 Restores the context set by C<&set_context>.
436 =cut
439 sub restore_context
441 my $self = shift;
443 if ($#context_stack < 0)
445 # Stack underflow.
446 die "Context stack underflow";
449 # Pop the old context and set it.
450 $context = pop @context_stack;
452 # FIXME - Should this return something, like maybe the context
453 # that was current when this was called?
456 =head2 config
458 $value = C4::Context->config("config_variable");
460 $value = C4::Context->config_variable;
462 Returns the value of a variable specified in the configuration file
463 from which the current context was created.
465 The second form is more compact, but of course may conflict with
466 method names. If there is a configuration variable called "new", then
467 C<C4::Config-E<gt>new> will not return it.
469 =cut
471 sub _common_config {
472 my $var = shift;
473 my $term = shift;
474 return if !defined($context->{$term});
475 # Presumably $self->{$term} might be
476 # undefined if the config file given to &new
477 # didn't exist, and the caller didn't bother
478 # to check the return value.
480 # Return the value of the requested config variable
481 return $context->{$term}->{$var};
484 sub config {
485 return _common_config($_[1],'config');
487 sub zebraconfig {
488 return _common_config($_[1],'server');
490 sub ModZebrations {
491 return _common_config($_[1],'serverinfo');
494 =head2 preference
496 $sys_preference = C4::Context->preference('some_variable');
498 Looks up the value of the given system preference in the
499 systempreferences table of the Koha database, and returns it. If the
500 variable is not set or does not exist, undef is returned.
502 In case of an error, this may return 0.
504 Note: It is impossible to tell the difference between system
505 preferences which do not exist, and those whose values are set to NULL
506 with this method.
508 =cut
510 # FIXME: running this under mod_perl will require a means of
511 # flushing the caching mechanism.
513 my %sysprefs;
514 my $use_syspref_cache = 1;
516 sub preference {
517 my $self = shift;
518 my $var = shift; # The system preference to return
520 if ($use_syspref_cache && exists $sysprefs{lc $var}) {
521 return $sysprefs{lc $var};
524 my $dbh = C4::Context->dbh or return 0;
526 my $value;
527 if ( defined $ENV{"OVERRIDE_SYSPREF_$var"} ) {
528 $value = $ENV{"OVERRIDE_SYSPREF_$var"};
529 } else {
530 my $syspref;
531 eval { $syspref = Koha::Config::SysPrefs->find( lc $var ) };
532 $value = $syspref ? $syspref->value() : undef;
535 $sysprefs{lc $var} = $value;
536 return $value;
539 sub boolean_preference {
540 my $self = shift;
541 my $var = shift; # The system preference to return
542 my $it = preference($self, $var);
543 return defined($it)? C4::Boolean::true_p($it): undef;
546 =head2 enable_syspref_cache
548 C4::Context->enable_syspref_cache();
550 Enable the in-memory syspref cache used by C4::Context. This is the
551 default behavior.
553 =cut
555 sub enable_syspref_cache {
556 my ($self) = @_;
557 $use_syspref_cache = 1;
560 =head2 disable_syspref_cache
562 C4::Context->disable_syspref_cache();
564 Disable the in-memory syspref cache used by C4::Context. This should be
565 used with Plack and other persistent environments.
567 =cut
569 sub disable_syspref_cache {
570 my ($self) = @_;
571 $use_syspref_cache = 0;
572 $self->clear_syspref_cache();
575 =head2 clear_syspref_cache
577 C4::Context->clear_syspref_cache();
579 cleans the internal cache of sysprefs. Please call this method if
580 you update the systempreferences table. Otherwise, your new changes
581 will not be seen by this process.
583 =cut
585 sub clear_syspref_cache {
586 %sysprefs = ();
589 =head2 set_preference
591 C4::Context->set_preference( $variable, $value );
593 This updates a preference's value both in the systempreferences table and in
594 the sysprefs cache.
596 =cut
598 sub set_preference {
599 my $self = shift;
600 my $var = lc(shift);
601 my $value = shift;
603 my $syspref = Koha::Config::SysPrefs->find( $var );
604 my $type = $syspref ? $syspref->type() : undef;
606 $value = 0 if ( $type && $type eq 'YesNo' && $value eq '' );
608 # force explicit protocol on OPACBaseURL
609 if ($var eq 'opacbaseurl' && substr($value,0,4) !~ /http/) {
610 $value = 'http://' . $value;
613 if ($syspref) {
614 $syspref = $syspref->set( { value => $value } )->store();
616 else {
617 $syspref = Koha::Config::SysPref->new( { variable => $var, value => $value } )->store();
620 if ($syspref) {
621 $sysprefs{$var} = $value;
625 # AUTOLOAD
626 # This implements C4::Config->foo, and simply returns
627 # C4::Context->config("foo"), as described in the documentation for
628 # &config, above.
630 # FIXME - Perhaps this should be extended to check &config first, and
631 # then &preference if that fails. OTOH, AUTOLOAD could lead to crappy
632 # code, so it'd probably be best to delete it altogether so as not to
633 # encourage people to use it.
634 sub AUTOLOAD
636 my $self = shift;
638 $AUTOLOAD =~ s/.*:://; # Chop off the package name,
639 # leaving only the function name.
640 return $self->config($AUTOLOAD);
643 =head2 Zconn
645 $Zconn = C4::Context->Zconn
647 Returns a connection to the Zebra database
649 C<$self>
651 C<$server> one of the servers defined in the koha-conf.xml file
653 C<$async> whether this is a asynchronous connection
655 =cut
657 sub Zconn {
658 my ($self, $server, $async ) = @_;
659 my $cache_key = join ('::', (map { $_ // '' } ($server, $async )));
660 if ( (!defined($ENV{GATEWAY_INTERFACE})) && defined($context->{"Zconn"}->{$cache_key}) && (0 == $context->{"Zconn"}->{$cache_key}->errcode()) ) {
661 # if we are running the script from the commandline, lets try to use the caching
662 return $context->{"Zconn"}->{$cache_key};
664 $context->{"Zconn"}->{$cache_key}->destroy() if defined($context->{"Zconn"}->{$cache_key}); #destroy old connection before making a new one
665 $context->{"Zconn"}->{$cache_key} = &_new_Zconn( $server, $async );
666 return $context->{"Zconn"}->{$cache_key};
669 =head2 _new_Zconn
671 $context->{"Zconn"} = &_new_Zconn($server,$async);
673 Internal function. Creates a new database connection from the data given in the current context and returns it.
675 C<$server> one of the servers defined in the koha-conf.xml file
677 C<$async> whether this is a asynchronous connection
679 C<$auth> whether this connection has rw access (1) or just r access (0 or NULL)
681 =cut
683 sub _new_Zconn {
684 my ( $server, $async ) = @_;
686 my $tried=0; # first attempt
687 my $Zconn; # connection object
688 my $elementSetName;
689 my $index_mode;
690 my $syntax;
692 $server //= "biblioserver";
694 if ( $server eq 'biblioserver' ) {
695 $index_mode = $context->{'config'}->{'zebra_bib_index_mode'} // 'dom';
696 } elsif ( $server eq 'authorityserver' ) {
697 $index_mode = $context->{'config'}->{'zebra_auth_index_mode'} // 'dom';
700 if ( $index_mode eq 'grs1' ) {
701 $elementSetName = 'F';
702 $syntax = ( $context->preference("marcflavour") eq 'UNIMARC' )
703 ? 'unimarc'
704 : 'usmarc';
706 } else { # $index_mode eq 'dom'
707 $syntax = 'xml';
708 $elementSetName = 'marcxml';
711 my $host = $context->{'listen'}->{$server}->{'content'};
712 my $user = $context->{"serverinfo"}->{$server}->{"user"};
713 my $password = $context->{"serverinfo"}->{$server}->{"password"};
714 eval {
715 # set options
716 my $o = new ZOOM::Options();
717 $o->option(user => $user) if $user && $password;
718 $o->option(password => $password) if $user && $password;
719 $o->option(async => 1) if $async;
720 $o->option(cqlfile=> $context->{"server"}->{$server}->{"cql2rpn"});
721 $o->option(cclfile=> $context->{"serverinfo"}->{$server}->{"ccl2rpn"});
722 $o->option(preferredRecordSyntax => $syntax);
723 $o->option(elementSetName => $elementSetName) if $elementSetName;
724 $o->option(databaseName => $context->{"config"}->{$server}||"biblios");
726 # create a new connection object
727 $Zconn= create ZOOM::Connection($o);
729 # forge to server
730 $Zconn->connect($host, 0);
732 # check for errors and warn
733 if ($Zconn->errcode() !=0) {
734 warn "something wrong with the connection: ". $Zconn->errmsg();
737 return $Zconn;
740 # _new_dbh
741 # Internal helper function (not a method!). This creates a new
742 # database connection from the data given in the current context, and
743 # returns it.
744 sub _new_dbh
747 ## $context
748 ## correct name for db_scheme
749 my $db_driver = $context->{db_driver};
751 my $db_name = $context->config("database");
752 my $db_host = $context->config("hostname");
753 my $db_port = $context->config("port") || '';
754 my $db_user = $context->config("user");
755 my $db_passwd = $context->config("pass");
756 # MJR added or die here, as we can't work without dbh
757 my $dbh = DBIx::Connector->connect(
758 "dbi:$db_driver:dbname=$db_name;host=$db_host;port=$db_port",
759 $db_user, $db_passwd,
761 'RaiseError' => $ENV{DEBUG} ? 1 : 0
765 # Check for the existence of a systempreference table; if we don't have this, we don't
766 # have a valid database and should not set RaiseError in order to allow the installer
767 # to run; installer will not run otherwise since we raise all db errors
769 eval {
770 local $dbh->{PrintError} = 0;
771 local $dbh->{RaiseError} = 1;
772 $dbh->do(qq{SELECT * FROM systempreferences WHERE 1 = 0 });
775 if ($@) {
776 $dbh->{RaiseError} = 0;
779 if ( $db_driver eq 'mysql' ) {
780 $dbh->{mysql_auto_reconnect} = 1;
783 my $tz = $ENV{TZ};
784 if ( $db_driver eq 'mysql' ) {
785 # Koha 3.0 is utf-8, so force utf8 communication between mySQL and koha, whatever the mysql default config.
786 # this is better than modifying my.cnf (and forcing all communications to be in utf8)
787 $dbh->{'mysql_enable_utf8'}=1; #enable
788 $dbh->do("set NAMES 'utf8'");
789 ($tz) and $dbh->do(qq(SET time_zone = "$tz"));
791 elsif ( $db_driver eq 'Pg' ) {
792 $dbh->do( "set client_encoding = 'UTF8';" );
793 ($tz) and $dbh->do(qq(SET TIME ZONE = "$tz"));
795 return $dbh;
798 =head2 dbh
800 $dbh = C4::Context->dbh;
802 Returns a database handle connected to the Koha database for the
803 current context. If no connection has yet been made, this method
804 creates one, and connects to the database.
806 This database handle is cached for future use: if you call
807 C<C4::Context-E<gt>dbh> twice, you will get the same handle both
808 times. If you need a second database handle, use C<&new_dbh> and
809 possibly C<&set_dbh>.
811 =cut
814 sub dbh
816 my $self = shift;
817 my $params = shift;
818 my $sth;
820 unless ( $params->{new} ) {
821 if ( defined($context->{db_driver}) && $context->{db_driver} eq 'mysql' && $context->{"dbh"} ) {
822 return $context->{"dbh"};
823 } elsif ( defined($context->{"dbh"}) && $context->{"dbh"}->ping() ) {
824 return $context->{"dbh"};
828 # No database handle or it died . Create one.
829 $context->{"dbh"} = &_new_dbh();
831 return $context->{"dbh"};
834 =head2 new_dbh
836 $dbh = C4::Context->new_dbh;
838 Creates a new connection to the Koha database for the current context,
839 and returns the database handle (a C<DBI::db> object).
841 The handle is not saved anywhere: this method is strictly a
842 convenience function; the point is that it knows which database to
843 connect to so that the caller doesn't have to know.
845 =cut
848 sub new_dbh
850 my $self = shift;
852 return &_new_dbh();
855 =head2 set_dbh
857 $my_dbh = C4::Connect->new_dbh;
858 C4::Connect->set_dbh($my_dbh);
860 C4::Connect->restore_dbh;
862 C<&set_dbh> and C<&restore_dbh> work in a manner analogous to
863 C<&set_context> and C<&restore_context>.
865 C<&set_dbh> saves the current database handle on a stack, then sets
866 the current database handle to C<$my_dbh>.
868 C<$my_dbh> is assumed to be a good database handle.
870 =cut
873 sub set_dbh
875 my $self = shift;
876 my $new_dbh = shift;
878 # Save the current database handle on the handle stack.
879 # We assume that $new_dbh is all good: if the caller wants to
880 # screw himself by passing an invalid handle, that's fine by
881 # us.
882 push @{$context->{"dbh_stack"}}, $context->{"dbh"};
883 $context->{"dbh"} = $new_dbh;
886 =head2 restore_dbh
888 C4::Context->restore_dbh;
890 Restores the database handle saved by an earlier call to
891 C<C4::Context-E<gt>set_dbh>.
893 =cut
896 sub restore_dbh
898 my $self = shift;
900 if ($#{$context->{"dbh_stack"}} < 0)
902 # Stack underflow
903 die "DBH stack underflow";
906 # Pop the old database handle and set it.
907 $context->{"dbh"} = pop @{$context->{"dbh_stack"}};
909 # FIXME - If it is determined that restore_context should
910 # return something, then this function should, too.
913 =head2 queryparser
915 $queryparser = C4::Context->queryparser
917 Returns a handle to an initialized Koha::QueryParser::Driver::PQF object.
919 =cut
921 sub queryparser {
922 my $self = shift;
923 unless (defined $context->{"queryparser"}) {
924 $context->{"queryparser"} = &_new_queryparser();
927 return
928 defined( $context->{"queryparser"} )
929 ? $context->{"queryparser"}->new
930 : undef;
933 =head2 _new_queryparser
935 Internal helper function to create a new QueryParser object. QueryParser
936 is loaded dynamically so as to keep the lack of the QueryParser library from
937 getting in anyone's way.
939 =cut
941 sub _new_queryparser {
942 my $qpmodules = {
943 'OpenILS::QueryParser' => undef,
944 'Koha::QueryParser::Driver::PQF' => undef
946 if ( can_load( 'modules' => $qpmodules ) ) {
947 my $QParser = Koha::QueryParser::Driver::PQF->new();
948 my $config_file = $context->config('queryparser_config');
949 $config_file ||= '/etc/koha/searchengine/queryparser.yaml';
950 if ( $QParser->load_config($config_file) ) {
951 # Set 'keyword' as the default search class
952 $QParser->default_search_class('keyword');
953 # TODO: allow indexes to be configured in the database
954 return $QParser;
957 return;
960 =head2 marcfromkohafield
962 $dbh = C4::Context->marcfromkohafield;
964 Returns a hash with marcfromkohafield.
966 This hash is cached for future use: if you call
967 C<C4::Context-E<gt>marcfromkohafield> twice, you will get the same hash without real DB access
969 =cut
972 sub marcfromkohafield
974 my $retval = {};
976 # If the hash already exists, return it.
977 return $context->{"marcfromkohafield"} if defined($context->{"marcfromkohafield"});
979 # No hash. Create one.
980 $context->{"marcfromkohafield"} = &_new_marcfromkohafield();
982 return $context->{"marcfromkohafield"};
985 # _new_marcfromkohafield
986 # Internal helper function (not a method!). This creates a new
987 # hash with stopwords
988 sub _new_marcfromkohafield
990 my $dbh = C4::Context->dbh;
991 my $marcfromkohafield;
992 my $sth = $dbh->prepare("select frameworkcode,kohafield,tagfield,tagsubfield from marc_subfield_structure where kohafield > ''");
993 $sth->execute;
994 while (my ($frameworkcode,$kohafield,$tagfield,$tagsubfield) = $sth->fetchrow) {
995 my $retval = {};
996 $marcfromkohafield->{$frameworkcode}->{$kohafield} = [$tagfield,$tagsubfield];
998 return $marcfromkohafield;
1001 =head2 stopwords
1003 $dbh = C4::Context->stopwords;
1005 Returns a hash with stopwords.
1007 This hash is cached for future use: if you call
1008 C<C4::Context-E<gt>stopwords> twice, you will get the same hash without real DB access
1010 =cut
1013 sub stopwords
1015 my $retval = {};
1017 # If the hash already exists, return it.
1018 return $context->{"stopwords"} if defined($context->{"stopwords"});
1020 # No hash. Create one.
1021 $context->{"stopwords"} = &_new_stopwords();
1023 return $context->{"stopwords"};
1026 # _new_stopwords
1027 # Internal helper function (not a method!). This creates a new
1028 # hash with stopwords
1029 sub _new_stopwords
1031 my $dbh = C4::Context->dbh;
1032 my $stopwordlist;
1033 my $sth = $dbh->prepare("select word from stopwords");
1034 $sth->execute;
1035 while (my $stopword = $sth->fetchrow_array) {
1036 $stopwordlist->{$stopword} = uc($stopword);
1038 $stopwordlist->{A} = "A" unless $stopwordlist;
1039 return $stopwordlist;
1042 =head2 userenv
1044 C4::Context->userenv;
1046 Retrieves a hash for user environment variables.
1048 This hash shall be cached for future use: if you call
1049 C<C4::Context-E<gt>userenv> twice, you will get the same hash without real DB access
1051 =cut
1054 sub userenv {
1055 my $var = $context->{"activeuser"};
1056 if (defined $var and defined $context->{"userenv"}->{$var}) {
1057 return $context->{"userenv"}->{$var};
1058 } else {
1059 return;
1063 =head2 set_userenv
1065 C4::Context->set_userenv($usernum, $userid, $usercnum,
1066 $userfirstname, $usersurname,
1067 $userbranch, $branchname, $userflags,
1068 $emailaddress, $branchprinter, $persona);
1070 Establish a hash of user environment variables.
1072 set_userenv is called in Auth.pm
1074 =cut
1077 sub set_userenv {
1078 shift @_;
1079 my ($usernum, $userid, $usercnum, $userfirstname, $usersurname, $userbranch, $branchname, $userflags, $emailaddress, $branchprinter, $persona, $shibboleth)=
1080 map { Encode::is_utf8( $_ ) ? $_ : Encode::decode('UTF-8', $_) } # CGI::Session doesn't handle utf-8, so we decode it here
1082 my $var=$context->{"activeuser"} || '';
1083 my $cell = {
1084 "number" => $usernum,
1085 "id" => $userid,
1086 "cardnumber" => $usercnum,
1087 "firstname" => $userfirstname,
1088 "surname" => $usersurname,
1089 #possibly a law problem
1090 "branch" => $userbranch,
1091 "branchname" => $branchname,
1092 "flags" => $userflags,
1093 "emailaddress" => $emailaddress,
1094 "branchprinter" => $branchprinter,
1095 "persona" => $persona,
1096 "shibboleth" => $shibboleth,
1098 $context->{userenv}->{$var} = $cell;
1099 return $cell;
1102 sub set_shelves_userenv {
1103 my ($type, $shelves) = @_ or return;
1104 my $activeuser = $context->{activeuser} or return;
1105 $context->{userenv}->{$activeuser}->{barshelves} = $shelves if $type eq 'bar';
1106 $context->{userenv}->{$activeuser}->{pubshelves} = $shelves if $type eq 'pub';
1107 $context->{userenv}->{$activeuser}->{totshelves} = $shelves if $type eq 'tot';
1110 sub get_shelves_userenv {
1111 my $active;
1112 unless ($active = $context->{userenv}->{$context->{activeuser}}) {
1113 $debug and warn "get_shelves_userenv cannot retrieve context->{userenv}->{context->{activeuser}}";
1114 return;
1116 my $totshelves = $active->{totshelves} or undef;
1117 my $pubshelves = $active->{pubshelves} or undef;
1118 my $barshelves = $active->{barshelves} or undef;
1119 return ($totshelves, $pubshelves, $barshelves);
1122 =head2 _new_userenv
1124 C4::Context->_new_userenv($session); # FIXME: This calling style is wrong for what looks like an _internal function
1126 Builds a hash for user environment variables.
1128 This hash shall be cached for future use: if you call
1129 C<C4::Context-E<gt>userenv> twice, you will get the same hash without real DB access
1131 _new_userenv is called in Auth.pm
1133 =cut
1136 sub _new_userenv
1138 shift; # Useless except it compensates for bad calling style
1139 my ($sessionID)= @_;
1140 $context->{"activeuser"}=$sessionID;
1143 =head2 _unset_userenv
1145 C4::Context->_unset_userenv;
1147 Destroys the hash for activeuser user environment variables.
1149 =cut
1153 sub _unset_userenv
1155 my ($sessionID)= @_;
1156 undef $context->{"activeuser"} if ($context->{"activeuser"} eq $sessionID);
1160 =head2 get_versions
1162 C4::Context->get_versions
1164 Gets various version info, for core Koha packages, Currently called from carp handle_errors() sub, to send to browser if 'DebugLevel' syspref is set to '2'.
1166 =cut
1170 # A little example sub to show more debugging info for CGI::Carp
1171 sub get_versions {
1172 my %versions;
1173 $versions{kohaVersion} = Koha::version();
1174 $versions{kohaDbVersion} = C4::Context->preference('version');
1175 $versions{osVersion} = join(" ", POSIX::uname());
1176 $versions{perlVersion} = $];
1178 no warnings qw(exec); # suppress warnings if unable to find a program in $PATH
1179 $versions{mysqlVersion} = `mysql -V`;
1180 $versions{apacheVersion} = `httpd -v`;
1181 $versions{apacheVersion} = `httpd2 -v` unless $versions{apacheVersion} ;
1182 $versions{apacheVersion} = `apache2 -v` unless $versions{apacheVersion} ;
1183 $versions{apacheVersion} = `/usr/sbin/apache2 -v` unless $versions{apacheVersion} ;
1185 return %versions;
1189 =head2 tz
1191 C4::Context->tz
1193 Returns a DateTime::TimeZone object for the system timezone
1195 =cut
1197 sub tz {
1198 my $self = shift;
1199 if (!defined $context->{tz}) {
1200 $context->{tz} = DateTime::TimeZone->new(name => 'local');
1202 return $context->{tz};
1206 =head2 IsSuperLibrarian
1208 C4::Context->IsSuperLibrarian();
1210 =cut
1212 sub IsSuperLibrarian {
1213 my $userenv = C4::Context->userenv;
1215 unless ( $userenv and exists $userenv->{flags} ) {
1216 # If we reach this without a user environment,
1217 # assume that we're running from a command-line script,
1218 # and act as a superlibrarian.
1219 carp("C4::Context->userenv not defined!");
1220 return 1;
1223 return ($userenv->{flags}//0) % 2;
1226 =head2 interface
1228 Sets the current interface for later retrieval in any Perl module
1230 C4::Context->interface('opac');
1231 C4::Context->interface('intranet');
1232 my $interface = C4::Context->interface;
1234 =cut
1236 sub interface {
1237 my ($class, $interface) = @_;
1239 if (defined $interface) {
1240 $interface = lc $interface;
1241 if ($interface eq 'opac' || $interface eq 'intranet') {
1242 $context->{interface} = $interface;
1243 } else {
1244 warn "invalid interface : '$interface'";
1248 return $context->{interface} // 'opac';
1252 __END__
1254 =head1 ENVIRONMENT
1256 =head2 C<KOHA_CONF>
1258 Specifies the configuration file to read.
1260 =head1 SEE ALSO
1262 XML::Simple
1264 =head1 AUTHORS
1266 Andrew Arensburger <arensb at ooblick dot com>
1268 Joshua Ferraro <jmf at liblime dot com>