gecko-dev/webshell/tests/viewer/winmain.cpp

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
 *
 * The contents of this file are subject to the Netscape Public License
 * Version 1.0 (the "NPL"); you may not use this file except in
 * compliance with the NPL.  You may obtain a copy of the NPL at
 * http://www.mozilla.org/NPL/
 *
 * Software distributed under the NPL is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
 * for the specific language governing rights and limitations under the
 * NPL.
 *
 * The Initial Developer of this code under the NPL is Netscape
 * Communications Corporation.  Portions created by Netscape are
 * Copyright (C) 1998 Netscape Communications Corporation.  All Rights
 * Reserved.
 */

// Win32 specific viewer driver

#include <windows.h>
#include <crtdbg.h>
#include <stdio.h>
#include "resources.h"
#include "jsconsres.h"
#include "JSConsole.h"
#include "nsViewer.h"
#include "nsGlobalVariables.h"
#include "nsIDocument.h"
#include "nsIURL.h"
#include "nsVoidArray.h"
#include "nsCRT.h"
#include "prenv.h"
#include "nsIScriptContext.h"
#include "nsIScriptContextOwner.h"
#include "nsITimer.h"

#include "nsIBrowserWindow.h"
#include "plevent.h"

#include <fstream.h>
#include "nsIParser.h"
#include "nsHTMLContentSinkStream.h"
#include "nsXIFDTD.h"

// Debug Robot options
static int gDebugRobotLoads = 5000;
static char gVerifyDir[_MAX_PATH];
static BOOL gVisualDebug = TRUE;

static NS_DEFINE_IID(kIScriptContextOwnerIID, NS_ISCRIPTCONTEXTOWNER_IID);

extern "C" BOOL CreateSiteWalkerDialog(HWND hParent, WindowData * aWinData);
// DebugRobot call
extern "C" NS_EXPORT int DebugRobot(
   nsVoidArray * workList, nsIWebShell * ww, int imax, char * verify_dir, void (*yieldProc)(const char *));

// Temporary Netlib stuff...
/* XXX: Don't include net.h... */
extern "C" {
extern int  NET_PollSockets();
};

#define DEBUG_EMPTY "(none)"


class nsWin32Viewer : public nsViewer {
    // From nsViewer
  public:
    virtual void AddMenu(nsIWidget* aMainWindow, PRBool aForPrintPreview);
    virtual void ShowConsole(WindowData* aWindata);
    virtual void DoDebugRobot(WindowData* aWindata);
    virtual void DoDebugSave(WindowData* aWindata);
    virtual void DoToggleSelection(WindowData* aWindata);
    virtual void DoSiteWalker(WindowData* aWindata);
    virtual void CopySelection(WindowData* aWindata);
    virtual void Destroy(WindowData* wd);
    virtual void CloseConsole();
    virtual void Stop();
    virtual void CrtSetDebug(PRUint32 aNewFlags);
    virtual void AddRelatedLink(char * name, char * url);
    virtual void ResetRelatedLinks();

      // Utilities
    virtual void CopyTextContent(WindowData* wd, HWND aHWnd);
};

static HANDLE gInstance, gPrevInstance;

//-----------------------------------------------------------------
// JSConsole support
//-----------------------------------------------------------------

// JSConsole window
JSConsole *gConsole = NULL;

static char* class1Name = "Viewer";
static char* class2Name = "PrintPreview";

void DestroyConsole()
{
 if (gConsole) {
    gConsole->SetNotification(NULL);
    delete gConsole;
    gConsole = NULL;
  }
}

HWND ghWnd = 0;
#define RL_MENU_POS 4

void nsWin32Viewer::AddRelatedLink(char * name, char * url)
{
   if (name) {
      HMENU hMenu = GetMenu(ghWnd);
      HMENU hDropDown = GetSubMenu(hMenu,RL_MENU_POS);
      int i = GetMenuItemCount(hDropDown);
      AppendMenu(hDropDown, MF_STRING, VIEWER_RL_BASE+i, (LPCSTR)name);
   }
}

