geekigeeki.py

   1 #!/usr/bin/python
   2 # -*- coding: utf-8 -*-
   3 #
   4 # Copyright 1999, 2000 Martin Pool <mbp@humbug.org.au>
   5 # Copyright 2002 Gerardo Poggiali
   6 # Copyright 2007, 2008 Bernie Innocenti <bernie@codewiz.org>
   7 #
   8 # This program is free software: you can redistribute it and/or modify
   9 # it under the terms of the GNU General Public License as published by
  10 # the Free Software Foundation, either version 3 of the License, or
  11 # (at your option) any later version.
  12 #
  13 # This program is distributed in the hope that it will be useful, but
  14 # WITHOUT ANY WARRANTY; without even the implied warranty of
  15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16 # General Public License for more details.
  17 #
  18 # You should have received a copy of the GNU General Public License
  19 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20
  21 __version__ = '$Id$'[4:12]
  22
  23 from time import clock
  24 start_time = clock()
  25
  26 import cgi, sys, os, re, errno, stat
  27 from os import path, environ
  28
  29 # Regular expression defining a WikiWord
  30 # (but this definition is also assumed in other places)
  31 word_re = re.compile(r"^\b((([A-Z][a-z0-9]+){2,}/)*([A-Z][a-z0-9]+){2,})\b$")
  32 # FIXME: we accept stuff like foo/../bar and we shouldn't
  33 file_re = re.compile(r"^\b([A-Za-z0-9_\-][A-Za-z0-9_\.\-/]*)\b$")
  34 img_re = re.compile(r"^.*\.(png|gif|jpg|jpeg)$", re.IGNORECASE)
  35 url_re = re.compile(r"^[a-z]{3,8}://[^\s'\"]+\S$")
  36 link_re = re.compile("(?:\[\[|{{)([^\s\|]+)(?:\s*\|\s*([^\]]+)|)(?:\]\]|}})")
  37
  38 title_done = False
  39
  40
  41 # CGI stuff ---------------------------------------------------------
  42
  43 def script_name():
  44     return environ.get('SCRIPT_NAME', '')
  45
  46 def privileged_path():
  47     return privileged_url or script_name()
  48
  49 def remote_user():
  50     user = environ.get('REMOTE_USER', '')
  51     if user is None or user == '' or user == 'anonymous':
  52         user = 'AnonymousCoward'
  53     return user
  54
  55 def remote_host():
  56     return environ.get('REMOTE_ADDR', '')
  57
  58 def get_hostname(addr):
  59     try:
  60         from socket import gethostbyaddr
  61         return gethostbyaddr(addr)[0] + ' (' + addr + ')'
  62     except Exception, er:
  63         return addr
  64
  65 def relative_url(pathname, privileged=False):
  66     if not (url_re.match(pathname) or pathname.startswith('/')):
  67         if privileged:
  68             url = privileged_path()
  69         else:
  70             url = script_name()
  71         pathname = url + '/' + pathname
  72     return pathname
  73
  74 def permalink(s):
  75     return re.sub(' ', '-', re.sub('[^a-z0-9_ ]', '', s.lower()).strip())
  76
  77 # Formatting stuff --------------------------------------------------
  78 def emit_header(mime_type="text/html"):
  79     print "Content-type: " + mime_type + "; charset=utf-8\n"
  80
  81 def send_guru(msg_text, msg_type):
  82     if not msg_text: return
  83     print '<pre id="guru" onclick="this.style.display = \'none\'" class="' + msg_type + '">'
  84     if msg_type == 'error':
  85         print '    Software Failure.  Press left mouse button to continue.\n'
  86     print msg_text
  87     if msg_type == 'error':
  88         print '      Guru Meditation #DEADBEEF.ABADC0DE'
  89     print '</pre>'
  90     # FIXME: This little JS snippet is harder to pass than ACID 3.0
  91     print """
  92     <script language="JavaScript" type="text/javascript">
  93         var guru = document.getElementById('guru');
  94         // Firefox 2.0 doesn't take border-color, but returns border-top-color fine
  95         var color = document.defaultView.getComputedStyle(guru,null).getPropertyValue('border-top-color');
  96
  97         function guruOn() {
  98             guru.style.setProperty('border-color', color, '');
  99             setTimeout('guruOff()', 1000);
 100         }
 101         function guruOff() {
 102             guru.style.setProperty('border-color', '#000000', '');
 103             setTimeout('guruOn()', 1000);
 104         }
 105         // Safari 2.0 returns this rgba crap
 106         // Konqueror 3.5.6 doesn't seem to support computed properties
 107         if (color && color != 'rgba(0, 0, 0, 0)') {
 108             //window.alert("enabled! color='" + color + "'");
 109             guruOn();
 110         }
 111     </script>"""
 112
 113 def send_title(name, text="Limbo", msg_text=None, msg_type='error'):
 114     global title_done
 115     if title_done: return
 116
 117     # Head
 118     emit_header()
 119     print '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"'
 120     print '  "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">'
 121     print '<html xmlns="http://www.w3.org/1999/xhtml" lang="en" xml:lang="en">'
 122
 123     site_name = globals().get('site_name', 'Unconfigured Site')
 124     print "<head><title>%s: %s</title>" % (site_name, text)
 125     print ' <meta http-equiv="Content-Type" content="application/xhtml+xml; charset=utf-8" />'
 126     if not name:
 127         print ' <meta name="robots" content="noindex,nofollow" />'
 128     for css in css_url:
 129         print ' <link rel="stylesheet" type="text/css" href="%s" />' % relative_url(css)
 130     print '</head>'
 131
 132     # Body
 133     if name and privileged_url is not None:
 134         print '<body ondblclick="location.href=\'' + privileged_path() + '?edit=' + name + '\'">'
 135     else:
 136         print '<body>'
 137
 138     title_done = True
 139     send_guru(msg_text, msg_type)
 140
 141     # Navbar
 142     print '<div class="nav">'
 143     if name:
 144         print '  <b>' + link_tag('?fullsearch=' + name, text, 'navlink') + '</b> '
 145     else:
 146         print '  <b>' + text + '</b> '
 147     print ' | ' + link_tag('FrontPage', 'Front Page', 'navlink')
 148     print ' | ' + link_tag('FindPage', 'Find Page', 'navlink')
 149     if 'history_url' in globals():
 150         print ' | <a href="' + history_url + '" class="navlink">Recent Changes</a>'
 151         if name:
 152             print ' | <a href="' + history_url + '?a=history;f=' + name + '" class="navlink">Page History</a>'
 153
 154     if name:
 155         print ' | ' + link_tag('?raw=' + name, 'Raw Text', 'navlink')
 156         if privileged_url is not None:
 157             print ' | ' + link_tag('?edit=' + name, 'Edit Page', 'navlink', privileged=True)
 158     else:
 159         print ' | <i>Immutable Page</i>'
 160
 161     user = remote_user()
 162     if user != 'AnonymousCoward':
 163         print ' | <span class="login"><i>logged in as <b>' + cgi.escape(user) + '</b></i></span>'
 164
 165     print '<hr /></div>'
 166
 167 def link_tag(params, text=None, ss_class=None, privileged=False):
 168     if text is None:
 169         text = params # default
 170     classattr = ''
 171     if ss_class:
 172         classattr += 'class="%s" ' % ss_class
 173         # Prevent crawlers from following links potentially added by spammers or to generated pages
 174         if ss_class == 'external' or ss_class == 'navlink':
 175             classattr += 'rel="nofollow" '
 176     elif url_re.match(params):
 177         classattr += 'rel="nofollow" '
 178     return '<a %shref="%s">%s</a>' % (classattr, relative_url(params, privileged=privileged), text)
 179
 180 # Search ---------------------------------------------------
 181
 182 def handle_fullsearch(needle):
 183     send_title(None, 'Full text search for "%s"' % (needle))
 184
 185     needle_re = re.compile(needle, re.IGNORECASE)
 186     hits = []
 187     all_pages = page_list()
 188     for page_name in all_pages:
 189         body = Page(page_name).get_raw_body()
 190         count = len(needle_re.findall(body))
 191         if count:
 192             hits.append((count, page_name))
 193
 194     # The default comparison for tuples compares elements in order,
 195     # so this sorts by number of hits
 196     hits.sort()
 197     hits.reverse()
 198
 199     print "<ul>"
 200     for (count, page_name) in hits:
 201         print '<li><p>' + Page(page_name).link_to()
 202         print ' . . . . ' + `count`
 203         print ['match', 'matches'][count != 1]
 204         print '</p></li>'
 205     print "</ul>"
 206
 207     print_search_stats(len(hits), len(all_pages))
 208
 209 def handle_titlesearch(needle):
 210     # TODO: check needle is legal -- but probably we can just accept any RE
 211     send_title(None, "Title search for \"" + needle + '"')
 212
 213     needle_re = re.compile(needle, re.IGNORECASE)
 214     all_pages = page_list()
 215     hits = filter(needle_re.search, all_pages)
 216
 217     print "<ul>"
 218     for filename in hits:
 219         print '<li><p>' + Page(filename).link_to() + "</p></li>"
 220     print "</ul>"
 221
 222     print_search_stats(len(hits), len(all_pages))
 223
 224 def print_search_stats(hits, searched):
 225     print "<p>%d hits out of %d pages searched.</p>" % (hits, searched)
 226
 227 def handle_raw(pagename):
 228     Page(pagename).send_raw()
 229
 230 def handle_edit(pagename):
 231     pg = Page(pagename)
 232     if 'save' in form:
 233         if form['file'].value:
 234             pg.save(form['file'].file.read())
 235         else:
 236             pg.save(form['savetext'].value.replace('\r\n', '\n'))
 237         pg.format()
 238     elif 'cancel' in form:
 239         pg.msg_text = 'Editing canceled'
 240         pg.msg_type = 'notice'
 241         pg.format()
 242     else: # preview or edit
 243         text = None
 244         if 'preview' in form:
 245             text = form['savetext'].value
 246         pg.send_editor(text)
 247
 248 def make_index_key():
 249     links = map(lambda ch: '<a href="#%s">%s</a>' % (ch, ch), 'abcdefghijklmnopqrstuvwxyz')
 250     return '<p><center>'+ ' | '.join(links) + '</center></p>'
 251
 252 def page_list(dir = None, re = word_re):
 253     return filter(re.match, os.listdir(dir or data_dir))
 254
 255 def send_footer(name, mod_string=None):
 256     if globals().get('debug_cgi', False):
 257         cgi.print_arguments()
 258         cgi.print_form(form)
 259         cgi.print_environ()
 260     print '<div id="footer"><hr />'
 261     print ('<p class="copyright"><span class="benchmark">generated in %0.3fs</span>' +
 262         ' by <a href="http://www.codewiz.org/wiki/GeekiGeeki">GeekiGeeki</a>' +
 263         ' version %s</p>') % (clock() - start_time, __version__)
 264     if mod_string:
 265         print '<p class="modified">last modified %s</p>' % mod_string
 266     print '</div></body></html>'
 267
 268 class WikiFormatter:
 269     """Object that turns Wiki markup into HTML.
 270
 271     All formatting commands can be parsed one line at a time, though
 272     some state is carried over between lines.
 273     """
 274     def __init__(self, raw):
 275         self.raw = raw
 276         self.h_level = 0
 277         self.in_pre = self.in_html = self.in_table = self.in_li = False
 278         self.in_header = True
 279         self.list_indents = []
 280         self.tr_cnt = 0
 281         self.styles = {
 282             #wiki   html   enabled?
 283             "//":  ["em",  False],
 284             "''":  ["em",  False],
 285             "**":  ["b",   False],
 286             "'''": ["b",   False],
 287             "##":  ["tt",  False],
 288             "``":  ["tt",  False],
 289             "__":  ["u",   False],
 290             "^^":  ["sup", False],
 291             ",,":  ["sub", False]
 292         }
 293
 294     def _b_repl(self, word):
 295         style = self.styles[word]
 296         style[1] = not style[1]
 297         return ['</', '<'][style[1]] + style[0] + '>'
 298
 299     def _tit_repl(self, word):
 300         if self.h_level:
 301             result = '</h%d><p>\n' % self.h_level
 302             self.h_level = 0
 303         else:
 304             self.h_level = len(word) - 1
 305             link = permalink(self.line)
 306             result = '\n</p><h%d id="%s"><a class="heading" href="#%s">¶</a> ' % (self.h_level, link, link)
 307         return result
 308
 309     def _br_repl(self, word):
 310         return '<br />'
 311
 312     def _rule_repl(self, word):
 313         return self._undent() + '\n<hr size="%d" noshade="noshade" />\n' % (len(word) - 2)
 314
 315     def _word_repl(self, word):
 316         return Page(word).link_to()
 317
 318     def _img_repl(self, word):
 319         path = relative_url(word)
 320         return '<a href="%s"><img border="0" src="%s" /></a>' % (path, path)
 321
 322     def _url_repl(self, word):
 323         if img_re.match(word):
 324             return '<a href="%s"><img border="0" src="%s" /></a>' % (word, word)
 325         else:
 326             return '<a href="%s" rel="nofollow" class="external">%s</a>' % (word, word)
 327
 328     def _macro_repl(self, word):
 329         m = re.compile("\<\<([^\s\|\>]+)(?:\s*\|\s*([^\>]+)|)\>\>").match(word)
 330         name = m.group(1)
 331         argv = [name]
 332         if m.group(2):
 333             argv.extend(m.group(2).split('|'))
 334         argv = map(str.strip, argv)
 335
 336         macro = globals().get('_macro_' + name)
 337         if not macro:
 338             execfile("macros/" + name + ".py", globals())
 339             macro = globals().get('_macro_' + name)
 340         if macro:
 341             return macro(argv)
 342         else:
 343             return '<strong class="error">&lt;&lt;' + '|'.join(argv) + '&gt;&gt;</strong>'
 344
 345     def _hurl_repl(self, word):
 346         m = link_re.match(word)
 347         name = m.group(1)
 348         descr = m.group(2) or name
 349
 350         if img_re.match(name):
 351             # DEPRECATED SYNTAX: use {{foo.jpg|descr}} instead
 352             _inl_repl(self, word)
 353         else:
 354             if img_re.match(descr):
 355                 descr = '<img border="0" src="' + descr + '" />'
 356
 357             return link_tag(name, descr, 'wikilink')
 358
 359     def _inl_repl(self, word):
 360         m = link_re.match(word)
 361         name = m.group(1)
 362         descr = m.group(2) or name
 363         name = relative_url(name)
 364         argv = descr.split('|')
 365         descr = argv.pop(0)
 366         if argv:
 367             args = '?' + '&amp;'.join(argv)
 368         else:
 369             args = ''
 370         if descr:
 371             # The "extthumb" nonsense works around a limitation of the HTML block model
 372             return '<div class="extthumb"><div class="thumb"><a href="%s"><img border="0" src="%s" alt="%s" /></a><div class="caption">%s</div></div></div>' \
 373                     % (name, name + args, descr, descr)
 374         else:
 375             return '<a href="%s"><img border="0" src="%s" /></a>' % (name, name + args)
 376
 377     def _email_repl(self, word):
 378         return '<a href="mailto:%s">%s</a>' % (word, word)
 379
 380     def _html_repl(self, word):
 381         self.in_html += 1
 382         return word; # Pass through
 383
 384     def _ent_repl(self, s):
 385         if self.in_html and s == '>':
 386             self.in_html -= 1
 387             return '>'
 388         return {'&': '&amp;',
 389                 '<': '&lt;',
 390                 '>': '&gt;'}[s]
 391
 392     def _li_repl(self, match):
 393         if self.in_li:
 394             return '</li><li>'
 395         else:
 396             self.in_li = True
 397             return '<li>'
 398
 399     def _pre_repl(self, word):
 400         if word == '{{{' and not self.in_pre:
 401             self.in_pre = True
 402             return '<pre>'
 403         elif self.in_pre:
 404             self.in_pre = False
 405             return '</pre>'
 406         return ''
 407
 408     def _hi_repl(self, word):
 409         return '<strong class="highlight ' + word + '">' + word + '</strong>'
 410
 411     def _tr_repl(self, word):
 412         out = ''
 413         if not self.in_table:
 414             self.in_table = True
 415             self.tr_cnt = 0
 416             out = '</p><table><tbody>\n'
 417         self.tr_cnt += 1
 418         out = out + '<tr class="' + ['even', 'odd'][self.tr_cnt % 2] + '">'
 419         return out + ['<td>', '<th>'][word.strip() == '||=']
 420
 421     def _td_repl(self, word):
 422         if self.in_table:
 423             return ['</td><td>', '</th><th>'][word.strip() == '||=']
 424         return ''
 425
 426     def _tre_repl(self, word):
 427         if self.in_table:
 428             return ['</td></tr>', '</th></tr>'][word.strip() == '||=']
 429         return ''
 430
 431     def _indent_level(self):
 432         return len(self.list_indents) and self.list_indents[-1]
 433
 434     def _indent_to(self, new_level):
 435         if self._indent_level() == new_level:
 436             return ''
 437         s = '</p>'
 438         while self._indent_level() > new_level:
 439             del(self.list_indents[-1])
 440             if self.in_li:
 441                 s += '</li>'
 442                 self.in_li = False # FIXME
 443             s += '</ul>\n'
 444         while self._indent_level() < new_level:
 445             self.list_indents.append(new_level)
 446             s += '<ul>\n'
 447         s += '<p>'
 448         return s
 449
 450     def _undent(self):
 451         res = '</p>'
 452         res += '</ul>' * len(self.list_indents)
 453         res += '<p>'
 454         self.list_indents = []
 455         return res
 456
 457     def replace(self, match):
 458         for type, hit in match.groupdict().items():
 459             if hit:
 460                 return getattr(self, '_' + type + '_repl')(hit)
 461         else:
 462             raise "Can't handle match " + `match`
 463
 464     def print_html(self):
 465         print '<div class="wiki"><p>'
 466
 467         # For each line, we scan through looking for magic
 468         # strings, outputting verbatim any intervening text
 469         # TODO: highlight search words (look at referrer)
 470         scan_re = re.compile(
 471             r"(?:"
 472             # Formatting
 473             + r"(?P<b>\*\*|'''|//|''|##|``|__|\^\^|,,)"
 474             + r"|(?P<tit>\={2,6})"
 475             + r"|(?P<br>\\\\)"
 476             + r"|(?P<rule>^-{3,})"
 477             + r"|(?P<hi>\b(FIXME|TODO|DONE)\b)"
 478
 479             # Links
 480             + r"|(?P<macro>\<\<([^\s\|\>]+)(?:\s*\|\s*([^\>]+)|)\>\>)"
 481             + r"|(?P<hurl>\[\[([^\s\|]+)(?:\s*\|\s*([^\]]+)|)\]\])"
 482
 483             # Inline HTML
 484             + r"|(?P<html><(/|)(br|hr|div|form|iframe|input|span))"
 485             + r"|(?P<ent>[<>&])"
 486
 487             # Auto links
 488             + r"|(?P<img>\b[a-zA-Z0-9_/-]+\.(png|gif|jpg|jpeg|bmp))" # LEGACY
 489             + r"|(?P<word>\b(?:[A-Z][a-z]+){2,}\b)" # LEGACY
 490             + r"|(?P<url>(http|https|ftp|mailto)\:[^\s'\"]+\S)" # LEGACY
 491             + r"|(?P<email>[-\w._+]+\@[\w.-]+)" # LEGACY
 492
 493             # Lists, divs, spans
 494             + r"|(?P<li>^\s+[\*#] +)"
 495             + r"|(?P<pre>\{\{\{|\s*\}\}\})"
 496             + r"|(?P<inl>\{\{([^\s\|]+)(?:\s*\|\s*([^\]]+)|)\}\})" #TODO
 497
 498             # Tables
 499             + r"|(?P<tr>^\s*\|\|(=|)\s*)"
 500             + r"|(?P<tre>\s*\|\|(=|)\s*$)"
 501             + r"|(?P<td>\s*\|\|(=|)\s*)"
 502             + r")")
 503         pre_re = re.compile(
 504             r"(?:"
 505             + r"(?P<pre>\s*\}\}\})"
 506             + r"|(?P<ent>[<>&])"
 507             + r")")
 508         blank_re = re.compile(r"^\s*$")
 509         indent_re = re.compile(r"^\s*")
 510         tr_re = re.compile(r"^\s*\|\|")
 511         eol_re = re.compile(r"\r?\n")
 512         for self.line in eol_re.split(self.raw.expandtabs()):
 513             # Skip ACLs
 514             if self.in_header:
 515                 if self.line.startswith('#'):
 516                     continue
 517                 self.in_header = False
 518
 519             if self.in_pre:
 520                 print re.sub(pre_re, self.replace, self.line)
 521             else:
 522                 if self.in_table and not tr_re.match(self.line):
 523                     self.in_table = False
 524                     print '</tbody></table><p>'
 525
 526                 if blank_re.match(self.line):
 527                     print '</p><p>'
 528                 else:
 529                     indent = indent_re.match(self.line)
 530                     print self._indent_to(len(indent.group(0)))
 531                     print re.sub(scan_re, self.replace, self.line)
 532
 533         if self.in_pre: print '</pre>'
 534         if self.in_table: print '</tbody></table><p>'
 535         print self._undent()
 536         print '</p></div>'
 537
 538 class Page:
 539     def __init__(self, page_name):
 540         self.page_name = page_name
 541         self.msg_text = ''
 542         self.msg_type = 'error'
 543
 544     def split_title(self):
 545         # look for the end of words and the start of a new word and insert a space there
 546         return re.sub('([a-z])([A-Z])', r'\1 \2', self.page_name)
 547
 548     def _filename(self):
 549         return path.join(data_dir, self.page_name)
 550
 551     def _tmp_filename(self):
 552         return path.join(data_dir, ('#' + self.page_name.replace('/','_') + '.' + `os.getpid()` + '#'))
 553
 554     def exists(self):
 555         try:
 556             os.stat(self._filename())
 557             return True
 558         except OSError, er:
 559             if er.errno == errno.ENOENT:
 560                 return False
 561             raise er
 562
 563     def link_to(self):
 564         word = self.page_name
 565         if self.exists():
 566             return link_tag(word, word, 'wikilink')
 567         else:
 568             return link_tag(word, nonexist_pfx + word, 'nonexistent')
 569
 570     def get_raw_body(self):
 571         try:
 572             return open(self._filename(), 'rb').read()
 573         except IOError, er:
 574             if er.errno == errno.ENOENT:
 575                 return '' # just doesn't exist, use default
 576             if er.errno == errno.EISDIR:
 577                 return self.format_dir()
 578             raise er
 579
 580     def format_dir(self):
 581         out = '== '
 582         path = ''
 583         for dir in self.page_name.split('/'):
 584             path = (path + '/' + dir) if path else dir
 585             out += '[[' + path + '|' + dir + ']]/'
 586         out += ' ==\n'
 587
 588         for file in page_list(self._filename(), file_re):
 589             if img_re.match(file):
 590                 if image_maxwidth:
 591                     maxwidth_arg = '|maxwidth=' + str(image_maxwidth)
 592                 out += '{{' + self.page_name + '/' + file + '|' + file + maxwidth_arg + '}}\n'
 593             else:
 594                 out += ' * [[' + self.page_name + '/' + file + ']]\n'
 595         return out
 596     def get_attrs(self):
 597         if 'attrs' in self.__dict__:
 598             return self.attrs
 599         self.attrs = {}
 600         try:
 601             file = open(self._filename(), 'rt')
 602             attr_re = re.compile(r"^#(\S*)(.*)$")
 603             for line in file:
 604                 m = attr_re.match(line)
 605                 if not m:
 606                     break
 607                 self.attrs[m.group(1)] = m.group(2).strip()
 608                 #print "bernie: attrs[" + m.group(1) + "] = " + m.group(2) + "<br>\n"
 609         except IOError, er:
 610             if er.errno != errno.ENOENT:
 611                 raise er
 612         return self.attrs
 613
 614     def get_attr(self, name, default):
 615         return self.get_attrs().get(name, default)
 616
 617     def can(self, action, default=True):
 618         acl = None
 619         try:
 620             #acl SomeUser:read,write All:read
 621             acl = self.get_attr("acl", None)
 622             for rule in acl.split():
 623                 (user, perms) = rule.split(':')
 624                 if user == remote_user() or user == "All":
 625                     return action in perms.split(',')
 626             return False
 627         except Exception, er:
 628             if acl:
 629                 self.msg_text = 'Illegal acl line: ' + acl
 630         return default
 631
 632     def can_write(self):
 633         return self.can("write", True)
 634
 635     def can_read(self):
 636         return self.can("read", True)
 637
 638     def send_naked(self):
 639         if self.can_read():
 640             WikiFormatter(self.get_raw_body()).print_html()
 641         else:
 642             send_guru("Read access denied by ACLs", "notice")
 643
 644     def format(self):
 645         page_name = None
 646         if self.can_write():
 647             page_name = self.page_name
 648
 649         #css foo.css bar.css
 650         global css_url
 651         css_url = self.get_attr("css", "").split() + css_url
 652
 653         send_title(page_name, self.split_title(), msg_text=self.msg_text, msg_type=self.msg_type)
 654         self.send_naked()
 655         send_footer(page_name, self._last_modified())
 656
 657     def _last_modified(self):
 658         try:
 659             from time import localtime, strftime
 660             modtime = localtime(os.stat(self._filename())[stat.ST_MTIME])
 661         except OSError, er:
 662             if er.errno != errno.ENOENT:
 663                 raise er
 664             return None
 665         return strftime(datetime_fmt, modtime)
 666
 667     def send_editor(self, preview=None):
 668         send_title(None, 'Edit ' + self.split_title(), msg_text=self.msg_text, msg_type=self.msg_type)
 669         if not self.can_write():
 670             send_guru("Write access denied by ACLs", "error")
 671             return
 672
 673         file = ''
 674         if 'file' in form:
 675             file = form['file'].value
 676
 677         print ('<p><b>Editing ' + self.page_name
 678             + ' for ' + cgi.escape(remote_user())
 679             + ' from ' + cgi.escape(get_hostname(remote_host()))
 680             + '</b></p>')
 681         print '<div class="editor"><form method="post" enctype="multipart/form-data" action="%s">' % relative_url(self.page_name)
 682         print '<input type="hidden" name="edit" value="%s">' % (self.page_name)
 683         print '<textarea wrap="off" spellcheck="true" id="editor" name="savetext" rows="17" cols="100">%s</textarea>' % (preview or self.get_raw_body())
 684         print 'Or upload a file: <input type="file" name="file" value="%s" />' % file
 685         print """
 686             <br />
 687             <input type="submit" name="save" value="Save" />
 688             <input type="submit" name="preview" value="Preview" />
 689             <input type="reset" value="Reset" />
 690             <input type="submit" name="cancel" value="Cancel" />
 691             <br />
 692             </form></div>"""
 693         print "<p>" + Page('EditingTips').link_to() + "</p>"
 694         if preview:
 695             print "<div class='preview'>"
 696             WikiFormatter(preview).print_html()
 697             print "</div>"
 698         send_footer(self.page_name)
 699
 700     def send_raw(self, mimetype='text/plain'):
 701         if self.can_read():
 702             body = self.get_raw_body()
 703             emit_header(mimetype)
 704             print body
 705         else:
 706             send_title(None, msg_text='Read access denied by ACLs', msg_type='notice')
 707
 708     def send_image(self, mimetype, args=[]):
 709         if 'maxwidth' in args:
 710             import subprocess
 711             emit_header(mimetype)
 712             sys.stdout.flush()
 713             subprocess.check_call(['gm', 'convert', self._filename(),
 714                 '-scale', args['maxwidth'].value + ' >', '-'])
 715         else:
 716             self.send_raw(mimetype)
 717
 718     def _write_file(self, data):
 719         tmp_filename = self._tmp_filename()
 720         open(tmp_filename, 'wb').write(data)
 721         name = self._filename()
 722         if os.name == 'nt':
 723             # Bad Bill!  POSIX rename ought to replace. :-(
 724             try:
 725                 os.remove(name)
 726             except OSError, er:
 727                 if er.errno != errno.ENOENT: raise er
 728         os.rename(tmp_filename, name)
 729
 730     def save(self, newdata):
 731         if not self.can_write():
 732             self.msg_text = 'Write access denied by ACLs'
 733             self.msg_type = 'error'
 734             return
 735
 736         self._write_file(newdata)
 737         rc = 0
 738         if post_edit_hook:
 739             # FIXME: what's the std way to perform shell quoting in python?
 740             cmd = ( post_edit_hook
 741                 + " '" + data_dir + '/' + self.page_name
 742                 + "' '" + remote_user()
 743                 + "' '" + remote_host() + "'"
 744             )
 745             out = os.popen(cmd)
 746             output = out.read()
 747             rc = out.close()
 748         if rc:
 749             self.msg_text += "Post-editing hook returned %d.\n" % rc
 750             self.msg_text += 'Command was: ' + cmd + '\n'
 751             if output:
 752                 self.msg_text += 'Output follows:\n' + output
 753         else:
 754             self.msg_text = 'Thank you for your contribution.  Your attention to detail is appreciated.'
 755             self.msg_type = 'success'
 756
 757 # Main ---------------------------------------------------------------
 758 try:
 759     execfile("geekigeeki.conf.py")
 760     form = cgi.FieldStorage()
 761
 762     for cmd in form:
 763         handler = globals().get('handle_' + cmd)
 764         if handler:
 765             handler(form[cmd].value)
 766             break
 767     else:
 768         path_info = environ.get('PATH_INFO', '')
 769         if len(path_info) and path_info[0] == '/':
 770             query = path_info[1:] or 'FrontPage'
 771         else:
 772             query = environ.get('QUERY_STRING', '') or 'FrontPage'
 773
 774         if file_re.match(query):
 775             if word_re.match(query):
 776                 Page(query).format()
 777             else:
 778                 from mimetypes import MimeTypes
 779                 type, encoding = MimeTypes().guess_type(query)
 780                 #type = type or 'text/plain'
 781                 #Page(query).send_raw(mimetype=type)
 782                 if type:
 783                     if type.startswith('image/'):
 784                         Page(query).send_image(mimetype=type,args=form)
 785                     else:
 786                         Page(query).send_raw(mimetype=type)
 787                 else:
 788                     Page(query).format()
 789         else:
 790             print "Status: 404 Not Found"
 791             send_title(None, msg_text='Can\'t work out query: ' + query)
 792             send_footer(None)
 793 except Exception:
 794     import traceback
 795     msg_text = traceback.format_exc()
 796     if title_done:
 797         send_guru(msg_text, "error")
 798     else:
 799         send_title(None, msg_text=msg_text)
 800     send_footer(None)
 801
 802 sys.stdout.flush()