// Script.Cpp : implementation file
//

#include "StdAfx.H"
#include "TestCon.H"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CScript

IMPLEMENT_DYNAMIC( CScript, CCmdTarget )

CScript::CScript( CScriptManager* pManager ) :
   m_pManager( pManager )
{
   ASSERT( m_pManager != NULL );
}

CScript::~CScript()
{
   Unload();
}

HRESULT CScript::AddNamedItem( LPCTSTR pszItemName )
{
   USES_CONVERSION;
   LPCOLESTR pszItemNameO;
   HRESULT hResult;

   pszItemNameO = T2COLE( pszItemName );

   hResult = m_pActiveScript->AddNamedItem( pszItemNameO,
	  SCRIPTITEM_ISSOURCE|SCRIPTITEM_ISVISIBLE );
   if( FAILED( hResult ) )
   {
	  return( hResult );
   }

   return( S_OK );
}

BOOL CScript::FindMacro( LPCTSTR pszMacroName )
{
   DISPID dispid;

   ASSERT( pszMacroName != NULL );

   return( m_mapMacros.Lookup( pszMacroName, dispid ) );
}

POSITION CScript::GetFirstMacroPosition()
{
   return( m_mapMacros.GetStartPosition() );
}

CString CScript::GetName()
{
   return( m_strScriptName );
}

CString CScript::GetNextMacroName( POSITION& posMacro )
{
   CString strMacro;
   DISPID dispid;

   m_mapMacros.GetNextAssoc( posMacro, strMacro, dispid );

   return( strMacro );
}

HRESULT CScript::LoadScript( LPCTSTR pszFileName, LPCTSTR pszScriptName )
{
   ITypeInfoPtr pTypeInfo;
   HRESULT hResult;
   CLSID clsid;
   CFile file;
   ULONG nFileSize;
   BSTR bstrScript;
   BOOL tSuccess;
   int iMethod;
   UINT nNames;
   BSTR bstrName;
   CString strName;
   CString strExtension;
   int iChar;

   ASSERT( pszFileName != NULL );
   ASSERT( pszScriptName != NULL );
   ASSERT( m_pActiveScript == NULL );

   iChar = lstrlen( pszFileName )-1;
   while( (iChar >= 0) && (pszFileName[iChar] != _T( '.' )) )
   {
	  iChar--;
   }

   if( (iChar >= 0) && (lstrcmpi( &pszFileName[iChar], _T( ".js" ) ) == 0) )
   {
	  hResult = CLSIDFromProgID( L"JScript", &clsid );
	  if( FAILED( hResult ) )
	  {
		 return( hResult );
	  }
   }
   else
   {
	  hResult = CLSIDFromProgID( L"VBScript", &clsid );
	  if( FAILED( hResult ) )
	  {
		 return( hResult );
	  }
   }

   hResult = m_pActiveScript.CreateInstance( clsid, NULL,
	  CLSCTX_INPROC_SERVER );
   if( FAILED( hResult ) )
   {
	  return( hResult );
   }

   m_pActiveScriptParse = m_pActiveScript;
   if( m_pActiveScriptParse == NULL )
   {
	  return( E_NOINTERFACE );
   }

   hResult = m_pActiveScript->SetScriptSite( &m_xActiveScriptSite );
   if( FAILED( hResult ) )
   {
	  return( hResult );
   }

   hResult = m_pActiveScriptParse->InitNew();
   if( FAILED( hResult ) )
   {
	  return( hResult );
   }

   CFileException error;
   tSuccess = file.Open( pszFileName, CFile::modeRead|CFile::shareDenyWrite,
	  &error );
   if( !tSuccess )
   {
	  return( HRESULT_FROM_WIN32( error.m_lOsError ) );
   }

   nFileSize = file.GetLength();
   nFileSize = file.Read( m_strScriptText.GetBuffer( nFileSize ), nFileSize );
   file.Close();
   m_strScriptText.ReleaseBuffer( nFileSize );
   bstrScript = m_strScriptText.AllocSysString();

   hResult = m_pActiveScriptParse->ParseScriptText( bstrScript, NULL, NULL,
	  NULL, DWORD( this ), 0, SCRIPTTEXT_ISVISIBLE, NULL, NULL );
   SysFreeString( bstrScript );
   if( FAILED( hResult ) )
   {
	  return( hResult );
   }

   hResult = m_pManager->AddNamedItems( m_pActiveScript );
   if( FAILED( hResult ) )
   {
	  return( hResult );
   }

   hResult = m_pActiveScript->SetScriptState( SCRIPTSTATE_CONNECTED );
   if( FAILED( hResult ) )
   {
	  return( hResult );
   }

   hResult = m_pActiveScript->GetScriptDispatch( NULL, &m_pDispatch );
   if( FAILED( hResult ) )
   {
	  return( hResult );
   }

   hResult = m_pDispatch->GetTypeInfo( 0, GetUserDefaultLCID(), &pTypeInfo );
   if( FAILED( hResult ) )
   {
	  return( hResult );
   }

   CSmartTypeAttr pTypeAttr( pTypeInfo );
   CSmartFuncDesc pFuncDesc( pTypeInfo );

   hResult = pTypeInfo->GetTypeAttr( &pTypeAttr );
   if( FAILED( hResult ) )
   {
	  return( hResult );
   }

   for( iMethod = 0; iMethod < pTypeAttr->cFuncs; iMethod++ )
   {
	  hResult = pTypeInfo->GetFuncDesc( iMethod, &pFuncDesc );
	  if( FAILED( hResult ) )
	  {
		 return( hResult );
	  }

	  if( (pFuncDesc->funckind == FUNC_DISPATCH) && (pFuncDesc->invkind ==
		 INVOKE_FUNC) && (pFuncDesc->cParams == 0) )
	  {
		 bstrName = NULL;
		 hResult = pTypeInfo->GetNames( pFuncDesc->memid, &bstrName, 1,
			&nNames );
		 if( FAILED( hResult ) )
		 {
			return( hResult );
		 }
		 ASSERT( nNames == 1 );

		 strName = bstrName;
		 SysFreeString( bstrName );
		 bstrName = NULL;

		 // Macros can't contain underscores, since those denote event handlers
		 if( strName.Find( _T( '_' ) ) == -1 )
		 {
			m_mapMacros.SetAt( strName, pFuncDesc->memid );
		 }
	  }

	  pFuncDesc.Release();
   }

   m_strScriptName = pszScriptName;

   return( S_OK );
}

