tools/package/pkg

   1 #!/usr/bin/env python
   2 # -*- coding: iso-8859-1 -*-
   3
   4 import os, sys, struct
   5 from stat import ST_SIZE
   6
   7 PKG_PREFIX                 = 'PKG'
   8 PKG_VERSION                = 1
   9
  10 PKG_HEADER_FORMAT          = '3cB'
  11
  12 PKG_FILE_PATHLENGTH_FORMAT = '!I'
  13 PKG_FILE_DATALENGTH_FORMAT = '!I'
  14
  15 def readHeader( file ):
  16     'readHeader( file ) -> version'
  17
  18     header = struct.unpack(
  19         PKG_HEADER_FORMAT,
  20         file.read( struct.calcsize( PKG_HEADER_FORMAT ) )
  21     )
  22     version = header[3]
  23
  24     size = struct.unpack(
  25         PKG_FILE_DATALENGTH_FORMAT,
  26         file.read( struct.calcsize( PKG_FILE_DATALENGTH_FORMAT ) )
  27     )
  28
  29     return version
  30
  31
  32 def writeHeader( file, version ):
  33     'writeHeader( file, version ) -> None'
  34
  35     file.write(
  36         struct.pack(
  37             PKG_HEADER_FORMAT,
  38             PKG_PREFIX[0], PKG_PREFIX[1], PKG_PREFIX[2], PKG_VERSION
  39         )
  40     )
  41
  42     file.write(
  43         struct.pack(
  44             PKG_FILE_DATALENGTH_FORMAT,
  45             0
  46         )
  47     )
  48
  49
  50 def readFile( file ):
  51     'readFile( file ) -> (path, data)'
  52
  53     pathLength = struct.unpack(
  54         PKG_FILE_PATHLENGTH_FORMAT,
  55         file.read( struct.calcsize( PKG_FILE_PATHLENGTH_FORMAT ) )
  56     )
  57
  58     path = file.read( pathLength[0] )
  59     file.read(1)
  60
  61     dataLength = struct.unpack(
  62         PKG_FILE_DATALENGTH_FORMAT,
  63         file.read( struct.calcsize( PKG_FILE_DATALENGTH_FORMAT ) )
  64     )
  65
  66     data = file.read( dataLength[0] )
  67
  68     return (path, data)
  69
  70
  71 def writeFile( file, path, data ):
  72     'writeFile( file, path, data ) -> None'
  73
  74     file.write( struct.pack( PKG_FILE_PATHLENGTH_FORMAT, len( path ) ) )
  75     file.write( path + "\0")
  76     file.write( struct.pack( PKG_FILE_DATALENGTH_FORMAT, len( data ) ) )
  77     file.write( data )
  78
  79
  80 def listFiles( path='.' ):
  81     'listFiles( path ) -> []'
  82
  83     files = []
  84     for name in os.listdir( path ):
  85         name = os.path.join( path, name )
  86
  87         if os.path.isdir( name ):
  88             files.extend( listFiles( name ) )
  89         else:
  90             files.append( os.path.normpath( name ) )
  91
  92     return files
  93
  94
  95 def unpack( source, destination ):
  96     'unpack( source, destination ) -> None'
  97
  98     if not os.path.exists( destination ):
  99         os.makedirs( destination )
 100
 101     inputSize = os.path.getsize( source )
 102     input     = open( source )
 103
 104     oldwd = os.getcwd()
 105     os.chdir( destination )
 106
 107     version = readHeader( input )
 108
 109     while input.tell() < inputSize:
 110         (path, data) = readFile( input )
 111
 112         print '« %s (%d bytes)' % (path, len( data ))
 113
 114         directory = os.path.dirname( path )
 115         if directory != '' and not os.path.exists( directory ):
 116             os.makedirs( directory )
 117
 118         output = open( path, 'w' )
 119         output.write( data )
 120         output.close()
 121
 122     os.chdir( oldwd )
 123
 124
 125 def pack( source, destination ):
 126     'pack( source, destination ) -> None'
 127
 128     output = open( destination, 'w' )
 129
 130     oldwd = os.getcwd()
 131     os.chdir( source )
 132
 133     files = listFiles()
 134
 135     writeHeader( output, PKG_VERSION )
 136
 137     for path in files:
 138         print '» %s (%d bytes)' % (path, os.path.getsize( path ))
 139
 140         input = open( path )
 141         writeFile( output, path, input.read() )
 142         input.close()
 143
 144     os.chdir( oldwd )
 145
 146     output.seek( struct.calcsize( PKG_HEADER_FORMAT ) )
 147     st = os.fstat( output.fileno() )
 148
 149     output.write(
 150         struct.pack(
 151             PKG_FILE_DATALENGTH_FORMAT,
 152             st[ST_SIZE]
 153         )
 154     )
 155
 156
 157     output.close()
 158
 159
 160 if __name__ == '__main__':
 161     mode        = sys.argv[1]
 162     source      = sys.argv[2]
 163     destination = sys.argv[3]
 164
 165     if   mode == 'c':
 166         pack( destination, source )
 167     elif mode == 'x':
 168         unpack( source, destination )
 169     else:
 170         print 'Invalid mode.'
 171