summaryrefslogtreecommitdiffstats
blob: e53246455a9d784dd8f094d1d74e21e3c638bf84 (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
154
155
156
157
158
159
160
161
162
163
164
165
#!/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, SWIFTEN_BASE_DEBUG, OTHER, SWIFTEN, LIMBER, SLIMBER, SWIFT_CONTROLLERS, SLUIFT, SWIFTOOLS, SWIFT = range(15)

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/Base/Debug.h"):
        return HeaderType.SWIFTEN_BASE_DEBUG

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

    if name.startswith("Limber"):
        return HeaderType.LIMBER

    if name.startswith("Slimber"):
        return HeaderType.SLIMBER

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

    if name.startswith("Sluift"):
        return HeaderType.SLUIFT

    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)