Source

highlight / server.py

Full commit
#!/usr/bin/env python

from cherrypy import expose, quickstart, tools
from pygments import highlight
from pygments.lexers import get_lexer_by_name, guess_lexer
from pygments.formatters import HtmlFormatter

class Highlighter:
    @expose
    def index(self, **kw):
        with open("index.html") as fo:
            return fo.read()

    @expose
    def pygmentize(self, **kw):
        code = kw.get("code", "")
        if not code:
            return ""

        lang = kw.get("lang", "").lower()
        if not lang:
            lexer = guess_lexer(code)
        else:
            lexer = get_lexer_by_name(lang)

        linenos = kw.get("linenos", 0)

        formatter = HtmlFormatter(linenos=linenos, cssclass="source")
        reply = highlight(code, lexer, formatter)
        callback = kw.get("callback", None)
        if callback:
            print "CALLBACK: %s" % callback
            reply = "%s(%s)" % (callback, reply)
        return reply

    @expose(alias="highlight.css")
    @tools.response_headers(headers=[("Content-Type", "text/css")])
    def css(self, **kw):
        return HtmlFormatter().get_style_defs()

    @expose(alias="highlight.js")
    @tools.response_headers(headers=[("Content-Type", "application/json")])
    def highlight(self, **kw):
        with open("jquery.js") as fo:
            jquery = fo.read()
        with open("highlight.js") as fo:
            script = fo.read()

        return "%s\n%s" % (jquery, script)

    @expose
    def demo(self, **kw):
        with open("demo.html") as fo:
            return fo.read()

if __name__ == "__main__":
    quickstart(Highlighter())