""" This module is used to build a graphical installer
from a p3d file. It will try to build installers for as
many platforms as possible. """

__all__ = ["InstallerMaker"]

import os, sys, subprocess, tarfile, shutil, time
from direct.directnotify.DirectNotifyGlobal import *
from pandac.PandaModules import Filename

class CachedFile:
    def __init__(self): self.str = ""
    def write(self, data): self.str += data

class InstallerMaker:
    notify = directNotify.newCategory("InstallerMaker")

    def __init__(self, shortname, fullname, p3dfile):
        self.shortname = shortname
        self.fullname = fullname
        self.p3dfile = p3dfile
        self.version = "1"

    def build(self):
        """ Creates the installer. Call this after you have set all the parameters. """
        self.__buildDEB()
        self.__buildNSIS()

    def __buildDEB(self):
        InstallerMaker.notify.info("Creating %s.deb..." % self.shortname)

        # Create a temporary directory and write the control file + launcher to it
        tempdir = Filename.temporary("", self.shortname + "_deb_", "") + "/"
        tempdir.makeDir()
        tempdir = tempdir.toOsSpecific()
        controlfile = open(os.path.join(tempdir, "control"), "w")
        controlfile.write("Package: %s\n" % self.shortname)
        controlfile.write("Version: %s\n" % self.version)
        controlfile.write("Section: games\n")
        controlfile.write("Priority: optional\n")
        controlfile.write("Architecture: all\n")
        controlfile.write("Depends: panda3d-runtime\n")
        controlfile.write("Description: %s\n" % self.fullname)
        controlfile.close()
        os.mkdir(os.path.join(tempdir, "usr"))
        os.mkdir(os.path.join(tempdir, "usr", "bin"))
        os.mkdir(os.path.join(tempdir, "usr", "share"))
        os.mkdir(os.path.join(tempdir, "usr", "share", "games"))
        launcherfile = open(os.path.join(tempdir, "usr", "bin", self.shortname), "w")
        launcherfile.write("#!/bin/sh\n")
        launcherfile.write("/usr/bin/panda3d /usr/share/games/%s/data.p3d\n" % self.shortname)
        launcherfile.close()
        shutil.copyfile(self.p3dfile, os.path.join(tempdir, "usr", "share", "games", "data.p3d"))

        # Create a control.tar.gz file in memory
        controltargz = CachedFile()
        controltarfile = tarfile.TarFile.gzopen("control.tar.gz", "w", controltargz)
        controltarfile.add(os.path.join(tempdir, "control"), "control")
        controltarfile.close()
        os.remove(os.path.join(tempdir, "control"))

        # Create the data.tar.gz file in the temporary directory
        datatarfile = tarfile.TarFile.gzopen(os.path.join(tempdir, "data.tar.gz"), "w")
        datatarfile.add(tempdir + "/usr", "/usr")
        datatarfile.close()

        # Open the deb file and write to it. It's actually
        # just an AR file, which is very easy to make.
        modtime = str(int(time.time())).ljust(11)
        if os.path.isfile(self.shortname + ".deb"):
            os.remove(self.shortname + ".deb")
        debfile = open(self.shortname + ".deb", "wb")
        debfile.write("!<arch>\x0A")
        debfile.write("debian-binary   %s 0     0     100644  4         \x60\x0A" % modtime)
        debfile.write("2.0\x0A")
        debfile.write("control.tar.gz  %s 0     0     100644  %s \x60\x0A" % (modtime, str(len(controltargz.str)).ljust(9)))
        debfile.write(controltargz.str)
        debfile.write("data.tar.gz     %s 0     0     100644  %s \x60\x0A" % (modtime, str(os.path.getsize(os.path.join(tempdir, "data.tar.gz"))).ljust(9)))

        # Copy everything from data.tar.gz to the deb file megabyte by megabyte.
        datatargz = open(os.path.join(tempdir, "data.tar.gz"), "rb")
        data = datatargz.read(1024 * 1024)
        while data != "":
            debfile.write(data)
            data = datatargz.read(1024 * 1024)
        datatargz.close()
        debfile.close()
        shutil.rmtree(tempdir)

    def __buildNSIS(self):
        # Check if we have makensis first
        makensis = None
        if (sys.platform.startswith("win")):
            for p in os.defpath.split(";") + os.environ["PATH"].split(";"):
                if os.path.isfile(os.path.join(p, "makensis.exe")):
                    makensis = os.path.join(p, "makensis.exe")
            if not makensis:
                import pandac
                makensis = os.path.dirname(os.path.dirname(pandac.__file__))
                makensis = os.path.join(makensis, "nsis", "makensis.exe")
                if not os.path.isfile(makensis): makensis = None
        else:
            for p in os.defpath.split(":") + os.environ["PATH"].split(":"):
                if os.path.isfile(os.path.join(p, "makensis")):
                    makensis = os.path.join(p, "makensis")
        
        if makensis == None:
            InstallerMaker.notify.warning("Makensis utility not found, no Windows installer will be built!")
            return
        InstallerMaker.notify.info("Creating %s.exe..." % self.shortname)

        tempfile = self.shortname + ".nsi"
        nsi = open(tempfile, "w")

        # Some global info
        nsi.write('Name "%s"\n' % self.fullname)
        nsi.write('OutFile "%s.exe"\n' % self.shortname)
        nsi.write('InstallDir "$PROGRAMFILES\%s"\n' % self.fullname)
        nsi.write('SetCompress auto\n')
        nsi.write('SetCompressor lzma\n')
        nsi.write('ShowInstDetails nevershow\n')
        nsi.write('ShowUninstDetails nevershow\n')
        nsi.write('InstType "Typical"\n')

        # Tell Vista that we require admin rights
        nsi.write('RequestExecutionLevel admin\n')
        nsi.write('\n')
        nsi.write('Function launch\n')
        nsi.write('  ExecShell "open" "$INSTDIR\%s.bat"\n' % self.shortname)
        nsi.write('FunctionEnd\n')
        nsi.write('\n')
        nsi.write('!include "MUI2.nsh"\n')
        nsi.write('!define MUI_ABORTWARNING\n')
        nsi.write('!define MUI_FINISHPAGE_RUN\n')
        nsi.write('!define MUI_FINISHPAGE_RUN_FUNCTION launch\n')
        nsi.write('!define MUI_FINISHPAGE_RUN_TEXT "Play %s"\n' % self.fullname)
        nsi.write('\n')
        nsi.write('Var StartMenuFolder\n')
        nsi.write('!insertmacro MUI_PAGE_WELCOME\n')
        nsi.write('!insertmacro MUI_PAGE_DIRECTORY\n')
        nsi.write('!insertmacro MUI_PAGE_STARTMENU Application $StartMenuFolder\n')
        nsi.write('!insertmacro MUI_PAGE_INSTFILES\n')
        nsi.write('!insertmacro MUI_PAGE_FINISH\n')
        nsi.write('!insertmacro MUI_UNPAGE_WELCOME\n')
        nsi.write('!insertmacro MUI_UNPAGE_CONFIRM\n')
        nsi.write('!insertmacro MUI_UNPAGE_INSTFILES\n')
        nsi.write('!insertmacro MUI_UNPAGE_FINISH\n')
        nsi.write('!insertmacro MUI_LANGUAGE "English"\n')

        # This section defines the installer.
        nsi.write('Section "Install"\n')
        nsi.write('  SetOutPath "$INSTDIR"\n')
        nsi.write('  WriteUninstaller "$INSTDIR\Uninstall.exe"\n')
        nsi.write('  ; Start menu items\n')
        nsi.write('  !insertmacro MUI_STARTMENU_WRITE_BEGIN Application\n')
        nsi.write('    CreateDirectory "$SMPROGRAMS\$StartMenuFolder"\n')
        nsi.write('    CreateShortCut "$SMPROGRAMS\$StartMenuFolder\Uninstall.lnk" "$INSTDIR\Uninstall.exe"\n')
        nsi.write('  !insertmacro MUI_STARTMENU_WRITE_END\n')
        nsi.write('SectionEnd\n')

        # This section defines the uninstaller.
        nsi.write('Section "Uninstall"\n')
        nsi.write('  Delete "$INSTDIR\Uninstall.exe"\n')
        nsi.write('  RMDir "$INSTDIR"\n')
        nsi.write('  ; Start menu items\n')
        nsi.write('  !insertmacro MUI_STARTMENU_GETFOLDER Application $StartMenuFolder\n')
        nsi.write('  Delete "$SMPROGRAMS\$StartMenuFolder\Uninstall.lnk"\n')
        nsi.write('  RMDir "$SMPROGRAMS\$StartMenuFolder"\n')
        nsi.write('SectionEnd')
        nsi.close()

        options = ["V2"]
        cmd = makensis
        for o in options:
            if sys.platform.startswith("win"):
                cmd += " /" + o
            else:
                cmd += " -" + o
        cmd += " " + tempfile
        os.system(cmd)

        os.remove(tempfile)