void nsWin32Viewer::ResetRelatedLinks()
{
   HMENU hMenu = GetMenu(ghWnd);
   HMENU hDropDown = GetSubMenu(hMenu,RL_MENU_POS);
   while (DeleteMenu(hDropDown,0,MF_BYPOSITION))
      ;
}

//-----------------------------------------------------------------
// CRT Debug
//-----------------------------------------------------------------

void nsWin32Viewer::CrtSetDebug(PRUint32 aNewFlags)
{
#ifndef MOZ_NO_DEBUG_RTL
  int oldFlags = _CrtSetDbgFlag(_CRTDBG_REPORT_FLAG);
  _CrtSetDbgFlag(aNewFlags);
  printf("Note: crt flags: old=%x new=%x\n", oldFlags, aNewFlags);
#endif
}


//-----------------------------------------------------------------
// Debug Robot support
//-----------------------------------------------------------------

void yieldProc(const char * str)
{
  // Process messages
  MSG msg;
  while (PeekMessage(&msg, NULL, 0, 0, PM_NOREMOVE)) {
    GetMessage(&msg, NULL, 0, 0);
    if (!JSConsole::sAccelTable ||
        !gConsole ||
        !gConsole->GetMainWindow() ||
        !TranslateAccelerator(gConsole->GetMainWindow(), JSConsole::sAccelTable, &msg)) {
      TranslateMessage(&msg);
      DispatchMessage(&msg);
      /* Pump Netlib... */
      NET_PollSockets();
    }
  }
}

/* Debug Robot Dialog options */

BOOL CALLBACK DlgProc(HWND hDlg, UINT msg, WPARAM wParam,LPARAM lParam)
{
   BOOL translated = FALSE;
   HWND hwnd;
   switch (msg)
   {
      case WM_INITDIALOG:
         {
            SetDlgItemInt(hDlg,IDC_PAGE_LOADS,5000,FALSE);
            char * text = PR_GetEnv("VERIFY_PARSER");
            SetDlgItemText(hDlg,IDC_VERIFICATION_DIRECTORY,text ? text : DEBUG_EMPTY);
            hwnd = GetDlgItem(hDlg,IDC_UPDATE_DISPLAY);
            SendMessage(hwnd,BM_SETCHECK,TRUE,0);
         }
         return FALSE;
      case WM_COMMAND:
         switch (LOWORD(wParam))
         {
            case IDOK:
               gDebugRobotLoads = GetDlgItemInt(hDlg,IDC_PAGE_LOADS,&translated,FALSE);
               GetDlgItemText(hDlg, IDC_VERIFICATION_DIRECTORY, gVerifyDir, sizeof(gVerifyDir));
               if (!strcmp(gVerifyDir,DEBUG_EMPTY))
                  gVerifyDir[0] = '\0';
               hwnd = GetDlgItem(hDlg,IDC_UPDATE_DISPLAY);
               gVisualDebug = (BOOL)SendMessage(hwnd,BM_GETCHECK,0,0);
               EndDialog(hDlg,IDOK);
               break;
            case IDCANCEL:
               EndDialog(hDlg,IDCANCEL);
               break;
         }
         break;
      default:
         return FALSE;
   }
   return TRUE;
}


BOOL CreateRobotDialog(HWND hParent)
{
   BOOL result = (DialogBox(gInstance,MAKEINTRESOURCE(IDD_DEBUGROBOT),hParent,(DLGPROC)DlgProc) == IDOK);
   return result;
}


void AddViewerMenu(HINSTANCE hInstance, nsIWidget* aWidget, LPCTSTR lpMenuName)
{
  HMENU menu = ::LoadMenu(hInstance,lpMenuName);
  HWND hwnd = aWidget->GetNativeData(NS_NATIVE_WIDGET);
  ghWnd = hwnd;
  ::SetMenu(hwnd, menu);
}

void nsWin32Viewer::AddMenu(nsIWidget* aMainWindow, PRBool aForPrintPreview)
{
   AddViewerMenu(gInstance, aMainWindow,
                 aForPrintPreview ? class2Name : class1Name);
}

