/*

 * Copyright (C) 2006, 2008 Apple Computer, Inc.  All rights reserved.

 * Copyright (C) 2009 Brent Fulgha.  All rights reserved.

 *

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

 * modification, are permitted provided that the following conditions

 * are met:

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

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

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

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

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

 *

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

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

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

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

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

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

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

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

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

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

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

 */

#include "stdafx.h"

#include "WinLauncher.h"

#include <WebKit/WebKitCOMAPI.h>

#include <commctrl.h>

#include <commdlg.h>

#include <objbase.h>

#include <shlwapi.h>

#include <wininet.h>

#include "PrintWebUIDelegate.h"

#define MAX_LOADSTRING 100

#define URLBAR_HEIGHT  24

// Global Variables:

HINSTANCE hInst;                                // current instance

HWND hMainWnd;

HWND hURLBarWnd;

long DefEditProc;

IWebView* gWebView = 0;

HWND gViewWindow = 0;

WinLauncherWebHost* gWebHost = 0;

PrintWebUIDelegate* gPrintDelegate = 0;

TCHAR szTitle[MAX_LOADSTRING];                    // The title bar text

TCHAR szWindowClass[MAX_LOADSTRING];            // the main window class name

// Forward declarations of functions included in this code module:

ATOM                MyRegisterClass(HINSTANCE hInstance);

BOOL                InitInstance(HINSTANCE, int);

LRESULT CALLBACK    WndProc(HWND, UINT, WPARAM, LPARAM);

INT_PTR CALLBACK    About(HWND, UINT, WPARAM, LPARAM);

LRESULT CALLBACK    MyEditProc(HWND, UINT, WPARAM, LPARAM);

static void loadURL(BSTR urlBStr);

HRESULT WinLauncherWebHost::updateAddressBar(IWebView* webView)

{

    IWebFrame* mainFrame = 0;

    IWebDataSource* dataSource = 0;

    IWebMutableURLRequest* request = 0;

    BSTR frameURL = 0;

    HRESULT hr = S_OK;

    hr = webView->mainFrame(&mainFrame);

    if (FAILED(hr))

        goto exit;

    hr = mainFrame->dataSource(&dataSource);

    if (FAILED(hr) || !dataSource)

        hr = mainFrame->provisionalDataSource(&dataSource);

    if (FAILED(hr) || !dataSource)

        goto exit;

    hr = dataSource->request(&request);

    if (FAILED(hr) || !request)

        goto exit;

    hr = request->mainDocumentURL(&frameURL);

    if (FAILED(hr))

        goto exit;

    SendMessage(hURLBarWnd, (UINT)WM_SETTEXT, 0, (LPARAM)frameURL);

exit:

    if (mainFrame)

        mainFrame->Release();

    if (dataSource)

        dataSource->Release();

    if (request)

        request->Release();

    SysFreeString(frameURL);

    return 0;

}

HRESULT STDMETHODCALLTYPE WinLauncherWebHost::QueryInterface(REFIID riid, void** ppvObject)

{

    *ppvObject = 0;

    if (IsEqualGUID(riid, IID_IUnknown))

        *ppvObject = static_cast<IWebFrameLoadDelegate*>(this);

    else if (IsEqualGUID(riid, IID_IWebFrameLoadDelegate))

        *ppvObject = static_cast<IWebFrameLoadDelegate*>(this);

    else

        return E_NOINTERFACE;

    AddRef();

    return S_OK;

}

ULONG STDMETHODCALLTYPE WinLauncherWebHost::AddRef(void)

{

    return ++m_refCount;

}

ULONG STDMETHODCALLTYPE WinLauncherWebHost::Release(void)

{

    ULONG newRef = --m_refCount;

    if (!newRef)

        delete(this);

    return newRef;

}

static void resizeSubViews()

{

    RECT rcClient;

    GetClientRect(hMainWnd, &rcClient);

    MoveWindow(hURLBarWnd, 0, 0, rcClient.right, URLBAR_HEIGHT, TRUE);

    MoveWindow(gViewWindow, 0, URLBAR_HEIGHT, rcClient.right, rcClient.bottom - URLBAR_HEIGHT, TRUE);

}

