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