Home | History | Annotate | Download | only in scripts 
      1 #!/usr/bin/env python
      2 #
      3 # Copyright (C) 2010 Google Inc. All rights reserved.
      4 #
      5 # Redistribution and use in source and binary forms, with or without
      6 # modification, are permitted provided that the following conditions are
      7 # met:
      8 #
      9 #         * Redistributions of source code must retain the above copyright
     10 # notice, this list of conditions and the following disclaimer.
     11 #         * Redistributions in binary form must reproduce the above
     12 # copyright notice, this list of conditions and the following disclaimer
     13 # in the documentation and/or other materials provided with the
     14 # distribution.
     15 #         * Neither the name of Google Inc. nor the names of its
     16 # contributors may be used to endorse or promote products derived from
     17 # this software without specific prior written permission.
     18 #
     19 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     20 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     21 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
     22 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
     23 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     24 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
     25 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     26 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     27 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     28 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     29 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     30 #
     31 
     32 # This script concatenates in place JS files in the order specified
     33 # using <script> tags in a given 'order.html' file.
     34 
     35 from __future__ import with_statement
     36 
     37 from HTMLParser import HTMLParser
     38 from cStringIO import StringIO
     39 
     40 import rjsmin
     41 import os.path
     42 import sys
     43 
     44 
     45 class OrderedJSFilesExtractor(HTMLParser):
     46 
     47     def __init__(self, order_html):
     48         HTMLParser.__init__(self)
     49         self.ordered_js_files = []
     50         self.feed(order_html)
     51 
     52     def handle_starttag(self, tag, attrs):
     53         if tag == 'script':
     54             attrs_dict = dict(attrs)
     55             if ('type' in attrs_dict and attrs_dict['type'] == 'text/javascript' and 'src' in attrs_dict):
     56                 self.ordered_js_files.append(attrs_dict['src'])
     57 
     58 class PathExpander:
     59 
     60     def __init__(self, paths):
     61         self.paths = paths
     62 
     63     def expand(self, filename):
     64         last_path = None
     65         expanded_name = None
     66         for path in self.paths:
     67             fname = "%s/%s" % (path, filename)
     68             if (os.access(fname, os.F_OK)):
     69                 if (last_path != None):
     70                     raise Exception('Ambiguous file %s: found in %s and %s' %
     71                                     (filename, last_path, path))
     72                 expanded_name = fname
     73                 last_path = path
     74         return expanded_name
     75 
     76 
     77 def main(argv):
     78 
     79     if len(argv) < 3:
     80         print('usage: %s order.html input_source_dir_1 input_source_dir_2 ... '
     81               'output_file' % argv[0])
     82         return 1
     83 
     84     output_file_name = argv.pop()
     85     input_order_file_name = argv[1]
     86     with open(input_order_file_name, 'r') as order_html:
     87         extractor = OrderedJSFilesExtractor(order_html.read())
     88 
     89     expander = PathExpander(argv[2:])
     90     output = StringIO()
     91 
     92     for input_file_name in extractor.ordered_js_files:
     93         full_path = expander.expand(input_file_name)
     94         if (full_path is None):
     95             raise Exception('File %s referenced in %s not found on any source paths, '
     96                             'check source tree for consistency' %
     97                             (input_file_name, input_order_file_name))
     98         output.write('/* %s */\n\n' % input_file_name)
     99         input_file = open(full_path, 'r')
    100         output.write(input_file.read())
    101         output.write('\n')
    102         input_file.close()
    103 
    104     if os.path.exists(output_file_name):
    105         os.remove(output_file_name)
    106     output_file = open(output_file_name, 'w')
    107     output_file.write(rjsmin.jsmin(output.getvalue()))
    108     output_file.close()
    109     output.close()
    110 
    111     # Touch output file directory to make sure that Xcode will copy
    112     # modified resource files.
    113     if sys.platform == 'darwin':
    114         output_dir_name = os.path.dirname(output_file_name)
    115         os.utime(output_dir_name, None)
    116 
    117 if __name__ == '__main__':
    118     sys.exit(main(sys.argv))
    119