init

Author: Ellipse0934

res/pygments/ptx.py

@@ -8,33 +8,68 @@ const Cthulhu = Base.PkgId(UUID("f68482b8-f384-11e8-15f7-abe071a5a75f"), "Cthulh
 # syntax highlighting
 #
 
-const _pygmentize = Ref{Union{String,Nothing}}()
-function pygmentize()
-    if !isassigned(_pygmentize)
-        _pygmentize[] = Sys.which("pygmentize")
-    end
-    return _pygmentize[]
-end
+# https://github.com/JuliaLang/julia/blob/dacd16f068fb27719b31effbe8929952ee2d5b32/stdlib/InteractiveUtils/src/codeview.jl
+const hlscheme = Dict{Symbol, Tuple{Bool, Union{Symbol, Int}}}(
+    :default     => (false, :normal), # e.g. comma, equal sign, unknown token
+    :comment     => (false, :light_black),
+    :label       => (false, :light_red),
+    :instruction => ( true, :light_cyan),
+    :type        => (false, :cyan),
+    :number      => (false, :yellow),
+    :bracket     => (false, :yellow),
+    :variable    => (false, :normal), # e.g. variable, register
+    :keyword     => (false, :light_magenta),
+    :funcname    => (false, :light_yellow),
+)
 
 function highlight(io::IO, code, lexer)
-    highlighter = pygmentize()
-    have_color = get(io, :color, false)
-    if highlighter === nothing || !have_color
+    if !haskey(io, :color)
         print(io, code)
+    elseif lexer == "ptx"
+        highlight_ptx(io, code)
     else
-        custom_lexer = joinpath(dirname(@__DIR__), "res", "pygments", "$lexer.py")
-        if isfile(custom_lexer)
-            lexer = `$custom_lexer -x`
-        end
-
-        pipe = open(`$highlighter -f terminal -P bg=dark -l $lexer`, "r+")
-        print(pipe, code)
-        close(pipe.in)
-        print(io, read(pipe, String))
+        print(io, code)
     end
-    return
 end
 
+function highlight_ptx(io::IO, code)
+    function get_token(s)
+        m = match(r"(\s*)([^\s]+)(.*)", s)
+        m isa RegexMatch && (return m.captures[1:3])
+        return nothing, nothing, nothing
+    end
+    print_tok(token, type) = Base.printstyled(io,
+                                              token,
+                                              bold = hlscheme[type][1],
+                                              color = hlscheme[type][2])
+    buf = IOBuffer(code)
+    while !eof(buf)
+        line = readline(buf)
+        indent, tok, line = get_token(line)
+        while (tok !== nothing)
+            print(io, indent)
+            if match(r"^\/\/", tok) isa RegexMatch         # Comment
+                print_tok(tok, :comment)
+                print_tok(line, :comment)
+                break
+            elseif match(r"^[\w]+:", tok) isa RegexMatch   # Label
+                print_tok(tok, :label)
+            elseif match(r"^[\w]+\.[\w]+(\.\w+)?", tok) isa RegexMatch # Instruction
+                print_tok(tok, :instruction)
+            elseif match(r"^\.[\w]+", tok) isa RegexMatch  # Directive
+                print_tok(tok, :type)
+            elseif match(r"^@!?%p.+", tok) isa RegexMatch  # Guard Predicate
+                print_tok(tok, :keyword)
+            elseif match(r"^%[\w]+", tok) isa RegexMatch   # Register
+                print_tok(tok, :number)
+            else                                           # Default
+                print_tok(tok, :default)
+            end
+            indent, tok, line = get_token(line)
+        end
+        print(io, '\n')
+    end
+end
 
 #
 # code_* replacements