geekigeeki.py

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