HRESULT CScript::RunMacro( LPCTSTR pszMacroName )
{
   DISPID dispid;
   COleDispatchDriver driver;
   BOOL tFound;

   ASSERT( pszMacroName != NULL );

   tFound = m_mapMacros.Lookup( pszMacroName, dispid );
   if( !tFound )
   {
	  return( DISP_E_MEMBERNOTFOUND );
   }

   driver.AttachDispatch( m_pDispatch, FALSE );
   try
   {
	  driver.InvokeHelper( dispid, DISPATCH_METHOD, VT_EMPTY, NULL, VTS_NONE );
   }
   catch( COleDispatchException* pException )
   {
	  CString strMessage;

	  AfxFormatString1( strMessage, IDS_DISPATCHEXCEPTION,
		 pException->m_strDescription );
	  AfxMessageBox( strMessage );
	  pException->Delete();
   }
   catch( COleException* pException )
   {
	  pException->Delete();
   }

   return( S_OK );
}

void CScript::Unload()
{
   if( m_pDispatch != NULL )
   {
	  m_pDispatch.Release();
   }
   if( m_pActiveScript != NULL )
   {
	  m_pActiveScript->SetScriptState( SCRIPTSTATE_DISCONNECTED );
	  m_pActiveScript->Close();
	  m_pActiveScript.Release();
   }
   if( m_pActiveScriptParse != NULL )
   {
	  m_pActiveScriptParse.Release();
   }
}


BEGIN_INTERFACE_MAP( CScript, CCmdTarget )
   INTERFACE_PART( CScript, IID_IActiveScriptSite, ActiveScriptSite )
   INTERFACE_PART( CScript, IID_IActiveScriptSiteWindow, ActiveScriptSiteWindow )
END_INTERFACE_MAP()

BEGIN_MESSAGE_MAP(CScript, CCmdTarget)
	//{{AFX_MSG_MAP(CScript)
		// NOTE - the ClassWizard will add and remove mapping macros here.
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CScript message handlers


///////////////////////////////////////////////////////////////////////////////
// IActiveScriptSite
///////////////////////////////////////////////////////////////////////////////

