r2wiki.py

import os
import re
import sys
import pydoc
try:
	from pygments import highlight
	from pygments.lexers import MarkdownLexer
	from pygments.formatters import TerminalFormatter
except ImportError:
	print 'Pygments dependency not met. pip install Pygments'
"""
Directions on how to setup is in the wiki itself
"""

try:
	src_dir = os.path.dirname(os.path.realpath(__file__))
	pattern = re.compile (sys.argv[1])

	found = ''

	for path, subdirs, files in os.walk(src_dir):
		for name in files:
			md_path = os.path.join(path, name)
			if not '.git' in md_path and md_path.endswith('.md'):
				with open(md_path, 'r') as f:
					for lines in f.readlines():
						if re.search(pattern, lines):
							if not any(filter in lines for filter in ['<p hidden>',
																	  '<!--',
																	  '<img src']):
								match = ((re.sub('\*\*', '', lines)).lstrip() + '\n')#.replace('`', "'")
								if match.startswith('- ['):
									split = match.split(']')
									found += max(split, key=len).replace('[', '') + \
											 ' [.](https://radare2.securisec.com%s' % split[-1].strip('(')
								else:
									if match.startswith('>'):
										match = ''.join(list(match)[1:])
									if match.startswith('#'):
										match = ''.join(list(match)[1:])
									if match.startswith(' _'):
										match = re.sub('_', '', ''.join(list(match)[1:]))
									found += match

	pydoc.pipepager(highlight(found, MarkdownLexer(), TerminalFormatter()), cmd='less -r')
except IndexError:
	print 'Usage: %s search_param' % sys.argv[0]