summaryrefslogtreecommitdiff
path: root/tools/notsd-fixup--includes
diff options
context:
space:
mode:
Diffstat (limited to 'tools/notsd-fixup--includes')
-rwxr-xr-xtools/notsd-fixup--includes86
1 files changed, 46 insertions, 40 deletions
diff --git a/tools/notsd-fixup--includes b/tools/notsd-fixup--includes
index a82247e175..afd616534d 100755
--- a/tools/notsd-fixup--includes
+++ b/tools/notsd-fixup--includes
@@ -231,41 +231,56 @@ class IncludeSection:
################################################################
# The main program loop
-def phase0(cache, filename, line, file=sys.stdout):
- global phase
- phase = phase0
+class Parser:
+ def __init__(self, cache, ifilename, ofilename):
+ self.cache = cache
+ self.ifilename = os.path.normpath(ifilename)
+ self.ofilename = ofilename
- if re.fullmatch('#include.*|typedef .*;', line):
- global includes
- includes = IncludeSection()
- phase1(cache, filename, line)
- else:
- print(line, file=file)
+ self.includes = None
+ self.phase = self.phase0
-def phase1(cache, filename, line, file=sys.stdout):
- global phase, includes
- phase = phase1
+ def phase0(self, line, ofile):
+ self.phase = self.phase0
- if line == '':
- includes.trailing_nl += '\n'
- elif line.startswith('#include'):
- includes.trailing_nl = ''
- match = re.fullmatch('^#include [<"]([^">]*)[">](.*)', line)
- if match:
- group, path = classify(cache, filename, match.group(1))
- includes.add(group, path, match.group(2))
+ if re.fullmatch('#include.*|typedef .*;', line):
+ self.includes = IncludeSection()
+ self.phase1(line, ofile)
else:
- sys.exit('panic: malformed #include line')
- elif re.fullmatch('typedef .*;', line):
- includes.trailing_nl = ''
- includes.typedef.append(line)
- else:
- includes.print(file=file)
- includes = None
- phase0(cache, filename, line, file=file)
+ print(line, file=ofile)
+
+ def phase1(self, line, ofile):
+ self.phase = self.phase1
+
+ if line == '':
+ self.includes.trailing_nl += '\n'
+ elif line.startswith('#include'):
+ self.includes.trailing_nl = ''
+ match = re.fullmatch('^#include [<"]([^">]*)[">](.*)', line)
+ if match:
+ group, path = classify(self.cache, self.ifilename, match.group(1))
+ self.includes.add(group, path, match.group(2))
+ else:
+ sys.exit('panic: malformed #include line')
+ elif re.fullmatch('typedef .*;', line):
+ self.includes.trailing_nl = ''
+ self.includes.typedef.append(line)
+ else:
+ self.includes.print(file=ofile)
+ self.includes = None
+ self.phase0(line, ofile)
-includes = None
-phase = phase0
+ def run(self):
+ print(' => {0} {1}'.format(
+ shlex.quote(__file__),
+ shlex.quote(self.ifilename),
+ ), file=sys.stderr)
+ with open(self.ofilename, 'w') as ofile:
+ with open(self.ifilename) as ifile:
+ for line in ifile:
+ self.phase(line.rstrip('\n'), ofile)
+ if self.includes:
+ self.includes.print(file=ofile)
def main(argv):
cache = Cache(__file__+'.cache')
@@ -279,16 +294,7 @@ def main(argv):
atexit.register(cleanup)
for filename in argv[1:]:
tmpfilename = os.path.join(os.path.dirname(filename), '.tmp.'+os.path.basename(filename)+'.tmp')
- print(' => {0} {1}'.format(
- shlex.quote(__file__),
- shlex.quote(filename),
- ), file=sys.stderr)
- with open(tmpfilename, 'w') as tmpfile:
- with open(filename) as f:
- for line in f:
- phase(cache, filename, line.rstrip('\n'), file=tmpfile)
- if includes:
- includes.print(file=tmpfile)
+ Parser(cache, filename, tmpfilename).run()
if not filecmp.cmp(filename, tmpfilename):
os.rename(tmpfilename, filename)
cleanup()