fpdfsdk/include/javascript/JS_Object.h

// Copyright 2014 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com

#ifndef FPDFSDK_INCLUDE_JAVASCRIPT_JS_OBJECT_H_
#define FPDFSDK_INCLUDE_JAVASCRIPT_JS_OBJECT_H_

#include "../fsdk_define.h"       // For FX_UINT
#include "../fsdk_mgr.h"          // For CPDFDoc_Environment
#include "../fx_systemhandler.h"  // For IFX_SystemHandler
#include "../jsapi/fxjs_v8.h"

class CPDFSDK_PageView;
class CJS_Object;
class CJS_Timer;
class CJS_Context;

class CJS_EmbedObj {
 public:
  CJS_EmbedObj(CJS_Object* pJSObject);
  virtual ~CJS_EmbedObj();

  virtual void TimerProc(CJS_Timer* pTimer){};

  CJS_Timer* BeginTimer(CPDFDoc_Environment* pApp, FX_UINT nElapse);
  void EndTimer(CJS_Timer* pTimer);

  CJS_Object* GetJSObject() { return m_pJSObject; };
  operator CJS_Object*() { return m_pJSObject; };

  CPDFSDK_PageView* JSGetPageView(IFXJS_Context* cc);
  int MsgBox(CPDFDoc_Environment* pApp,
             CPDFSDK_PageView* pPageView,
             const FX_WCHAR* swMsg,
             const FX_WCHAR* swTitle = NULL,
             FX_UINT nType = 0,
             FX_UINT nIcon = 0);
  void Alert(CJS_Context* pContext, const FX_WCHAR* swMsg);

 protected:
  CJS_Object* m_pJSObject;
};

class CJS_Object {
 public:
  CJS_Object(JSFXObject pObject);
  virtual ~CJS_Object(void);

  void MakeWeak();
  void Dispose();

  virtual FX_BOOL IsType(const FX_CHAR* sClassName) { return TRUE; };
  virtual CFX_ByteString GetClassName() { return ""; };

  virtual FX_BOOL InitInstance(IFXJS_Context* cc) { return TRUE; };
  virtual FX_BOOL ExitInstance() { return TRUE; };

  operator JSFXObject() {
    return v8::Local<v8::Object>::New(m_pIsolate, m_pObject);
  }
  operator CJS_EmbedObj*() { return m_pEmbedObj; };

  void SetEmbedObject(CJS_EmbedObj* pObj) { m_pEmbedObj = pObj; };
  CJS_EmbedObj* GetEmbedObject() { return m_pEmbedObj; };

  static CPDFSDK_PageView* JSGetPageView(IFXJS_Context* cc);
  static int MsgBox(CPDFDoc_Environment* pApp,
                    CPDFSDK_PageView* pPageView,
                    const FX_WCHAR* swMsg,
                    const FX_WCHAR* swTitle = NULL,
                    FX_UINT nType = 0,
                    FX_UINT nIcon = 0);
  static void Alert(CJS_Context* pContext, const FX_WCHAR* swMsg);

  v8::Isolate* GetIsolate() { return m_pIsolate; }

 protected:
  CJS_EmbedObj* m_pEmbedObj;
  v8::Global<v8::Object> m_pObject;
  v8::Isolate* m_pIsolate;
};

struct JS_TIMER_MAP {
  FX_UINT nID;
  CJS_Timer* pTimer;
};

typedef CFX_ArrayTemplate<JS_TIMER_MAP*> CTimerMapArray;

struct JS_TIMER_MAPARRAY {
 public:
  JS_TIMER_MAPARRAY() {}

  ~JS_TIMER_MAPARRAY() { Reset(); }

  void Reset() {
    for (int i = 0, sz = m_Array.GetSize(); i < sz; i++)
      delete m_Array.GetAt(i);

    m_Array.RemoveAll();
  }

  void SetAt(FX_UINT nIndex, CJS_Timer* pTimer) {
    int i = Find(nIndex);

    if (i >= 0) {
      if (JS_TIMER_MAP* pMap = m_Array.GetAt(i))
        pMap->pTimer = pTimer;
    } else {
      if (JS_TIMER_MAP* pMap = new JS_TIMER_MAP) {
        pMap->nID = nIndex;
        pMap->pTimer = pTimer;
        m_Array.Add(pMap);
      }
    }
  }

