138 lines
4.9 KiB
Python
Executable File
138 lines
4.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys,os,re
|
|
from argparse import ArgumentParser
|
|
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
|
import bc
|
|
|
|
__author__ = "Ville Rantanen <ville.q.rantanen@gmail.com>"
|
|
__version__ = "0.1"
|
|
|
|
''' Rules modified from mistune project '''
|
|
|
|
def setup_options():
|
|
''' Create command line options '''
|
|
usage='''
|
|
Color notation renderer in ANSI codes
|
|
Special syntaxes:
|
|
* Colors: insert string ${C}, where C is one of KRGBYMCWkrgbymcwSUZ
|
|
|
|
'''
|
|
|
|
parser=ArgumentParser(description=usage,
|
|
epilog=__author__)
|
|
|
|
parser.add_argument("-v","--version",action="version",version=__version__)
|
|
parser.add_argument("-D",action="store_true",dest="debug",default=False,
|
|
help="Debug mode")
|
|
parser.add_argument("--no-color","-n",action="store_false",dest="color",default=True,
|
|
help="Disable color.")
|
|
parser.add_argument("-z",action="store_true",dest="zero",default=False,
|
|
help="Reset coloring at the end of each line.")
|
|
parser.add_argument("filename",type=str,
|
|
help="File to show, - for stdin")
|
|
opts=parser.parse_args()
|
|
return opts
|
|
|
|
block_match={
|
|
'block_code': (re.compile(r'^( {4}[^*])(.*)$'),'${c}\\1\\2','${Z}${c}'),
|
|
'multiline_code' : (re.compile(r'^ *(`{3,}|~{3,}) *(\S*)?'),'${c}\\1\\2','${Z}${c}'), # ```lang
|
|
'block_quote': (re.compile(r'^(>[ >]* )'),'${K}\\1${Z}','${Z}'),
|
|
|
|
'hrule': (re.compile(r'^ {0,3}[-*_]([-*_]){2,}$'),False,'${Z}'),
|
|
'heading' : (re.compile(r'^ *(#{1,6}) *([^\n]+?) *#* *(?:\n+|$)'),'${W}\\1 ${U}\\2','${W}${U}'),
|
|
'lheading' : (re.compile(r'^(=+|-+)$'),'${W}\\1','${W}',(-1,re.compile(r'^([^\n]+)'),'${W}\\1')),
|
|
'list_bullet':(re.compile(r'^( *)([*+-]|[\d\.]+)( +)'),'\\1${y}\\2${Z}\\3','${Z}'),
|
|
'list_loose': (None,False,'${Z}'),
|
|
'text': (re.compile(r'^[^\n]+'),False,'${Z}'),
|
|
'empty': (re.compile(r'^$'),False,'${Z}'),
|
|
}
|
|
blocks=['block_quote', 'multiline_code','hrule', 'heading','lheading','list_bullet', 'block_code', 'text', 'empty']
|
|
|
|
inline_match={
|
|
'bold1': (re.compile(r'(^| )(_[^_]+_)'), '${W}\\1\\2'), # _bold_
|
|
'bold2': (re.compile(r'(^| )(\*{1,2}[^\*]+\*{1,2})'), '${W}\\1\\2'), # **bold**
|
|
'code': (re.compile(r'([`]+[^`]+[`]+)'),'${c}\\1'), # `code`
|
|
'link': (re.compile(r'(\[[^\]]+\]\([^\)]+\))'),'${B}${U}\\1'), # [text](link)
|
|
'image': (re.compile(r'(!\[[^\]]+\]\([^\)]+\))'),'${r}\\1'), # 
|
|
'underline': (re.compile(r'(^|\W)(__)([^_]+)(__)'), '\\1\\2${U}\\3${Z}\\4'), # __underline__
|
|
'strikethrough': (re.compile(r'(~~)([^~]+)(~~)'),'\\1${st}\\2${so}\\3'), # ~~strike~~
|
|
}
|
|
inlines=['bold1','bold2','code','image','link','underline','strikethrough']
|
|
|
|
|
|
bc=bc.ansi()
|
|
opts=setup_options()
|
|
if opts.filename=="-":
|
|
f=sys.stdin
|
|
else:
|
|
f=open(opts.filename,'r')
|
|
data=[]
|
|
# Read data
|
|
for row in f:
|
|
if not row:
|
|
continue
|
|
row=row.decode('utf-8').rstrip("\n\r ")
|
|
data.append([None,row])
|
|
|
|
block='text'
|
|
new_block='text'
|
|
multiline_block=False
|
|
# Parse styles
|
|
for i,line in enumerate(data):
|
|
row=line[1]
|
|
if line[0] is not None:
|
|
# Previous lines have set the style already
|
|
continue
|
|
for match in blocks:
|
|
if block_match[match][0].match(row):
|
|
new_block=match
|
|
if match.startswith('multiline'):
|
|
if multiline_block:
|
|
multiline_block=False
|
|
else:
|
|
multiline_block=match
|
|
break
|
|
if multiline_block:
|
|
new_block=multiline_block
|
|
# Lists must end with empty line
|
|
if new_block not in ('empty','list_bullet') and block.startswith('list_'):
|
|
new_block='list_loose'
|
|
|
|
if len(block_match[match])>3:
|
|
# Style sets block in previous or next lines
|
|
data[i+block_match[match][3][0]][0]=new_block
|
|
data[i+block_match[match][3][0]][1]=block_match[match][3][1].sub(
|
|
block_match[match][3][2], data[i+block_match[match][3][0]][1])
|
|
data[i][0]=new_block
|
|
if new_block!=block:
|
|
block=new_block
|
|
|
|
# Start inserting colors, and printing
|
|
for i,line in enumerate(data):
|
|
row=line[1]
|
|
block=line[0]
|
|
multiline_block=block.startswith('multiline')
|
|
if not multiline_block:
|
|
sys.stdout.write(bc.Z)
|
|
#print(multiline_block)
|
|
if block_match[block][1]:
|
|
row=block_match[block][0].sub(block_match[block][1],row)
|
|
if not (multiline_block or match=='block_code'):
|
|
for match in inlines:
|
|
if inline_match[match][0].search(row):
|
|
row=inline_match[match][0].sub(inline_match[match][1]+block_match[block][2],row)
|
|
if opts.color:
|
|
colored=bc.color_string(row)
|
|
else:
|
|
colored=bc.nocolor_string(row)
|
|
if opts.debug:
|
|
multistr="*" if multiline_block else " "
|
|
colored="{:<18}{:}:".format(data[i][0],multistr)+colored
|
|
sys.stdout.write(colored.encode('utf-8'))
|
|
if opts.zero:
|
|
sys.stdout.write(bc.Z)
|
|
sys.stdout.write("\n")
|
|
|
|
|