| OLD | NEW | 
|---|
| 1 #!/usr/bin/python | 1 #!/usr/bin/python | 
| 2 # Copyright (c) 2009 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2009 The Chromium Authors. All rights reserved. | 
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be | 
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. | 
| 5 | 5 | 
| 6 """Rewrites paths in -I, -L and other option to be relative to a sysroot.""" | 6 """Rewrites paths in -I, -L and other option to be relative to a sysroot.""" | 
| 7 | 7 | 
| 8 import sys | 8 import sys | 
| 9 import os | 9 import os | 
|  | 10 import optparse | 
| 10 | 11 | 
| 11 REWRITE_PREFIX = ['-I', | 12 REWRITE_PREFIX = ['-I', | 
| 12                   '-idirafter', | 13                   '-idirafter', | 
| 13                   '-imacros', | 14                   '-imacros', | 
| 14                   '-imultilib', | 15                   '-imultilib', | 
| 15                   '-include', | 16                   '-include', | 
| 16                   '-iprefix', | 17                   '-iprefix', | 
| 17                   '-iquote', | 18                   '-iquote', | 
| 18                   '-isystem', | 19                   '-isystem', | 
| 19                   '-L'] | 20                   '-L'] | 
| 20 | 21 | 
| 21 def RewritePath(path, sysroot): | 22 def RewritePath(path, opts): | 
| 22   """Rewrites a path by prefixing it with the sysroot if it is absolute.""" | 23   """Rewrites a path by stripping the prefix and prepending the sysroot.""" | 
|  | 24   sysroot = opts.sysroot | 
|  | 25   prefix = opts.strip_prefix | 
| 23   if os.path.isabs(path) and not path.startswith(sysroot): | 26   if os.path.isabs(path) and not path.startswith(sysroot): | 
|  | 27     if path.startswith(prefix): | 
|  | 28       path = path[len(prefix):] | 
| 24     path = path.lstrip('/') | 29     path = path.lstrip('/') | 
| 25     return os.path.join(sysroot, path) | 30     return os.path.join(sysroot, path) | 
| 26   else: | 31   else: | 
| 27     return path | 32     return path | 
| 28 | 33 | 
| 29 def RewriteLine(line, sysroot): | 34 def RewriteLine(line, opts): | 
| 30   """Rewrites all the paths in recognized options.""" | 35   """Rewrites all the paths in recognized options.""" | 
| 31   args = line.split() | 36   args = line.split() | 
| 32   count = len(args) | 37   count = len(args) | 
| 33   i = 0 | 38   i = 0 | 
| 34   while i < count: | 39   while i < count: | 
| 35     for prefix in REWRITE_PREFIX: | 40     for prefix in REWRITE_PREFIX: | 
| 36       # The option can be either in the form "-I /path/to/dir" or | 41       # The option can be either in the form "-I /path/to/dir" or | 
| 37       # "-I/path/to/dir" so handle both. | 42       # "-I/path/to/dir" so handle both. | 
| 38       if args[i] == prefix: | 43       if args[i] == prefix: | 
| 39         i += 1 | 44         i += 1 | 
| 40         try: | 45         try: | 
| 41           args[i] = RewritePath(args[i], sysroot) | 46           args[i] = RewritePath(args[i], opts) | 
| 42         except IndexError: | 47         except IndexError: | 
| 43           sys.stderr.write('Missing argument following %s\n' % prefix) | 48           sys.stderr.write('Missing argument following %s\n' % prefix) | 
| 44           break | 49           break | 
| 45       elif args[i].startswith(prefix): | 50       elif args[i].startswith(prefix): | 
| 46         args[i] = prefix + RewritePath(args[i][len(prefix):], sysroot) | 51         args[i] = prefix + RewritePath(args[i][len(prefix):], opts) | 
| 47     i += 1 | 52     i += 1 | 
| 48 | 53 | 
| 49   return ' '.join(args) | 54   return ' '.join(args) | 
| 50 | 55 | 
| 51 def main(argv): | 56 def main(argv): | 
| 52   try: | 57   parser = optparse.OptionParser() | 
| 53     sysroot = argv[1] | 58   parser.add_option('-s', '--sysroot', default='/', help='sysroot to prepend') | 
| 54   except IndexError: | 59   parser.add_option('-p', '--strip-prefix', default='', help='prefix to strip') | 
| 55     sys.stderr.write('usage: %s /path/to/sysroot\n' % argv[0]) | 60   opts, args = parser.parse_args(argv[1:]) | 
| 56     return 1 |  | 
| 57 | 61 | 
| 58   for line in sys.stdin.readlines(): | 62   for line in sys.stdin.readlines(): | 
| 59     line = RewriteLine(line.strip(), sysroot) | 63     line = RewriteLine(line.strip(), opts) | 
| 60     print line | 64     print line | 
| 61   return 0 | 65   return 0 | 
| 62 | 66 | 
| 63 if __name__ == '__main__': | 67 if __name__ == '__main__': | 
| 64   sys.exit(main(sys.argv)) | 68   sys.exit(main(sys.argv)) | 
| OLD | NEW | 
|---|