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