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