Remove if checks after new.
[pdfium.git] / fpdfsdk / include / javascript / JS_Object.h
1 // Copyright 2014 PDFium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #ifndef FPDFSDK_INCLUDE_JAVASCRIPT_JS_OBJECT_H_
8 #define FPDFSDK_INCLUDE_JAVASCRIPT_JS_OBJECT_H_
9
10 #include "../fsdk_define.h"       // For FX_UINT
11 #include "../fsdk_mgr.h"          // For CPDFDoc_Environment
12 #include "../fx_systemhandler.h"  // For IFX_SystemHandler
13 #include "../jsapi/fxjs_v8.h"
14
15 class CPDFSDK_PageView;
16 class CJS_Object;
17 class CJS_Timer;
18 class CJS_Context;
19
20 class CJS_EmbedObj {
21  public:
22   CJS_EmbedObj(CJS_Object* pJSObject);
23   virtual ~CJS_EmbedObj();
24
25   virtual void TimerProc(CJS_Timer* pTimer){};
26
27   CJS_Timer* BeginTimer(CPDFDoc_Environment* pApp, FX_UINT nElapse);
28   void EndTimer(CJS_Timer* pTimer);
29
30   CJS_Object* GetJSObject() { return m_pJSObject; };
31   operator CJS_Object*() { return m_pJSObject; };
32
33   CPDFSDK_PageView* JSGetPageView(IFXJS_Context* cc);
34   int MsgBox(CPDFDoc_Environment* pApp,
35              CPDFSDK_PageView* pPageView,
36              const FX_WCHAR* swMsg,
37              const FX_WCHAR* swTitle = NULL,
38              FX_UINT nType = 0,
39              FX_UINT nIcon = 0);
40   void Alert(CJS_Context* pContext, const FX_WCHAR* swMsg);
41
42  protected:
43   CJS_Object* m_pJSObject;
44 };
45
46 class CJS_Object {
47  public:
48   CJS_Object(JSFXObject pObject);
49   virtual ~CJS_Object(void);
50
51   void MakeWeak();
52   void Dispose();
53
54   virtual FX_BOOL IsType(const FX_CHAR* sClassName) { return TRUE; };
55   virtual CFX_ByteString GetClassName() { return ""; };
56
57   virtual FX_BOOL InitInstance(IFXJS_Context* cc) { return TRUE; };
58   virtual FX_BOOL ExitInstance() { return TRUE; };
59
60   operator JSFXObject() {
61     return v8::Local<v8::Object>::New(m_pIsolate, m_pObject);
62   }
63   operator CJS_EmbedObj*() { return m_pEmbedObj; };
64
65   void SetEmbedObject(CJS_EmbedObj* pObj) { m_pEmbedObj = pObj; };
66   CJS_EmbedObj* GetEmbedObject() { return m_pEmbedObj; };
67
68   static CPDFSDK_PageView* JSGetPageView(IFXJS_Context* cc);
69   static int MsgBox(CPDFDoc_Environment* pApp,
70                     CPDFSDK_PageView* pPageView,
71                     const FX_WCHAR* swMsg,
72                     const FX_WCHAR* swTitle = NULL,
73                     FX_UINT nType = 0,
74                     FX_UINT nIcon = 0);
75   static void Alert(CJS_Context* pContext, const FX_WCHAR* swMsg);
76
77   v8::Isolate* GetIsolate() { return m_pIsolate; }
78
79  protected:
80   CJS_EmbedObj* m_pEmbedObj;
81   v8::Global<v8::Object> m_pObject;
82   v8::Isolate* m_pIsolate;
83 };
84
85 struct JS_TIMER_MAP {
86   FX_UINT nID;
87   CJS_Timer* pTimer;
88 };
89
90 typedef CFX_ArrayTemplate<JS_TIMER_MAP*> CTimerMapArray;
91
92 struct JS_TIMER_MAPARRAY {
93  public:
94   JS_TIMER_MAPARRAY() {}
95
96   ~JS_TIMER_MAPARRAY() { Reset(); }
97
98   void Reset() {
99     for (int i = 0, sz = m_Array.GetSize(); i < sz; i++)
100       delete m_Array.GetAt(i);
101
102     m_Array.RemoveAll();
103   }
104
105   void SetAt(FX_UINT nIndex, CJS_Timer* pTimer) {
106     int i = Find(nIndex);
107
108     if (i >= 0) {
109       if (JS_TIMER_MAP* pMap = m_Array.GetAt(i))
110         pMap->pTimer = pTimer;
111     } else {
112       JS_TIMER_MAP* pMap = new JS_TIMER_MAP;
113       pMap->nID = nIndex;
114       pMap->pTimer = pTimer;
115       m_Array.Add(pMap);
116     }
117   }
118
119   CJS_Timer* GetAt(FX_UINT nIndex) {
120     int i = Find(nIndex);
121
122     if (i >= 0) {
123       if (JS_TIMER_MAP* pMap = m_Array.GetAt(i))
124         return pMap->pTimer;
125     }
126     return NULL;
127   }
128
129   void RemoveAt(FX_UINT nIndex) {
130     int i = Find(nIndex);
131
132     if (i >= 0) {
133       delete m_Array.GetAt(i);
134       m_Array.RemoveAt(i);
135     }
136     // To prevent potential fake memory leak reported by vc6.
137     if (m_Array.GetSize() == 0)
138       m_Array.RemoveAll();
139   }
140
141   int Find(FX_UINT nIndex) {
142     for (int i = 0, sz = m_Array.GetSize(); i < sz; i++) {
143       if (JS_TIMER_MAP* pMap = m_Array.GetAt(i)) {
144         if (pMap->nID == nIndex)
145           return i;
146       }
147     }
148
149     return -1;
150   }
151
152   CTimerMapArray m_Array;
153 };
154
155 JS_TIMER_MAPARRAY& GetTimeMap();
156
157 class CJS_Runtime;
158
159 class CJS_Timer {
160  public:
161   CJS_Timer(CJS_EmbedObj* pObj, CPDFDoc_Environment* pApp)
162       : m_nTimerID(0),
163         m_pEmbedObj(pObj),
164         m_bProcessing(FALSE),
165         m_dwStartTime(0),
166         m_dwTimeOut(0),
167         m_dwElapse(0),
168         m_pRuntime(NULL),
169         m_nType(0),
170         m_pApp(pApp) {}
171
172   virtual ~CJS_Timer() { KillJSTimer(); }
173
174  public:
175   FX_UINT SetJSTimer(FX_UINT nElapse) {
176     if (m_nTimerID)
177       KillJSTimer();
178     IFX_SystemHandler* pHandler = m_pApp->GetSysHandler();
179     m_nTimerID = pHandler->SetTimer(nElapse, TimerProc);
180     GetTimeMap().SetAt(m_nTimerID, this);
181     m_dwElapse = nElapse;
182     return m_nTimerID;
183   };
184
185   void KillJSTimer() {
186     if (m_nTimerID) {
187       IFX_SystemHandler* pHandler = m_pApp->GetSysHandler();
188       pHandler->KillTimer(m_nTimerID);
189       GetTimeMap().RemoveAt(m_nTimerID);
190       m_nTimerID = 0;
191     }
192   };
193
194   void SetType(int nType) { m_nType = nType; }
195
196   int GetType() const { return m_nType; }
197
198   void SetStartTime(FX_DWORD dwStartTime) { m_dwStartTime = dwStartTime; }
199
200   FX_DWORD GetStartTime() const { return m_dwStartTime; }
201
202   void SetTimeOut(FX_DWORD dwTimeOut) { m_dwTimeOut = dwTimeOut; }
203
204   FX_DWORD GetTimeOut() const { return m_dwTimeOut; }
205
206   void SetRuntime(CJS_Runtime* pRuntime) { m_pRuntime = pRuntime; }
207
208   CJS_Runtime* GetRuntime() const { return m_pRuntime; }
209
210   void SetJScript(const CFX_WideString& script) { m_swJScript = script; }
211
212   CFX_WideString GetJScript() const { return m_swJScript; }
213
214   static void TimerProc(int idEvent) {
215     if (CJS_Timer* pTimer = GetTimeMap().GetAt(idEvent)) {
216       if (!pTimer->m_bProcessing) {
217         pTimer->m_bProcessing = TRUE;
218         if (pTimer->m_pEmbedObj)
219           pTimer->m_pEmbedObj->TimerProc(pTimer);
220         pTimer->m_bProcessing = FALSE;
221       } else {
222         //      TRACE(L"BUSY!\n");
223       }
224     }
225   };
226
227  private:
228   FX_UINT m_nTimerID;
229   CJS_EmbedObj* m_pEmbedObj;
230   FX_BOOL m_bProcessing;
231
232   // data
233   FX_DWORD m_dwStartTime;
234   FX_DWORD m_dwTimeOut;
235   FX_DWORD m_dwElapse;
236   CJS_Runtime* m_pRuntime;
237   CFX_WideString m_swJScript;
238   int m_nType;  // 0:Interval; 1:TimeOut
239
240   CPDFDoc_Environment* m_pApp;
241 };
242
243 #endif  // FPDFSDK_INCLUDE_JAVASCRIPT_JS_OBJECT_H_