1
0
Fork 0
personal-style/main.py

91 lines
1.8 KiB
Python

import re
import subprocess
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('project')
RULES_MAIN = [
r"\b(System\.out\.println)\b",
r"^\s*var\b\s*\w*\s*=\s*[^n ][^e][^w]",
r"^\s*var\b\s*\w*\s*=\s*\r",
]
DUMB_WORDS = [
'derp',
'herp',
'fjerp',
'todo',
'attempts to',
'actual',
'technically',
'so',
"big integer",
# Misspelled words
'concatinate', # Concatenate
]
def re_select(options):
return r'(?:' + r'|'.join(options) + r')'
def re_delimit(rule):
return r'\b' + rule + r'\b'
def re_case_insensitive_sub(m):
m = m.group(0)
l = m.lower()
u = m.upper()
if l == u == m:
return m
else:
return '[' + l + u + ']'
def re_case_insensitive(rule):
return re.sub(r'\w', re_case_insensitive_sub, rule)
def re_sentence(sentence):
words = sentence.split(' ')
re = '\s+'.join(re_case_insensitive(w) for w in words)
return re
RE_DOC_COULD_BE_LINK = r'^(\s*(?:\/\*\*|\*).*)(\b[A-Z][a-z]\w*[A-Z]\w*)([ ,.])'
RULES_BOTH = [
r"\.size\(\)\s*[=!<]=\s*0",
r"\b(hasSize\(0\))\b",
r"\b(\.size\(\)\)\.isEqualTo\()\b",
r"\.get\([^)(]+\)\)\.isEqualTo",
r"\.isEqualTo\((List|Map|Set)\.of\(",
r"\bCollectors\.toMap\b",
r"\bCollectors\.toList\b",
re_delimit(re_select([re_sentence(w) for w in DUMB_WORDS])),
#RE_DOC_COULD_BE_LINK,
]
def construct_cmd(rules: list[str], target: str) -> list[str]:
return ['ag', '--java', '(' + re_select(rules) + ')', target]
def main():
args = parser.parse_args()
target = args.project
result1 = subprocess.run(construct_cmd(RULES_MAIN, target + '/main'))
result2 = subprocess.run(construct_cmd(RULES_BOTH, target))
if result1.returncode == 0 or result2.returncode == 0:
exit(1)
if __name__ == '__main__':
main()