void EnumTypeLib(ITypeLib *typeLib, EnumTypeLibProc pfn)
{
UINT count = typeLib->GetTypeInfoCount();
for (UINT i = 0; i < count; i++)
{
ITypeInfoPtr typeInfo;
typeLib->GetTypeInfo(i, &typeInfo);
TYPEATTR *typeAttr = NULL;
typeInfo->GetTypeAttr(&typeAttr);
pfn(typeInfo, typeAttr);
typeInfo->ReleaseTypeAttr(typeAttr);
}
}
HRESULT GetDispatchProperty(IDispatch* pDisp, LPOLESTR lpName, VARIANT* pRes)
{
#ifdef _DEBUG
ITypeInfoPtr pTypeInfo;
LPTYPEATTR pTypeAttr;
pDisp->GetTypeInfo(1, GetSystemDefaultLCID(), &pTypeInfo);
pTypeInfo->GetTypeAttr(&pTypeAttr);
pTypeInfo->ReleaseTypeAttr(pTypeAttr);
#endif
DISPID DispId = 0;
HRESULT hr = pDisp->GetIDsOfNames(IID_NULL, &lpName, 1, GetSystemDefaultLCID(), &DispId);
if (hr != S_OK)
return hr;
_variant_t varRes;
unsigned int ArgError;
DISPPARAMS Params;
Params.cArgs = 0;
Params.rgvarg = NULL;
Params.cNamedArgs = 0;
Params.rgdispidNamedArgs = NULL;
hr = pDisp->Invoke(DispId, IID_NULL, GetSystemDefaultLCID(), DISPATCH_PROPERTYGET, &Params, pRes, NULL, &ArgError);
return hr;
}
HRESULT CScript::LoadScript( LPCTSTR pszFileName, LPCTSTR pszScriptName )
{
ITypeInfoPtr pTypeInfo;
HRESULT hResult;
CLSID clsid;
CFile file;
ULONGLONG nFileSize;
BSTR bstrScript;
BOOL tSuccess;
int iMethod;
UINT nNames;
BSTR bstrName;
CString strName;
CString strExtension;
int iChar;
ENSURE( pszFileName != NULL );
ENSURE( pszScriptName != NULL );
ENSURE( m_pActiveScript == NULL );
iChar = lstrlen( pszFileName )-1;
while( (iChar >= 0) && (pszFileName[iChar] != _T( '.' )) )
{
iChar--;
}
if( (iChar >= 0) && (_tcsicmp( &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();
if( nFileSize > INT_MAX )
{
return( E_OUTOFMEMORY );
}
nFileSize = file.Read( m_strScriptText.GetBuffer( ULONG( nFileSize ) ), ULONG( nFileSize ) );
file.Close();
m_strScriptText.ReleaseBuffer( ULONG( 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 );
}