//-----------------------------------------------------------------
// nsWin32Viewer Implementation
//-----------------------------------------------------------------

void nsWin32Viewer::ShowConsole(WindowData* aWinData)
{
    HWND hWnd = aWinData->windowWidget->GetNativeData(NS_NATIVE_WIDGET);
    if (!gConsole) {

      // load the accelerator table for the console
      if (!JSConsole::sAccelTable) {
        JSConsole::sAccelTable = LoadAccelerators(gInstance,
                                                  MAKEINTRESOURCE(ACCELERATOR_TABLE));
      }
      
      nsIScriptContextOwner *owner = nsnull;
      nsIScriptContext *context = nsnull;        
      if (NS_OK == aWinData->observer->QueryInterface(kIScriptContextOwnerIID, (void **)&owner)) {
        if (NS_OK == owner->GetScriptContext(&context)) {

          // create the console
          gConsole = JSConsole::CreateConsole();
          gConsole->SetContext(context);
          // lifetime of the context is still unclear at this point.
          // Anyway, as long as the web widget is alive the context is alive.
          // Maybe the context shouldn't even be RefCounted
          context->Release();
          gConsole->SetNotification(DestroyConsole);
        }
        
        NS_RELEASE(owner);
      }
      else {
        MessageBox(hWnd, "Unable to load JavaScript", "Viewer Error", MB_ICONSTOP);
      }
    }
}

void nsWin32Viewer::CloseConsole()
{
  DestroyConsole();
}

void nsWin32Viewer::DoSiteWalker(WindowData* aWinData)
{
    if ((nsnull != aWinData) && (nsnull != aWinData->observer)) {
        HWND hWnd = aWinData->windowWidget->GetNativeData(NS_NATIVE_WIDGET);
        CreateSiteWalkerDialog(hWnd, aWinData);
    }
   
}

void nsWin32Viewer::DoDebugRobot(WindowData* aWindata)
{
  if ((nsnull != aWindata) && (nsnull != aWindata->observer)) {
    if (CreateRobotDialog(aWindata->windowWidget->GetNativeData(NS_NATIVE_WIDGET)))
    {
#if XXX_fix_me
      nsIDocument* doc = aWindata->observer->mWebWidget->GetDocument();
      if (nsnull!=doc) {
        const char * str = doc->GetDocumentURL()->GetSpec();
        nsVoidArray * gWorkList = new nsVoidArray();
        gWorkList->AppendElement(new nsString(str));
        DebugRobot( 
          gWorkList, 
          gVisualDebug ? aWindata->observer->mWebWidget : nsnull, 
          gDebugRobotLoads, 
          PL_strdup(gVerifyDir),
          yieldProc);
      }
#endif
    }
  }
}


void nsWin32Viewer::DoDebugSave(WindowData* aWindata)
{
  if ((nsnull != aWindata) && (nsnull != aWindata->observer)) 
  {
    nsIWebShell*  webShell = nsnull;
    nsIPresShell* presShell = nsnull;
    nsIDocument*  doc = nsnull;
    
    nsIWebShell*  webShell = aWindata->observer->mWebWidget;  
    
    if (webShell != nsnull)
      presShell = GetPresShell(aWebWidget);
    
    if (nsnull != presShell)
      doc = shell->GetDocument();

 
    if (doc != nsnull)
    {
      nsString buffer;
      doc->CreateXIF(buffer,PR_FALSE);


      nsIParser* parser;
      nsresult rv = NS_NewParser(&parser);
      if (NS_OK == rv) {
        nsIHTMLContentSink* sink = nsnull;
        
        rv = NS_New_HTML_ContentSinkStream(&sink);

        if (NS_OK == rv) {

#if defined(WIN32)
/*
          const char* filename="c:\\temp\\save.html";
          ofstream    out(filename);
          sink->SetOutputStream(out);
*/
#endif
          parser->SetContentSink(sink);
          
          nsIDTD* dtd = nsnull;
          rv = NS_NewXIFDTD(&dtd);
          if (NS_OK == rv) 
          {
            parser->RegisterDTD(dtd);
            dtd->SetContentSink(sink);
            dtd->SetParser(parser);
            parser->Parse(buffer, PR_TRUE);           
           }
          NS_IF_RELEASE(dtd);
          NS_IF_RELEASE(sink);
        }
        NS_RELEASE(parser);
      }
    }
  }
}


