123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
import re, os, sys
|
|
|
|
valid_extensions = ['c', 'cpp', 'h', 'hpp', 'mm', 'lua', 'rb', 'proto',
|
|
'init', 'init-example']
|
|
path_blacklist = [
|
|
'library/include/df/',
|
|
'plugins/stonesense/allegro',
|
|
'plugins/isoworld/allegro',
|
|
'plugins/isoworld/agui',
|
|
'depends/',
|
|
'.git/',
|
|
'build',
|
|
]
|
|
|
|
def valid_file(filename):
|
|
return len(filter(lambda ext: filename.endswith('.' + ext), valid_extensions)) and \
|
|
not len(filter(lambda path: path.replace('\\', '/') in filename.replace('\\', '/'), path_blacklist))
|
|
|
|
success = True
|
|
def error(msg):
|
|
global success
|
|
success = False
|
|
sys.stderr.write(msg + '\n')
|
|
|
|
class LinterError(Exception): pass
|
|
|
|
class Linter(object):
|
|
def check(self, lines):
|
|
failures = []
|
|
for i, line in enumerate(lines):
|
|
if not self.check_line(line):
|
|
failures.append(i + 1)
|
|
if len(failures):
|
|
raise LinterError('%s: %s' % (self.msg, self.display_lines(failures, len(lines))))
|
|
|
|
def fix(self, lines):
|
|
for i in range(len(lines)):
|
|
lines[i] = self.fix_line(lines[i])
|
|
|
|
def display_lines(self, lines, total):
|
|
if len(lines) == total - 1:
|
|
return 'entire file'
|
|
if not len(lines):
|
|
# should never happen
|
|
return 'nowhere'
|
|
if len(lines) == 1:
|
|
return 'line %i' % lines[0]
|
|
s = 'lines '
|
|
range_start = range_end = lines[0]
|
|
for i, line in enumerate(lines):
|
|
if line > range_end + 1:
|
|
if range_start == range_end:
|
|
s += ('%i, ' % range_end)
|
|
else:
|
|
s += ('%i-%i, ' % (range_start, range_end))
|
|
range_start = range_end = line
|
|
if i == len(lines) - 1:
|
|
s += ('%i' % line)
|
|
else:
|
|
range_end = line
|
|
if i == len(lines) - 1:
|
|
s += ('%i-%i, ' % (range_start, range_end))
|
|
return s.rstrip(' ').rstrip(',')
|
|
|
|
class NewlineLinter(Linter):
|
|
msg = 'Contains DOS-style newlines'
|
|
def check_line(self, line):
|
|
return '\r' not in line
|
|
def fix_line(self, line):
|
|
return line.replace('\r', '')
|
|
|
|
class TrailingWhitespaceLinter(Linter):
|
|
msg = 'Contains trailing whitespace'
|
|
def check_line(self, line):
|
|
line = line.replace('\r', '')
|
|
return not line.endswith(' ') and not line.endswith('\t')
|
|
def fix_line(self, line):
|
|
return line.rstrip('\t ')
|
|
|
|
class TabLinter(Linter):
|
|
msg = 'Contains tabs'
|
|
def check_line(self, line):
|
|
return '\t' not in line
|
|
def fix_line(self, line):
|
|
return line.replace('\t', ' ')
|
|
|
|
linters = [NewlineLinter(), TrailingWhitespaceLinter(), TabLinter()]
|
|
|
|
def main():
|
|
root_path = os.path.abspath(sys.argv[1] if len(sys.argv) > 1 else '.')
|
|
fix = (len(sys.argv) > 2 and sys.argv[2] == '--fix')
|
|
global path_blacklist
|
|
path_blacklist = map(lambda s: os.path.join(root_path, s), path_blacklist)
|
|
|
|
for cur, dirnames, filenames in os.walk(root_path):
|
|
for filename in filenames:
|
|
full_path = os.path.join(cur, filename)
|
|
rel_path = full_path.replace(root_path, '.')
|
|
if not valid_file(full_path):
|
|
continue
|
|
lines = []
|
|
with open(full_path, 'rb') as f:
|
|
lines = f.read().split('\n')
|
|
for linter in linters:
|
|
try:
|
|
linter.check(lines)
|
|
except LinterError as e:
|
|
error('%s: %s' % (rel_path, e))
|
|
if fix:
|
|
linter.fix(lines)
|
|
contents = '\n'.join(lines)
|
|
with open(full_path, 'wb') as f:
|
|
f.write(contents)
|
|
|
|
if success:
|
|
print('All linters completed successfully')
|
|
sys.exit(0)
|
|
else:
|
|
sys.exit(1)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|