Home | History | Annotate | Download | only in build 
      1 #!/usr/bin/env python
      2 #
      3 # Copyright (C) 2012 The Android Open Source Project
      4 #
      5 # Licensed under the Apache License, Version 2.0 (the 'License');
      6 # you may not use this file except in compliance with the License.
      7 # You may obtain a copy of the License at
      8 #
      9 #      http://www.apache.org/licenses/LICENSE-2.0
     10 #
     11 # Unless required by applicable law or agreed to in writing, software
     12 # distributed under the License is distributed on an 'AS IS' BASIS,
     13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     14 # See the License for the specific language governing permissions and
     15 # limitations under the License.
     16 #
     17 
     18 # prepare_pdk_tree.py target_dir [-m manifest] pdk_groups
     19 # Ex: prepare_pdk_tree.py ../tmp/pdk grouper
     20 # create mount_pdk.sh and umount_pdk.sh, which mounts/umounts pdk sources.
     21 
     22 import os
     23 import re
     24 import sys
     25 import subprocess
     26 
     27 
     28 class ManifestHandler(object):
     29 
     30     def __init__(self):
     31         # current pattern
     32         self.current = 0
     33         self.patterns = [re.compile('path=\"([^\"]*)\".*groups=\"([^\"]*)\"'), \
     34                          re.compile('groups=\"([^\"]*)\".*path=\"([^\"]*)\"')]
     35 
     36     def getAttribs(self, line):
     37         attrib = [None, None] # list of path, groups
     38         m = self.patterns[self.current].search(line)
     39         # if match fails, try both pattens and change default one
     40         # if match founds
     41         if m is None:
     42             notCurrent = 1 - self.current
     43             mOther = self.patterns[notCurrent].search(line)
     44             if mOther is not None:
     45                 # toggle
     46                 self.current = notCurrent
     47                 m = mOther
     48         if m is not None:
     49             if (self.current == 0):
     50                 attrib[0] = m.group(1)
     51                 attrib[1] = m.group(2)
     52             else:
     53                 attrib[0] = m.group(2)
     54                 attrib[1] = m.group(1)
     55         return attrib
     56 
     57 def isInGroups(groupsAttrib, groups):
     58     if groupsAttrib is None:
     59         return False
     60     for group in groups:
     61         if group in groupsAttrib:
     62             return True
     63     return False
     64 
     65 def getPDKDirs(manifest, groups):
     66     subdirs = []
     67     handler = ManifestHandler()
     68     f = open(manifest, 'r')
     69     for line in f:
     70         [path, groupsAttrib] = handler.getAttribs(line)
     71         if isInGroups(groupsAttrib, groups):
     72             subdirs.append(path)
     73     f.close()
     74     return subdirs
     75 
     76 def create_symbolic_link(src_top, dest_top, dir_name):
     77     src_full = src_top + "/" + dir_name
     78     dest_full = dest_top + "/" + dir_name
     79     #print "create symbolic link from " + dest_full + " to " + src_full
     80     # remove existing link first to prevent recursive loop
     81     os.system("rm -rf " + dest_full)
     82     os.system("ln -s " + src_full + " " + dest_full)
     83 
     84 # The only file not from manifest.
     85 copy_files_list = [ "Makefile" ]
     86 MOUNT_FILE = 'mount_pdk.sh'
     87 UMOUNT_FILE = 'umount_pdk.sh'
     88 SH_HEADER = "#!/bin/bash\n#Auto-generated file, do not edit!\n"
     89 
     90 def main(argv):
     91     manifestFile = ".repo/manifest.xml"
     92     groups = ["pdk"]
     93     if len(argv) < 2:
     94         print "create_pdk_tree.py target_dir [-m manifest] pdk_groups"
     95         print " ex) create_pdk_tree.py ../tmp grouper"
     96         print " Note that pdk group is included by default"
     97         print " Do not create target_dir under the current source tree. This will cause build error."
     98         sys.exit(1)
     99     targetDir = argv[1]
    100     argc = 2
    101     if len(argv) > 2:
    102         if argv[2] == "-m":
    103             manifestFile = argv[3]
    104             argc += 2
    105     while argc < len(argv):
    106         groups.append(argv[argc])
    107         argc += 1
    108     sourceDir = os.path.abspath('.')
    109     targetDir = os.path.abspath(targetDir)
    110 
    111     p = subprocess.Popen("mount", stdout = subprocess.PIPE)
    112     targetMounted = False
    113     for line in p.stdout:
    114         if targetDir in line:
    115             targetMounted = True
    116     p.stdout.close()
    117 
    118     if targetMounted:
    119         print "target dir already mounted"
    120         if os.path.exists(targetDir + '/' + UMOUNT_FILE):
    121             print "Use existing file", UMOUNT_FILE, "to unmount"
    122             sys.exit(1)
    123         else:
    124             print "Will create scripts, but may need manual unmount"
    125 
    126     subdirs = getPDKDirs(manifestFile, groups)
    127     print subdirs
    128     os.system("mkdir -p " + targetDir)
    129     mountf = open(targetDir + '/' + MOUNT_FILE, 'w+')
    130     mountf.write(SH_HEADER)
    131     umountf = open(targetDir + '/' + UMOUNT_FILE, 'w+')
    132     umountf.write(SH_HEADER)
    133     for subdir in subdirs:
    134         os.system("mkdir -p " + targetDir + '/' + subdir)
    135         mountf.write("mount --bind " + sourceDir + "/" + subdir + " " + targetDir + "/" + subdir + \
    136                         "\n")
    137         umountf.write("umount " + sourceDir + "/" + subdir + "\n")
    138     for file_name in copy_files_list:
    139         create_symbolic_link(sourceDir, targetDir, file_name)
    140     mountf.close()
    141     umountf.close()
    142     os.system("chmod 700 " + targetDir + '/' + MOUNT_FILE)
    143     os.system("chmod 700 " + targetDir + '/' + UMOUNT_FILE)
    144 
    145 if __name__ == '__main__':
    146     main(sys.argv)
    147