STDMETHODIMP_( ULONG ) CScript::XActiveScriptSite::AddRef()
{
   METHOD_PROLOGUE_( CScript, ActiveScriptSite )

   TRACE( "XActiveScriptSite::AddRef()\n" );

   return( pThis->ExternalAddRef() );
}

STDMETHODIMP_( ULONG ) CScript::XActiveScriptSite::Release()
{
   METHOD_PROLOGUE_( CScript, ActiveScriptSite )

   TRACE( "XActiveScriptSite::Release()\n" );

   return( pThis->ExternalRelease() );
}

STDMETHODIMP CScript::XActiveScriptSite::QueryInterface( REFIID iid,
   void** ppInterface )
{
   METHOD_PROLOGUE_( CScript, ActiveScriptSite )

   return( pThis->ExternalQueryInterface( &iid, ppInterface ) );
}

STDMETHODIMP CScript::XActiveScriptSite::GetDocVersionString(
   BSTR* pbstrVersion )
{
   (void)pbstrVersion;

   METHOD_PROLOGUE( CScript, ActiveScriptSite )

   return( E_NOTIMPL );
}

STDMETHODIMP CScript::XActiveScriptSite::GetItemInfo( LPCOLESTR pszName,
   DWORD dwReturnMask, IUnknown** ppItem, ITypeInfo** ppTypeInfo )
{
   USES_CONVERSION;
   LPCTSTR pszNameT;
   HRESULT hResult;

   METHOD_PROLOGUE( CScript, ActiveScriptSite )

   if( ppItem != NULL )
   {
	  *ppItem = NULL;
   }
   if( ppTypeInfo != NULL )
   {
	  *ppTypeInfo = NULL;
   }

   if( dwReturnMask&SCRIPTINFO_IUNKNOWN )
   {
	  if( ppItem == NULL )
	  {
		 return( E_POINTER );
	  }
   }
   if( dwReturnMask&SCRIPTINFO_ITYPEINFO )
   {
	  if( ppTypeInfo == NULL )
	  {
		 return( E_POINTER );
	  }
   }

   pszNameT = OLE2CT( pszName );
   TRACE( "XActiveScriptSite::GetItemInfo( %s )\n", pszNameT );

   if( dwReturnMask&SCRIPTINFO_IUNKNOWN )
   {
	  hResult = pThis->m_pManager->GetItemDispatch( pszNameT, ppItem );
	  if( FAILED( hResult ) )
	  {
		 return( hResult );
	  }
   }
   if( dwReturnMask&SCRIPTINFO_ITYPEINFO )
   {
	  pThis->m_pManager->GetItemTypeInfo( pszNameT, ppTypeInfo );
   }

   return( S_OK );
}

STDMETHODIMP CScript::XActiveScriptSite::GetLCID( LCID* plcid )
{
   METHOD_PROLOGUE( CScript, ActiveScriptSite )

   if( plcid == NULL )
   {
	  return( E_POINTER );
   }

   *plcid = GetUserDefaultLCID();

   return( S_OK );
}

STDMETHODIMP CScript::XActiveScriptSite::OnEnterScript()
{
   METHOD_PROLOGUE( CScript, ActiveScriptSite )

   return( S_OK );
}

STDMETHODIMP CScript::XActiveScriptSite::OnLeaveScript()
{
   METHOD_PROLOGUE( CScript, ActiveScriptSite )

   return( S_OK );
}

STDMETHODIMP CScript::XActiveScriptSite::OnScriptError(
   IActiveScriptError* pError )
{
   USES_CONVERSION;
   HRESULT hResult;
   CExcepInfo excep;
   CString strMessage;
   int nResult;
   DWORD dwContext;
   ULONG iLine;
   LONG iChar;
   BSTR bstrSourceLineText;

   METHOD_PROLOGUE( CScript, ActiveScriptSite )

   TRACE( "XActiveScriptSite::OnScriptError()\n" );

   ASSERT( pError != NULL );

   hResult = pError->GetSourcePosition( &dwContext, &iLine, &iChar );
   if( SUCCEEDED( hResult ) )
   {
	  TRACE( "Error at line %u, character %d\n", iLine, iChar );
   }
   bstrSourceLineText = NULL;
   hResult = pError->GetSourceLineText( &bstrSourceLineText );
   if( SUCCEEDED( hResult ) )
   {
	  TRACE( "Source Text: %s\n", OLE2CT( bstrSourceLineText ) );
	  SysFreeString( bstrSourceLineText );
	  bstrSourceLineText = NULL;
   }
   hResult = pError->GetExceptionInfo( &excep );
   if( SUCCEEDED( hResult ) )
   {
	  AfxFormatString2( strMessage, IDS_SCRIPTERRORFORMAT, OLE2CT(
		 excep.bstrSource ), OLE2CT( excep.bstrDescription ) );
	  nResult = AfxMessageBox( strMessage, MB_YESNO );
	  if( nResult == IDYES )
	  {
		 return( S_OK );
	  }
	  else
	  {
		 return( E_FAIL );
	  }
   }

   return( E_FAIL );
}

