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|bmp|ico)$", 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', writable=False):
 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
 129     for link in link_urls:
 130         rel, href = link
 131         print ' <link rel="%s" href="%s" />' % (rel, relative_url(href))
 132
 133     if name and writable and privileged_url is not None:
 134         print ' <link rel="alternate" type="application/x-wiki" title="Edit this page" href="%s" />' \
 135             % (privileged_path() + '?edit=' + name)
 136
 137     if history_url is not None:
 138         print ' <link rel="alternate" type="application/rss+xml" title="RSS" href="%s" />' \
 139             % relative_url(history_url + '?a=rss')
 140
 141     print '</head>'
 142
 143     # Body
 144     if name and writable and privileged_url is not None:
 145         print '<body ondblclick="location.href=\'' + privileged_path() + '?edit=' + name + '\'">'
 146     else:
 147         print '<body>'
 148
 149     title_done = True
 150     send_guru(msg_text, msg_type)
 151
 152     # Navbar
 153     print '<div class="nav">'
 154     if name:
 155         print '  <b>' + link_tag('?fullsearch=' + name, text, 'navlink') + '</b> '
 156     else:
 157         print '  <b>' + text + '</b> '
 158     print ' | ' + link_tag('FrontPage', 'Home', 'navlink')
 159     print ' | ' + link_tag('FindPage', 'Find Page', 'navlink')
 160     if 'history_url' in globals():
 161         print ' | <a href="' + relative_url(history_url) + '" class="navlink">Recent Changes</a>'
 162         if name:
 163             print ' | <a href="' + relative_url(history_url + '?a=history;f=' + name) + '" class="navlink">Page History</a>'
 164
 165     if name:
 166         print ' | ' + link_tag('?raw=' + name, 'Raw Text', 'navlink')
 167         if privileged_url is not None:
 168             if writable:
 169                 print ' | ' + link_tag('?edit=' + name, 'Edit', 'navlink', privileged=True)
 170             else:
 171                 print ' | ' + link_tag(name, 'Login', 'navlink', privileged=True)
 172
 173     else:
 174         print ' | <i>Immutable Page</i>'
 175
 176     user = remote_user()
 177     if user != 'AnonymousCoward':
 178         print ' | <span class="login"><i>logged in as <b>' + cgi.escape(user) + '</b></i></span>'
 179
 180     print '<hr /></div>'
 181
 182 def link_tag(params, text=None, ss_class=None, privileged=False):
 183     if text is None:
 184         text = params # default
 185     classattr = ''
 186     if ss_class:
 187         classattr += 'class="%s" ' % ss_class
 188         # Prevent crawlers from following links potentially added by spammers or to generated pages
 189         if ss_class == 'external' or ss_class == 'navlink':
 190             classattr += 'rel="nofollow" '
 191     elif url_re.match(params):
 192         classattr += 'rel="nofollow" '
 193     return '<a %shref="%s">%s</a>' % (classattr, relative_url(params, privileged=privileged), text)
 194
 195 # Search ---------------------------------------------------
 196
 197 def handle_fullsearch(needle):
 198     send_title(None, 'Full text search for "%s"' % (needle))
 199
 200     needle_re = re.compile(needle, re.IGNORECASE)
 201     hits = []
 202     all_pages = page_list()
 203     for page_name in all_pages:
 204         body = Page(page_name).get_raw_body()
 205         count = len(needle_re.findall(body))
 206         if count:
 207             hits.append((count, page_name))
 208
 209     # The default comparison for tuples compares elements in order,
 210     # so this sorts by number of hits
 211     hits.sort()
 212     hits.reverse()
 213
 214     print "<ul>"
 215     for (count, page_name) in hits:
 216         print '<li><p>' + Page(page_name).link_to()
 217         print ' . . . . ' + `count`
 218         print ['match', 'matches'][count != 1]
 219         print '</p></li>'
 220     print "</ul>"
 221
 222     print_search_stats(len(hits), len(all_pages))
 223
 224 def handle_titlesearch(needle):
 225     # TODO: check needle is legal -- but probably we can just accept any RE
 226     send_title(None, "Title search for \"" + needle + '"')
 227
 228     needle_re = re.compile(needle, re.IGNORECASE)
 229     all_pages = page_list()
 230     hits = filter(needle_re.search, all_pages)
 231
 232     print "<ul>"
 233     for filename in hits:
 234         print '<li><p>' + Page(filename).link_to() + "</p></li>"
 235     print "</ul>"
 236
 237     print_search_stats(len(hits), len(all_pages))
 238
 239 def print_search_stats(hits, searched):
 240     print "<p>%d hits out of %d pages searched.</p>" % (hits, searched)
 241
 242 def handle_raw(pagename):
 243     Page(pagename).send_raw()
 244
 245 def handle_edit(pagename):
 246     pg = Page(pagename)
 247     if 'save' in form:
 248         if form['file'].value:
 249             pg.save(form['file'].file.read())
 250         else:
 251             pg.save(form['savetext'].value.replace('\r\n', '\n'))
 252         pg.format()
 253     elif 'cancel' in form:
 254         pg.msg_text = 'Editing canceled'
 255         pg.msg_type = 'notice'
 256         pg.format()
 257     else: # preview or edit
 258         text = None
 259         if 'preview' in form:
 260             text = form['savetext'].value
 261         pg.send_editor(text)
 262
 263 def make_index_key():
 264     links = map(lambda ch: '<a href="#%s">%s</a>' % (ch, ch), 'abcdefghijklmnopqrstuvwxyz')
 265     return '<p><center>'+ ' | '.join(links) + '</center></p>'
 266
 267 def page_list(dir = None, re = word_re):
 268     return filter(re.match, os.listdir(dir or data_dir))
 269
 270 def send_footer(name, mod_string=None):
 271     if globals().get('debug_cgi', False):
 272         cgi.print_arguments()
 273         cgi.print_form(form)
 274         cgi.print_environ()
 275     print '<div id="footer"><hr />'
 276     print ('<p class="copyright"><span class="benchmark">generated in %0.3fs</span>' +
 277         ' by <a href="http://www.codewiz.org/wiki/GeekiGeeki">GeekiGeeki</a>' +
 278         ' version %s</p>') % (clock() - start_time, __version__)
 279     if mod_string:
 280         print '<p class="modified">last modified %s</p>' % mod_string
 281     print '</div></body></html>'
 282
 283 class WikiFormatter:
 284     """Object that turns Wiki markup into HTML.
 285
 286     All formatting commands can be parsed one line at a time, though
 287     some state is carried over between lines.
 288     """
 289     def __init__(self, raw):
 290         self.raw = raw
 291         self.h_level = 0
 292         self.in_pre = self.in_html = self.in_table = self.in_li = False
 293         self.in_header = True
 294         self.list_indents = []
 295         self.tr_cnt = 0
 296         self.styles = {
 297             #wiki   html   enabled?
 298             "//":  ["em",  False],
 299             "''":  ["em",  False],
 300             "**":  ["b",   False],
 301             "'''": ["b",   False],
 302             "##":  ["tt",  False],
 303             "``":  ["tt",  False],
 304             "__":  ["u",   False],
 305             "^^":  ["sup", False],
 306             ",,":  ["sub", False]
 307         }
 308
 309     def _b_repl(self, word):
 310         style = self.styles[word]
 311         style[1] = not style[1]
 312         return ['</', '<'][style[1]] + style[0] + '>'
 313
 314     def _tit_repl(self, word):
 315         if self.h_level:
 316             result = '</h%d><p>\n' % self.h_level
 317             self.h_level = 0
 318         else:
 319             self.h_level = len(word) - 1
 320             link = permalink(self.line)
 321             result = '\n</p><h%d id="%s"><a class="heading" href="#%s">¶</a> ' % (self.h_level, link, link)
 322         return result
 323
 324     def _br_repl(self, word):
 325         return '<br />'
 326
 327     def _rule_repl(self, word):
 328         return self._undent() + '\n<hr size="%d" noshade="noshade" />\n' % (len(word) - 2)
 329
 330     def _word_repl(self, word):
 331         return Page(word).link_to()
 332
 333     def _img_repl(self, word):
 334         path = relative_url(word)
 335         return '<a href="%s"><img border="0" src="%s" /></a>' % (path, path)
 336
 337     def _url_repl(self, word):
 338         if img_re.match(word):
 339             return '<a href="%s"><img border="0" src="%s" /></a>' % (word, word)
 340         else:
 341             return '<a href="%s" rel="nofollow" class="external">%s</a>' % (word, word)
 342
 343     def _macro_repl(self, word):
 344         m = re.compile("\<\<([^\s\|\>]+)(?:\s*\|\s*([^\>]+)|)\>\>").match(word)
 345         name = m.group(1)
 346         argv = [name]
 347         if m.group(2):
 348             argv.extend(m.group(2).split('|'))
 349         argv = map(str.strip, argv)
 350
 351         macro = globals().get('_macro_' + name)
 352         if not macro:
 353             execfile("macros/" + name + ".py", globals())
 354             macro = globals().get('_macro_' + name)
 355         if macro:
 356             return macro(argv)
 357         else:
 358             return '<strong class="error">&lt;&lt;' + '|'.join(argv) + '&gt;&gt;</strong>'
 359
 360     def _hurl_repl(self, word):
 361         m = link_re.match(word)
 362         name = m.group(1)
 363         if m.group(2) is None:
 364             descr = name
 365         elif img_re.match(m.group(2)):
 366             descr = '<img border="0" src="' + descr + '" />'
 367         else:
 368             descr = m.group(2)
 369
 370         return link_tag(name, descr, 'wikilink')
 371
 372     def _inl_repl(self, word):
 373         m = link_re.match(word)
 374         name = m.group(1)
 375         descr = m.group(2) or name
 376         name = relative_url(name)
 377         argv = descr.split('|')
 378         descr = argv.pop(0)
 379
 380         if argv:
 381             args = '?' + '&amp;'.join(argv)
 382         else:
 383             args = ''
 384
 385         if descr:
 386             # The "extthumb" nonsense works around a limitation of the HTML block model
 387             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>' \
 388                     % (name, name + args, descr, descr)
 389         else:
 390             return '<a href="%s"><img border="0" src="%s" /></a>' % (name, name + args)
 391
 392     def _email_repl(self, word):
 393         return '<a href="mailto:%s">%s</a>' % (word, word)
 394
 395     def _html_repl(self, word):
 396         self.in_html += 1
 397         return word; # Pass through
 398
 399     def _ent_repl(self, s):
 400         if self.in_html and s == '>':
 401             self.in_html -= 1
 402             return '>'
 403         return {'&': '&amp;',
 404                 '<': '&lt;',
 405                 '>': '&gt;'}[s]
 406
 407     def _li_repl(self, match):
 408         if self.in_li:
 409             return '</li><li>'
 410         else:
 411             self.in_li = True
 412             return '<li>'
 413
 414     def _pre_repl(self, word):
 415         if word == '{{{' and not self.in_pre:
 416             self.in_pre = True
 417             return '<pre>'
 418         elif self.in_pre:
 419             self.in_pre = False
 420             return '</pre>'
 421         return ''
 422
 423     def _hi_repl(self, word):
 424         return '<strong class="highlight ' + word + '">' + word + '</strong>'
 425
 426     def _tr_repl(self, word):
 427         out = ''
 428         if not self.in_table:
 429             self.in_table = True
 430             self.tr_cnt = 0
 431             out = '</p><table><tbody>\n'
 432         self.tr_cnt += 1
 433         out = out + '<tr class="' + ['even', 'odd'][self.tr_cnt % 2] + '">'
 434         return out + ['<td>', '<th>'][word.strip() == '||=']
 435
 436     def _td_repl(self, word):
 437         if self.in_table:
 438             return ['</td><td>', '</th><th>'][word.strip() == '||=']
 439         return ''
 440
 441     def _tre_repl(self, word):
 442         if self.in_table:
 443             return ['</td></tr>', '</th></tr>'][word.strip() == '||=']
 444         return ''
 445
 446     def _indent_level(self):
 447         return len(self.list_indents) and self.list_indents[-1]
 448
 449     def _indent_to(self, new_level):
 450         if self._indent_level() == new_level:
 451             return ''
 452         s = '</p>'
 453         while self._indent_level() > new_level:
 454             del(self.list_indents[-1])
 455             if self.in_li:
 456                 s += '</li>'
 457                 self.in_li = False # FIXME
 458             s += '</ul>\n'
 459         while self._indent_level() < new_level:
 460             self.list_indents.append(new_level)
 461             s += '<ul>\n'
 462         s += '<p>'
 463         return s
 464
 465     def _undent(self):
 466         res = '</p>'
 467         res += '</ul>' * len(self.list_indents)
 468         res += '<p>'
 469         self.list_indents = []
 470         return res
 471
 472     def replace(self, match):
 473         for type, hit in match.groupdict().items():
 474             if hit:
 475                 return getattr(self, '_' + type + '_repl')(hit)
 476         else:
 477             raise "Can't handle match " + `match`
 478
 479     def print_html(self):
 480         print '<div class="wiki"><p>'
 481
 482         # For each line, we scan through looking for magic
 483         # strings, outputting verbatim any intervening text
 484         # TODO: highlight search words (look at referrer)
 485         scan_re = re.compile(
 486             r"(?:"
 487             # Formatting
 488             + r"(?P<b>\*\*|'''|//|''|##|``|__|\^\^|,,)"
 489             + r"|(?P<tit>\={2,6})"
 490             + r"|(?P<br>\\\\)"
 491             + r"|(?P<rule>^-{3,})"
 492             + r"|(?P<hi>\b(FIXME|TODO|DONE)\b)"
 493
 494             # Links
 495             + r"|(?P<macro>\<\<([^\s\|\>]+)(?:\s*\|\s*([^\>]+)|)\>\>)"
 496             + r"|(?P<hurl>\[\[([^\s\|]+)(?:\s*\|\s*([^\]]+)|)\]\])"
 497
 498             # Inline HTML
 499             + r"|(?P<html><(/|)(br|hr|div|form|iframe|input|span))"
 500             + r"|(?P<ent>[<>&])"
 501
 502             # Auto links (LEGACY)
 503             + r"|(?P<img>\b[a-zA-Z0-9_/-]+\.(png|gif|jpg|jpeg|bmp|ico))"
 504             + r"|(?P<word>\b(?:[A-Z][a-z]+){2,}\b)"
 505             + r"|(?P<url>(http|https|ftp|mailto)\:[^\s'\"]+\S)"
 506             + r"|(?P<email>[-\w._+]+\@[\w.-]+)"
 507
 508             # Lists, divs, spans
 509             + r"|(?P<li>^\s+[\*#] +)"
 510             + r"|(?P<pre>\{\{\{|\s*\}\}\})"
 511             + r"|(?P<inl>\{\{([^\s\|]+)(?:\s*\|\s*([^\]]+)|)\}\})"
 512
 513             # Tables
 514             + r"|(?P<tr>^\s*\|\|(=|)\s*)"
 515             + r"|(?P<tre>\s*\|\|(=|)\s*$)"
 516             + r"|(?P<td>\s*\|\|(=|)\s*)"
 517             + r")")
 518         pre_re = re.compile(
 519             r"(?:"
 520             + r"(?P<pre>\s*\}\}\})"
 521             + r"|(?P<ent>[<>&])"
 522             + r")")
 523         blank_re = re.compile(r"^\s*$")
 524         indent_re = re.compile(r"^\s*")
 525         tr_re = re.compile(r"^\s*\|\|")
 526         eol_re = re.compile(r"\r?\n")
 527         for self.line in eol_re.split(self.raw.expandtabs()):
 528             # Skip ACLs
 529             if self.in_header:
 530                 if self.line.startswith('#'):
 531                     continue
 532                 self.in_header = False
 533
 534             if self.in_pre:
 535                 print re.sub(pre_re, self.replace, self.line)
 536             else:
 537                 if self.in_table and not tr_re.match(self.line):
 538                     self.in_table = False
 539                     print '</tbody></table><p>'
 540
 541                 if blank_re.match(self.line):
 542                     print '</p><p>'
 543                 else:
 544                     indent = indent_re.match(self.line)
 545                     print self._indent_to(len(indent.group(0)))
 546                     print re.sub(scan_re, self.replace, self.line)
 547
 548         if self.in_pre: print '</pre>'
 549         if self.in_table: print '</tbody></table><p>'
 550         print self._undent()
 551         print '</p></div>'
 552
 553 class Page:
 554     def __init__(self, page_name):
 555         self.page_name = page_name
 556         self.msg_text = ''
 557         self.msg_type = 'error'
 558
 559     def split_title(self):
 560         # look for the end of words and the start of a new word and insert a space there
 561         return re.sub('([a-z])([A-Z])', r'\1 \2', self.page_name)
 562
 563     def _filename(self):
 564         return path.join(data_dir, self.page_name)
 565
 566     def _tmp_filename(self):
 567         return path.join(data_dir, ('#' + self.page_name.replace('/','_') + '.' + `os.getpid()` + '#'))
 568
 569     def exists(self):
 570         try:
 571             os.stat(self._filename())
 572             return True
 573         except OSError, er:
 574             if er.errno == errno.ENOENT:
 575                 return False
 576             raise er
 577
 578     def link_to(self):
 579         word = self.page_name
 580         if self.exists():
 581             return link_tag(word, word, 'wikilink')
 582         else:
 583             return link_tag(word, nonexist_pfx + word, 'nonexistent')
 584
 585     def get_raw_body(self):
 586         try:
 587             return open(self._filename(), 'rb').read()
 588         except IOError, er:
 589             if er.errno == errno.ENOENT:
 590                 return '' # just doesn't exist, use default
 591             if er.errno == errno.EISDIR:
 592                 return self.format_dir()
 593             raise er
 594
 595     def format_dir(self):
 596         out = '== '
 597         path = ''
 598         for dir in self.page_name.split('/'):
 599             path = (path + '/' + dir) if path else dir
 600             out += '[[' + path + '|' + dir + ']]/'
 601         out += ' ==\n'
 602
 603         for file in page_list(self._filename(), file_re):
 604             if img_re.match(file):
 605                 if image_maxwidth:
 606                     maxwidth_arg = '|maxwidth=' + str(image_maxwidth)
 607                 out += '{{' + self.page_name + '/' + file + '|' + file + maxwidth_arg + '}}\n'
 608             else:
 609                 out += ' * [[' + self.page_name + '/' + file + ']]\n'
 610         return out
 611     def get_attrs(self):
 612         if 'attrs' in self.__dict__:
 613             return self.attrs
 614         self.attrs = {}
 615         try:
 616             file = open(self._filename(), 'rt')
 617             attr_re = re.compile(r"^#(\S*)(.*)$")
 618             for line in file:
 619                 m = attr_re.match(line)
 620                 if not m:
 621                     break
 622                 self.attrs[m.group(1)] = m.group(2).strip()
 623                 #print "bernie: attrs[" + m.group(1) + "] = " + m.group(2) + "<br>\n"
 624         except IOError, er:
 625             if er.errno != errno.ENOENT and er.errno != errno.EISDIR:
 626                 raise er
 627         return self.attrs
 628
 629     def get_attr(self, name, default):
 630         return self.get_attrs().get(name, default)
 631
 632     def can(self, action, default=True):
 633         acl = None
 634         try:
 635             #acl SomeUser:read,write All:read
 636             acl = self.get_attr("acl", None)
 637             for rule in acl.split():
 638                 (user, perms) = rule.split(':')
 639                 if user == remote_user() or user == "All":
 640                     return action in perms.split(',')
 641             return False
 642         except Exception, er:
 643             if acl:
 644                 self.msg_text = 'Illegal acl line: ' + acl
 645         return default
 646
 647     def can_write(self):
 648         return self.can("write", True)
 649
 650     def can_read(self):
 651         return self.can("read", True)
 652
 653     def send_naked(self):
 654         if self.can_read():
 655             WikiFormatter(self.get_raw_body()).print_html()
 656         else:
 657             send_guru("Read access denied by ACLs", "notice")
 658
 659     def format(self):
 660         #css foo.css
 661         value = self.get_attr("css", None)
 662         if value:
 663             global link_urls
 664             link_urls += { "stylesheet": value }
 665
 666         send_title(self.page_name, self.split_title(),
 667             msg_text=self.msg_text, msg_type=self.msg_type, writable=self.can_write())
 668         self.send_naked()
 669         send_footer(self.page_name, self._last_modified())
 670
 671     def _last_modified(self):
 672         try:
 673             from time import localtime, strftime
 674             modtime = localtime(os.stat(self._filename())[stat.ST_MTIME])
 675         except OSError, er:
 676             if er.errno != errno.ENOENT:
 677                 raise er
 678             return None
 679         return strftime(datetime_fmt, modtime)
 680
 681     def send_editor(self, preview=None):
 682         send_title(None, 'Edit ' + self.split_title(), msg_text=self.msg_text, msg_type=self.msg_type)
 683         if not self.can_write():
 684             send_guru("Write access denied by ACLs", "error")
 685             return
 686
 687         file = ''
 688         if 'file' in form:
 689             file = form['file'].value
 690
 691         print ('<p><b>Editing ' + self.page_name
 692             + ' for ' + cgi.escape(remote_user())
 693             + ' from ' + cgi.escape(get_hostname(remote_host()))
 694             + '</b></p>')
 695         print '<div class="editor"><form method="post" enctype="multipart/form-data" action="%s">' % relative_url(self.page_name)
 696         print '<input type="hidden" name="edit" value="%s">' % (self.page_name)
 697         print '<textarea wrap="off" spellcheck="true" id="editor" name="savetext" rows="17" cols="100">%s</textarea>' % (preview or self.get_raw_body())
 698         print 'Or upload a file: <input type="file" name="file" value="%s" />' % file
 699         print """
 700             <br />
 701             <input type="submit" name="save" value="Save" />
 702             <input type="submit" name="preview" value="Preview" />
 703             <input type="reset" value="Reset" />
 704             <input type="submit" name="cancel" value="Cancel" />
 705             <br />
 706             </form></div>"""
 707         print "<p>" + Page('EditingTips').link_to() + "</p>"
 708         if preview:
 709             print "<div class='preview'>"
 710             WikiFormatter(preview).print_html()
 711             print "</div>"
 712         send_footer(self.page_name)
 713
 714     def send_raw(self, mimetype='text/plain'):
 715         if self.can_read():
 716             body = self.get_raw_body()
 717             emit_header(mimetype)
 718             print body
 719         else:
 720             send_title(None, msg_text='Read access denied by ACLs', msg_type='notice')
 721
 722     def send_image(self, mimetype, args=[]):
 723         if 'maxwidth' in args:
 724             import subprocess
 725             emit_header(mimetype)
 726             sys.stdout.flush()
 727             subprocess.check_call(['gm', 'convert', self._filename(),
 728                 '-scale', args['maxwidth'].value + ' >', '-'])
 729         else:
 730             self.send_raw(mimetype)
 731
 732     def _write_file(self, data):
 733         tmp_filename = self._tmp_filename()
 734         open(tmp_filename, 'wb').write(data)
 735         name = self._filename()
 736         if os.name == 'nt':
 737             # Bad Bill!  POSIX rename ought to replace. :-(
 738             try:
 739                 os.remove(name)
 740             except OSError, er:
 741                 if er.errno != errno.ENOENT: raise er
 742         os.rename(tmp_filename, name)
 743
 744     def save(self, newdata):
 745         if not self.can_write():
 746             self.msg_text = 'Write access denied by ACLs'
 747             self.msg_type = 'error'
 748             return
 749
 750         self._write_file(newdata)
 751         rc = 0
 752         if post_edit_hook:
 753             # FIXME: what's the std way to perform shell quoting in python?
 754             cmd = ( post_edit_hook
 755                 + " '" + data_dir + '/' + self.page_name
 756                 + "' '" + remote_user()
 757                 + "' '" + remote_host() + "'"
 758             )
 759             out = os.popen(cmd)
 760             output = out.read()
 761             rc = out.close()
 762         if rc:
 763             self.msg_text += "Post-editing hook returned %d.\n" % rc
 764             self.msg_text += 'Command was: ' + cmd + '\n'
 765             if output:
 766                 self.msg_text += 'Output follows:\n' + output
 767         else:
 768             self.msg_text = 'Thank you for your contribution.  Your attention to detail is appreciated.'
 769             self.msg_type = 'success'
 770
 771 # Main ---------------------------------------------------------------
 772 try:
 773     execfile("geekigeeki.conf.py")
 774     form = cgi.FieldStorage()
 775
 776     for cmd in form:
 777         handler = globals().get('handle_' + cmd)
 778         if handler:
 779             handler(form[cmd].value)
 780             break
 781     else:
 782         path_info = environ.get('PATH_INFO', '')
 783         if len(path_info) and path_info[0] == '/':
 784             query = path_info[1:] or 'FrontPage'
 785         else:
 786             query = environ.get('QUERY_STRING', '') or 'FrontPage'
 787
 788         if file_re.match(query):
 789             if word_re.match(query):
 790                 Page(query).format()
 791             else:
 792                 from mimetypes import MimeTypes
 793                 type, encoding = MimeTypes().guess_type(query)
 794                 #type = type or 'text/plain'
 795                 #Page(query).send_raw(mimetype=type)
 796                 if type:
 797                     if type.startswith('image/'):
 798                         Page(query).send_image(mimetype=type,args=form)
 799                     else:
 800                         Page(query).send_raw(mimetype=type)
 801                 else:
 802                     Page(query).format()
 803         else:
 804             print "Status: 404 Not Found"
 805             send_title(None, msg_text='Can\'t work out query: ' + query)
 806             send_footer(None)
 807 except Exception:
 808     import traceback
 809     msg_text = traceback.format_exc()
 810     if title_done:
 811         send_guru(msg_text, "error")
 812     else:
 813         send_title(None, msg_text=msg_text)
 814     send_footer(None)
 815
 816 sys.stdout.flush()