#!/usr/bin/env python3 
import argparse
import zipfile
from os import listdir, path

SRC = path.join(path.dirname(path.abspath(__file__)), 'DeviceSupport')
DEVICE_SUPPORT_PATH='Contents/Developer/Platforms/iPhoneOS.platform/DeviceSupport'

def unzip_file(name, target):
  f = path.join(SRC, name + '.zip')
  zip_ref = zipfile.ZipFile(f, 'r')
  zip_ref.extractall(target)
  zip_ref.close()

def process(xcode, version):
  target = path.join(xcode, DEVICE_SUPPORT_PATH)
  exist = listdir(target)
  all_files = [i.replace('.zip', '') for i in listdir(SRC) if i.endswith('.zip')]
  new_files = list(set(all_files) - set(exist))

  if version:
      new_files = list(filter(lambda x : version in x, new_files))

  for i in new_files:
    print ('Unzip file "{}.zip" to {}'.format(i, target))
    unzip_file(i, target)
  print ('\nUpdate successfully for {}'.format(xcode))

if __name__ == '__main__':
  parser = argparse.ArgumentParser()
  parser.add_argument(
    '-t',
    type=str,
    dest='target',
    default='/Applications/Xcode.app',
    help='The path for Xcode'
  )
  parser.add_argument(
    '-v',
    type=str,
    dest='version',
    default=None,
    help='Specific version (default is all)'
  )
  args = parser.parse_args()
  process(args.target, args.version)