1#!/usr/bin/env python3
 2
 3# This script uses Pygments and Python3. You must have both installed
 4# for this to work.
 5#
 6# http://pygments.org/
 7# http://python.org/
 8#
 9# It may be used with the source-filter or repo.source-filter settings
10# in cgitrc.
11#
12# The following environment variables can be used to retrieve the
13# configuration of the repository for which this script is called:
14# CGIT_REPO_URL        ( = repo.url       setting )
15# CGIT_REPO_NAME       ( = repo.name      setting )
16# CGIT_REPO_PATH       ( = repo.path      setting )
17# CGIT_REPO_OWNER      ( = repo.owner     setting )
18# CGIT_REPO_DEFBRANCH  ( = repo.defbranch setting )
19# CGIT_REPO_SECTION    ( = section        setting )
20# CGIT_REPO_CLONE_URL  ( = repo.clone-url setting )
21
22
23import sys
24from pygments import highlight
25from pygments.util import ClassNotFound
26from pygments.lexers import TextLexer
27from pygments.lexers import guess_lexer
28from pygments.lexers import guess_lexer_for_filename
29from pygments.formatters import HtmlFormatter
30
31
32data = sys.stdin.read()
33filename = sys.argv[1]
34formatter = HtmlFormatter(style='pastie')
35
36try:
37	lexer = guess_lexer_for_filename(filename, data)
38except ClassNotFound:
39	# check if there is any shebang
40	if data[0:2] == '#!':
41		lexer = guess_lexer(data)
42	else:
43		lexer = TextLexer()
44except TypeError:
45	lexer = TextLexer()
46
47# highlight! :-)
48# printout pygments' css definitions as well
49sys.stdout.write('<style>')
50sys.stdout.write(formatter.get_style_defs('.highlight'))
51sys.stdout.write('</style>')
52sys.stdout.write(highlight(data, lexer, formatter, outfile=None))