int APIENTRY _tWinMain(HINSTANCE hInstance,

                     HINSTANCE hPrevInstance,

                     LPTSTR    lpCmdLine,

                     int       nCmdShow)

{

#ifdef _CRTDBG_MAP_ALLOC

    _CrtSetReportFile(_CRT_WARN, _CRTDBG_FILE_STDERR);

    _CrtSetReportMode(_CRT_WARN, _CRTDBG_MODE_FILE);

#endif

    UNREFERENCED_PARAMETER(hPrevInstance);

    UNREFERENCED_PARAMETER(lpCmdLine);

     // TODO: Place code here.

    MSG msg;

    HACCEL hAccelTable;

    INITCOMMONCONTROLSEX InitCtrlEx;

    InitCtrlEx.dwSize = sizeof(INITCOMMONCONTROLSEX);

    InitCtrlEx.dwICC  = 0x00004000; //ICC_STANDARD_CLASSES;

    InitCommonControlsEx(&InitCtrlEx);

    // Initialize global strings

    LoadString(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);

    LoadString(hInstance, IDC_WINLAUNCHER, szWindowClass, MAX_LOADSTRING);

    MyRegisterClass(hInstance);

    // Perform application initialization:

    if (!InitInstance (hInstance, nCmdShow))

        return FALSE;

    // Init COM

    OleInitialize(NULL);

    hURLBarWnd = CreateWindow(L"EDIT", 0,

                        WS_CHILD | WS_VISIBLE | WS_BORDER | ES_LEFT | ES_AUTOVSCROLL, 

                        0, 0, 0, 0,

                        hMainWnd,

                        0,

                        hInstance, 0);

    DefEditProc = GetWindowLong(hURLBarWnd, GWL_WNDPROC);

    SetWindowLong(hURLBarWnd, GWL_WNDPROC,(long)MyEditProc);

    SetFocus(hURLBarWnd);

    HRESULT hr = WebKitCreateInstance(CLSID_WebView, 0, IID_IWebView, (void**)&gWebView);

    if (FAILED(hr))

        goto exit;

    gWebHost = new WinLauncherWebHost();

    gWebHost->AddRef();

    hr = gWebView->setFrameLoadDelegate(gWebHost);

    if (FAILED(hr))

        goto exit;

    gPrintDelegate = new PrintWebUIDelegate;

    gPrintDelegate->AddRef();

    hr = gWebView->setUIDelegate(gPrintDelegate);

    if (FAILED (hr))

        goto exit;

    hr = gWebView->setHostWindow((OLE_HANDLE) hMainWnd);

    if (FAILED(hr))

        goto exit;

    RECT clientRect;

    GetClientRect(hMainWnd, &clientRect);

    hr = gWebView->initWithFrame(clientRect, 0, 0);

    if (FAILED(hr))

        goto exit;

    IWebFrame* frame;

    hr = gWebView->mainFrame(&frame);

    if (FAILED(hr))

        goto exit;

    static BSTR defaultHTML = SysAllocString(TEXT("<p style=\"background-color: #00FF00\">Testing</p><img src=\"http://webkit.org/images/icon-gold.png\" alt=\"Face\"><div style=\"border: solid blue\" contenteditable=\"true\">div with blue border</div><ul><li>foo<li>bar<li>baz</ul>"));

    frame->loadHTMLString(defaultHTML, 0);

    frame->Release();

    IWebViewPrivate* viewExt;

    hr = gWebView->QueryInterface(IID_IWebViewPrivate, (void**)&viewExt);

    if (FAILED(hr))

        goto exit;

    hr = viewExt->viewWindow((OLE_HANDLE*) &gViewWindow);

    viewExt->Release();

    if (FAILED(hr) || !gViewWindow)

        goto exit;

    resizeSubViews();

    ShowWindow(gViewWindow, nCmdShow);

    UpdateWindow(gViewWindow);

    hAccelTable = LoadAccelerators(hInstance, MAKEINTRESOURCE(IDC_WINLAUNCHER));

    // Main message loop:

    while (GetMessage(&msg, NULL, 0, 0)) {

        if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg)) {

            TranslateMessage(&msg);

            DispatchMessage(&msg);

        }

    }

