1#!/usr/bin/env python 2# 3# Copyright 2009, Google Inc. 4# All rights reserved. 5# 6# Redistribution and use in source and binary forms, with or without 7# modification, are permitted provided that the following conditions are 8# met: 9# 10# * Redistributions of source code must retain the above copyright 11# notice, this list of conditions and the following disclaimer. 12# * Redistributions in binary form must reproduce the above 13# copyright notice, this list of conditions and the following disclaimer 14# in the documentation and/or other materials provided with the 15# distribution. 16# * Neither the name of Google Inc. nor the names of its 17# contributors may be used to endorse or promote products derived from 18# this software without specific prior written permission. 19# 20# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 21# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 22# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 23# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 24# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 25# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 26# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 27# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 28# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 29# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 30# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 31 32"""fuse_gtest_files.py v0.2.0 33Fuses Google Test source code into a .h file and a .cc file. 34 35SYNOPSIS 36 fuse_gtest_files.py [GTEST_ROOT_DIR] OUTPUT_DIR 37 38 Scans GTEST_ROOT_DIR for Google Test source code, and generates 39 two files: OUTPUT_DIR/gtest/gtest.h and OUTPUT_DIR/gtest/gtest-all.cc. 40 Then you can build your tests by adding OUTPUT_DIR to the include 41 search path and linking with OUTPUT_DIR/gtest/gtest-all.cc. These 42 two files contain everything you need to use Google Test. Hence 43 you can "install" Google Test by copying them to wherever you want. 44 45 GTEST_ROOT_DIR can be omitted and defaults to the parent 46 directory of the directory holding this script. 47 48EXAMPLES 49 ./fuse_gtest_files.py fused_gtest 50 ./fuse_gtest_files.py path/to/unpacked/gtest fused_gtest 51 52This tool is experimental. In particular, it assumes that there is no 53conditional inclusion of Google Test headers. Please report any 54problems to googletestframework@googlegroups.com. You can read 55http://code.google.com/p/googletest/wiki/GoogleTestAdvancedGuide for 56more information. 57""" 58 59__author__ = 'wan@google.com (Zhanyong Wan)' 60 61import os 62import re 63import sets 64import sys 65 66# We assume that this file is in the scripts/ directory in the Google 67# Test root directory. 68DEFAULT_GTEST_ROOT_DIR = os.path.join(os.path.dirname(__file__), '..') 69 70# Regex for matching '#include "gtest/..."'. 71INCLUDE_GTEST_FILE_REGEX = re.compile(r'^\s*#\s*include\s*"(gtest/.+)"') 72 73# Regex for matching '#include "src/..."'. 74INCLUDE_SRC_FILE_REGEX = re.compile(r'^\s*#\s*include\s*"(src/.+)"') 75 76# Where to find the source seed files. 77GTEST_H_SEED = 'include/gtest/gtest.h' 78GTEST_SPI_H_SEED = 'include/gtest/gtest-spi.h' 79GTEST_ALL_CC_SEED = 'src/gtest-all.cc' 80 81# Where to put the generated files. 82GTEST_H_OUTPUT = 'gtest/gtest.h' 83GTEST_ALL_CC_OUTPUT = 'gtest/gtest-all.cc' 84 85 86def VerifyFileExists(directory, relative_path): 87 """Verifies that the given file exists; aborts on failure. 88 89 relative_path is the file path relative to the given directory. 90 """ 91 92 if not os.path.isfile(os.path.join(directory, relative_path)): 93 print 'ERROR: Cannot find %s in directory %s.' % (relative_path, 94 directory) 95 print ('Please either specify a valid project root directory ' 96 'or omit it on the command line.') 97 sys.exit(1) 98 99 100def ValidateGTestRootDir(gtest_root): 101 """Makes sure gtest_root points to a valid gtest root directory. 102 103 The function aborts the program on failure. 104 """ 105 106 VerifyFileExists(gtest_root, GTEST_H_SEED) 107 VerifyFileExists(gtest_root, GTEST_ALL_CC_SEED) 108 109 110def VerifyOutputFile(output_dir, relative_path): 111 """Verifies that the given output file path is valid. 112 113 relative_path is relative to the output_dir directory. 114 """ 115 116 # Makes sure the output file either doesn't exist or can be overwritten. 117 output_file = os.path.join(output_dir, relative_path) 118 if os.path.exists(output_file): 119 # TODO(wan@google.com): The following user-interaction doesn't 120 # work with automated processes. We should provide a way for the 121 # Makefile to force overwriting the files. 122 print ('%s already exists in directory %s - overwrite it? (y/N) ' % 123 (relative_path, output_dir)) 124 answer = sys.stdin.readline().strip() 125 if answer not in ['y', 'Y']: 126 print 'ABORTED.' 127 sys.exit(1) 128 129 # Makes sure the directory holding the output file exists; creates 130 # it and all its ancestors if necessary. 131 parent_directory = os.path.dirname(output_file) 132 if not os.path.isdir(parent_directory): 133 os.makedirs(parent_directory) 134 135 136def ValidateOutputDir(output_dir): 137 """Makes sure output_dir points to a valid output directory. 138 139 The function aborts the program on failure. 140 """ 141 142 VerifyOutputFile(output_dir, GTEST_H_OUTPUT) 143 VerifyOutputFile(output_dir, GTEST_ALL_CC_OUTPUT) 144 145 146def FuseGTestH(gtest_root, output_dir): 147 """Scans folder gtest_root to generate gtest/gtest.h in output_dir.""" 148 149 output_file = file(os.path.join(output_dir, GTEST_H_OUTPUT), 'w') 150 processed_files = sets.Set() # Holds all gtest headers we've processed. 151 152 def ProcessFile(gtest_header_path): 153 """Processes the given gtest header file.""" 154 155 # We don't process the same header twice. 156 if gtest_header_path in processed_files: 157 return 158 159 processed_files.add(gtest_header_path) 160 161 # Reads each line in the given gtest header. 162 for line in file(os.path.join(gtest_root, gtest_header_path), 'r'): 163 m = INCLUDE_GTEST_FILE_REGEX.match(line) 164 if m: 165 # It's '#include "gtest/..."' - let's process it recursively. 166 ProcessFile('include/' + m.group(1)) 167 else: 168 # Otherwise we copy the line unchanged to the output file. 169 output_file.write(line) 170 171 ProcessFile(GTEST_H_SEED) 172 output_file.close() 173 174 175def FuseGTestAllCcToFile(gtest_root, output_file): 176 """Scans folder gtest_root to generate gtest/gtest-all.cc in output_file.""" 177 178 processed_files = sets.Set() 179 180 def ProcessFile(gtest_source_file): 181 """Processes the given gtest source file.""" 182 183 # We don't process the same #included file twice. 184 if gtest_source_file in processed_files: 185 return 186 187 processed_files.add(gtest_source_file) 188 189 # Reads each line in the given gtest source file. 190 for line in file(os.path.join(gtest_root, gtest_source_file), 'r'): 191 m = INCLUDE_GTEST_FILE_REGEX.match(line) 192 if m: 193 if 'include/' + m.group(1) == GTEST_SPI_H_SEED: 194 # It's '#include "gtest/gtest-spi.h"'. This file is not 195 # #included by "gtest/gtest.h", so we need to process it. 196 ProcessFile(GTEST_SPI_H_SEED) 197 else: 198 # It's '#include "gtest/foo.h"' where foo is not gtest-spi. 199 # We treat it as '#include "gtest/gtest.h"', as all other 200 # gtest headers are being fused into gtest.h and cannot be 201 # #included directly. 202 203 # There is no need to #include "gtest/gtest.h" more than once. 204 if not GTEST_H_SEED in processed_files: 205 processed_files.add(GTEST_H_SEED) 206 output_file.write('#include "%s"\n' % (GTEST_H_OUTPUT,)) 207 else: 208 m = INCLUDE_SRC_FILE_REGEX.match(line) 209 if m: 210 # It's '#include "src/foo"' - let's process it recursively. 211 ProcessFile(m.group(1)) 212 else: 213 output_file.write(line) 214 215 ProcessFile(GTEST_ALL_CC_SEED) 216 217 218def FuseGTestAllCc(gtest_root, output_dir): 219 """Scans folder gtest_root to generate gtest/gtest-all.cc in output_dir.""" 220 221 output_file = file(os.path.join(output_dir, GTEST_ALL_CC_OUTPUT), 'w') 222 FuseGTestAllCcToFile(gtest_root, output_file) 223 output_file.close() 224 225 226def FuseGTest(gtest_root, output_dir): 227 """Fuses gtest.h and gtest-all.cc.""" 228 229 ValidateGTestRootDir(gtest_root) 230 ValidateOutputDir(output_dir) 231 232 FuseGTestH(gtest_root, output_dir) 233 FuseGTestAllCc(gtest_root, output_dir) 234 235 236def main(): 237 argc = len(sys.argv) 238 if argc == 2: 239 # fuse_gtest_files.py OUTPUT_DIR 240 FuseGTest(DEFAULT_GTEST_ROOT_DIR, sys.argv[1]) 241 elif argc == 3: 242 # fuse_gtest_files.py GTEST_ROOT_DIR OUTPUT_DIR 243 FuseGTest(sys.argv[1], sys.argv[2]) 244 else: 245 print __doc__ 246 sys.exit(1) 247 248 249if __name__ == '__main__': 250 main() 251