开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 查看源码
SCRIPT.CPP
资源名称：MSDN_VC98.zip　[点击查看]
上传用户：bangxh
上传日期：2007-01-31
资源大小：42235k
文件大小：13k
源码类别：

Windows编程

开发平台：

Visual C++

SCRIPT.CPP：源码内容
  1. // Script.Cpp : implementation file
  2. //
  4. #include "StdAfx.H"
  5. #include "TestCon.H"
  7. #ifdef _DEBUG
  8. #define new DEBUG_NEW
  9. #undef THIS_FILE
  10. static char THIS_FILE[] = __FILE__;
  11. #endif
  13. /////////////////////////////////////////////////////////////////////////////
  14. // CScript
  16. IMPLEMENT_DYNAMIC( CScript, CCmdTarget )
  18. CScript::CScript( CScriptManager* pManager ) :
  19.    m_pManager( pManager )
  20. {
  21.    ASSERT( m_pManager != NULL );
  22. }
  24. CScript::~CScript()
  25. {
  26.    Unload();
  27. }
  29. HRESULT CScript::AddNamedItem( LPCTSTR pszItemName )
  30. {
  31.    USES_CONVERSION;
  32.    LPCOLESTR pszItemNameO;
  33.    HRESULT hResult;
  35.    pszItemNameO = T2COLE( pszItemName );
  37.    hResult = m_pActiveScript->AddNamedItem( pszItemNameO,
  38.   SCRIPTITEM_ISSOURCE|SCRIPTITEM_ISVISIBLE );
  39.    if( FAILED( hResult ) )
  40.    {
  41.   return( hResult );
  42.    }
  44.    return( S_OK );
  45. }
  47. BOOL CScript::FindMacro( LPCTSTR pszMacroName )
  48. {
  49.    DISPID dispid;
  51.    ASSERT( pszMacroName != NULL );
  53.    return( m_mapMacros.Lookup( pszMacroName, dispid ) );
  54. }
  56. POSITION CScript::GetFirstMacroPosition()
  57. {
  58.    return( m_mapMacros.GetStartPosition() );
  59. }
  61. CString CScript::GetName()
  62. {
  63.    return( m_strScriptName );
  64. }
  66. CString CScript::GetNextMacroName( POSITION& posMacro )
  67. {
  68.    CString strMacro;
  69.    DISPID dispid;
  71.    m_mapMacros.GetNextAssoc( posMacro, strMacro, dispid );
  73.    return( strMacro );
  74. }
  76. HRESULT CScript::LoadScript( LPCTSTR pszFileName, LPCTSTR pszScriptName )
  77. {
  78.    ITypeInfoPtr pTypeInfo;
  79.    HRESULT hResult;
  80.    CLSID clsid;
  81.    CFile file;
  82.    ULONG nFileSize;
  83.    BSTR bstrScript;
  84.    BOOL tSuccess;
  85.    int iMethod;
  86.    UINT nNames;
  87.    BSTR bstrName;
  88.    CString strName;
  89.    CString strExtension;
  90.    int iChar;
  92.    ASSERT( pszFileName != NULL );
  93.    ASSERT( pszScriptName != NULL );
  94.    ASSERT( m_pActiveScript == NULL );
  96.    iChar = lstrlen( pszFileName )-1;
  97.    while( (iChar >= 0) && (pszFileName[iChar] != _T( '.' )) )
  98.    {
  99.   iChar--;
  100.    }
  102.    if( (iChar >= 0) && (lstrcmpi( &pszFileName[iChar], _T( ".js" ) ) == 0) )
  103.    {
  104.   hResult = CLSIDFromProgID( L"JScript", &clsid );
  105.   if( FAILED( hResult ) )
  106.   {
  107.  return( hResult );
  108.   }
  109.    }
  110.    else
  111.    {
  112.   hResult = CLSIDFromProgID( L"VBScript", &clsid );
  113.   if( FAILED( hResult ) )
  114.   {
  115.  return( hResult );
  116.   }
  117.    }
  119.    hResult = m_pActiveScript.CreateInstance( clsid, NULL,
  120.   CLSCTX_INPROC_SERVER );
  121.    if( FAILED( hResult ) )
  122.    {
  123.   return( hResult );
  124.    }
  126.    m_pActiveScriptParse = m_pActiveScript;
  127.    if( m_pActiveScriptParse == NULL )
  128.    {
  129.   return( E_NOINTERFACE );
  130.    }
  132.    hResult = m_pActiveScript->SetScriptSite( &m_xActiveScriptSite );
  133.    if( FAILED( hResult ) )
  134.    {
  135.   return( hResult );
  136.    }
  138.    hResult = m_pActiveScriptParse->InitNew();
  139.    if( FAILED( hResult ) )
  140.    {
  141.   return( hResult );
  142.    }
  144.    CFileException error;
  145.    tSuccess = file.Open( pszFileName, CFile::modeRead|CFile::shareDenyWrite,
  146.   &error );
  147.    if( !tSuccess )
  148.    {
  149.   return( HRESULT_FROM_WIN32( error.m_lOsError ) );
  150.    }
  152.    nFileSize = file.GetLength();
  153.    nFileSize = file.Read( m_strScriptText.GetBuffer( nFileSize ), nFileSize );
  154.    file.Close();
  155.    m_strScriptText.ReleaseBuffer( nFileSize );
  156.    bstrScript = m_strScriptText.AllocSysString();
  158.    hResult = m_pActiveScriptParse->ParseScriptText( bstrScript, NULL, NULL,
  159.   NULL, DWORD( this ), 0, SCRIPTTEXT_ISVISIBLE, NULL, NULL );
  160.    SysFreeString( bstrScript );
  161.    if( FAILED( hResult ) )
  162.    {
  163.   return( hResult );
  164.    }
  166.    hResult = m_pManager->AddNamedItems( m_pActiveScript );
  167.    if( FAILED( hResult ) )
  168.    {
  169.   return( hResult );
  170.    }
  172.    hResult = m_pActiveScript->SetScriptState( SCRIPTSTATE_CONNECTED );
  173.    if( FAILED( hResult ) )
  174.    {
  175.   return( hResult );
  176.    }
  178.    hResult = m_pActiveScript->GetScriptDispatch( NULL, &m_pDispatch );
  179.    if( FAILED( hResult ) )
  180.    {
  181.   return( hResult );
  182.    }
  184.    hResult = m_pDispatch->GetTypeInfo( 0, GetUserDefaultLCID(), &pTypeInfo );
  185.    if( FAILED( hResult ) )
  186.    {
  187.   return( hResult );
  188.    }
  190.    CSmartTypeAttr pTypeAttr( pTypeInfo );
  191.    CSmartFuncDesc pFuncDesc( pTypeInfo );
  193.    hResult = pTypeInfo->GetTypeAttr( &pTypeAttr );
  194.    if( FAILED( hResult ) )
  195.    {
  196.   return( hResult );
  197.    }
  199.    for( iMethod = 0; iMethod < pTypeAttr->cFuncs; iMethod++ )
  200.    {
  201.   hResult = pTypeInfo->GetFuncDesc( iMethod, &pFuncDesc );
  202.   if( FAILED( hResult ) )
  203.   {
  204.  return( hResult );
  205.   }
  207.   if( (pFuncDesc->funckind == FUNC_DISPATCH) && (pFuncDesc->invkind ==
  208.  INVOKE_FUNC) && (pFuncDesc->cParams == 0) )
  209.   {
  210.  bstrName = NULL;
  211.  hResult = pTypeInfo->GetNames( pFuncDesc->memid, &bstrName, 1,
  212. &nNames );
  213.  if( FAILED( hResult ) )
  214.  {
  215. return( hResult );
  216.  }
  217.  ASSERT( nNames == 1 );
  219.  strName = bstrName;
  220.  SysFreeString( bstrName );
  221.  bstrName = NULL;
  223.  // Macros can't contain underscores, since those denote event handlers
  224.  if( strName.Find( _T( '_' ) ) == -1 )
  225.  {
  226. m_mapMacros.SetAt( strName, pFuncDesc->memid );
  227.  }
  228.   }
  230.   pFuncDesc.Release();
  231.    }
  233.    m_strScriptName = pszScriptName;
  235.    return( S_OK );
  236. }
  238. HRESULT CScript::RunMacro( LPCTSTR pszMacroName )
  239. {
  240.    DISPID dispid;
  241.    COleDispatchDriver driver;
  242.    BOOL tFound;
  244.    ASSERT( pszMacroName != NULL );
  246.    tFound = m_mapMacros.Lookup( pszMacroName, dispid );
  247.    if( !tFound )
  248.    {
  249.   return( DISP_E_MEMBERNOTFOUND );
  250.    }
  252.    driver.AttachDispatch( m_pDispatch, FALSE );
  253.    try
  254.    {
  255.   driver.InvokeHelper( dispid, DISPATCH_METHOD, VT_EMPTY, NULL, VTS_NONE );
  256.    }
  257.    catch( COleDispatchException* pException )
  258.    {
  259.   CString strMessage;
  261.   AfxFormatString1( strMessage, IDS_DISPATCHEXCEPTION,
  262.  pException->m_strDescription );
  263.   AfxMessageBox( strMessage );
  264.   pException->Delete();
  265.    }
  266.    catch( COleException* pException )
  267.    {
  268.   pException->Delete();
  269.    }
  271.    return( S_OK );
  272. }
  274. void CScript::Unload()
  275. {
  276.    if( m_pDispatch != NULL )
  277.    {
  278.   m_pDispatch.Release();
  279.    }
  280.    if( m_pActiveScript != NULL )
  281.    {
  282.   m_pActiveScript->SetScriptState( SCRIPTSTATE_DISCONNECTED );
  283.   m_pActiveScript->Close();
  284.   m_pActiveScript.Release();
  285.    }
  286.    if( m_pActiveScriptParse != NULL )
  287.    {
  288.   m_pActiveScriptParse.Release();
  289.    }
  290. }
  293. BEGIN_INTERFACE_MAP( CScript, CCmdTarget )
  294.    INTERFACE_PART( CScript, IID_IActiveScriptSite, ActiveScriptSite )
  295.    INTERFACE_PART( CScript, IID_IActiveScriptSiteWindow, ActiveScriptSiteWindow )
  296. END_INTERFACE_MAP()
  298. BEGIN_MESSAGE_MAP(CScript, CCmdTarget)
  299. //{{AFX_MSG_MAP(CScript)
  300. // NOTE - the ClassWizard will add and remove mapping macros here.
  301. //}}AFX_MSG_MAP
  302. END_MESSAGE_MAP()
  304. /////////////////////////////////////////////////////////////////////////////
  305. // CScript message handlers
  308. ///////////////////////////////////////////////////////////////////////////////
  309. // IActiveScriptSite
  310. ///////////////////////////////////////////////////////////////////////////////
  312. STDMETHODIMP_( ULONG ) CScript::XActiveScriptSite::AddRef()
  313. {
  314.    METHOD_PROLOGUE_( CScript, ActiveScriptSite )
  316.    TRACE( "XActiveScriptSite::AddRef()n" );
  318.    return( pThis->ExternalAddRef() );
  319. }
  321. STDMETHODIMP_( ULONG ) CScript::XActiveScriptSite::Release()
  322. {
  323.    METHOD_PROLOGUE_( CScript, ActiveScriptSite )
  325.    TRACE( "XActiveScriptSite::Release()n" );
  327.    return( pThis->ExternalRelease() );
  328. }
  330. STDMETHODIMP CScript::XActiveScriptSite::QueryInterface( REFIID iid,
  331.    void** ppInterface )
  332. {
  333.    METHOD_PROLOGUE_( CScript, ActiveScriptSite )
  335.    return( pThis->ExternalQueryInterface( &iid, ppInterface ) );
  336. }
  338. STDMETHODIMP CScript::XActiveScriptSite::GetDocVersionString(
  339.    BSTR* pbstrVersion )
  340. {
  341.    (void)pbstrVersion;
  343.    METHOD_PROLOGUE( CScript, ActiveScriptSite )
  345.    return( E_NOTIMPL );
  346. }
  348. STDMETHODIMP CScript::XActiveScriptSite::GetItemInfo( LPCOLESTR pszName,
  349.    DWORD dwReturnMask, IUnknown** ppItem, ITypeInfo** ppTypeInfo )
  350. {
  351.    USES_CONVERSION;
  352.    LPCTSTR pszNameT;
  353.    HRESULT hResult;
  355.    METHOD_PROLOGUE( CScript, ActiveScriptSite )
  357.    if( ppItem != NULL )
  358.    {
  359.   *ppItem = NULL;
  360.    }
  361.    if( ppTypeInfo != NULL )
  362.    {
  363.   *ppTypeInfo = NULL;
  364.    }
  366.    if( dwReturnMask&SCRIPTINFO_IUNKNOWN )
  367.    {
  368.   if( ppItem == NULL )
  369.   {
  370.  return( E_POINTER );
  371.   }
  372.    }
  373.    if( dwReturnMask&SCRIPTINFO_ITYPEINFO )
  374.    {
  375.   if( ppTypeInfo == NULL )
  376.   {
  377.  return( E_POINTER );
  378.   }
  379.    }
  381.    pszNameT = OLE2CT( pszName );
  382.    TRACE( "XActiveScriptSite::GetItemInfo( %s )n", pszNameT );
  384.    if( dwReturnMask&SCRIPTINFO_IUNKNOWN )
  385.    {
  386.   hResult = pThis->m_pManager->GetItemDispatch( pszNameT, ppItem );
  387.   if( FAILED( hResult ) )
  388.   {
  389.  return( hResult );
  390.   }
  391.    }
  392.    if( dwReturnMask&SCRIPTINFO_ITYPEINFO )
  393.    {
  394.   pThis->m_pManager->GetItemTypeInfo( pszNameT, ppTypeInfo );
  395.    }
  397.    return( S_OK );
  398. }
  400. STDMETHODIMP CScript::XActiveScriptSite::GetLCID( LCID* plcid )
  401. {
  402.    METHOD_PROLOGUE( CScript, ActiveScriptSite )
  404.    if( plcid == NULL )
  405.    {
  406.   return( E_POINTER );
  407.    }
  409.    *plcid = GetUserDefaultLCID();
  411.    return( S_OK );
  412. }
  414. STDMETHODIMP CScript::XActiveScriptSite::OnEnterScript()
  415. {
  416.    METHOD_PROLOGUE( CScript, ActiveScriptSite )
  418.    return( S_OK );
  419. }
  421. STDMETHODIMP CScript::XActiveScriptSite::OnLeaveScript()
  422. {
  423.    METHOD_PROLOGUE( CScript, ActiveScriptSite )
  425.    return( S_OK );
  426. }
  428. STDMETHODIMP CScript::XActiveScriptSite::OnScriptError(
  429.    IActiveScriptError* pError )
  430. {
  431.    USES_CONVERSION;
  432.    HRESULT hResult;
  433.    CExcepInfo excep;
  434.    CString strMessage;
  435.    int nResult;
  436.    DWORD dwContext;
  437.    ULONG iLine;
  438.    LONG iChar;
  439.    BSTR bstrSourceLineText;
  441.    METHOD_PROLOGUE( CScript, ActiveScriptSite )
  443.    TRACE( "XActiveScriptSite::OnScriptError()n" );
  445.    ASSERT( pError != NULL );
  447.    hResult = pError->GetSourcePosition( &dwContext, &iLine, &iChar );
  448.    if( SUCCEEDED( hResult ) )
  449.    {
  450.   TRACE( "Error at line %u, character %dn", iLine, iChar );
  451.    }
  452.    bstrSourceLineText = NULL;
  453.    hResult = pError->GetSourceLineText( &bstrSourceLineText );
  454.    if( SUCCEEDED( hResult ) )
  455.    {
  456.   TRACE( "Source Text: %sn", OLE2CT( bstrSourceLineText ) );
  457.   SysFreeString( bstrSourceLineText );
  458.   bstrSourceLineText = NULL;
  459.    }
  460.    hResult = pError->GetExceptionInfo( &excep );
  461.    if( SUCCEEDED( hResult ) )
  462.    {
  463.   AfxFormatString2( strMessage, IDS_SCRIPTERRORFORMAT, OLE2CT(
  464.  excep.bstrSource ), OLE2CT( excep.bstrDescription ) );
  465.   nResult = AfxMessageBox( strMessage, MB_YESNO );
  466.   if( nResult == IDYES )
  467.   {
  468.  return( S_OK );
  469.   }
  470.   else
  471.   {
  472.  return( E_FAIL );
  473.   }
  474.    }
  476.    return( E_FAIL );
  477. }
  479. STDMETHODIMP CScript::XActiveScriptSite::OnScriptTerminate(
  480.    const VARIANT* pvarResult, const EXCEPINFO* pExcepInfo )
  481. {
  482.    (void)pvarResult;
  483.    (void)pExcepInfo;
  485.    METHOD_PROLOGUE( CScript, ActiveScriptSite )
  487.    TRACE( "XActiveScriptSite::OnScriptTerminate()n" );
  489.    return( S_OK );
  490. }
  492. STDMETHODIMP CScript::XActiveScriptSite::OnStateChange(
  493.    SCRIPTSTATE eState )
  494. {
  495.    METHOD_PROLOGUE( CScript, ActiveScriptSite )
  497.    TRACE( "XActiveScriptSite::OnStateChange()n" );
  499.    switch( eState )
  500.    {
  501.    case SCRIPTSTATE_UNINITIALIZED:
  502.   TRACE( "tSCRIPTSTATE_UNINITIALIZEDn" );
  503.   break;
  505.    case SCRIPTSTATE_INITIALIZED:
  506.   TRACE( "tSCRIPTSTATE_INITIALIZEDn" );
  507.   break;
  509.    case SCRIPTSTATE_STARTED:
  510.   TRACE( "tSCRIPTSTATE_STARTEDn" );
  511.   break;
  513.    case SCRIPTSTATE_CONNECTED:
  514.   TRACE( "tSCRIPTSTATE_CONNECTEDn" );
  515.   break;
  517.    case SCRIPTSTATE_DISCONNECTED:
  518.   TRACE( "tSCRIPTSTATE_DISCONNECTEDn" );
  519.   break;
  521.    case SCRIPTSTATE_CLOSED:
  522.   TRACE( "tSCRIPTSTATE_CLOSEDn" );
  523.   break;
  525.    default:
  526.   TRACE( "tUnknown SCRIPTSTATE valuen" );
  527.   ASSERT( FALSE );
  528.   break;
  529.    }
  531.    return( S_OK );
  532. }
  535. ///////////////////////////////////////////////////////////////////////////////
  536. // IActiveScriptSiteWindow
  537. ///////////////////////////////////////////////////////////////////////////////
  539. STDMETHODIMP_( ULONG ) CScript::XActiveScriptSiteWindow::AddRef()
  540. {
  541.    METHOD_PROLOGUE_( CScript, ActiveScriptSiteWindow )
  543.    TRACE( "XActiveScriptSiteWindow::AddRef()n" );
  545.    return( pThis->ExternalAddRef() );
  546. }
  548. STDMETHODIMP_( ULONG ) CScript::XActiveScriptSiteWindow::Release()
  549. {
  550.    METHOD_PROLOGUE_( CScript, ActiveScriptSiteWindow )
  552.    TRACE( "XActiveScriptSiteWindow::Release()n" );
  554.    return( pThis->ExternalRelease() );
  555. }
  557. STDMETHODIMP CScript::XActiveScriptSiteWindow::QueryInterface(
  558.    REFIID iid, void** ppInterface )
  559. {
  560.    METHOD_PROLOGUE_( CScript, ActiveScriptSiteWindow )
  562.    return( pThis->ExternalQueryInterface( &iid, ppInterface ) );
  563. }
  565. STDMETHODIMP CScript::XActiveScriptSiteWindow::EnableModeless(
  566.    BOOL tEnable )
  567. {
  568.    (void)tEnable;
  570.    METHOD_PROLOGUE( CScript, ActiveScriptSiteWindow )
  572.    return( E_NOTIMPL );
  573. }
  575. STDMETHODIMP CScript::XActiveScriptSiteWindow::GetWindow(
  576.    HWND* phWindow )
  577. {
  578.    METHOD_PROLOGUE( CScript, ActiveScriptSiteWindow )
  580.    if( phWindow == NULL )
  581.    {
  582.   return( E_POINTER );
  583.    }
  585.    *phWindow = AfxGetMainWnd()->GetSafeHwnd();
  587.    return( S_OK );
  588. }