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