exit:

    gPrintDelegate->Release();

    gWebView->Release();

    shutDownWebKit();

#ifdef _CRTDBG_MAP_ALLOC

    _CrtDumpMemoryLeaks();

#endif

    // Shut down COM.

    OleUninitialize();

    return static_cast<int>(msg.wParam);

}

ATOM MyRegisterClass(HINSTANCE hInstance)

{

    WNDCLASSEX wcex;

    wcex.cbSize = sizeof(WNDCLASSEX);

    wcex.style          = CS_HREDRAW | CS_VREDRAW;

    wcex.lpfnWndProc    = WndProc;

    wcex.cbClsExtra     = 0;

    wcex.cbWndExtra     = 0;

    wcex.hInstance      = hInstance;

    wcex.hIcon          = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_WINLAUNCHER));

    wcex.hCursor        = LoadCursor(NULL, IDC_ARROW);

    wcex.hbrBackground  = 0;

    wcex.lpszMenuName   = MAKEINTRESOURCE(IDC_WINLAUNCHER);

    wcex.lpszClassName  = szWindowClass;

    wcex.hIconSm        = LoadIcon(wcex.hInstance, MAKEINTRESOURCE(IDI_SMALL));

    return RegisterClassEx(&wcex);

}

BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)

{

   hInst = hInstance; // Store instance handle in our global variable

   hMainWnd = CreateWindow(szWindowClass, szTitle, WS_OVERLAPPEDWINDOW,

      CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, hInstance, NULL);

   if (!hMainWnd)

      return FALSE;

   ShowWindow(hMainWnd, nCmdShow);

   UpdateWindow(hMainWnd);

   return TRUE;

}

static BOOL CALLBACK AbortProc(HDC hDC, int Error)

{

    MSG msg;

    while (::PeekMessage(&msg, 0, 0, 0, PM_REMOVE)) {

        ::TranslateMessage(&msg);

        ::DispatchMessage(&msg);

    }

    return TRUE;

}

static HDC getPrinterDC()

{

    PRINTDLG pdlg;

    memset(&pdlg, 0, sizeof(PRINTDLG));

    pdlg.lStructSize = sizeof(PRINTDLG);

    pdlg.Flags = PD_PRINTSETUP | PD_RETURNDC;

    ::PrintDlg(&pdlg);

    return pdlg.hDC;

}

static void initDocStruct(DOCINFO* di, TCHAR* docname)

{

    memset(di, 0, sizeof(DOCINFO));

    di->cbSize = sizeof(DOCINFO);

    di->lpszDocName = docname;

}

void PrintView(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)

{

    HDC printDC = getPrinterDC();

    if (!printDC) {

        ::MessageBox(0, _T("Error creating printing DC"), _T("Error"), MB_APPLMODAL | MB_OK);

        return;

    }

    if (::SetAbortProc(printDC, AbortProc) == SP_ERROR) {

        ::MessageBox(0, _T("Error setting up AbortProc"), _T("Error"), MB_APPLMODAL | MB_OK);

        return;

    }

    IWebFrame* frame = 0;

    if (FAILED(gWebView->mainFrame(&frame)))

        goto exit;

    IWebFramePrivate* framePrivate = 0;

    if (FAILED(frame->QueryInterface(&framePrivate)))

        goto exit;

    framePrivate->setInPrintingMode(TRUE, printDC);

    UINT pageCount = 0;

    framePrivate->getPrintedPageCount(printDC, &pageCount);

    DOCINFO di;

    initDocStruct(&di, _T("WebKit Doc"));

    ::StartDoc(printDC, &di);

    // FIXME: Need CoreGraphics implementation

    void* graphicsContext = 0;

    for (size_t page = 1; page <= pageCount; ++page) {

        ::StartPage(printDC);

        framePrivate->spoolPages(printDC, page, page, graphicsContext);

        ::EndPage(printDC);

    }

    framePrivate->setInPrintingMode(FALSE, printDC);

    ::EndDoc(printDC);

    ::DeleteDC(printDC);

exit:

    if (frame)

        frame->Release();

    if (framePrivate)

        framePrivate->Release();

}

LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)

{

    int wmId, wmEvent;

    PAINTSTRUCT ps;

    HDC hdc;

    switch (message) {

    case WM_COMMAND:

        wmId    = LOWORD(wParam);

        wmEvent = HIWORD(wParam);

        // Parse the menu selections:

        switch (wmId) {

            case IDM_ABOUT:

                DialogBox(hInst, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, About);

                break;

            case IDM_EXIT:

                DestroyWindow(hWnd);

                break;

            case IDM_PRINT:

                PrintView(hWnd, message, wParam, lParam);

                break;

            default:

                return DefWindowProc(hWnd, message, wParam, lParam);

        }

        break;

    case WM_DESTROY:

        PostQuitMessage(0);

        break;

    case WM_SIZE:

        if (!gWebView)

            break;

        resizeSubViews();

        break; 

    default:

        return DefWindowProc(hWnd, message, wParam, lParam);

    }

    return 0;

}

#define MAX_URL_LENGTH  1024

LRESULT CALLBACK MyEditProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)

{

    switch (message) {

        case WM_CHAR:

            if (wParam == 13) { // Enter Key

                wchar_t strPtr[MAX_URL_LENGTH];

                *((LPWORD)strPtr) = MAX_URL_LENGTH; 

                int strLen = SendMessage(hDlg, EM_GETLINE, 0, (LPARAM)strPtr);

                BSTR bstr = SysAllocStringLen(strPtr, strLen);

                loadURL(bstr);

                SysFreeString(bstr);

                return 0;

            } else

                return (LRESULT)CallWindowProc((WNDPROC)DefEditProc,hDlg,message,wParam,lParam);

            break;

        default:

             return (LRESULT)CallWindowProc((WNDPROC)DefEditProc,hDlg,message,wParam,lParam);

        break;

    }

    return 0;

}

// Message handler for about box.

INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)

{

    UNREFERENCED_PARAMETER(lParam);

    switch (message) {

    case WM_INITDIALOG:

        return (INT_PTR)TRUE;

    case WM_COMMAND:

        if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL) {

            EndDialog(hDlg, LOWORD(wParam));

            return (INT_PTR)TRUE;

        }

        break;

    }

    return (INT_PTR)FALSE;

}

static void loadURL(BSTR urlBStr)

{

    IWebFrame* frame = 0;

    IWebMutableURLRequest* request = 0;

    static BSTR methodBStr = SysAllocString(TEXT("GET"));

    if (urlBStr && urlBStr[0] && (PathFileExists(urlBStr) || PathIsUNC(urlBStr))) {

        TCHAR fileURL[INTERNET_MAX_URL_LENGTH];

        DWORD fileURLLength = sizeof(fileURL)/sizeof(fileURL[0]);

        if (SUCCEEDED(UrlCreateFromPath(urlBStr, fileURL, &fileURLLength, 0)))

            SysReAllocString(&urlBStr, fileURL);

    }

    HRESULT hr = gWebView->mainFrame(&frame);

    if (FAILED(hr))

        goto exit;

    hr = WebKitCreateInstance(CLSID_WebMutableURLRequest, 0, IID_IWebMutableURLRequest, (void**)&request);

    if (FAILED(hr))

        goto exit;

    hr = request->initWithURL(urlBStr, WebURLRequestUseProtocolCachePolicy, 60);

    if (FAILED(hr))

        goto exit;

    hr = request->setHTTPMethod(methodBStr);

    if (FAILED(hr))

        goto exit;

    hr = frame->loadRequest(request);

    if (FAILED(hr))

        goto exit;

    SetFocus(gViewWindow);

exit:

    if (frame)

        frame->Release();

    if (request)

        request->Release();

}