void nsWin32Viewer::DoToggleSelection(WindowData* aWindata)
{
  if ((nsnull != aWindata) && (nsnull != aWindata->observer)) 
  {
    nsIWebShell*  webShell = nsnull;
    nsIPresShell* presShell = nsnull;
    nsIDocument*  doc = nsnull;
    
    nsIWebShell*  webShell = aWindata->observer->mWebWidget;  
    
    if (webShell != nsnull)
      presShell = GetPresShell(aWebWidget);
    
    if (nsnull != presShell)
      doc = shell->GetDocument();
 
    if (doc != nsnull)
    {
      PRBool  current = doc->GetDisplaySelection();
      doc->SetDisplaySelection(!current);
    }
  }
}

// Selects all the Content
void nsWin32Viewer::CopyTextContent(WindowData* wd, HWND aHWnd)
{
  HGLOBAL     hGlobalMemory;
  PSTR        pGlobalMemory;

  if (wd->observer != nsnull) {
#ifdef XXX_fix_me
    nsIDocument* doc = wd->observer->mWebWidget->GetDocument();
    if (doc != nsnull) {
      // Get Text from Selection
      nsString text;
      doc->GetSelectionText(text);

      // Copy text to Global Memory Area
      hGlobalMemory = (HGLOBAL)GlobalAlloc(GHND, text.Length()+1);
      if (hGlobalMemory != NULL) {
        pGlobalMemory = (PSTR) GlobalLock(hGlobalMemory);
        char * str = text.ToNewCString();
        char * s   = str;
        for (int i=0;i<text.Length();i++) {
          *pGlobalMemory++ = *s++;
        }
        delete str;

        // Put data on Clipboard
        GlobalUnlock(hGlobalMemory);
        OpenClipboard(aHWnd);
        EmptyClipboard();
        SetClipboardData(CF_TEXT, hGlobalMemory);
        CloseClipboard();
      }

      NS_IF_RELEASE(doc);
    }
#endif
  }
}

void nsWin32Viewer::CopySelection(WindowData* aWindata)
{
  CopyTextContent(aWindata, aWindata->windowWidget->GetNativeData(NS_NATIVE_WIDGET));
}

void nsWin32Viewer::Stop()
{
  PostQuitMessage(0);
}

void nsWin32Viewer::Destroy(WindowData* wd)
{
  CloseConsole();
  nsViewer::Destroy(wd);
}

static nsITimer* gNetTimer;

static void
PollNet(nsITimer *aTimer, void *aClosure)
{
  NET_PollSockets();
  NS_IF_RELEASE(gNetTimer);
  if (NS_OK == NS_NewTimer(&gNetTimer)) {
    gNetTimer->Init(PollNet, nsnull, 1000 / 50);
  }
}

int PASCAL
RunViewer(HANDLE instance, HANDLE prevInstance, LPSTR cmdParam, int nCmdShow, nsWin32Viewer* aViewer)
{
  gInstance = instance;
  gPrevInstance = prevInstance;

  SetViewer(aViewer);

  nsIWidget *mainWindow = nsnull;
  nsDocLoader* dl = aViewer->SetupViewer(&mainWindow, 0, 0);
 
  // Process messages
  MSG msg;
  PollNet(0, 0);
  while (::GetMessage(&msg, NULL, 0, 0)) {
    if (!JSConsole::sAccelTable ||
        !gConsole ||
        !gConsole->GetMainWindow() ||
        !TranslateAccelerator(gConsole->GetMainWindow(), JSConsole::sAccelTable, &msg)) {
      TranslateMessage(&msg);
      DispatchMessage(&msg);
      NET_PollSockets();
    }
  }

  aViewer->CleanupViewer(dl);

  return msg.wParam;
}


