#!/usr/bin/python
import sys, glob, os, subprocess, time, shutil
sys.path.append('src')
from subprocess import check_call as _check_call
from functools import partial
from pyvix.vix import *

PREFIX = "/var/www/vhosts/kovidgoyal.net/subdomains/libprs500"
DOWNLOADS = PREFIX+"/httpdocs/downloads"
DOCS = PREFIX+"/httpdocs/apidocs"
HTML2LRF = "src/libprs500/ebooks/lrf/html/demo"
check_call = partial(_check_call, shell=True)
h = Host(hostType=VIX_SERVICEPROVIDER_VMWARE_WORKSTATION)

def build_windows():
    files = glob.glob('dist/*.exe')
    for file in files:
        os.unlink(file)
    
    
    
    vm = h.openVM('/mnt/extra/vmware/Windows Vista/Windows Vista.vmx')
    vm.powerOn() 
    if not vm.waitForToolsInGuest():
        print >>sys.stderr, 'Windows is not booting up'
        sys.exit(1)
    
    
    
    vm.loginInGuest('kovid', 'et tu brutus')
    vm.loginInGuest(VIX_CONSOLE_USER_NAME, '')
    vm.runProgramInGuest('C:\\Users\kovid\Desktop\libprs500.bat', '')
    vm.runProgramInGuest('C:\Windows\system32\shutdown.exe', '/s')
    
    if not glob.glob('dist/*.exe'):
        raise Exception('Windows build has failed')
    return os.path.basename(glob.glob('dist/*.exe')[-1])

def build_osx():
    files = glob.glob('dist/*.dmg')
    for file in files:
            os.unlink(file)
    
    vm = h.openVM('/mnt/extra/vmware/Mac OSX/Mac OSX.vmx')
    vm.powerOn()
    c = 20 * 60
    print 'Waiting (minutes):',
    while c > 0:
        if glob.glob('dist/*.dmg'):
            break
        time.sleep(10)
        c -= 10
        if c%60==0:
            print c, ',',
    print
        
    if not glob.glob('dist/*.dmg'):
        raise Exception('OSX build has failed')
    vm.powerOff()
    return os.path.basename(glob.glob('dist/*.dmg')[-1])
    


def upload_demo():
    f = open(os.path.join(HTML2LRF, 'demo_ext.html'), 'w')
    f.write("<h2>The HTML</h2><pre>\n")
    f.write(open(os.path.join(HTML2LRF, 'demo.html')).read())
    f.write('\n</pre>')
    f.close()
    check_call('''html2lrf --title='Demonstration of html2lrf' --author='Kovid Goyal' --header --output=/tmp/html2lrf.lrf %s/demo.html'''%(HTML2LRF,))
    check_call('''scp /tmp/html2lrf.lrf castalia:%s/'''%(DOWNLOADS,))

def upload_installers(exe, dmg):
    check_call('''ssh castalia rm -f %s/libprs500\*.exe'''%(DOWNLOADS,))
    check_call('''scp dist/%s castalia:%s/'''%(exe, DOWNLOADS))
    check_call('''ssh castalia rm -f %s/libprs500\*.dmg'''%(DOWNLOADS,))
    check_call('''scp dist/%s castalia:%s/'''%(dmg, DOWNLOADS))
    check_call('''ssh castalia chmod a+r %s/\*'''%(DOWNLOADS,))
    check_call('''ssh castalia /root/bin/update-installer-links %s %s'''%(exe, dmg))

def upload_docs():
    check_call('''epydoc --config epydoc.conf''')
    check_call('''scp -r docs/html castalia:%s/'''%(DOCS,))
    check_call('''epydoc -v --config epydoc-pdf.conf''')
    check_call('''scp docs/pdf/api.pdf castalia:%s/'''%(DOCS,))

    

def main():
    shutil.rmtree('build')
    os.mkdir('build')
    shutil.rmtree('docs')
    os.mkdir('docs')
    check_call("sudo python setup.py develop", shell=True)
    upload_demo()
    print 'Building OSX installer...'
    dmg = build_osx()
    print 'Building Windows installer...'
    exe = build_windows()
    print 'Uploading installers...'
    upload_installers(exe, dmg)
    print 'Uploading to PyPI'
    check_call('''python setup.py register sdist bdist_egg upload''')
    upload_docs()
    check_call('''rm -rf dist/* build/*''')
    
if __name__ == '__main__':
    main()