Editorial Workflows

import editor
import console
import os
import re
import sys
import codecs
import workflow
from StringIO import StringIO

theme = editor.get_theme()
workflow.set_variable('CSS', workflow.get_variable('CSS Dark' if theme == 'Dark' else 'CSS Light'))

p = editor.get_path()
term = workflow.get_variable('Search Term')
pattern = re.compile(re.escape(term), flags=re.IGNORECASE)
from urllib import quote
dir = os.path.split(p)[0]
valid_extensions = set(['.txt', '.md', '.markd', '.text', '.mdown', '.taskpaper'])
html = StringIO()
match_count = 0
for w in os.walk(dir):
	dir_path = w[0]
	filenames = w[2]
	for name in filenames:
		full_path = os.path.join(dir_path, name)
		ext = os.path.splitext(full_path)[1]
		if ext.lower() not in valid_extensions:
			continue
		found_snippets = []
		i = 0
		try:
			with codecs.open(full_path, 'r', 'utf-8') as f:
				for line in f:
					for match in re.finditer(pattern, line):
						start = max(0, match.start(0) - 100)
						end = min(len(line)-1, match.end(0) + 100)
						snippet = (line[start:match.start(0)],
						           match.group(0),
						           line[match.end(0):end],
						           match.start(0) + i,
						           match.end(0) + i)
						found_snippets.append(snippet)
					i += len(line)
		except UnicodeDecodeError, e:
			pass
		if len(found_snippets) > 0:
			match_count += 1
			root, rel_path = editor.to_relative_path(full_path)
			ed_url = 'editorial://open/' + quote(rel_path.encode('utf-8')) + '?root=' + root
			html.write('<h2><a href="' + ed_url + '">' + name + '</a></h2>')
			for snippet in found_snippets:
				start = snippet[3]
				end = snippet[4]
				select_url = 'editorial://open/' + quote(rel_path.encode('utf-8')) + '?root=' + root
				select_url += '&selection=' + str(start) + '-' + str(end)
				html.write('<a class="result-box" href="' + select_url + '">' + snippet[0] + '<span class="highlight">' + snippet[1] + '</span>' + snippet[2] + '</a>')
if match_count == 0:
	html.write('<p>No matches found.</p>')

workflow.set_output(html.getvalue())