IkiWiki/Plugin/tex.pm

   1 #!/usr/bin/perl
   2 # Licensed under GPL v2 or greater
   3 # (c) 2007 Patrick Winnertz <patrick.winnertz@skolelinux.org>
   4
   5 package IkiWiki::Plugin::tex;
   6 use warnings;
   7 use strict;
   8 use Digest::MD5 qw(md5_hex);
   9 use File::Temp qw(tempdir);
  10 use URI::Escape qw(uri_escape);
  11 use IkiWiki 2.00;
  12
  13 sub import { #{{{
  14         hook(type => "preprocess", id => "tex", call => \&preprocess);
  15 } #}}}
  16
  17 sub preprocess (@) { #{{{
  18         my %params = @_;
  19
  20         my $height = $params{height};
  21         if (! defined $height || ! length $height) {
  22                 $height = 12;
  23         }
  24         else {
  25                 $height =~ s#(\d+)#$1#;
  26         }
  27
  28         my $code = $params{code};
  29         if (! defined $code && ! length $code) {
  30                 return "[[tex ".gettext("missing tex code"). "]]";
  31         }
  32
  33         if (check($code)) {
  34                 return create($code, check_height($height), \%params);
  35         }
  36         else {
  37                 return "[[tex ".gettext("code includes disallowed latex commands"). "]]";
  38         }
  39 } #}}}
  40
  41 sub check_height ($) { #{{{
  42         # Since latex doesn't support unlimited scaling this function
  43         # returns the closest supported size.
  44         my $height =shift;
  45
  46         my @allowed=(8,9,10,11,12,14,17,20);
  47
  48         my $ret;
  49         my $fit;
  50         foreach my $val (@allowed) {
  51                 my $f = abs($val - $height);
  52                 if (! defined($fit) || $f < $fit ) {
  53                         $ret=$val;
  54                         $fit=$f;
  55                 }
  56         }
  57         return $ret;
  58 } #}}}
  59
  60 sub create ($$$) { #{{{
  61         # This function calls the image generating function and returns
  62         # the <img .. /> for the generated image.
  63         my $code = shift;
  64         my $height = shift;
  65         my $params = shift;
  66
  67         if (! defined($height) and not length($height) ) {
  68                 $height = 12;
  69         }
  70
  71         my $digest = md5_hex($code, $height);
  72
  73         my $teximgdir = "/teximages";
  74         my $imglink = "$teximgdir/$digest.png";
  75         my $imglog = "$teximgdir/$digest.log";
  76         will_render($params->{destpage}, $imglink);
  77         will_render($params->{destpage}, $imglog);
  78
  79         my $imgurl;
  80         my $logurl;
  81         if (! $params->{preview}) {
  82                 $imgurl = urlto($imglink, $params->{destpage});
  83                 $logurl = urlto($imglog, $params->{destpage});
  84         }
  85         else {
  86                 $imgurl="$config{url}/$teximgdir/$digest.png";
  87                 $logurl="$config{url}/$teximgdir/$digest.log";
  88         }
  89
  90         if (-e "$config{destdir}/$imglink" ||
  91             gen_image($code, $height, $digest, $teximgdir)) {
  92                 return qq{<img src="$imgurl" alt="}.uri_escape($code).qq{" class="teximage" />};
  93         }
  94         else {
  95                 return qq{[[tex <a href="$logurl">}.gettext("failed to generate image from code")."</a>]]";
  96         }
  97 } #}}}
  98
  99 sub gen_image ($$$$) { #{{{
 100         # Actually creates the image.
 101         my $code = shift;
 102         my $height = shift;
 103         my $digest = shift;
 104         my $imagedir = shift;
 105
 106         #TODO This should move into the setup file.
 107         my $tex = '\documentclass['.$height.'pt]{scrartcl}';
 108         $tex .= '\usepackage[version=3]{mhchem}';
 109         $tex .= '\usepackage{amsmath}';
 110         $tex .= '\usepackage{amsfonts}';
 111         $tex .= '\usepackage{amssymb}';
 112         $tex .= '\pagestyle{empty}';
 113         $tex .= '\begin{document}';
 114         $tex .= '$$'.$code.'$$';
 115         $tex .= '\end{document}';
 116
 117         my $tmp = eval { create_tmp_dir($digest) };
 118         if (! $@ &&
 119             writefile("$digest.tex", $tmp, $tex) &&
 120             system("cd $tmp; latex --interaction=nonstopmode $tmp/$digest.tex > /dev/null") == 0 &&
 121             system("dvips -E $tmp/$digest.dvi -o $tmp/$digest.ps 2> $tmp/$digest.log") == 0 &&
 122             # ensure destination directory exists
 123             writefile("$imagedir/$digest.png", $config{destdir}, "") &&
 124             system("convert -density 120  -trim -transparent \"#FFFFFF\" $tmp/$digest.ps $config{destdir}/$imagedir/$digest.png > $tmp/$digest.log") == 0) {
 125                 return 1;
 126         }
 127         else {
 128                 # store failure log
 129                 my $log;
 130                 {
 131                         open(my $f, '<', "$tmp/$digest.log");
 132                         local $/=undef;
 133                         $log = <$f>;
 134                         close($f);
 135                 }
 136                 writefile("$digest.log", "$config{destdir}/$imagedir", $log);
 137
 138                 return 0;
 139         }
 140 } #}}}
 141
 142 sub create_tmp_dir ($) { #{{{
 143         # Create a temp directory, it will be removed when ikiwiki exits.
 144         my $base = shift;
 145
 146         my $template = $base.".XXXXXXXXXX";
 147         my $tmpdir = tempdir($template, TMPDIR => 1, CLEANUP => 1);
 148         return $tmpdir;
 149 } #}}}
 150
 151 sub check ($) { #{{{
 152         # Check if the code is ok
 153         my $code = shift;
 154
 155         my @badthings = (
 156                 qr/\$\$/,
 157                 qr/\\include/,
 158                 qr/\\includegraphic/,
 159                 qr/\\usepackage/,
 160                 qr/\\newcommand/,
 161                 qr/\\renewcommand/,
 162                 qr/\\def/,
 163                 qr/\\input/,
 164                 qr/\\open/,
 165                 qr/\\loop/,
 166                 qr/\\errorstopmode/,
 167                 qr/\\scrollmode/,
 168                 qr/\\batchmode/,
 169                 qr/\\read/,
 170                 qr/\\write/,
 171         );
 172
 173         foreach my $thing (@badthings) {
 174                 if ($code =~ m/$thing/ ) {
 175                         return 0;
 176                 }
 177         }
 178         return 1;
 179 } #}}}
 180
 181 1