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():
 268     return _macro_search("titlesearch")
 269
 270 def _macro_FullSearch():
 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="Go" /></form>""" % (type, default)
 279
 280 def _macro_WordIndex():
 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():
 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.list_indents = []
 341         self.in_pre = False
 342         self.in_table = False
 343         self.tr_cnt = 0
 344         self.in_var = False
 345         self.in_header = True
 346
 347     def _emph_repl(self, word):
 348         if len(word) == 3:
 349             self.is_b = not self.is_b
 350             return ['</strong>', '<strong>'][self.is_b]
 351         else:
 352             self.is_em = not self.is_em
 353             return ['</em>', '<em>'][self.is_em]
 354
 355     def _tit_repl(self, word):
 356         if self.h_level:
 357             result = "</h%d>" % self.h_level
 358             self.h_level = 0
 359         else:
 360             self.h_level = len(word) - 1
 361             result = "<h%d>" % self.h_level
 362         return result
 363
 364     def _rule_repl(self, word):
 365         s = self._undent()
 366         if len(word) <= 3:
 367             s = s + "\n<hr size='1' noshade=\"noshade\" />\n"
 368         else:
 369             s = s + "\n<hr size='%d' noshade=\"noshade\" />\n" % (len(word) - 2 )
 370         return s
 371
 372     def _word_repl(self, word):
 373         return Page(word).link_to()
 374
 375     def _img_repl(self, word):
 376         return '<img border="0" src="%s/%s" />' % (script_name(), word)
 377
 378     def _url_repl(self, word):
 379         if img_re.match(word):
 380             return '<img border="0" src="%s" />' % word
 381         else:
 382             return '<a href="%s" rel="nofollow" class="external">%s</a>' % (word, word)
 383
 384     def _hurl_repl(self, word):
 385         m = re.compile("\[\[(\S+)\ (.+)\]\]").match(word)
 386         anchor = m.group(1)
 387         descr = m.group(2)
 388         if img_re.match(anchor):
 389             return '<img border="0" src="%s" alt="%s" />' % (anchor, descr)
 390         elif url_re.match(anchor):
 391             return '<a href="%s" rel="nofollow" class="external">%s</a>' % (anchor, descr)
 392         elif anchor.startswith('/'):
 393             return '<a href="%s">%s</a>' % (anchor, descr)
 394         else:
 395             return link_tag(anchor, descr)
 396
 397     def _email_repl(self, word):
 398         return '<a href="mailto:%s">%s</a>' % (word, word)
 399
 400
 401     def _ent_repl(self, s):
 402         return {'&': '&amp;',
 403                 '<': '&lt;',
 404                 '>': '&gt;'}[s]
 405
 406
 407     def _li_repl(self, match):
 408         return '<li>'
 409
 410
 411     def _pre_repl(self, word):
 412         if word == '{{{' and not self.in_pre:
 413             self.in_pre = True
 414             return '<pre>'
 415         elif self.in_pre:
 416             self.in_pre = False
 417             return '</pre>'
 418         return ''
 419
 420     def _hi_repl(self, word):
 421         if word == 'FIXME':
 422             cl = 'error'
 423         elif word == 'DONE':
 424             cl = 'success'
 425         elif word == 'TODO':
 426             cl = 'notice'
 427         return '<strong class="highlight ' + cl + '">' + word + '</strong>'
 428
 429     def _var_repl(self, word):
 430         if word == '{{' and not self.in_var:
 431             self.in_var = True
 432             return '<code>'
 433         elif self.in_var:
 434             self.in_var = False
 435             return '</code>'
 436         return ''
 437
 438     def _tr_repl(self, word):
 439         out = ''
 440         if not self.in_table:
 441             self.in_table = True
 442             self.tr_cnt = 0
 443             out = '</p><table><tbody>\n'
 444         self.tr_cnt += 1
 445         return out + '<tr class="' + ['even', 'odd'][self.tr_cnt % 2] + '"><td>'
 446
 447     def _tre_repl(self, word):
 448         if self.in_table:
 449             return '</td></tr>'
 450         return ''
 451
 452     def _td_repl(self, word):
 453         if self.in_table:
 454             return '</td><td>'
 455         return ''
 456
 457     def _macro_repl(self, word):
 458         macro_name = word[2:-2]
 459         # TODO: Somehow get the default value into the search field
 460         return apply(globals()['_macro_' + macro_name], ())
 461
 462     def _indent_level(self):
 463         return len(self.list_indents) and self.list_indents[-1]
 464
 465     def _indent_to(self, new_level):
 466         if self._indent_level() == new_level:
 467             return ''
 468         s = '</p>'
 469         while self._indent_level() > new_level:
 470             del(self.list_indents[-1])
 471             s += '</ul>\n'
 472         while self._indent_level() < new_level:
 473             self.list_indents.append(new_level)
 474             s += '<ul>\n'
 475         s += '<p>'
 476         return s
 477
 478     def _undent(self):
 479         res = '</p>'
 480         res += '</ul>' * len(self.list_indents)
 481         res += '<p>'
 482         self.list_indents = []
 483         return res
 484
 485     def replace(self, match):
 486         for type, hit in match.groupdict().items():
 487             if hit:
 488                 return apply(getattr(self, '_' + type + '_repl'), (hit,))
 489         else:
 490             raise "Can't handle match " + `match`
 491
 492     def print_html(self):
 493         print "<div class='wiki'><p>"
 494
 495         # For each line, we scan through looking for magic
 496         # strings, outputting verbatim any intervening text
 497         # TODO: highlight search words (look at referer)
 498         scan_re = re.compile(
 499             r"(?:"
 500             # Formatting
 501             + r"(?P<emph>'{2,3})"
 502             + r"|(?P<tit>\={2,6})"
 503             + r"|(?P<rule>^-{3,})"
 504             + r"|(?P<ent>[<>&])"
 505             + r"|(?P<hi>\b(FIXME|TODO|DONE)\b)"
 506
 507             # Links
 508             + r"|(?P<img>\b[a-zA-Z0-9_-]+\.(png|gif|jpg|jpeg|bmp))"
 509             + r"|(?P<word>\b(?:[A-Z][a-z]+){2,}\b)"
 510             + r"|(?P<hurl>\[\[\S+\s+.+\]\])"
 511             + r"|(?P<url>(http|ftp|nntp|news|mailto)\:[^\s'\"]+\S)"
 512             + r"|(?P<email>[-\w._+]+\@[\w.-]+)"
 513
 514             # Lists, divs, spans
 515             + r"|(?P<li>^\s+\*)"
 516             + r"|(?P<pre>(\{\{\{|\s*\}\}\}))"
 517             + r"|(?P<var>(\{\{|\}\}))"
 518
 519             # Tables
 520             + r"|(?P<tr>^\s*\|\|\s*)"
 521             + r"|(?P<tre>\s*\|\|\s*$)"
 522             + r"|(?P<td>\s*\|\|\s*)"
 523
 524             # Macros
 525             + r"|(?P<macro>\[\[(TitleSearch|FullSearch|WordIndex|TitleIndex)\]\])"
 526             + r")")
 527         pre_re = re.compile(
 528             r"(?:"
 529             + r"(?P<pre>\s*\}\}\})"
 530             + r")")
 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         raw = string.expandtabs(self.raw)
 536         for line in eol_re.split(raw):
 537             # Skip ACLs
 538             if self.in_header:
 539                 if line.startswith('#'):
 540                    continue
 541                 self.in_header = False
 542
 543             if self.in_pre:
 544                 print re.sub(pre_re, self.replace, line)
 545             else:
 546                 if self.in_table and not tr_re.match(line):
 547                     self.in_table = False
 548                     print '</tbody></table><p>'
 549
 550                 if blank_re.match(line):
 551                     print '</p><p>'
 552                 else:
 553                     indent = indent_re.match(line)
 554                     print self._indent_to(len(indent.group(0)))
 555                     print re.sub(scan_re, self.replace, line)
 556
 557         if self.in_pre: print '</pre>'
 558         if self.in_table: print '</tbody></table><p>'
 559         print self._undent()
 560         print "</p></div>"
 561
 562 # ----------------------------------------------------------
 563 class Page:
 564     def __init__(self, page_name):
 565         self.page_name = page_name
 566         self.msg = ''
 567         self.msg_type = 'error'
 568         self.attrs = {}
 569
 570     def split_title(self):
 571         # look for the end of words and the start of a new word,
 572         # and insert a space there
 573         return re.sub('([a-z])([A-Z])', r'\1 \2', self.page_name)
 574
 575     def _text_filename(self):
 576         return path.join(text_dir, self.page_name)
 577
 578     def _tmp_filename(self):
 579         return path.join(text_dir, ('#' + self.page_name + '.' + `os.getpid()` + '#'))
 580
 581     def exists(self):
 582         try:
 583             os.stat(self._text_filename())
 584             return 1
 585         except OSError, er:
 586             if er.errno == errno.ENOENT:
 587                 return 0
 588             else:
 589                 raise er
 590
 591     def link_to(self):
 592         word = self.page_name
 593         if self.exists():
 594             return link_tag(word, word, 'wikilink')
 595         else:
 596             if nonexist_qm:
 597                 return link_tag(word, '?', 'nonexistent') + word
 598             else:
 599                 return link_tag(word, word, 'nonexistent')
 600
 601
 602     def get_raw_body(self):
 603         try:
 604             return open(self._text_filename(), 'rt').read()
 605         except IOError, er:
 606             if er.errno == errno.ENOENT:
 607                 # just doesn't exist, use default
 608                 return 'Describe %s here.' % self.page_name
 609             else:
 610                 raise er
 611
 612     def get_attrs(self):
 613         if self.attrs:
 614             return self.attrs
 615         try:
 616             file = open(self._text_filename(), 'rt')
 617             attr_re = re.compile(r"^#(\S*)(.*)$")
 618             for line in file:
 619                 m = attr_re.match(line)
 620                 if not m:
 621                     break
 622                 self.attrs[m.group(1)] = m.group(2).strip()
 623                 #print "bernie: attrs[" + m.group(1) + "] = " + m.group(2) + "<br>\n"
 624         except IOError, er:
 625             if er.errno != errno.ENOENT:
 626                 raise er
 627         return self.attrs
 628
 629     def can_edit(self):
 630         attrs = self.get_attrs()
 631         try:
 632             # SomeUser:read,write All:read
 633             acl = attrs["acl"]
 634             for rule in acl.split():
 635                 (user,perms) = acl.split(':')
 636                 if user == remote_user() or user == "All":
 637                     if 'write' in perms.split(','):
 638                         return True
 639             return False
 640         except:
 641             pass
 642         return True
 643
 644     def send_page(self):
 645         page_name = None
 646         if self.can_edit():
 647             page_name = self.page_name
 648         send_title(page_name, self.split_title(), msg=self.msg, msg_type=self.msg_type)
 649         PageFormatter(self.get_raw_body()).print_html()
 650         send_footer(page_name, self._last_modified())
 651
 652     def _last_modified(self):
 653         if not self.exists():
 654             return None
 655         from time import localtime, strftime
 656         modtime = localtime(os.stat(self._text_filename())[stat.ST_MTIME])
 657         return strftime(datetime_fmt, modtime)
 658
 659     def send_editor(self, preview=None):
 660         send_title(None, 'Edit ' + self.split_title(), msg=self.msg, msg_type=self.msg_type)
 661
 662         print ('<p><b>Editing ' + self.page_name
 663             + ' for ' + cgi.escape(remote_user())
 664             + ' from ' + cgi.escape(get_hostname(remote_host()))
 665             + '</b></p>')
 666         print '<div class="editor"><form method="post" action="%s/%s">' % (script_name(), self.page_name)
 667         print '<input type="hidden" name="savepage" value="%s">' % (self.page_name)
 668         print """<textarea wrap="virtual" id="editor" name="savetext" rows="17" cols="80">%s</textarea>""" % (preview or self.get_raw_body())
 669         print """
 670             <br />
 671             <input type="submit" name="save" value="Save" />
 672             <input type="submit" name="preview" value="Preview" />
 673             <input type="reset" value="Reset" />
 674             <input type="submit" name="cancel" value="Cancel" />
 675             <br />
 676             </form></div>"""
 677         print "<p>" + Page('EditingTips').link_to() + "</p>"
 678         if preview:
 679             print "<div class='preview'>"
 680             PageFormatter(preview).print_html()
 681             print "</div>"
 682         send_footer(self.page_name)
 683
 684     def send_raw(self):
 685         emit_header("text/plain")
 686         print self.get_raw_body()
 687
 688     def _write_file(self, text):
 689         tmp_filename = self._tmp_filename()
 690         open(tmp_filename, 'wt').write(text.replace('\r\n', '\n'))
 691         text = self._text_filename()
 692         if os.name == 'nt':
 693             # Bad Bill!  POSIX rename ought to replace. :-(
 694             try:
 695                 os.remove(text)
 696             except OSError, er:
 697                 if er.errno <> errno.ENOENT: raise er
 698         os.rename(tmp_filename, text)
 699
 700     def save_text(self, newtext):
 701         self._write_file(newtext)
 702         rc = 0
 703         if post_edit_hook:
 704             # FIXME: what's the std way to perform shell quoting in python?
 705             cmd = ( post_edit_hook
 706                 + " '" + text_dir + '/' + self.page_name
 707                 + "' '" + remote_user()
 708                 + "' '" + remote_host() + "'"
 709             )
 710             out = os.popen(cmd)
 711             msg = out.read()
 712             rc = out.close()
 713         if rc:
 714             self.msg += "Post-editing hook returned %d.\n" % rc
 715             self.msg += 'Command was: ' + cmd + '\n'
 716             if msg:
 717                 self.msg += 'Output follows:\n' + msg
 718         else:
 719             self.msg = 'Thankyou for your contribution.  Your attention to detail is appreciated.'
 720             self.msg_type = 'success'
 721
 722 def send_verbatim(filename, mime_type='application/octet-stream'):
 723     pathname = path.join(text_dir, filename)
 724     data = open(pathname, 'rb').read()
 725     emit_header(mime_type)
 726     sys.stdout.write(data)
 727
 728 # Main ---------------------------------------------------------------
 729 try:
 730     # Configuration values
 731     site_name = 'Codewiz'
 732
 733     # set to None for read-only sites
 734     # leave empty ('') to allow anonymous edits
 735     # otherwise, set to a URL that requires authentication
 736     privileged_url = 'https://www.codewiz.org/~bernie/wiki'
 737
 738     data_dir = '/home/bernie/public_html/wiki'
 739     text_dir = path.join(data_dir, 'text')
 740     css_url = '../wikidata/geekigeeki.css'  # optional stylesheet link
 741     history_url = '../wikigit/wiki.git'
 742     post_edit_hook = './post_edit_hook.sh'
 743     datetime_fmt = '%a %d %b %Y %I:%M %p'
 744     allow_edit = True                       # Is it possible to edit pages?
 745     show_hosts = True                       # show hostnames?
 746     nonexist_qm = False                     # show '?' for nonexistent?
 747     debug_cgi = False                       # Set to True for CGI var dump
 748
 749     form = cgi.FieldStorage()
 750
 751     handlers = { 'fullsearch':  do_fullsearch,
 752                  'titlesearch': do_titlesearch,
 753                  'edit':        do_edit,
 754                  'raw':         do_raw,
 755                  'savepage':    do_savepage }
 756
 757     for cmd in handlers.keys():
 758         if form.has_key(cmd):
 759             apply(handlers[cmd], (form[cmd].value,))
 760             break
 761     else:
 762         path_info = environ.get('PATH_INFO', '')
 763         if len(path_info) and path_info[0] == '/':
 764             query = path_info[1:] or 'FrontPage'
 765         else:
 766             query = environ.get('QUERY_STRING', '') or 'FrontPage'
 767
 768         if file_re.match(query):
 769             if word_re.match(query):
 770                 Page(query).send_page()
 771             elif img_re.match(query):
 772                 #FIXME: use correct mime type
 773                 send_verbatim(query, 'image/jpeg')
 774             else:
 775                 send_verbatim(query)
 776         else:
 777             # TODO: return 404?
 778             send_title(None, msg='Can\'t work out query: ' + query)
 779 except:
 780     import traceback
 781     msg=traceback.format_exc()
 782     if title_done:
 783         send_guru(msg, "error")
 784     else:
 785         send_title(None, msg=msg)
 786     send_footer(None)
 787
 788 sys.stdout.flush()