895 lines
30 KiB
Python
895 lines
30 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright (C) 2015-2018 Andrew Hamilton. All rights reserved.
|
|
# Licensed under the Artistic License 2.0.
|
|
|
|
import ast
|
|
import asyncio
|
|
import contextlib
|
|
import enum
|
|
import functools
|
|
import gzip
|
|
import hashlib
|
|
import math
|
|
import os
|
|
import os.path
|
|
import pickle
|
|
import pwd
|
|
import stat
|
|
import subprocess
|
|
import tempfile
|
|
import time
|
|
import traceback
|
|
|
|
import PIL.Image
|
|
import pygments
|
|
import pygments.lexers
|
|
import pygments.styles
|
|
import toml
|
|
|
|
import vigil.fill3 as fill3
|
|
import vigil.gut as gut
|
|
import vigil.lscolors as lscolors
|
|
import vigil.termstr as termstr
|
|
|
|
|
|
CACHE_PATH = ".vigil"
|
|
|
|
|
|
if "PYGMENT_STYLE" not in os.environ:
|
|
os.environ["PYGMENT_STYLE"] = "native"
|
|
|
|
|
|
class Status(enum.IntEnum):
|
|
|
|
ok = 1
|
|
problem = 2
|
|
normal = 3
|
|
error = 4
|
|
not_applicable = 5
|
|
running = 6
|
|
pending = 7
|
|
paused = 8
|
|
timed_out = 9
|
|
|
|
|
|
_STATUS_COLORS = {Status.ok: termstr.Color.green,
|
|
Status.problem: termstr.Color.red,
|
|
Status.normal: termstr.Color.white,
|
|
Status.not_applicable: termstr.Color.grey_100,
|
|
Status.running: termstr.Color.light_blue,
|
|
Status.paused: termstr.Color.yellow,
|
|
Status.timed_out: termstr.Color.purple}
|
|
STATUS_MEANINGS = [
|
|
(Status.normal, "Normal"), (Status.ok, "Ok"),
|
|
(Status.problem, "Problem"), (Status.not_applicable, "Not applicable"),
|
|
(Status.running, "Running"), (Status.paused, "Paused"),
|
|
(Status.timed_out, "Timed out"), (Status.pending, "Pending"),
|
|
(Status.error, "Error")
|
|
]
|
|
_STATUS_TO_TERMSTR = {
|
|
status: termstr.TermStr(" ", termstr.CharStyle(bg_color=color))
|
|
for status, color in _STATUS_COLORS.items()}
|
|
_STATUS_TO_TERMSTR[Status.error] = termstr.TermStr(
|
|
"E", termstr.CharStyle(bg_color=termstr.Color.red))
|
|
_STATUS_TO_TERMSTR[Status.pending] = "."
|
|
STATUS_CURSOR_COLORS = {Status.ok: termstr.Color.black,
|
|
Status.problem: termstr.Color.white,
|
|
Status.normal: termstr.Color.black,
|
|
Status.not_applicable: termstr.Color.white,
|
|
Status.running: termstr.Color.white,
|
|
Status.paused: termstr.Color.black,
|
|
Status.timed_out: termstr.Color.white}
|
|
|
|
|
|
def get_ls_color_codes():
|
|
if "LS_COLORS" not in os.environ:
|
|
script = os.path.join(os.path.dirname(__file__), "LS_COLORS.sh")
|
|
with open(script) as file_:
|
|
codes = file_.readline().strip()[len("LS_COLORS='"):-len("';")]
|
|
os.environ["LS_COLORS"] = codes
|
|
return lscolors.get_color_codes(os.environ)
|
|
|
|
|
|
_LS_COLOR_CODES = get_ls_color_codes()
|
|
TIMEOUT = 60
|
|
|
|
|
|
def _printable(text):
|
|
return "".join(char if ord(char) > 31 or char in ["\n", "\t"] else "#"
|
|
for char in text)
|
|
|
|
|
|
def _fix_input(input_):
|
|
input_str = (input_.decode("utf-8", errors="replace")
|
|
if isinstance(input_, bytes) else input_)
|
|
return _printable(input_str).expandtabs(tabsize=4)
|
|
|
|
|
|
def _do_command(command, **kwargs):
|
|
completed_process = subprocess.run(command, stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE, **kwargs)
|
|
return (_fix_input(completed_process.stdout),
|
|
_fix_input(completed_process.stderr), completed_process.returncode)
|
|
|
|
|
|
def _run_command(command, success_status=None, error_status=None,
|
|
timeout=None):
|
|
success_status = Status.ok if success_status is None else success_status
|
|
error_status = Status.problem if error_status is None else error_status
|
|
stdout, stderr, returncode = _do_command(command, timeout=timeout)
|
|
result_status = success_status if returncode == 0 else error_status
|
|
return result_status, fill3.Text(stdout + stderr)
|
|
|
|
|
|
def deps(**kwargs):
|
|
def decorating_func(func):
|
|
for key, value in kwargs.items():
|
|
setattr(func, key, value)
|
|
return func
|
|
return decorating_func
|
|
|
|
|
|
def _syntax_highlight(text, lexer, style):
|
|
def _parse_rgb(hex_rgb):
|
|
if hex_rgb.startswith("#"):
|
|
hex_rgb = hex_rgb[1:]
|
|
return tuple(eval("0x"+hex_rgb[index:index+2]) for index in [0, 2, 4])
|
|
|
|
def _char_style_for_token_type(token_type, default_bg_color,
|
|
default_style):
|
|
try:
|
|
token_style = style.style_for_token(token_type)
|
|
except KeyError:
|
|
return default_style
|
|
fg_color = (termstr.Color.black if token_style["color"] is None
|
|
else _parse_rgb(token_style["color"]))
|
|
bg_color = (default_bg_color if token_style["bgcolor"] is None
|
|
else _parse_rgb(token_style["bgcolor"]))
|
|
return termstr.CharStyle(fg_color, bg_color, token_style["bold"],
|
|
token_style["italic"],
|
|
token_style["underline"])
|
|
default_bg_color = _parse_rgb(style.background_color)
|
|
default_style = termstr.CharStyle(bg_color=default_bg_color)
|
|
text = fill3.join(
|
|
"", [termstr.TermStr(text, _char_style_for_token_type(
|
|
token_type, default_bg_color, default_style))
|
|
for token_type, text in pygments.lex(text, lexer)])
|
|
return fill3.Text(text, pad_char=termstr.TermStr(" ").bg_color(
|
|
default_bg_color))
|
|
|
|
|
|
def _syntax_highlight_using_path(text, path):
|
|
lexer = pygments.lexers.get_lexer_for_filename(path, text)
|
|
style = pygments.styles.get_style_by_name(os.environ["PYGMENT_STYLE"])
|
|
return _syntax_highlight(text, lexer, style)
|
|
|
|
|
|
def pygments_(path):
|
|
with open(path) as file_:
|
|
try:
|
|
text = file_.read()
|
|
except UnicodeDecodeError:
|
|
return Status.not_applicable, fill3.Text("Not unicode")
|
|
else:
|
|
try:
|
|
source_widget = _syntax_highlight_using_path(_fix_input(text),
|
|
path)
|
|
except pygments.util.ClassNotFound:
|
|
return Status.normal, fill3.Text(text)
|
|
return Status.normal, source_widget
|
|
|
|
|
|
def linguist(path):
|
|
# Dep: ruby?, ruby-dev, libicu-dev, cmake, "gem install github-linguist"
|
|
return _run_command(["linguist", path], Status.normal)
|
|
|
|
|
|
def _permissions_in_octal(permissions):
|
|
result = []
|
|
for part_index in range(3):
|
|
index = part_index * 3 + 1
|
|
part = permissions[index:index+3]
|
|
digit = sum(2 ** (2 - index) for index, element in enumerate(part)
|
|
if element != "-")
|
|
result.append(str(digit))
|
|
return "".join(result)
|
|
|
|
|
|
def _pretty_bytes(bytes):
|
|
if bytes == 0:
|
|
return "0 B"
|
|
units = ("B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB")
|
|
unit_index = int(math.floor(math.log(bytes, 1024)))
|
|
power = math.pow(1024, unit_index)
|
|
conversion = round(bytes/power, 2)
|
|
return f"{conversion} {units[unit_index]}"
|
|
|
|
|
|
def _md5(path):
|
|
with open(path, "rb") as file:
|
|
return hashlib.md5(file.read()).hexdigest()
|
|
|
|
|
|
@deps(deps={"file", "coreutils"}, executables={"file", "sha1sum"})
|
|
def metadata(path):
|
|
|
|
def detail(value, unit):
|
|
result = f" ({value})" if unit is None else f" ({value} {unit})"
|
|
return termstr.TermStr(result).fg_color(termstr.Color.grey_100)
|
|
is_symlink = "yes" if os.path.islink(path) else "no"
|
|
stat_result = os.stat(path)
|
|
permissions = stat.filemode(stat_result.st_mode)
|
|
hardlinks = str(stat_result.st_nlink)
|
|
group = [pwd.getpwuid(stat_result.st_gid).pw_name,
|
|
detail(stat_result.st_gid, "gid")]
|
|
owner = [pwd.getpwuid(stat_result.st_uid).pw_name,
|
|
detail(stat_result.st_uid, "uid")]
|
|
modified, created, access = [
|
|
[time.asctime(time.gmtime(seconds)), detail(int(seconds), "secs")]
|
|
for seconds in (stat_result.st_mtime, stat_result.st_ctime,
|
|
stat_result.st_atime)]
|
|
size = [_pretty_bytes(stat_result.st_size),
|
|
detail(stat_result.st_size, "bytes")]
|
|
stdout, *rest = _do_command(
|
|
["file", "--dereference", "--brief", "--uncompress", "--mime", path])
|
|
mime_type = stdout
|
|
stdout, *rest = _do_command(
|
|
["file", "--dereference", "--brief", "--uncompress", path])
|
|
file_type = stdout
|
|
md5sum = _md5(path)
|
|
stdout, *rest = _do_command(["sha1sum", path])
|
|
sha1sum = stdout.split()[0]
|
|
permissions_value = [permissions,
|
|
detail(_permissions_in_octal(permissions), None)]
|
|
text = []
|
|
for line in [
|
|
("size", size), ("permissions", permissions_value), None,
|
|
("modified time", modified), ("creation time", created),
|
|
("access time", access), None,
|
|
("owner", owner), ("group", group), None,
|
|
("hardlinks", hardlinks), ("symlink", is_symlink), None,
|
|
("md5", md5sum), ("sha1", sha1sum), None,
|
|
("mime type", mime_type.strip()),
|
|
("file type", file_type.strip())]:
|
|
if line is None:
|
|
text.append("\n")
|
|
else:
|
|
name, value = line
|
|
name = termstr.TermStr(name + ":").fg_color(
|
|
termstr.Color.light_blue).ljust(16)
|
|
text.append(name + fill3.join("", value) + "\n")
|
|
return (Status.normal, fill3.Text(fill3.join("", text)))
|
|
|
|
|
|
@deps(deps={"python3-pygments"}, url="python3-pygments")
|
|
def contents(path):
|
|
root, ext = splitext(path)
|
|
if ext == "":
|
|
with open(path) as file_:
|
|
return Status.normal, fill3.Text(_fix_input(file_.read()))
|
|
else:
|
|
return pygments_(path)
|
|
|
|
|
|
def _is_python_syntax_correct(path, python_version):
|
|
if python_version == "python":
|
|
stdin, stdout, returncode = _do_command(
|
|
["python", "-c",
|
|
f"__import__('compiler').parse(open('{path}').read())"])
|
|
return returncode == 0
|
|
else: # python3
|
|
with open(path) as f:
|
|
source = f.read()
|
|
try:
|
|
ast.parse(source)
|
|
except:
|
|
return False
|
|
return True
|
|
|
|
|
|
def _python_version(path): # Need a better hueristic
|
|
for version in ["python3", "python"]:
|
|
if _is_python_syntax_correct(path, version):
|
|
return version
|
|
return "python3"
|
|
|
|
|
|
@deps(deps={"python"},
|
|
url="https://en.wikipedia.org/wiki/Python_syntax_and_semantics")
|
|
def python_syntax(path):
|
|
status = (Status.ok if _is_python_syntax_correct(path, "python") or
|
|
_is_python_syntax_correct(path, "python3") else Status.problem)
|
|
return status, fill3.Text("")
|
|
|
|
|
|
def _has_shebang_line(path):
|
|
with open(path, "rb") as file_:
|
|
return file_.read(2) == "#!"
|
|
|
|
|
|
def _is_python_test_file(path):
|
|
path = str(os.path.basename(path))
|
|
return path.endswith("_test.py") or path.startswith("test_")
|
|
|
|
|
|
@deps(deps={"python", "python3"},
|
|
url="https://docs.python.org/3/library/unittest.html")
|
|
def python_unittests(path):
|
|
if _is_python_test_file(path):
|
|
command = ([path] if _has_shebang_line(path)
|
|
else [_python_version(path), path])
|
|
stdout, stderr, returncode = _do_command(command, timeout=TIMEOUT)
|
|
status = Status.ok if returncode == 0 else Status.problem
|
|
return status, fill3.Text(stdout + "\n" + stderr)
|
|
else:
|
|
return Status.not_applicable, fill3.Text("No tests.")
|
|
|
|
|
|
@deps(deps={"python", "python3"},
|
|
url="https://docs.python.org/3/library/pydoc.html")
|
|
def pydoc(path):
|
|
stdout, stderr, returncode = _do_command(
|
|
[_python_version(path), "-m", "pydoc", path], timeout=TIMEOUT)
|
|
status = Status.normal if returncode == 0 else Status.not_applicable
|
|
if not stdout.startswith("Help on module"):
|
|
status = Status.not_applicable
|
|
stdout = stdout.replace(os.getcwd() + "/", "")
|
|
return status, fill3.Text(_fix_input(stdout))
|
|
|
|
|
|
@deps(deps={"mypy"}, url="mypy", executables={"mypy"})
|
|
def mypy(path):
|
|
stdout, stderr, returncode = _do_command(["mypy", path], timeout=TIMEOUT)
|
|
status = Status.ok if returncode == 0 else Status.normal
|
|
return status, fill3.Text(stdout)
|
|
|
|
|
|
def _colorize_coverage_report(text):
|
|
line_color = {"> ": termstr.Color.green, "! ": termstr.Color.red,
|
|
" ": None}
|
|
return fill3.join("", [termstr.TermStr(line).fg_color(line_color[line[:2]])
|
|
for line in text.splitlines(keepends=True)])
|
|
|
|
|
|
@deps(deps={"python-coverage", "python3-coverage"}, url="python3-coverage")
|
|
def python_coverage(path):
|
|
# FIX: Also use test_*.py files.
|
|
test_path = path[:-(len(".py"))] + "_test.py"
|
|
if os.path.exists(test_path):
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
coverage_cmd = [_python_version(path), "-m", "coverage"]
|
|
coverage_path = os.path.join(temp_dir, "coverage")
|
|
env = os.environ.copy()
|
|
env["COVERAGE_FILE"] = coverage_path
|
|
stdout, *rest = _do_command(
|
|
coverage_cmd + ["run", test_path], env=env, timeout=TIMEOUT)
|
|
path = os.path.normpath(path)
|
|
stdout, *rest = _do_command(
|
|
coverage_cmd + ["annotate", "--directory", temp_dir, path],
|
|
env=env)
|
|
flat_path = path.replace("/", "_")
|
|
with open(os.path.join(temp_dir, flat_path + ",cover"), "r") as f:
|
|
stdout = f.read()
|
|
return Status.normal, fill3.Text(_colorize_coverage_report(stdout))
|
|
else:
|
|
return Status.not_applicable, fill3.Text(
|
|
"No corresponding test file: " + os.path.normpath(test_path))
|
|
|
|
|
|
@deps(deps={"python-pycodestyle", "python3-pycodestyle"},
|
|
url="python-pycodestyle")
|
|
def pycodestyle(path):
|
|
return _run_command([_python_version(path), "-m", "pycodestyle", path])
|
|
|
|
|
|
@deps(deps={"python-pydocstyle", "python3-pydocstyle"},
|
|
url="python3-pydocstyle")
|
|
def pydocstyle(path):
|
|
return _run_command([_python_version(path), "-m", "pydocstyle", path])
|
|
|
|
|
|
@deps(deps={"python-pyflakes", "python3-pyflakes"}, url="pyflakes")
|
|
def pyflakes(path):
|
|
return _run_command([_python_version(path), "-m", "pyflakes", path])
|
|
|
|
|
|
@deps(deps={"pylint", "pylint3"}, url="pylint3")
|
|
def pylint(path):
|
|
return _run_command([_python_version(path), "-m", "pylint",
|
|
"--errors-only", path])
|
|
|
|
|
|
@deps(url="https://github.com/ahamilton/vigil/blob/master/gut.py")
|
|
def python_gut(path):
|
|
with open(path) as module_file:
|
|
output = gut.gut_module(module_file.read())
|
|
source_widget = _syntax_highlight_using_path(_fix_input(output), path)
|
|
return Status.normal, source_widget
|
|
|
|
|
|
@deps(deps={"python", "python3"},
|
|
url="https://docs.python.org/3/library/modulefinder.html")
|
|
def python_modulefinder(path):
|
|
return _run_command([_python_version(path), "-m", "modulefinder", path],
|
|
Status.normal)
|
|
|
|
|
|
@deps(deps={"python", "python3"},
|
|
url="https://docs.python.org/3/library/dis.html")
|
|
def dis(path):
|
|
return _run_command([_python_version(path), "-m", "dis", path],
|
|
Status.normal)
|
|
|
|
|
|
def _get_mccabe_line_score(line, python_version):
|
|
position, function_name, score = line.split()
|
|
return int(score if python_version == "python3" else score[:-1])
|
|
|
|
|
|
def _colorize_mccabe(text, python_version):
|
|
return fill3.join("", [
|
|
termstr.TermStr(line).fg_color(termstr.Color.yellow)
|
|
if _get_mccabe_line_score(line, python_version) > 10 else line
|
|
for line in text.splitlines(keepends=True)])
|
|
|
|
|
|
@deps(deps={"python-mccabe", "python3-mccabe"}, url="python3-mccabe")
|
|
def python_mccabe(path):
|
|
python_version = _python_version(path)
|
|
stdout, *rest = _do_command([python_version, "-m", "mccabe", path])
|
|
max_score = 0
|
|
with contextlib.suppress(ValueError): # When there are no lines
|
|
max_score = max(_get_mccabe_line_score(line, python_version)
|
|
for line in stdout.splitlines())
|
|
status = Status.problem if max_score > 10 else Status.ok
|
|
return status, fill3.Text(_colorize_mccabe(stdout, python_version))
|
|
|
|
|
|
def python_tidy(path): # Deps: found on internet?
|
|
stdout, *rest = _do_command(["python", "python-tidy.py", path])
|
|
return Status.normal, _syntax_highlight_using_path(stdout, path)
|
|
|
|
|
|
# FIX: Reenable when pydisasm is not causing problems
|
|
# @deps(deps={"pip3/xdis"}, executables={"pydisasm"},
|
|
# url="https://pypi.python.org/pypi/xdis")
|
|
# def pydisasm(path):
|
|
# return _run_command(["pydisasm", path], Status.normal,
|
|
# Status.not_applicable)
|
|
|
|
|
|
@deps(deps={"python-bandit", "python3-bandit"}, url="python3-bandit")
|
|
def bandit(path):
|
|
python_version = _python_version(path)
|
|
stdout, stderr, returncode = _do_command(
|
|
[python_version, "-m", "bandit.cli.main", "-f", "txt", path],
|
|
timeout=TIMEOUT)
|
|
status = Status.ok if returncode == 0 else Status.normal
|
|
text_without_timestamp = "".join(stdout.splitlines(keepends=True)[2:])
|
|
return status, fill3.Text(text_without_timestamp)
|
|
|
|
|
|
def _perl_version(path):
|
|
# stdout, stderr, returncode = _do_command(["perl", "-c", path])
|
|
# return "perl6" if "Perl v6.0.0 required" in stderr else "perl"
|
|
return "perl"
|
|
|
|
|
|
@deps(deps={"perl"}, url="https://en.wikipedia.org/wiki/Perl")
|
|
def perl_syntax(path):
|
|
return _run_command([_perl_version(path), "-c", path])
|
|
|
|
|
|
@deps(deps={"perl-doc"}, url="http://perldoc.perl.org/",
|
|
executables={"perldoc"})
|
|
def perldoc(path):
|
|
stdout, stderr, returncode = _do_command(["perldoc", "-t", path])
|
|
return ((Status.normal, fill3.Text(stdout)) if returncode == 0
|
|
else (Status.not_applicable, fill3.Text(stderr)))
|
|
|
|
|
|
@deps(deps={"perltidy"}, url="http://perltidy.sourceforge.net/",
|
|
executables={"perltidy"})
|
|
def perltidy(path):
|
|
stdout, *rest = _do_command(["perltidy", "-st", path])
|
|
return Status.normal, _syntax_highlight_using_path(stdout, path)
|
|
|
|
|
|
# def perl6_syntax(path):
|
|
# return _run_command(["perl6", "-c", path])
|
|
# perl6_syntax.deps={"rakudo"}
|
|
|
|
|
|
@deps(deps={"splint"}, url="splint", executables={"splint"})
|
|
def splint(path):
|
|
stdout, stderr, returncode = _do_command(["splint", "-preproc", path])
|
|
status = Status.ok if returncode == 0 else Status.problem
|
|
return status, fill3.Text(stdout + stderr)
|
|
|
|
|
|
@deps(deps={"tidy"}, url="tidy", executables={"tidy"})
|
|
def html_syntax(path):
|
|
# Maybe only show errors
|
|
stdout, stderr, returncode = _do_command(["tidy", path])
|
|
status = Status.ok if returncode == 0 else Status.problem
|
|
return status, fill3.Text(stderr)
|
|
|
|
|
|
@deps(deps={"tidy"}, url="tidy", executables={"tidy"})
|
|
def tidy(path):
|
|
stdout, *rest = _do_command(["tidy", path])
|
|
return Status.normal, fill3.Text(stdout)
|
|
|
|
|
|
@deps(deps={"bcpp"}, executables={"bcpp"})
|
|
def bcpp(path):
|
|
stdout, stderr, returncode = _do_command(["bcpp", "-fi", path])
|
|
status = Status.normal if returncode == 0 else Status.problem
|
|
return status, _syntax_highlight_using_path(stdout, path)
|
|
|
|
|
|
@deps(deps={"uncrustify"}, url="uncrustify", executables={"uncrustify"})
|
|
def uncrustify(path):
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
config_path = os.path.join(temp_dir, "uncrustify.cfg")
|
|
stdout, stderr, returncode = _do_command(
|
|
["uncrustify", "--detect", "-f", path, "-o", config_path])
|
|
if returncode == 0:
|
|
stdout, stderr, returncode = _do_command(
|
|
["uncrustify", "-c", config_path, "-f", path])
|
|
status = Status.normal if returncode == 0 else Status.problem
|
|
return status, _syntax_highlight_using_path(stdout, path)
|
|
|
|
|
|
def _pil_pixels(pil_image):
|
|
data = list(pil_image.getdata())
|
|
width = pil_image.width
|
|
return [data[row_index*width:(row_index+1)*width]
|
|
for row_index in range(pil_image.height)]
|
|
|
|
|
|
MAX_IMAGE_SIZE = 80
|
|
|
|
|
|
def _resize_image(image, new_width):
|
|
scale = new_width / image.width
|
|
return image.resize((int(image.width * scale), int(image.height * scale)),
|
|
PIL.Image.ANTIALIAS)
|
|
|
|
|
|
@deps(deps={"python3-pil"}, url="python3-pil")
|
|
def pil(path):
|
|
with open(path, "rb") as image_file:
|
|
with PIL.Image.open(image_file).convert("RGB") as image:
|
|
if image.width > (MAX_IMAGE_SIZE // 2):
|
|
image = _resize_image(image, MAX_IMAGE_SIZE // 2)
|
|
text = " " * 2 * image.width
|
|
result = []
|
|
for row in _pil_pixels(image):
|
|
row_style = []
|
|
for pixel in row:
|
|
style = termstr.CharStyle(bg_color=pixel)
|
|
row_style.extend([style, style])
|
|
result.append(termstr.TermStr(text, tuple(row_style)))
|
|
return Status.normal, fill3.Fixed(result)
|
|
|
|
|
|
@deps(deps={"python3-pil"}, url="python3-pil")
|
|
def pil_half(path):
|
|
with open(path, "rb") as image_file:
|
|
with PIL.Image.open(image_file).convert("RGB") as image:
|
|
if image.width > MAX_IMAGE_SIZE:
|
|
image = _resize_image(image, MAX_IMAGE_SIZE)
|
|
text = "▀" * image.width
|
|
rows = _pil_pixels(image)
|
|
if image.height % 2 == 1:
|
|
rows.append([None] * image.width)
|
|
result = fill3.Fixed([
|
|
termstr.TermStr(text, tuple(termstr.CharStyle(
|
|
fg_color=top_pixel, bg_color=bottom_pixel)
|
|
for top_pixel, bottom_pixel in zip(rows[index],
|
|
rows[index+1])))
|
|
for index in range(0, image.height, 2)])
|
|
return Status.normal, result
|
|
|
|
|
|
@deps(deps={"git"}, url="https://git-scm.com/docs/git-blame",
|
|
executables={"git"})
|
|
def git_blame(path): # FIX: Add to tools_test.py
|
|
stdout, stderr, returncode = _do_command(
|
|
["git", "blame", "--show-stats", "--date=short", path])
|
|
if returncode == 0:
|
|
return Status.normal, fill3.Text(stdout)
|
|
else:
|
|
return Status.not_applicable, fill3.Text("")
|
|
|
|
|
|
@deps(deps={"git"}, url="https://git-scm.com/docs/git-log",
|
|
executables={"git"})
|
|
def git_log(path):
|
|
stdout, stderr, returncode = _do_command(["git", "log", "--stat", path])
|
|
if returncode == 0:
|
|
return Status.normal, fill3.Text(stdout)
|
|
else:
|
|
return Status.not_applicable, fill3.Text("")
|
|
|
|
|
|
@deps(deps={"golang-golang-x-tools"}, url="golang-golang-x-tools",
|
|
executables={"godoc"})
|
|
def godoc(path):
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
symlink_path = os.path.join(temp_dir, "file.go")
|
|
os.symlink(os.path.abspath(path), symlink_path)
|
|
stdout, stderr, returncode = _do_command(["godoc", "."], cwd=temp_dir)
|
|
os.remove(symlink_path)
|
|
return Status.normal, fill3.Text(stdout)
|
|
|
|
|
|
def make_tool_function(dependencies, url, command, success_status=None,
|
|
error_status=None):
|
|
command = command.split()
|
|
executables = set([command[0]])
|
|
success_status = None if success_status is None else Status[success_status]
|
|
error_status = None if error_status is None else Status[error_status]
|
|
@deps(deps=set(dependencies), url=url, executables=executables)
|
|
def func(path):
|
|
return _run_command(command + [path], success_status, error_status)
|
|
return func
|
|
|
|
|
|
tools_toml_path = os.path.join(os.path.dirname(__file__), "tools.toml")
|
|
with open(tools_toml_path) as tools_toml_file:
|
|
tools_toml = toml.load(tools_toml_file)
|
|
tools_for_extensions = tools_toml["tools_for_extensions"]
|
|
del tools_toml["tools_for_extensions"]
|
|
for tool_name, tool_toml in tools_toml.items():
|
|
tool_func = make_tool_function(**tool_toml)
|
|
tool_func.__name__ = tool_func.__qualname__ = tool_name
|
|
globals()[tool_name] = tool_func
|
|
|
|
#############################
|
|
|
|
|
|
LOG_PATH = os.path.join(os.getcwd(), "vigil.log")
|
|
|
|
|
|
def log_error(message=None):
|
|
message = traceback.format_exc() if message is None else message + "\n"
|
|
with open(LOG_PATH, "a") as log_file:
|
|
log_file.write(message)
|
|
|
|
|
|
def lru_cache_with_eviction(maxsize=128, typed=False):
|
|
versions = {}
|
|
make_key = functools._make_key
|
|
|
|
def evict(*args, **kwds):
|
|
key = make_key(args, kwds, typed)
|
|
if key in versions:
|
|
versions[key] += 1
|
|
|
|
def decorating_function(user_function):
|
|
|
|
def remove_version(*args, **kwds):
|
|
return user_function(*args[1:], **kwds)
|
|
new_func = functools.lru_cache(maxsize=maxsize, typed=typed)(
|
|
remove_version)
|
|
|
|
def add_version(*args, **kwds):
|
|
key = make_key(args, kwds, typed)
|
|
return new_func(*((versions.setdefault(key, 0),) + args), **kwds)
|
|
add_version.versions = versions
|
|
add_version.cache_info = new_func.cache_info
|
|
add_version.evict = evict
|
|
return functools.update_wrapper(add_version, user_function)
|
|
return decorating_function
|
|
|
|
|
|
def dump_pickle_safe(object_, path, protocol=pickle.HIGHEST_PROTOCOL,
|
|
open=open):
|
|
tmp_path = path + ".tmp"
|
|
try:
|
|
with open(tmp_path, "wb") as file_:
|
|
pickle.dump(object_, file_, protocol=protocol)
|
|
except (OSError, KeyboardInterrupt):
|
|
os.remove(tmp_path)
|
|
else:
|
|
os.rename(tmp_path, path)
|
|
|
|
|
|
def status_to_str(status):
|
|
return (_STATUS_TO_TERMSTR[status] if isinstance(status, enum.Enum)
|
|
else status)
|
|
|
|
|
|
class Result:
|
|
|
|
def __init__(self, path, tool):
|
|
self.path = path
|
|
self.tool = tool
|
|
self.pickle_path = os.path.join(CACHE_PATH, path + "-" + tool.__name__)
|
|
self.scroll_position = (0, 0)
|
|
self.is_completed = False
|
|
self.is_placeholder = True
|
|
self.status = Status.pending
|
|
|
|
@property
|
|
@lru_cache_with_eviction(maxsize=50)
|
|
def result(self):
|
|
unknown_label = fill3.Text("?")
|
|
if self.is_placeholder:
|
|
return unknown_label
|
|
try:
|
|
with gzip.open(self.pickle_path, "rb") as pickle_file:
|
|
return pickle.load(pickle_file)
|
|
except FileNotFoundError:
|
|
return unknown_label
|
|
|
|
@result.setter
|
|
def result(self, value):
|
|
os.makedirs(os.path.dirname(self.pickle_path), exist_ok=True)
|
|
dump_pickle_safe(value, self.pickle_path, open=gzip.open)
|
|
Result.result.fget.evict(self)
|
|
|
|
def set_status(self, status):
|
|
self.status = status
|
|
self.entry.appearance_cache = None
|
|
|
|
async def run(self, log, appearance_changed_event, runner):
|
|
self.is_placeholder = False
|
|
tool_name = tool_name_colored(self.tool, self.path)
|
|
path = path_colored(self.path)
|
|
log.log_message(["Running ", tool_name, " on ", path, "..."])
|
|
self.set_status(Status.running)
|
|
if runner.is_already_paused:
|
|
runner.is_already_paused = False
|
|
runner.pause()
|
|
appearance_changed_event.set()
|
|
start_time = time.time()
|
|
new_status = await runner.run_tool(self.path, self.tool)
|
|
Result.result.fget.evict(self)
|
|
end_time = time.time()
|
|
self.set_status(new_status)
|
|
appearance_changed_event.set()
|
|
self.is_completed = True
|
|
log.log_message(
|
|
["Finished running ", tool_name, " on ", path, ". ",
|
|
status_to_str(new_status),
|
|
f" {round(end_time - start_time, 2)} secs"])
|
|
|
|
def reset(self):
|
|
self.is_placeholder = True
|
|
self.set_status(Status.pending)
|
|
|
|
def appearance_min(self):
|
|
return [status_to_str(self.status)]
|
|
|
|
def as_html(self):
|
|
html, styles = termstr.TermStr(status_to_str(self.status)).as_html()
|
|
return (f'<a title="{self.tool.__name__}" '
|
|
f'href="{self.path}/{self.tool.__name__}">{html}</a>', styles)
|
|
|
|
|
|
def generic_tools():
|
|
return [contents, metadata]
|
|
|
|
|
|
TOOLS_FOR_EXTENSIONS = []
|
|
for extensions, tool_names in tools_for_extensions:
|
|
TOOLS_FOR_EXTENSIONS.append(
|
|
(extensions, [globals()[tool_name] for tool_name in tool_names]))
|
|
|
|
|
|
@functools.lru_cache(maxsize=1)
|
|
def _tools_for_extension():
|
|
result = {}
|
|
for extensions, tools in TOOLS_FOR_EXTENSIONS:
|
|
for extension in extensions:
|
|
result[extension] = tools
|
|
return result
|
|
|
|
|
|
def tools_all():
|
|
tools_ = set(generic_tools())
|
|
tools_.add(git_blame)
|
|
tools_.add(git_log)
|
|
for tool_list in _tools_for_extension().values():
|
|
tools_.update(set(tool_list))
|
|
return tools_
|
|
|
|
|
|
def tool_dependencies(tool):
|
|
try:
|
|
return tool.deps
|
|
except AttributeError:
|
|
return set()
|
|
|
|
|
|
def dependencies():
|
|
dependencies_all = set()
|
|
for tool in tools_all():
|
|
dependencies_all.update(tool_dependencies(tool))
|
|
return dependencies_all
|
|
|
|
|
|
def splitext(path):
|
|
root, ext = os.path.splitext(path)
|
|
if "." in root:
|
|
for compound_ext in [".tar.gz", ".tar.bz2"]:
|
|
if path.endswith(compound_ext):
|
|
return path[:-len(compound_ext)], path[-len(compound_ext):]
|
|
return root, ext
|
|
|
|
|
|
def tools_for_path(path):
|
|
git_tools = [git_blame, git_log] if os.path.exists(".git") else []
|
|
root, ext = splitext(path)
|
|
extra_tools = [] if ext == "" else _tools_for_extension().get(ext[1:], [])
|
|
return generic_tools() + git_tools + extra_tools
|
|
|
|
|
|
def run_tool_no_error(path, tool):
|
|
try:
|
|
status, result = tool(path)
|
|
except subprocess.TimeoutExpired:
|
|
status, result = Status.timed_out, fill3.Text("Timed out")
|
|
except:
|
|
status, result = Status.error, _syntax_highlight(
|
|
traceback.format_exc(), pygments.lexers.PythonTracebackLexer(),
|
|
pygments.styles.get_style_by_name(os.environ["PYGMENT_STYLE"]))
|
|
return status, result
|
|
|
|
|
|
def _convert_lscolor_code_to_charstyle(lscolor_code):
|
|
parts = lscolor_code.split(";")
|
|
if len(parts) == 1:
|
|
is_bold = parts[0] == "1"
|
|
fg_color = None
|
|
elif len(parts) == 2:
|
|
is_bold = False
|
|
fg_color = int(parts[1])
|
|
else:
|
|
is_bold = len(parts) == 4 and parts[3] == "1"
|
|
fg_color = int(parts[2])
|
|
return termstr.CharStyle(fg_color=fg_color, is_bold=is_bold)
|
|
|
|
|
|
def _charstyle_of_path(path):
|
|
color_code = lscolors.color_code_for_path(path, _LS_COLOR_CODES)
|
|
return (termstr.CharStyle() if color_code is None else
|
|
_convert_lscolor_code_to_charstyle(color_code))
|
|
|
|
|
|
@functools.lru_cache(maxsize=100)
|
|
def path_colored(path):
|
|
char_style = _charstyle_of_path(path)
|
|
path = path[2:]
|
|
dirname, basename = os.path.split(path)
|
|
if dirname == "":
|
|
return termstr.TermStr(basename, char_style)
|
|
else:
|
|
dirname = dirname + os.path.sep
|
|
return (termstr.TermStr(dirname, _charstyle_of_path(dirname)) +
|
|
termstr.TermStr(basename, char_style))
|
|
|
|
|
|
@functools.lru_cache(maxsize=100)
|
|
def tool_name_colored(tool, path):
|
|
char_style = (termstr.CharStyle(is_bold=True) if tool in generic_tools()
|
|
else _charstyle_of_path(path))
|
|
return termstr.TermStr(tool.__name__, char_style)
|
|
|
|
|
|
@functools.lru_cache()
|
|
def get_homepage_of_package(package):
|
|
line = subprocess.getoutput(f"dpkg-query --status {package}|grep Homepage")
|
|
return line.split()[1]
|
|
|
|
|
|
def url_of_tool(tool):
|
|
try:
|
|
url = tool.url
|
|
return url if url.startswith("http") else get_homepage_of_package(url)
|
|
except AttributeError:
|
|
return None
|