STDMETHODIMP CScript::XActiveScriptSite::OnScriptTerminate(
   const VARIANT* pvarResult, const EXCEPINFO* pExcepInfo )
{
   (void)pvarResult;
   (void)pExcepInfo;

   METHOD_PROLOGUE( CScript, ActiveScriptSite )

   TRACE( "XActiveScriptSite::OnScriptTerminate()\n" );

   return( S_OK );
}

STDMETHODIMP CScript::XActiveScriptSite::OnStateChange(
   SCRIPTSTATE eState )
{
   METHOD_PROLOGUE( CScript, ActiveScriptSite )

   TRACE( "XActiveScriptSite::OnStateChange()\n" );

   switch( eState )
   {
   case SCRIPTSTATE_UNINITIALIZED:
	  TRACE( "\tSCRIPTSTATE_UNINITIALIZED\n" );
	  break;

   case SCRIPTSTATE_INITIALIZED:
	  TRACE( "\tSCRIPTSTATE_INITIALIZED\n" );
	  break;

   case SCRIPTSTATE_STARTED:
	  TRACE( "\tSCRIPTSTATE_STARTED\n" );
	  break;

   case SCRIPTSTATE_CONNECTED:
	  TRACE( "\tSCRIPTSTATE_CONNECTED\n" );
	  break;

   case SCRIPTSTATE_DISCONNECTED:
	  TRACE( "\tSCRIPTSTATE_DISCONNECTED\n" );
	  break;

   case SCRIPTSTATE_CLOSED:
	  TRACE( "\tSCRIPTSTATE_CLOSED\n" );
	  break;

   default:
	  TRACE( "\tUnknown SCRIPTSTATE value\n" );
	  ASSERT( FALSE );
	  break;
   }

   return( S_OK );
}


///////////////////////////////////////////////////////////////////////////////
// IActiveScriptSiteWindow
///////////////////////////////////////////////////////////////////////////////

STDMETHODIMP_( ULONG ) CScript::XActiveScriptSiteWindow::AddRef()
{
   METHOD_PROLOGUE_( CScript, ActiveScriptSiteWindow )

   TRACE( "XActiveScriptSiteWindow::AddRef()\n" );

   return( pThis->ExternalAddRef() );
}

STDMETHODIMP_( ULONG ) CScript::XActiveScriptSiteWindow::Release()
{
   METHOD_PROLOGUE_( CScript, ActiveScriptSiteWindow )

   TRACE( "XActiveScriptSiteWindow::Release()\n" );

   return( pThis->ExternalRelease() );
}

STDMETHODIMP CScript::XActiveScriptSiteWindow::QueryInterface(
   REFIID iid, void** ppInterface )
{
   METHOD_PROLOGUE_( CScript, ActiveScriptSiteWindow )

   return( pThis->ExternalQueryInterface( &iid, ppInterface ) );
}

STDMETHODIMP CScript::XActiveScriptSiteWindow::EnableModeless(
   BOOL tEnable )
{
   (void)tEnable;

   METHOD_PROLOGUE( CScript, ActiveScriptSiteWindow )

   return( E_NOTIMPL );
}

STDMETHODIMP CScript::XActiveScriptSiteWindow::GetWindow(
   HWND* phWindow )
{
   METHOD_PROLOGUE( CScript, ActiveScriptSiteWindow )

   if( phWindow == NULL )
   {
	  return( E_POINTER );
   }

   *phWindow = AfxGetMainWnd()->GetSafeHwnd();

   return( S_OK );
}