Mac/scripts/binhextree.py

   1 #
   2 # binhextree - Recursively descend a directory and
   3 # pack all resource files.
   4 #
   5 # Actually it doesn't binhex anymore, it only copies projects.
   6 #
   7 # Jack Jansen, CWI, August 1995.
   8 #
   9
  10 import os
  11 import binhex
  12 import sys
  13 import macostools
  14 import macfs
  15
  16 import aetools
  17 from Metrowerks_Shell_Suite import Metrowerks_Shell_Suite
  18 from Required_Suite import Required_Suite
  19
  20 class MwShell(aetools.TalkTo, Metrowerks_Shell_Suite, Required_Suite):
  21         pass
  22
  23 # Top-level directory
  24 TOP=''
  25
  26 # Where to put CW projects, relative to TOP
  27 CWDIR=':Mac:mwerks:projects'
  28 # From which folders to put projects there
  29 CWDIRDIRS=['build.mac', 'build.macstand', 'build.macfreeze', 'PlugIns']
  30
  31 # Helper routines
  32 def binhexit(path, name):
  33         dstfile = path + '.hqx'
  34         if os.path.exists(dstfile):
  35                 print 'Compare', path,'...',
  36                 if binhexcompare(path, dstfile):
  37                         print 'Identical, skipped.'
  38                         return
  39                 else:
  40                         print 'Not up-to-date.'
  41         print 'Binhexing', path
  42         binhex.binhex(path, dstfile)
  43
  44 def binhexcompare(source, hqxfile):
  45         """(source, hqxfile) - Check whether the two files match (forks only)"""
  46         ifp = binhex.HexBin(hqxfile)
  47
  48         sfp = open(source, 'rb')
  49         while 1:
  50                 d = ifp.read(128000)
  51                 d2 = sfp.read(128000)
  52                 if d <> d2:
  53                         return 0
  54                 if not d: break
  55         sfp.close()
  56         ifp.close_data()
  57
  58         d = ifp.read_rsrc(128000)
  59         if d:
  60                 sfp = binhex.openrsrc(source, 'rb')
  61                 d2 = sfp.read(128000)
  62                 if d <> d2:
  63                         return 0
  64                 while 1:
  65                         d = ifp.read_rsrc(128000)
  66                         d2 = sfp.read(128000)
  67                         if d <> d2:
  68                                 return 0
  69                         if not d: break
  70         return 1
  71
  72 # Project files to handle
  73 project_files = {}
  74
  75 def hexbincwprojects(creator):
  76         """Compact and hexbin all files remembered with a given creator"""
  77         cw_running = 0
  78         for fss in project_files[creator]:
  79                 srcfile = fss.as_pathname()
  80
  81                 old_style = 0
  82                 if srcfile[-1] == 'µ':
  83                         dstfile = srcfile[:-1]+'mu.hqx'
  84                         old_style = 1
  85                 elif srcfile[-3] == '.mu':
  86                         dstfile = srcfile + '.hqx'
  87                 elif ord(srcfile[-1]) >= 128:
  88                         dstfile = srcfile[:-1]+`ord(srcfile[-1])`+'.hqx'
  89                 else:
  90                         dstfile = srcfile + '.hqx'
  91
  92                 if os.path.exists(dstfile) and \
  93                                 os.stat(dstfile)[8] >= os.stat(srcfile)[8]:
  94                         print 'Skip', dstfile,'- Up-to-date'
  95                         continue
  96                 print 'Compacting', dstfile
  97                 if old_style:
  98                         if not cw_running:
  99                                 try:
 100                                         mgr = MwShell(creator, start=1)
 101                                 except 'foo':
 102                                         print 'Not handled:', creator
 103                                         return
 104                                 cw_running = 1
 105                         mgr.open(fss)
 106                         mgr.Reset_File_Paths()
 107                         mgr.Remove_Binaries()
 108                         mgr.Close_Project()
 109
 110                 print 'Binhexing', dstfile
 111                 binhex.binhex(srcfile, dstfile)
 112         if cw_running:
 113                 mgr.quit()
 114
 115 def copycwproject(path, name):
 116         """Copy CW project (if needed) and remember for hexbinning"""
 117         global project_files
 118
 119         dstdir = os.path.join(TOP, CWDIR)
 120         if path[:len(dstdir)] == dstdir:
 121                 return
 122         srcdir = os.path.split(path)[0]
 123         srcdir = os.path.split(srcdir)[1]
 124         if srcdir in CWDIRDIRS:
 125                 if not os.path.exists(dstdir):
 126                         print dstdir
 127                         print 'No CW-project dir, skip', name
 128                         return
 129                 dstfile = os.path.join(dstdir, os.path.join(srcdir, name))
 130         else:
 131                 if path[-2:] == '.µ':
 132                         dstfile = path[:-2]+ '.mu'
 133                 elif path[-4:] == '.prj':
 134                         dstfile = None
 135                 else:
 136                         return
 137
 138         if dstfile:
 139                 # If the destination doesn't exists or is older that the source
 140                 # we copy and remember it
 141
 142                 if os.path.exists(dstfile) and \
 143                                 os.stat(dstfile)[8] >= os.stat(path)[8]:
 144                         print 'Not copying', path,'- Up-to-date'
 145                 else:
 146                         print 'Copy', path
 147                         macostools.copy(path, dstfile)
 148         else:
 149                 dstfile = path
 150
 151         fss = macfs.FSSpec(dstfile)
 152         creator = fss.GetCreatorType()[0]
 153
 154         if project_files.has_key(creator):
 155                 project_files[creator].append(fss)
 156         else:
 157                 project_files[creator] = [fss]
 158
 159 def copycwexpfile(path, name):
 160         """Copy CW export file"""
 161         global project_files
 162
 163         dstdir = os.path.join(TOP, CWDIR)
 164         if path[:len(dstdir)] == dstdir:
 165                 return
 166         srcdir = os.path.split(path)[0]
 167         srcdir = os.path.split(srcdir)[1]
 168         if srcdir in CWDIRDIRS:
 169                 if not os.path.exists(dstdir):
 170                         print dstdir
 171                         print 'No CW-project dir, skip', name
 172                         return
 173                 dstfile = os.path.join(dstdir, os.path.join(srcdir, name))
 174         else:
 175                 if path[-6:] != '.µ.exp':
 176                         return
 177                 dstfile = path[:-6] + '.mu.exp'
 178         if dstfile[-6:] == '.µ.exp':
 179                 dstfile = dstfile[:-6]+'.mu.exp'
 180
 181         # If the destination doesn't exists or is older that the source
 182         # we copy and remember it
 183
 184         if os.path.exists(dstfile) and \
 185                         os.stat(dstfile)[8] >= os.stat(path)[8]:
 186                 print 'Not copying', path,'- Up-to-date'
 187         else:
 188                 print 'Copy', path
 189                 macostools.copy(path, dstfile)
 190
 191 extensions = [
 192 ##      ('.rsrc', binhexit),
 193 ##      ('.gif', binhexit),
 194         ('.µ', copycwproject),
 195         ('.prj', copycwproject),
 196         ('.prj.exp', copycwexpfile),
 197         ('.µ.exp', copycwexpfile)
 198         ]
 199
 200 def walker(arg, top, names):
 201         lnames = names[:]
 202         for n in lnames:
 203                 if n[0] == '(' and n[-1] == ')':
 204                         names.remove(n)
 205                         continue
 206                 for ext, handler in extensions:
 207                         if n[-len(ext):] == ext:
 208                                 name = os.path.join(top, n)
 209                                 handler(name, n)
 210
 211 def dodir(name):
 212         global TOP, project_files
 213         TOP = name
 214         os.path.walk(name, walker, None)
 215
 216 ##      for creator in project_files.keys():
 217 ##              hexbincwprojects(creator)
 218         project_files = {}
 219
 220 def main():
 221         if len(sys.argv) > 1:
 222                 for dir in sys.argv[1:]:
 223                         dodir(dir)
 224         elif os.name == 'mac':
 225                 import macfs
 226                 dir, ok = macfs.GetDirectory('Folder to search:')
 227                 if not ok:
 228                         sys.exit(0)
 229                 dodir(dir.as_pathname())
 230         else:
 231                 print 'Usage: hexbintree dir ...'
 232                 sys.exit(1)
 233         if os.name == 'mac':
 234                 sys.exit(1)   # Keep window
 235         else:
 236                 sys.exit(0)
 237
 238 if __name__ == '__main__':
 239         main()
 240