import os
import sys
import platform
import zipfile
import shutil
from optparse import OptionParser

parser = OptionParser(usage='cctbx_upload.py [options]')
parser.add_option('--dest',
		  dest='dest',
                  action='store',
		  help='specifies where the upload goes to - [release], dev, next')
option, args = parser.parse_args()

if option.dest == 'trunk':
  destination = 'distro@10.8.0.1:/var/distro/bin_dir_trunk/'
else:
  destination = 'distro@10.8.0.1:/var/distro/bin_dir-1.5/'

tmp_folder = '~/tmp/cctbx/'
if sys.platform[:3] == 'win':
  upload_cmd = r'"C:\Program Files\PuTTY\pscp.exe"'
else:
  upload_cmd = 'scp'

build_def = {
  'linux-64bit':
#    (('/mnt/hgfs/cctbx/cctbx_latest/build_lin64/', '/mnt/hgfs/cctbx/cctbx_latest/modules/cctbx_project/',
#      '/tmp/cctbx/', 'cctbx-linux64.zip', '-j4'),),
  (('/mnt/devel/cctbx/cctbx_latest/build_lin64_py38/', '/mnt/devel/cctbx/cctbx_latest/modules/cctbx_project/',
    '/tmp/cctbx/', 'cctbx-linux64.zip', '-j6'),),
'darwin-64bit':
  (('~/build/svn/cctbx/build_mac64_py38/', '~/build/svn/cctbx/modules/cctbx_project/',
    '/tmp/cctbx/', 'cctbx-mac64.zip', '-j3'),),

'win32-32bit':
  # [('e:/cctbx/cctbx_latest/build_win32/', 'e:/cctbx/cctbx_latest/modules/cctbx_project/',
  #   'e:/tmp/cctbx/', 'cctbx-win32-sse2.zip', '-j6')],
  [('d:/devel/cctbx/cctbx_latest/build_win32_py38/', 'd:/devel/cctbx/cctbx_latest/modules/cctbx_project/',
    'd:/tmp/cctbx/', 'cctbx-win32-sse2.zip', '-j10')],
'win32-64bit':
  #[('e:/cctbx/cctbx_latest/build_win64/', 'e:/cctbx/cctbx_latest/modules/cctbx_project/',
  #  'e:/tmp/cctbx/', 'cctbx-win64.zip', '-j6')],
  [('d:/devel/cctbx/cctbx_latest/build_win64_py38/', 'd:/devel/cctbx/cctbx_latest/modules/cctbx_project/',
    'd:/tmp/cctbx/', 'cctbx-win64.zip', '-j10')],
}

if option.dest == 'next':
  pass
# build ALL win on win64
build_def['win32-64bit'] += build_def['win32-32bit']

def compile(_platform):
  cwd = os.getcwd()
  zip_names = []
  try:
    bd = build_def[_platform]
    for d in bd:
      build_dir = os.path.expanduser(d[0])
      src_dir = os.path.expanduser(d[1])
      tmp_dir = os.path.expanduser(d[2])
      bin_dir = build_dir+'bin/'
      os.chdir(build_dir)
      if os.system(bin_dir + 'libtbx.scons ' + d[4]) != 0:
        print('Scons returned non zero...')
        return False
      bundle_dir = tmp_dir + 'bundle'
      if os.path.exists(bundle_dir):
        shutil.rmtree(bundle_dir)
      os.mkdir(bundle_dir)
      os.chdir(bundle_dir)
      if os.system(bin_dir+'libtbx.python ' + src_dir + 'libtbx/bundle/copy_all.py cctbx') != 0:
        print('Failed to create a distribution...')
        return False

      zip_name = tmp_dir + d[3]
      print('Creating ' +  zip_name + ':')
      root_dir_len = len(bundle_dir.replace('\\', '/') + '/')
      dest_zip = zipfile.ZipFile(zip_name,
                                mode='w', compression=zipfile.ZIP_DEFLATED)
      for dir_path, dir_names, file_names in os.walk(bundle_dir):
        dir_path = dir_path.replace('\\', '/')
        for f in file_names:
          dest_zip.write(dir_path+'/'+f, 'cctbx/' + dir_path[root_dir_len:] + '/' + f)
      dest_zip.close()
      os.chdir(cwd)
      zip_names.append(zip_name)
      shutil.rmtree(bundle_dir)
    return zip_names
  finally:
    os.chdir(cwd)

if __name__ == '__main__':
  plat = sys.platform + '-' + platform.architecture()[0]
  print('Compiling for: ' + plat)
  zip_names = compile(plat)
  if zip_names == False or len(zip_names) == 0:
    print('Compilation failed. Aborting...')
    sys.exit(1)

  up_str = upload_cmd
  for f in zip_names: up_str += ' ' + f
  up_str = up_str +  ' ' + destination
  print('Uploading; ' + up_str)
  sys.exit(0)
  if os.system(up_str) != 0:
    print('Upload has failed...')
  else:
    print('Done...')