Tricks of the Windows Gam…ming Gurus (2nd Edition)/Disc2.iso/msdn_vcb/samples/vc98/addins/booknote/markit.cpp

C/C++ Source or Header | 1998-04-02 | 4KB | 168 lines

// MarkIt.cpp : Implementation of CMarkIt
#include "stdafx.h"
#include "Booknote.h"
#include "MarkIt.h"
#include "dlgnote.h"

CString strMainKey = _T("Software\\Microsoft\\devstudio\\6.0\\AddIns\\Booknote.MarkIt.1\\");

/////////////////////////////////////////////////////////////////////////////
// CMarkIt
CMarkIt::CMarkIt()
{
    CRegKey regKey;
    long lRes;
    unsigned long lSize;

    lRes = regKey.Create(HKEY_CURRENT_USER, strMainKey);
    regKey.QueryValue(m_strFile.GetBuffer(_MAX_PATH), _T("LogFile"), &lSize);
    m_strFile.ReleaseBuffer();
    if (m_strFile.IsEmpty())
    {
        m_strFile = _T("c:\\booklog.txt");
    }
}

CMarkIt::~CMarkIt()
{
    CRegKey regKey;
    long lRes;

    lRes = regKey.Create(HKEY_CURRENT_USER, strMainKey);
    regKey.SetValue(m_strFile, _T("LogFile"));
}


HRESULT CMarkIt::OnConnection(IApplication* pApp, VARIANT_BOOL bFirstTime, long dwAddInID, VARIANT_BOOL* bOnConnection)
{
    HRESULT hr = S_OK;
    m_spApplication = pApp;
    m_dwAddInID = dwAddInID;

    hr = pApp->SetAddInInfo((long)_Module.GetModuleInstance(),
        static_cast<IMarkIt*>(this), IDB_TOOLBAR_MEDIUM_MARKIT, IDB_TOOLBAR_LARGE_MARKIT, dwAddInID);
    LPCTSTR szCommand = _T("BookNote");
    VARIANT_BOOL bRet;
    if (SUCCEEDED(hr))
    {
        hr = pApp->AddCommand(CComBSTR(_T("BookNote\nBookNote\nBookNote\nBookNote")),CComBSTR(_T("BookNote")), 0, dwAddInID, &bRet);
    }

    // Add toolbar buttons only if this is the first time the add-in
    // is being loaded.  Toolbar buttons are automatically remembered
    // by Developer Studio from session to session, so we should only
    // add the toolbar buttons once.
    if (bFirstTime)
    {
        if (SUCCEEDED(hr))
        {
            hr = pApp->AddCommandBarButton(dsGlyph, CComBSTR(_T("BookNote")), dwAddInID);
        }
    }

    *bOnConnection = SUCCEEDED(hr) ? VARIANT_TRUE :VARIANT_FALSE;
    return hr;
}

HRESULT CMarkIt::OnDisconnection(VARIANT_BOOL bLastTime)
{
    return S_OK;
}

HRESULT CMarkIt::BookNote()
{
    AFX_MANAGE_STATE(AfxGetStaticModuleState());
    m_spApplication->EnableModeless(VARIANT_FALSE);
    CComQIPtr<ITextDocument, &IID_ITextDocument> pDocument;
    CComPtr<IDispatch> pDispDocument;
    CDlgNote dlgNote;
    int iRes;
    CTime timeNow;
    CComBSTR bstrTime;

    m_spApplication->get_ActiveDocument(&pDispDocument);
    pDocument = pDispDocument;
    pDispDocument = NULL;
    if (pDocument)
    {
        CComBSTR bstrName;
        pDocument->get_FullName(&bstrName);
        CComQIPtr<ITextSelection, &IID_ITextSelection> pSelection;
        CComPtr<IDispatch> pDispSelection;
        pDocument->get_Selection(&pDispSelection);
        pSelection = pDispSelection;
        pDispSelection = NULL;
        if (pSelection)
        {
            long col, line;
            CComBSTR bstr(2048,LPCSTR(" "));
            CComBSTR bstrNote;
            HRESULT hr;
            hr = pSelection->get_CurrentColumn(&col);
            hr =  pSelection->get_CurrentLine(&line);

            dlgNote.m_strLogFile = m_strFile;
            dlgNote.SetMarkIt(this);
            iRes = dlgNote.DoModal();
            if (iRes == IDOK)
            {
                m_strFile = dlgNote.m_strLogFile;
                // produce output like "c:\test\foo.cpp(37) : warning C4310: cast truncates constant value"
                bstrNote = dlgNote.m_strNote;
                timeNow = CTime::GetCurrentTime();
                bstrTime = timeNow.Format(_T("%I:%M:%S %p  %d/%m/%Y"));
                swprintf(bstr,L"%s(%d): (%s) %s", bstrName, line, bstrTime, bstrNote);
                if (dlgNote.m_fEcho)
                {
                    m_spApplication->PrintToOutputWindow(bstr);
                }

                CString strOut;
                strOut = bstr;
                CStdioFile log;
                try
                {
                    if (!dlgNote.m_strLogFile.IsEmpty() && log.Open(dlgNote.m_strLogFile, CFile::modeNoTruncate | CFile::modeWrite | CFile::modeCreate))
                    {
                        log.SeekToEnd();
                        strOut += _T("\n");
                        log.WriteString(strOut);
                        log.Close();
                    }
                }
                catch (CFileException *eFile)
                {
                    CString strMsg;
                    CString strErr;
                    eFile->GetErrorMessage(strErr.GetBuffer(_MAX_PATH), _MAX_PATH);
                    strErr.ReleaseBuffer();
                    strMsg.Format("%s \n    %s", strErr, dlgNote.m_strLogFile);
                    MessageBox(NULL, strMsg, _T("BOOKNOTE"), MB_OK);
                    eFile->Delete();
                }
            }

        }
    }
    m_spApplication->EnableModeless((VARIANT_BOOL)VARIANT_TRUE);
    return S_OK;
}


HRESULT CMarkIt::Dump(LPCTSTR szFile)
{
    CStdioFile file;
    CString str;
    HRESULT hr = E_FAIL;

    if (file.Open(szFile, CFile::modeRead))
    {
        while (file.ReadString(str))
        {
            CComBSTR bstr = str;
            hr = m_spApplication->PrintToOutputWindow(bstr);
        }
    }
    return (hr);
}