import sys, os

# Template used then the program is a GUI program

WINMAINTEMPLATE = """

#include <windows.h>

int WINAPI WinMain(

    HINSTANCE hInstance,      // handle to current instance

    HINSTANCE hPrevInstance,  // handle to previous instance

    LPSTR lpCmdLine,          // pointer to command line

    int nCmdShow              // show state of window

    )

{

    extern int Py_FrozenMain(int, char **);

    PyImport_FrozenModules = _PyImport_FrozenModules;

    return Py_FrozenMain(__argc, __argv);

}

"""

SERVICETEMPLATE = """

extern int PythonService_main(int, char **);

int main( int argc, char **argv)

{

    PyImport_FrozenModules = _PyImport_FrozenModules;

    return PythonService_main(argc, argv);

}

"""

subsystem_details = {

    # -s flag        : (C entry point template), (is it __main__?), (is it a DLL?)

    'console'        : (None,                    1,                 0),

    'windows'        : (WINMAINTEMPLATE,         1,                 0),

    'service'        : (SERVICETEMPLATE,         0,                 0),

    'com_dll'        : ("",                      0,                 1),

}

def get_custom_entry_point(subsystem):

    try:

        return subsystem_details[subsystem][:2]

    except KeyError:

        raise ValueError, "The subsystem %s is not known" % subsystem

def makemakefile(outfp, vars, files, target):

    save = sys.stdout

    try:

        sys.stdout = outfp

        realwork(vars, files, target)

    finally:

        sys.stdout = save

def realwork(vars, moddefns, target):

    version_suffix = "%r%r" % sys.version_info[:2]

    print "# Makefile for Microsoft Visual C++ generated by freeze.py script"

    print

    print 'target = %s' % target

    print 'pythonhome = %s' % vars['prefix']

    print

    print 'DEBUG=0 # Set to 1 to use the _d versions of Python.'

    print '!IF $(DEBUG)'

    print 'debug_suffix=_d'

    print 'c_debug=/Zi /Od /DDEBUG /D_DEBUG'

    print 'l_debug=/DEBUG'

    print 'temp_dir=Build\\Debug'

    print '!ELSE'

    print 'debug_suffix='

    print 'c_debug=/Ox'

    print 'l_debug='

    print 'temp_dir=Build\\Release'

    print '!ENDIF'

    print

    print '# The following line assumes you have built Python using the standard instructions'

    print '# Otherwise fix the following line to point to the library.'

    print 'pythonlib = "$(pythonhome)/pcbuild/python%s$(debug_suffix).lib"' % version_suffix

    print

    # We only ever write one "entry point" symbol - either

    # "main" or "WinMain".  Therefore, there is no need to

    # pass a subsystem switch to the linker as it works it

    # out all by itself.  However, the subsystem _does_ determine

    # the file extension and additional linker flags.

    target_link_flags = ""

    target_ext = ".exe"

    if subsystem_details[vars['subsystem']][2]:

        target_link_flags = "-dll"

        target_ext = ".dll"

    print "# As the target uses Python%s.dll, we must use this compiler option!" % version_suffix

    print "cdl = /MD"

    print

    print "all: $(target)$(debug_suffix)%s" % (target_ext)

    print

    print '$(temp_dir):'

    print '  if not exist $(temp_dir)\. mkdir $(temp_dir)'

    print

    objects = []

    libs = ["shell32.lib", "comdlg32.lib", "wsock32.lib", "user32.lib", "oleaut32.lib"]

    for moddefn in moddefns:

        print "# Module", moddefn.name

        for file in moddefn.sourceFiles:

            base = os.path.basename(file)

            base, ext = os.path.splitext(base)

            objects.append(base + ".obj")

            print '$(temp_dir)\%s.obj: "%s"' % (base, file)

            print "\t@$(CC) -c -nologo /Fo$* $(cdl) $(c_debug) /D BUILD_FREEZE",

            print '"-I$(pythonhome)/Include"  "-I$(pythonhome)/PC" \\'

            print "\t\t$(cflags) $(cdebug) $(cinclude) \\"

            extra = moddefn.GetCompilerOptions()

            if extra:

                print "\t\t%s \\" % (' '.join(extra),)

            print '\t\t"%s"' % file

            print

        # Add .lib files this module needs

        for modlib in moddefn.GetLinkerLibs():

            if modlib not in libs:

                libs.append(modlib)

    print "ADDN_LINK_FILES=",

    for addn in vars['addn_link']: print '"%s"' % (addn),

    print ; print

    print "OBJS=",

    for obj in objects: print '"$(temp_dir)\%s"' % (obj),

    print ; print

    print "LIBS=",

    for lib in libs: print '"%s"' % (lib),

    print ; print

    print "$(target)$(debug_suffix)%s: $(temp_dir) $(OBJS)" % (target_ext)

    print "\tlink -out:$(target)$(debug_suffix)%s %s" % (target_ext, target_link_flags),

    print "\t$(OBJS) \\"

    print "\t$(LIBS) \\"

    print "\t$(ADDN_LINK_FILES) \\"

    print "\t$(pythonlib) $(lcustom) $(l_debug)\\"

    print "\t$(resources)"

    print

    print "clean:"

    print "\t-rm -f *.obj"

    print "\t-rm -f $(target).exe"