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) from None


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(r' 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(r'$(temp_dir)\%s.obj: "%s"' % (base, file))
 print("\t@$(CC) -c -nologo /Fo$* $(cdl) $(c_debug) /D BUILD_FREEZE", end=' ')
 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=", end=' ')
 for addn in vars['addn_link']: print('"%s"' % (addn), end=' ')
 print() ; print()

 print("OBJS=", end=' ')
 for obj in objects: print(r'"$(temp_dir)\%s"' % (obj), end=' ')
 print() ; print()

 print("LIBS=", end=' ')
 for lib in libs: print('"%s"' % (lib), end=' ')
 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()
 print("clean:")
 print("\t-del /f *.obj")
 print("\t-del /f $(target).exe")