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