1# -*- coding: utf-8 -*-
2
3#-------------------------------------------------------------------------
4# drawElements Quality Program utilities
5# --------------------------------------
6#
7# Copyright (c) 2017 The Khronos Group Inc.
8#
9# Licensed under the Apache License, Version 2.0 (the "License");
10# you may not use this file except in compliance with the License.
11# You may obtain a copy of the License at
12#
13#      http://www.apache.org/licenses/LICENSE-2.0
14#
15# Unless required by applicable law or agreed to in writing, software
16# distributed under the License is distributed on an "AS IS" BASIS,
17# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18# See the License for the specific language governing permissions and
19# limitations under the License.
20#
21#-------------------------------------------------------------------------
22
23import os
24import sys
25import codecs
26from optparse import OptionParser
27
28FILE_PATTERNS		= ["*.hpp", "*.h", "*.cpp", "*.py"]
29IGNORE_FILES		= set()
30CHECK_END_COMMENT	= True
31
32def hasBOM (file):
33	with open(file, 'rb') as f:
34		line0 = f.readline()
35		if line0.startswith(codecs.BOM_UTF8):
36			return True
37	return False
38
39def removeBOM (file):
40	with open(file, 'r+b') as f:
41		chunk = f.read(1024)
42		if chunk.startswith(codecs.BOM_UTF8):
43			chunk = chunk[3:]
44		else:
45			return
46		readpos = 1024;
47		writepos = 0;
48		while chunk:
49			f.seek(writepos, os.SEEK_SET)
50			f.write(chunk)
51			writepos += len(chunk)
52			f.seek(readpos, os.SEEK_SET)
53			chunk = f.read(1024)
54			readpos += len(chunk)
55		f.truncate(readpos-3)
56
57def getFileList (path):
58	if os.path.isfile(path):
59		yield path
60	elif os.path.isdir(path):
61		for root, dirs, files in os.walk(path):
62			for file in files:
63				yield os.path.join(root, file)
64
65def checkBOMs (files, fix):
66	correct = True
67	for file in files:
68		if hasBOM(file):
69			if fix:
70				removeBOM(file)
71				print("File %s contained BOM and was fixed" % file)
72			else:
73				correct = False
74				print("File %s contains BOM" % file)
75	return correct
76
77if __name__ == "__main__":
78	parser = OptionParser()
79	parser.add_option("-x", "--fix", action="store_true", dest="fix", default=False, help="attempt to fix BOMs")
80
81	(options, args)	= parser.parse_args()
82	fix				= options.fix
83
84	print("Checking BOMs...")
85	for dir in args:
86		checkBOMs(getFileList(os.path.normpath(dir)), fix)