//------------------------------------------------------------------
// Win32 Main
//------------------------------------------------------------------

void main(int argc, char **argv)
{
#ifdef XP_PC
  PL_InitializeEventsLib("");
#endif
  nsWin32Viewer* viewer = new nsWin32Viewer();
  viewer->ProcessArguments(argc, argv);
  RunViewer(GetModuleHandle(NULL), NULL, 0, SW_SHOW, viewer);
}

int PASCAL
WinMain(HANDLE instance, HANDLE prevInstance, LPSTR cmdParam, int nCmdShow)
{
  nsWin32Viewer* viewer = new nsWin32Viewer();
  return(RunViewer(instance, prevInstance, cmdParam, nCmdShow, viewer));
}

static WindowData * gWinData;
static int gTop100Pointer = 0;
static char * gTop100List[] = {
   "http://www.yahoo.com",
   "http://www.netscape.com",
   "http://www.microsoft.com",
   "http://www.excite.com",
   "http://www.mckinley.com",
   "http://www.city.net",
   "http://www.webcrawler.com",
   "http://www.mirabilis.com",
   "http://www.infoseek.com",
   "http://www.pathfinder.com",
   "http://www.warnerbros.com",
   "http://www.cnn.com",
   "http://www.altavista.digital.com",
   "http://www.altavista.com",
   "http://www.usatoday.com",
   "http://www.disney.com",
   "http://www.starwave.com",
   "http://www.hotwired.com",
   "http://www.hotbot.com",
   "http://www.lycos.com",
   "http://www.pointcom.com",
   "http://www.cnet.com",
   "http://www.search.com",
   "http://www.news.com",
   "http://www.download.com",
   "http://www.geocities.com",
   "http://www.aol.com",
   "http://members.aol.com",
   "http://www.imdb.com",
   "http://uk.imdb.com",
   "http://macromedia.com",
   "http://www.infobeat.com",
   "http://www.fxweb.com",
   "http://www.whowhere.com",
   "http://www.real.com",
   "http://www.sportsline.com",
   "http://www.dejanews.com",
   "http://www.the-park.com",
   "http://www.cmpnet.com",
   "http://www.go2net.com",
   "http://www.metacrawler.com",
   "http://www.playsite.com",
   "http://www.stocksite.com",
   "http://www.sony.com",
   "http://www.music.sony.com",
   "http://www.station.sony.com",
   "http://www.scea.sony.com",
   "http://www.infospace.com",
   "http://www.zdnet.com",
   "http://www.hotfiles.com",
   "http://www.chathouse.com",
   "http://www.looksmart.com",
   "http://www.iamginegames.com",
   "http://www.macaddict.com",
   "http://www.rsac.org",
   "http://www.apple.com",
   "http://www.beseen.com",
   "http://www.dogpile.com",
   "http://www.xoom.com",
   "http://www.tucows.com",
   "http://www.freethemes.com",
   "http://www.winfiles.com",
   "http://www.vservers.com",
   "http://www.mtv.com",
   "http://www.the-xfiles.com",
   "http://www.datek.com",
   "http://www.cyberthrill.com",
   "http://www.surplusdirect.com",
   "http://www.tomshardware.com",
   "http://www.bigyellow.com",
   "http://www.100hot.com",
   "http://www.messagemates.com",
   "http://www.onelist.com",
   "http://www.bluemountain.com",
   "http://www.ea.com",
   "http://www.bullfrog.co.uk",
   "http://www.travelocity.com",
   "http://www.ibm.com",
   "http://www.bigcharts.com",
   "http://www.davesclassics.com",
   "http://www.goto.com",
   "http://www.weather.com",
   "http://www.gamespot.com",
   "http://www.bloomberg.com",
   "http://www.winzip.com",
   "http://www.filez.com",
   "http://www.westwood.com",
   "http://www.internet.com",
   "http://www.cardmaster.com",
   "http://www.creaf.com",
   "http://netaddress.usa.net",
   "http://www.occ.com",
   "http://www.as.org",
   "http://www.amazon.com",
   "http://www.drudgereport.com",
   "http://www.hardradio.com",
   "http://www.intel.com",
   "http://www.mp3.com",
   "http://www.ebay.com",
   "http://www.msn.com",
   "http://www.fifa.com",
   "http://www.attitude.com",
   "http://www.happypuppy.com",
   "http://www.gamesdomain.com",
   "http://www.onsale.com",
   "http://www.tm.com",
   "http://www.xlnc1.com",
   "http://www.greatsports.com",
   "http://www.discovery.com",
   "http://www.nai.com",
   "http://www.nasa.gov",
   "http://www.ogr.com",
   "http://www.warzone.com",
   "http://www.gamestats.com",
   "http://www.winamp.com",
   "http://java.sun.com",
   "http://www.hp.com",
   "http://www.cdnow.com",
   "http://www.nytimes.com",
   "http://www.majorleaguebaseball.com",
   "http://www.washingtonpost.com",
   "http://www.planetquake.com",
   "http://www.wsj.com",
   "http://www.slashdot.org",
   "http://www.adobe.com",
   "http://www.quicken.com",
   "http://www.talkcity.com",
   "http://www.developer.com",
   "http://www.mapquest.com",
   0
   };



