'''
 ====================================================================
 Copyright (c) 2003-2006 Barry A Scott.  All rights reserved.

 This software is licensed as described in the file LICENSE.txt,
 which you should have received as part of this distribution.

 ====================================================================

    wb_shell_win32_commands.py

'''
import win32process
import win32con
import win32api
import os
import string
import subprocess

def getTerminalProgramList():
    return ['CMD']

def getFileBrowserProgramList():
    return ['Explorer']

def EditFile( app, project_info, filename ):
    p = app.prefs.getEditor()

    if p.editor_image:
        if p.editor_options:
            command_line = '"%s" %s "%s"' % \
                (p.editor_image, p.editor_options, filename)
        else:
            command_line = '"%s" "%s"' % \
                (p.editor_image, filename)
    else:
        command_line = '"notepad.exe" "%s"' % filename

    app.log.info( command_line )
    CreateProcess( app, command_line, project_info.getWorkingDir() )

def GuiDiffFiles( app, options ):
    cmd_line = '"%s" %s' % (app.prefs.getDiffTool().gui_diff_tool, options)
    app.log.info( cmd_line )
    CreateProcess( app, cmd_line, os.path.curdir )

def ShellDiffFiles( app, options ):
    cmd_line = '"%s" %s' % (app.prefs.getDiffTool().shell_diff_tool, options)
    app.log.info( cmd_line )
    return __run_command_with_output( cmd_line )

def ShellOpen( app, project_info, filename ):
    app.log.info( 'Open %s' % filename )
    try:
        win32api.ShellExecute( 0, 'open',
                    filename, '',
                    project_info.getWorkingDir(),
                    win32con.SW_SHOWNORMAL )
    except win32api.error, e:
        if e[0] == 31:
            app.log.error( 'Unable to shell open %s\n'
                 'Is an application associated with this file type?' % filename )
        else:
            app.log.error( 'Unable to shell open %s - %s' % (filename, e[2]) )

def CommandShell( app, project_info ):
    shell_script_filename = os.path.join( os.environ['TEMP'], 'wb_shell_tmp.cmd' )

    p = app.prefs.getShell()
    working_dir = project_info.getWorkingDir()

    # calc a title that is leaf to root so that the leaf shows up in a task bar first
    title = []
    pi = project_info
    while pi:
        title.append( pi.project_name )
        pi = pi.parent

    f  = open( shell_script_filename, 'w' )
    f.write( "@title %s\n" % string.join( title, ' ' ) )
    f.write( "@set PYTHONPATH=\n" )
    f.write( '@cd %s\n' % working_dir )
    f.write( '@echo on\n' )

    if len( p.shell_init_command ) > 0:
        f.write( 'call %s\n' % p.shell_init_command )
    f.close()

    command_line = '"%s" /k %s' % (os.environ['ComSpec'], shell_script_filename)

    app.log.info( command_line )
    CreateProcess( app, command_line, working_dir )

def FileBrowser( app, project_info ):
    command_line = 'explorer.exe /e,%s' % project_info.getWorkingDir()

    app.log.info( command_line )
    CreateProcess( app, command_line, project_info.getWorkingDir() )


def CreateProcess( app, command_line, current_dir ):
    if not ensureDirectory( app, current_dir ):
        return

    try:
        si = win32process.STARTUPINFO()
        h_process, h_thread, process_id, thread_id = win32process.CreateProcess\
            (
            None,
            command_line,
            None,               # processAttributes
            None,               # threadAttributes , 
            0,                  # bInheritHandles ,
            win32con.CREATE_NEW_CONSOLE,    # dwCreationFlags ,
            None,               # newEnvironment ,
            current_dir,        # currentDirectory ,
            si                  # startupinfo
            )
    except win32process.error, detail:
        app.log.error( 'Create process failed for command - %s\n'
                        'Reason %s' % (command_line, detail) )

def ensureDirectory( app, current_dir ):
    if not os.path.exists( current_dir ):
        try:
            os.makedirs( current_dir )
            app.log.info( 'Created directory %s' % current_dir )
        except IOError, detail:
            app.log.error( 'Create directory %s - %s' % (current_dir, detail) )
            return 0

    elif not os.path.isdir( current_dir ):
        app.log.error( '%s is not a directory' % current_dir )
        return 0

    return 1

def __run_command_with_output( command_line ):
    err_prefix = 'error running %s' % command_line
    try:
        proc = subprocess.Popen(
                    command_line,
                    bufsize=-1,
                    stdin=subprocess.PIPE,
                    stdout=subprocess.PIPE,
                    stderr=subprocess.STDOUT
                    )
        all_output = proc.stdout.read()
        proc.wait()

        return all_output

    except EnvironmentError, e:
        return '%s - %s' % (err_prefix, str(e))