ikiwiki/plugins/rst

103 lines
3.1 KiB
Python
Executable File

#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# rstproc — xml-rpc-based ikiwiki plugin to process RST files
#
# TODO: the top of this file should be converted to a python library for
# ikiwiki plugins
#
# based a little bit on rst.pm by Sergio Talens-Oliag, but only a little bit. :)
#
# Copyright © martin f. krafft <madduck@madduck.net>
# Released under the terms of the GNU GPL version 2
__name__ = 'rstproc'
__description__ = 'xml-rpc-based ikiwiki plugin to process RST files'
__version__ = '0.2'
__author__ = 'martin f. krafft <madduck@madduck.net>'
__copyright__ = 'Copyright © ' + __author__
__licence__ = 'GPLv2'
from docutils.core import publish_string;
import posix
import select
import sys
import xmlrpclib
import xml.parsers.expat
from SimpleXMLRPCServer import SimpleXMLRPCDispatcher
def write(s):
# no comment
sys.stdout.write(s)
sys.stdout.flush()
def debug(s):
print >>sys.stderr, __name__ + ':DEBUG:' + s
sys.stderr.flush()
def rpc_read(processor):
acc = ''
ret = None
while True:
line = sys.stdin.readline()
if line is None: continue
if len(line) == 0: sys.exit(posix.EX_OK)
# debug('read line: ' + line)
acc += line
try:
ret = processor(acc)
# debug('processed: ' + acc)
# debug('got back: ' + ret.__class__.__name__)
return ret
except xml.parsers.expat.ExpatError:
# debug('request invalid or incomplete: ' + acc)
pass
return None
def rpc_call(cmd, **kwargs):
call = xmlrpclib.dumps(sum(kwargs.items(), ()), cmd)
write(call + '\n')
resp = rpc_read(lambda resp: resp)
class SimpleStdinOutXMLRPCHandler(SimpleXMLRPCDispatcher):
def __init__(self, allow_none=False, encoding=None):
# see http://bugs.debian.org/470645
if SimpleXMLRPCDispatcher.__init__.func_code.co_argcount == 1:
# python2.4 and before only took one argument
SimpleXMLRPCDispatcher.__init__(self)
else:
SimpleXMLRPCDispatcher.__init__(self, allow_none, encoding)
def process_request(self, req):
write(self._marshaled_dispatch(req))
def handle_request(self):
def processor(req):
self.process_request(req)
while True:
ret = rpc_read(processor)
if ret is not None: return ret
def rst2html(*kwargs):
# FIXME arguments should be treated as a hash, the order could change
# at any time and break this.
html = publish_string(kwargs[3], writer_name='html',
settings_overrides = { 'halt_level': 6
, 'file_insertion_enabled': 0
, 'raw_enabled': 1
})
content = html.split('<div class="document">', 1)[1]
content = content.split('</div>\n</body>')[:-1][0].strip()
# debug('content = ' + content)
return content
def importme():
rpc_call('hook', type='htmlize', id='rst', call='rst2html')
handler = SimpleStdinOutXMLRPCHandler()
handler.register_function(importme, name='import')
handler.register_function(rst2html)
while True:
handler.handle_request()