Update style.

[stagit.git] / highlight

diff --git a/highlight b/highlight
index c67e70319b31a8f834597897aa9cea7fd1a56ec6..ba17566d7d88492e957142078fd95ccb381312b7 100755 (executable)
--- a/highlight
+++ b/highlight
@@ -1,29 +1,39 @@
-#!/usr/bin/env ruby
+#!/usr/bin/env python3
 
-require 'linguist'
-require 'pygments'
+import pygments
+from pygments import highlight
+from pygments.formatters import HtmlFormatter
+from pygments.lexers import guess_lexer, guess_lexer_for_filename
 
-stdin = ARGF.file
+from sys import stdin, stderr
 
-filename = stdin.readline.strip  # Read first line (filename).
-contents = stdin.read            # Read rest (code).
+filename = stdin.readline().strip()
+contents = stdin.read()
 
-detected = Linguist::FileBlob.new(filename).language
+lexer=None
 
-# Debugging
-#puts "File #{filename}"
-#puts "Code:
-#{contents}"
-#print "Language: "
-#pp detected
+try:
+    lexer = guess_lexer_for_filename(filename, contents)
+except pygments.util.ClassNotFound:
+    try:
+        lexer = guess_lexer(contents)
+    except pygments.util.ClassNotFound:
+        pass
 
-html = Pygments.highlight(contents,
-  :lexer => detected.name,
-  :formatter => 'html',
-  :options => {
-    :encoding => 'utf-8',
-    :linenos => 'table',
-    :lineanchors => 'loc',
-    :anchorlinenos => true})
+if lexer is None:
+    from pygments.lexers import TextLexer
+    lexer = TextLexer
 
-puts html
+FORMAT = HtmlFormatter(
+    style='murphy',
+    cssclass='highlight',
+    linenos='table',
+    lineanchors='loc',
+    anchorlinenos=True)
+
+print(highlight(contents, lexer, FORMAT))
+print('<style>')
+print(FORMAT.get_style_defs('.highlight'))
+print('</style>')
+
+print("Filename: {}; Lexer: {}.".format(filename, lexer), file=stderr)