summaryrefslogtreecommitdiffstats
blob: 81c7b179c523ca6cb74f6999b80a58595994ebfb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
#!/usr/bin/env python

import sys;
import os;
import re;
from sets import Set

filename = sys.argv[1]

inPlace = False
if "-i" in sys.argv:
	inPlace = True

filename_base = os.path.basename(filename)
(filename_name, filename_ext) = os.path.splitext(filename_base)

c_stdlib_headers = Set(["assert.h",  "limits.h",  "signal.h",  "stdlib.h", "ctype.h", "locale.h",  "stdarg.h", "string.h", "errno.h", "math.h", "stddef.h", "time.h", "float.h",  "setjmp.h", "stdio.h", "iso646.h", "wchar.h", "wctype.h", "complex.h", "inttypes.h", "stdint.h", "tgmath.h", "fenv.h", "stdbool.h"])

cpp_stdlib_headers = Set(["algorithm", "fstream", "list", "regex", "typeindex", "array", "functional", "locale", "set", "typeinfo", "atomic", "future", "map", "sstream", "type_traits", "bitset", "initializer_list", "memory", "stack", "unordered_map", "chrono", "iomanip", "mutex", "stdexcept", "unordered_set", "codecvt", "ios", "new", "streambuf", "utility", "complex", "iosfwd", "numeric", "string", "valarray", "condition_variable", "iostream", "ostream", "strstream", "vector", "deque", "istream", "queue", "system_error", "exception", "iterator", "random", "thread", "forward_list", "limits", "ratio", "tuple", "cassert", "ciso646", "csetjmp", "cstdio", "ctime", "cctype", "climits", "csignal", "cstdlib", "cwchar", "cerrno", "clocale", "cstdarg", "cstring", "cwctype", "cfloat", "cmath", "cstddef"])

class HeaderType:
		PRAGMA_ONCE, CORRESPONDING_HEADER, C_STDLIB, CPP_STDLIB, BOOST, QT, OTHER, SWIFTEN, SWIFT_CONTROLLERS, SWIFTOOLS, SWIFT = range(11)

def findHeaderBlock(lines):
	start = False
	end = False
	lastLine = None

	for idx, line in enumerate(lines):
		if not start and line.startswith("#"):
			start = idx
		elif start and (not end) and (not line.startswith("#")) and line.strip():
			end = idx-1
			break
	if not end:
		end = len(lines)
	return (start, end)

def lineToFileName(line):
	match = re.match( r'#include "(.*)"', line)
	if match:
		return match.group(1)
	match = re.match( r'#include <(.*)>', line)
	if match:
		return match.group(1)
	return False

def fileNameToHeaderType(name):
	if name.endswith(filename_name + ".h"):
		return HeaderType.CORRESPONDING_HEADER

	if name in c_stdlib_headers:
		return HeaderType.C_STDLIB

	if name in cpp_stdlib_headers:
		return HeaderType.CPP_STDLIB

	if name.startswith("boost"):
		return HeaderType.BOOST

	if name.startswith("Q"):
		return HeaderType.QT

	if name.startswith("Swiften"):
		return HeaderType.SWIFTEN

	if name.startswith("Swift/Controllers"):
		return HeaderType.SWIFT_CONTROLLERS

	if name.startswith("SwifTools"):
		return HeaderType.SWIFTOOLS

	if name.startswith("Swift"):
		return HeaderType.SWIFT

	return HeaderType.OTHER

def serializeHeaderGroups(groups):
	headerList = []
	for group in range(0, HeaderType.SWIFT + 1):
		if group in groups:
			# sorted and without duplicates
			headers = sorted(list(set(groups[group])))
			headerList.extend(headers)
			headerList.extend(["\n"])
	headerList.pop()
	return headerList

def overwriteFile(filename, content):
	with open(filename, 'w') as f:
		for line in content:
			f.write(line)

def cleanHeaderFile(content, headerStart, headerEnd, headerGroups):
	del content[headerStart:headerEnd]
	newHeaders = serializeHeaderGroups(headerGroups)
	content[headerStart:1] = newHeaders

	if inPlace :
		overwriteFile(filename, content)
	else :
		for line in content:
			print line,

def cleanImplementationFile(content, headerStart, headerEnd, headerGroups):
	del content[headerStart:headerEnd]
	newHeaders = serializeHeaderGroups(headerGroups)
	content[headerStart:1] = newHeaders

	if inPlace :
		overwriteFile(filename, content)
	else :
		for line in content:
			print line,


containsComplexPreprocessorDirectives = False

with open(filename) as f:
	content = f.readlines()

(headerStart, headerEnd) = findHeaderBlock(content)

headerGroups = {}

for line in content[headerStart:headerEnd]:
	if line.strip():
		if line.strip().startswith("#pragma once"):
			headerType = HeaderType.PRAGMA_ONCE
		elif line.strip().startswith("#if") or line.strip().startswith("#def") or line.strip().startswith("#undef") or line.strip().startswith("#pragma "):
			containsComplexPreprocessorDirectives = True
			break
		else:
			#print line
			headerType = fileNameToHeaderType(lineToFileName(line))

		#filename = lineToFileName(line)
		if headerType in headerGroups:
			headerGroups[headerType].append(line)
		else:
			headerGroups[headerType] = [line]

if containsComplexPreprocessorDirectives:
	print "Cannot format headers containing preprocessor #if, #pragma, #define or #undef statements!"
	exit(1)

if filename_base.endswith(".h"):
	if not HeaderType.PRAGMA_ONCE in headerGroups:
		print "Missing #pragma once!"
		exit(2)
	cleanHeaderFile(content, headerStart, headerEnd, headerGroups)
elif filename_base.endswith(".cpp") or filename_base.endswith(".mm"):
	cleanImplementationFile(content, headerStart, headerEnd, headerGroups)