//+-------------------------------------------------------------------------
//
//  Microsoft Windows
//  Copyright (C) Microsoft Corporation, 1997.
//
//  File:       cmdline.cpp
//
//  Contents:   commandline helper routines.
//
//  Classes:    
//
//  Notes:      
//
//  History:    17-Nov-97   rogerg      Created.
//
//--------------------------------------------------------------------------

#include "precomp.h"


//+---------------------------------------------------------------------------
//
//  Member:     CCmdLine::CCmdLine, public
//
//  Synopsis:  Constructor.
//
//  Arguments:  
//
//  Returns:    
//
//  Modifies:   
//
//  History:    17-Nov-97       rogerg        Created.
//
//----------------------------------------------------------------------------

CCmdLine::CCmdLine()
{
    m_cmdLineFlags = 0;
    m_pszJobFile = NULL;
}


//+---------------------------------------------------------------------------
//
//  Member:     CCmdLine::~CCmdLine, public
//
//  Synopsis:  destructor.
//
//  Arguments:  
//
//  Returns:    
//
//  Modifies:   
//
//  History:    07-Jul-98       rogerg        Created.
//
//----------------------------------------------------------------------------

CCmdLine::~CCmdLine()
{
    if (m_pszJobFile)
    {
        FREE(m_pszJobFile);
    }
}


//+---------------------------------------------------------------------------
//
//  Member:     CCmdLine::ParseCommandLine, public
//
//  Synopsis:   Parses the command line passed to the Application
//		and Sets the member variables accordingly.
//
//  Arguments:  
//
//  Returns:    
//
//  Modifies:   
//
//  History:    17-Nov-97       rogerg        Created.
//
//----------------------------------------------------------------------------

void CCmdLine::ParseCommandLine()
{
    char *lpszRemaining;

    m_cmdLineFlags = 0;

    // start at 1 -- the first is the exe
    for (int i=1; i< __argc; i++)
    {
 	    if (MatchOption(__argv[i], "Embedding"))
    	{
 	        m_cmdLineFlags |= CMDLINE_COMMAND_EMBEDDING;
    	}
 	    else if (MatchOption(__argv[i],"Register"))
    	{
 	        m_cmdLineFlags |= CMDLINE_COMMAND_REGISTER;
    	}
 	    else if (MatchOption(__argv[i], "logon") )
	    {
 	        m_cmdLineFlags |= CMDLINE_COMMAND_LOGON;
	    }
 	    else if (MatchOption(__argv[i],"logoff") )
	    {
 	        m_cmdLineFlags |= CMDLINE_COMMAND_LOGOFF;
	    }
 	    else if (MatchOption(__argv[i],"DllRegisterServer") )
	    {
 	        m_cmdLineFlags |= CMDLINE_COMMAND_REGISTER;
	    }
 	    else if (MatchOption(__argv[i],"Idle") )
	    {
            // pretend idle is the idle schedule firing for this 
            // user so same code path is exceriside if command
            // line invoked or TS launched us.
            
            m_cmdLineFlags |= CMDLINE_COMMAND_SCHEDULE;

            // m_pszJobFile will be scheduleguid_UserName

            TCHAR szDomainUserName[MAX_DOMANDANDMACHINENAMESIZE];
            GetDefaultDomainAndUserName(szDomainUserName,TEXT("_"), MAX_DOMANDANDMACHINENAMESIZE);

            DWORD dwJobFileLen = ( lstrlen(WSZGUID_IDLESCHEDULE)            /* guid of schedule */
                                        + 1                                /* space for separator */
                                        + lstrlen(szDomainUserName)          /* space for domainUserNmae */
                                        + 1                                /* space for NULL */
                                 );

            m_pszJobFile = (WCHAR *) ALLOC(dwJobFileLen * sizeof(WCHAR));

            if (m_pszJobFile)
            {
                StringCchCopy(m_pszJobFile, dwJobFileLen, WSZGUID_IDLESCHEDULE);
                StringCchCat(m_pszJobFile, dwJobFileLen, TEXT("_"));
                StringCchCat(m_pszJobFile, dwJobFileLen, szDomainUserName);            
            }
        }
 	    else if (MatchOption(__argv[i], "Schedule=", FALSE, &lpszRemaining))
	    {
            m_cmdLineFlags |= CMDLINE_COMMAND_SCHEDULE;
           
            // command lines are always in ANSI so convert jobname to WCHAR
            
            ULONG ulJobFileSize = lstrlenA(lpszRemaining) + 1;
            m_pszJobFile = (WCHAR *) ALLOC(ulJobFileSize*sizeof(WCHAR));

            // UP to Schedule method on invoke to handle if jobfile is null.
            if (m_pszJobFile)
            {
                MultiByteToWideChar(CP_ACP, 0, lpszRemaining, -1, m_pszJobFile, ulJobFileSize);
            }
	    }
	    else
	    {
	        AssertSz( i== 0, "Unknown Command Line"); // unknown command line
	    }
    }
}


//+---------------------------------------------------------------------------
//
//  Member:     CCmdLine::MatchOption, private
//
//  Synopsis:   given a command line and an option determines if the command
//		line matches the option up until the end of the option text
//		if there is additional text after the option text a pointer
//		to it will be returned in lpRemaining, else lpRemaining wil
//		be set to NULL.
//
//		!!!CmdLine options always come in in ANSI
//
//  Arguments:  [lpsz] - command line value to match
//		[lpszOption] - Option string to match command line too.
//		[fExactMatch] - when true, the command line must contain the same
//				number of characters as the Option (i.e.) There shouldn't
//				be any remaining characters
//		[lpszRemaining] - Points to any remaining part of the command after the match
//
//  Returns:    
//
//  Modifies:   
//
//  History:    17-Nov-97       rogerg        Created.
//
//----------------------------------------------------------------------------

BOOL CCmdLine::MatchOption(LPSTR lpsz, LPSTR lpszOption,BOOL fExactMatch,LPSTR *lpszRemaining)
{
    if (lpszRemaining)
	    *lpszRemaining = '\0';

    if ( lpsz && (lpsz[0] == TEXT('-') || lpsz[0] == TEXT('/')) )
    {
        int nRet = 0;

        lpsz++;

        while (! (nRet = toupper(*lpsz) - toupper(*lpszOption)) && *lpsz)
        {
	        lpsz++;
	        lpszOption++;
        }

        if (*lpszOption || (*lpsz && fExactMatch) )
	        return FALSE;

        if (lpszRemaining)
	        *lpszRemaining = lpsz;

        return(TRUE);
    }

    return FALSE;
}


//+---------------------------------------------------------------------------
//
//  Member:     CCmdLine::MatchOption, private
//
//  Synopsis:   given a command line and an option determines if the command
//		line matches exactly matches the option 
//
//  Arguments:  [lpsz] - command line value to match
//		[lpszOption] - Option string to match command line too.
//
//  Returns:    
//
//  Modifies:   
//
//  History:    17-Nov-97       rogerg        Created.
//
//----------------------------------------------------------------------------

BOOL CCmdLine::MatchOption(LPSTR lpsz, LPSTR lpszOption)
{
    return MatchOption(lpsz, lpszOption, TRUE /* fExactmatch */, NULL);
}