  CJS_Timer* GetAt(FX_UINT nIndex) {
    int i = Find(nIndex);

    if (i >= 0) {
      if (JS_TIMER_MAP* pMap = m_Array.GetAt(i))
        return pMap->pTimer;
    }
    return NULL;
  }

  void RemoveAt(FX_UINT nIndex) {
    int i = Find(nIndex);

    if (i >= 0) {
      delete m_Array.GetAt(i);
      m_Array.RemoveAt(i);
    }
    // To prevent potential fake memory leak reported by vc6.
    if (m_Array.GetSize() == 0)
      m_Array.RemoveAll();
  }

  int Find(FX_UINT nIndex) {
    for (int i = 0, sz = m_Array.GetSize(); i < sz; i++) {
      if (JS_TIMER_MAP* pMap = m_Array.GetAt(i)) {
        if (pMap->nID == nIndex)
          return i;
      }
    }

    return -1;
  }

  CTimerMapArray m_Array;
};

JS_TIMER_MAPARRAY& GetTimeMap();

class CJS_Runtime;

class CJS_Timer {
 public:
  CJS_Timer(CJS_EmbedObj* pObj, CPDFDoc_Environment* pApp)
      : m_nTimerID(0),
        m_pEmbedObj(pObj),
        m_bProcessing(FALSE),
        m_dwStartTime(0),
        m_dwTimeOut(0),
        m_dwElapse(0),
        m_pRuntime(NULL),
        m_nType(0),
        m_pApp(pApp) {}

  virtual ~CJS_Timer() { KillJSTimer(); }

 public:
  FX_UINT SetJSTimer(FX_UINT nElapse) {
    if (m_nTimerID)
      KillJSTimer();
    IFX_SystemHandler* pHandler = m_pApp->GetSysHandler();
    m_nTimerID = pHandler->SetTimer(nElapse, TimerProc);
    GetTimeMap().SetAt(m_nTimerID, this);
    m_dwElapse = nElapse;
    return m_nTimerID;
  };

  void KillJSTimer() {
    if (m_nTimerID) {
      if (m_pApp == NULL) {
        GetTimeMap().RemoveAt(m_nTimerID);
        m_nTimerID = 0;
        return;
      }
      IFX_SystemHandler* pHandler = m_pApp->GetSysHandler();
      pHandler->KillTimer(m_nTimerID);
      GetTimeMap().RemoveAt(m_nTimerID);
      m_nTimerID = 0;
    }
  };

  void SetType(int nType) { m_nType = nType; }

  int GetType() const { return m_nType; }

  void SetStartTime(FX_DWORD dwStartTime) { m_dwStartTime = dwStartTime; }

  FX_DWORD GetStartTime() const { return m_dwStartTime; }

  void SetTimeOut(FX_DWORD dwTimeOut) { m_dwTimeOut = dwTimeOut; }

  FX_DWORD GetTimeOut() const { return m_dwTimeOut; }

  void SetRuntime(CJS_Runtime* pRuntime) { m_pRuntime = pRuntime; }

  CJS_Runtime* GetRuntime() const { return m_pRuntime; }

  void SetJScript(const CFX_WideString& script) { m_swJScript = script; }

  CFX_WideString GetJScript() const { return m_swJScript; }

  static void TimerProc(int idEvent) {
    if (CJS_Timer* pTimer = GetTimeMap().GetAt(idEvent)) {
      if (!pTimer->m_bProcessing) {
        pTimer->m_bProcessing = TRUE;
        if (pTimer->m_pEmbedObj)
          pTimer->m_pEmbedObj->TimerProc(pTimer);
        pTimer->m_bProcessing = FALSE;
      } else {
        //	TRACE(L"BUSY!\n");
      }
    }
  };

 private:
  FX_UINT m_nTimerID;
  CJS_EmbedObj* m_pEmbedObj;
  FX_BOOL m_bProcessing;

  // data
  FX_DWORD m_dwStartTime;
  FX_DWORD m_dwTimeOut;
  FX_DWORD m_dwElapse;
  CJS_Runtime* m_pRuntime;
  CFX_WideString m_swJScript;
  int m_nType;  // 0:Interval; 1:TimeOut

  CPDFDoc_Environment* m_pApp;
};

#endif  // FPDFSDK_INCLUDE_JAVASCRIPT_JS_OBJECT_H_