1 /*

     2  * Copyright (C) 2009 Apple Inc. All Rights Reserved.

     3  * Copyright (C) 2009 Brent Fulgham. All Rights Reserved.

     4  *

     5  * Redistribution and use in source and binary forms, with or without

     6  * modification, are permitted provided that the following conditions

     7  * are met:

     8  * 1. Redistributions of source code must retain the above copyright

     9  *    notice, this list of conditions and the following disclaimer.

    10  * 2. Redistributions in binary form must reproduce the above copyright

    11  *    notice, this list of conditions and the following disclaimer in the

    12  *    documentation and/or other materials provided with the distribution.

    13  *

    14  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY

    15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

    16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

    17  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR

    18  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

    19  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

    20  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

    21  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY

    22  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

    23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

    24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 

    25  */

    26 

    27 #include "stdafx.h"

    28 #include "PrintWebUIDelegate.h"

    29 

    30 #include <commctrl.h>

    31 #include <commdlg.h>

    32 #include <objbase.h>

    33 #include <shlwapi.h>

    34 #include <wininet.h>

    35 

    36 #include <WebKit/WebKitCOMAPI.h>

    37 

    38 static const int MARGIN = 20;

    39 

    40 HRESULT PrintWebUIDelegate::QueryInterface(REFIID riid, void** ppvObject)

    41 {

    42     *ppvObject = 0;

    43     if (IsEqualIID(riid, IID_IUnknown))

    44         *ppvObject = static_cast<IWebUIDelegate*>(this);

    45     else if (IsEqualIID(riid, IID_IWebUIDelegate))

    46         *ppvObject = static_cast<IWebUIDelegate*>(this);

    47     else

    48         return E_NOINTERFACE;

    49 

    50     AddRef();

    51     return S_OK;

    52 }

    53 

    54 ULONG PrintWebUIDelegate::AddRef(void)

    55 {

    56     return ++m_refCount;

    57 }

    58 

    59 ULONG PrintWebUIDelegate::Release(void)

    60 {

    61     ULONG newRef = --m_refCount;

    62     if (!newRef)

    63         delete this;

    64 

    65     return newRef;

    66 }

    67 

    68 HRESULT PrintWebUIDelegate::webViewPrintingMarginRect(IWebView* view, RECT* rect)

    69 {

    70     if (!view || !rect)

    71         return E_POINTER;

    72 

    73     IWebFrame* mainFrame = 0;

    74     if (FAILED(view->mainFrame(&mainFrame)))

    75         return E_FAIL;

    76 

    77     IWebFramePrivate* privateFrame = 0;

    78     if (FAILED(mainFrame->QueryInterface(&privateFrame))) {

    79         mainFrame->Release();

    80         return E_FAIL;

    81     }

    82 

    83     privateFrame->frameBounds(rect);

    84 

    85     rect->left += MARGIN;

    86     rect->top += MARGIN;

    87     HDC dc = ::GetDC(0);

    88     rect->right = (::GetDeviceCaps(dc, LOGPIXELSX) * 6.5) - MARGIN;

    89     rect->bottom = (::GetDeviceCaps(dc, LOGPIXELSY) * 11) - MARGIN;

    90     ::ReleaseDC(0, dc);

    91 

    92     privateFrame->Release();

    93     mainFrame->Release();

    94 

    95     return S_OK;

    96 }

    97 

    98 HRESULT PrintWebUIDelegate::webViewHeaderHeight(IWebView* webView, float* height)

    99 {

   100    if (!webView || !height)

   101       return E_POINTER;

   102    

   103    HDC dc = ::GetDC(0);

   104 

   105    TEXTMETRIC textMetric;

   106    ::GetTextMetrics(dc, &textMetric);

   107    ::ReleaseDC(0, dc);

   108 

   109    *height = 1.1 * textMetric.tmHeight;

   110 

   111    return S_OK;

   112 }

   113 

   114 HRESULT PrintWebUIDelegate::webViewFooterHeight(IWebView* webView, float* height)

   115 {

   116    if (!webView || !height)

   117       return E_POINTER;

   118 

   119    HDC dc = ::GetDC(0);

   120 

   121    TEXTMETRIC textMetric;

   122    ::GetTextMetrics(dc, &textMetric);

   123    ::ReleaseDC(0, dc);

   124 

   125    *height = 1.1 * textMetric.tmHeight;

   126 

   127    return S_OK;

   128 }

   129 

   130 HRESULT PrintWebUIDelegate::drawHeaderInRect( 

   131             /* [in] */ IWebView *webView,

   132             /* [in] */ RECT *rect,

   133             /* [in] */ OLE_HANDLE drawingContext)

   134 {

   135    if (!webView || !rect)

   136       return E_POINTER;

   137 

   138    // Turn off header for now.

   139    HDC dc = reinterpret_cast<HDC>(drawingContext);

   140 

   141    HFONT hFont = (HFONT)::GetStockObject(ANSI_VAR_FONT);

   142    HFONT hOldFont = (HFONT)::SelectObject(dc, hFont);

   143 

   144    LPCTSTR header(_T("[Sample Header]"));

   145    int length = _tcslen(header);

   146 

   147    int rc = ::DrawText(dc, header, length, rect, DT_LEFT | DT_NOCLIP | DT_VCENTER  | DT_SINGLELINE);

   148    ::SelectObject(dc, hOldFont);

   149 

   150    if (!rc)

   151       return E_FAIL;

   152 

   153    ::MoveToEx(dc, rect->left, rect->bottom, 0);

   154    HPEN hPen = (HPEN)::GetStockObject(BLACK_PEN);

   155    HPEN hOldPen = (HPEN)::SelectObject(dc, hPen);

   156    ::LineTo(dc, rect->right, rect->bottom);

   157    ::SelectObject(dc, hOldPen);

   158 

   159    return S_OK;

   160 }

   161 

   162 HRESULT PrintWebUIDelegate::drawFooterInRect( 

   163             /* [in] */ IWebView *webView,

   164             /* [in] */ RECT *rect,

   165             /* [in] */ OLE_HANDLE drawingContext,

   166             /* [in] */ UINT pageIndex,

   167             /* [in] */ UINT pageCount)

   168 {

   169    if (!webView || !rect)

   170       return E_POINTER;

   171 

   172    HDC dc = reinterpret_cast<HDC>(drawingContext);

   173 

   174    HFONT hFont = (HFONT)::GetStockObject(ANSI_VAR_FONT);

   175    HFONT hOldFont = (HFONT)::SelectObject(dc, hFont);

   176 

   177    LPCTSTR footer(_T("[Sample Footer]"));

   178    int length = _tcslen(footer);

   179 

   180    // Add a line, 1/10th inch above the footer text from left margin to right margin.

   181    ::MoveToEx(dc, rect->left, rect->top, 0);

   182    HPEN hPen = (HPEN)::GetStockObject(BLACK_PEN);

   183    HPEN hOldPen = (HPEN)::SelectObject(dc, hPen);

   184    ::LineTo(dc, rect->right, rect->top);

   185    ::SelectObject(dc, hOldPen);

   186 

   187    int rc = ::DrawText(dc, footer, length, rect, DT_LEFT | DT_NOCLIP | DT_VCENTER  | DT_SINGLELINE);

   188    ::SelectObject(dc, hOldFont);

   189 

   190    if (!rc)

   191       return E_FAIL;

   192 

   193    return S_OK;

   194 }