Fix UTF-8 issues in reading templates.
[pr0n] / perl / Sesse / pr0n / Templates.pm
index 3e65a9a9217767f9bb7592a4e6f917023c3aacea..ecac13f623d3c6e29e5bd59fc3a8e086c129de61 100644 (file)
@@ -14,12 +14,11 @@ BEGIN {
        %EXPORT_TAGS = qw();
        @EXPORT_OK   = qw();
 }
        %EXPORT_TAGS = qw();
        @EXPORT_OK   = qw();
 }
-our %dirs;
+our %dirs = ();
 
 sub update_dirs {
        my $r = shift;
 
 sub update_dirs {
        my $r = shift;
-       my $base = $r->dir_config('TemplateBase');
-       %dirs = ();
+       my $base = $Sesse::pr0n::Config::template_base;
        
        for my $dir (<$base/*>) {
                next unless -d $dir;
        
        for my $dir (<$base/*>) {
                next unless -d $dir;
@@ -33,11 +32,11 @@ sub update_dirs {
 sub r_to_dir {
        my $r = shift;
 
 sub r_to_dir {
        my $r = shift;
 
-       if (!defined(%dirs)) {
+       if (scalar(keys %dirs) == 0) {
                update_dirs($r);
        }
        
                update_dirs($r);
        }
        
-       my $site = $r->get_server_name();
+       my $site = Sesse::pr0n::Common::get_server_name($r);
        if (defined($dirs{$site})) {
                return $site;
        } else {
        if (defined($dirs{$site})) {
                return $site;
        } else {
@@ -56,8 +55,8 @@ sub fetch_template {
 
        my $newcache = {};
 
 
        my $newcache = {};
 
-       my $base = $r->dir_config('TemplateBase');
-       open TEMPLATE, "<$base/$dir/$template"
+       my $base = $Sesse::pr0n::Config::template_base;
+       open TEMPLATE, "<:utf8", "$base/$dir/$template"
                or ($dir ne 'default' and open TEMPLATE, "<$base/default/$template")
                or Sesse::pr0n::Common::error($r, "Couldn't open $dir/$template: $!");
 
                or ($dir ne 'default' and open TEMPLATE, "<$base/default/$template")
                or Sesse::pr0n::Common::error($r, "Couldn't open $dir/$template: $!");
 
@@ -85,8 +84,8 @@ sub process_template {
 }
 
 sub print_template {
 }
 
 sub print_template {
-       my ($r, $template, $args) = @_;
-       $r->print(process_template($r, $template, $args));
+       my ($r, $io, $template, $args) = @_;
+       $io->print(Encode::encode_utf8(process_template($r, $template, $args)));
 }
 
 1;
 }
 
 1;