BOOL CALLBACK SiteWalkerDlgProc(HWND hDlg, UINT msg, WPARAM wParam,LPARAM lParam)
{
   BOOL translated = FALSE;
   switch (msg)
   {
      case WM_INITDIALOG:
         {
            SetDlgItemText(hDlg,IDC_SITE_NAME, gTop100List[gTop100Pointer]);
            EnableWindow(GetDlgItem(hDlg,ID_SITE_PREVIOUS),TRUE);
            if (gWinData && gWinData->mViewer)
               gWinData->mViewer->GoTo(gTop100List[gTop100Pointer]);
         }
         return FALSE;
      case WM_COMMAND:
         switch (LOWORD(wParam))
         {
            case ID_SITE_NEXT:
               {
                  char * p = gTop100List[++gTop100Pointer];
                  if (p) {
                     EnableWindow(GetDlgItem(hDlg,ID_SITE_NEXT),TRUE);
                     SetDlgItemText(hDlg,IDC_SITE_NAME, p);
                     if (gWinData && gWinData->mViewer)
                        gWinData->mViewer->GoTo(gTop100List[gTop100Pointer]);
                  }
                  else  {
                     EnableWindow(GetDlgItem(hDlg,ID_SITE_NEXT),FALSE);
                     EnableWindow(GetDlgItem(hDlg,ID_SITE_PREVIOUS),TRUE);
                     SetDlgItemText(hDlg,IDC_SITE_NAME, "[END OF LIST]");
                  }
               }
               break;
            case ID_SITE_PREVIOUS:
               {
                  if (gTop100Pointer > 0) {
                     EnableWindow(GetDlgItem(hDlg,ID_SITE_PREVIOUS),TRUE);
                     SetDlgItemText(hDlg,IDC_SITE_NAME, gTop100List[--gTop100Pointer]);
                     if (gWinData && gWinData->mViewer)
                        gWinData->mViewer->GoTo(gTop100List[gTop100Pointer]);
                  }
                  else  {
                     EnableWindow(GetDlgItem(hDlg,ID_SITE_PREVIOUS),FALSE);
                     EnableWindow(GetDlgItem(hDlg,ID_SITE_NEXT),TRUE);
                  }
               }
               break;
            case ID_EXIT:
               EndDialog(hDlg,IDCANCEL);
               break;
         }
         break;
      default:
         return FALSE;
   }
   return TRUE;
}


BOOL CreateSiteWalkerDialog(HWND hParent, WindowData * aWinData)
{
   gWinData = aWinData;
   BOOL result = (DialogBox(gInstance,MAKEINTRESOURCE(IDD_SITEWALKER),hParent,(DLGPROC)SiteWalkerDlgProc) == IDOK);
   return result;
}