841bd197e0166f9a933e87bfc0a046d9a5e77c52
[pdfium.git] / fpdfsdk / src / fsdk_mgr.cpp
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 #include <algorithm>
8
9 #include "../../public/fpdf_ext.h"
10 #include "../../third_party/base/nonstd_unique_ptr.h"
11 #include "../include/formfiller/FFL_FormFiller.h"
12 #include "../include/fsdk_define.h"
13 #include "../include/fsdk_mgr.h"
14 #include "../include/javascript/IJavaScript.h"
15
16 #if _FX_OS_ == _FX_ANDROID_
17 #include "time.h"
18 #else
19 #include <ctime>
20 #endif
21
22 class CFX_SystemHandler : public IFX_SystemHandler {
23  public:
24   explicit CFX_SystemHandler(CPDFDoc_Environment* pEnv)
25       : m_pEnv(pEnv), m_nCharSet(-1) {}
26   ~CFX_SystemHandler() override {}
27
28  public:
29   // IFX_SystemHandler
30   void InvalidateRect(FX_HWND hWnd, FX_RECT rect) override;
31   void OutputSelectedRect(void* pFormFiller, CPDF_Rect& rect) override;
32   FX_BOOL IsSelectionImplemented() override;
33   CFX_WideString GetClipboardText(FX_HWND hWnd) override { return L""; }
34   FX_BOOL SetClipboardText(FX_HWND hWnd, CFX_WideString string) override {
35     return FALSE;
36   }
37   void ClientToScreen(FX_HWND hWnd, int32_t& x, int32_t& y) override {}
38   void ScreenToClient(FX_HWND hWnd, int32_t& x, int32_t& y) override {}
39   void SetCursor(int32_t nCursorType) override;
40   FX_HMENU CreatePopupMenu() override { return NULL; }
41   FX_BOOL AppendMenuItem(FX_HMENU hMenu,
42                          int32_t nIDNewItem,
43                          CFX_WideString string) override {
44     return FALSE;
45   }
46   FX_BOOL EnableMenuItem(FX_HMENU hMenu,
47                          int32_t nIDItem,
48                          FX_BOOL bEnabled) override {
49     return FALSE;
50   }
51   int32_t TrackPopupMenu(FX_HMENU hMenu,
52                          int32_t x,
53                          int32_t y,
54                          FX_HWND hParent) override {
55     return -1;
56   }
57   void DestroyMenu(FX_HMENU hMenu) override {}
58   CFX_ByteString GetNativeTrueTypeFont(int32_t nCharset) override;
59   FX_BOOL FindNativeTrueTypeFont(int32_t nCharset,
60                                  CFX_ByteString sFontFaceName) override;
61   CPDF_Font* AddNativeTrueTypeFontToPDF(CPDF_Document* pDoc,
62                                         CFX_ByteString sFontFaceName,
63                                         uint8_t nCharset) override;
64   int32_t SetTimer(int32_t uElapse, TimerCallback lpTimerFunc) override;
65   void KillTimer(int32_t nID) override;
66   FX_BOOL IsSHIFTKeyDown(FX_DWORD nFlag) override {
67     return m_pEnv->FFI_IsSHIFTKeyDown(nFlag);
68   }
69   FX_BOOL IsCTRLKeyDown(FX_DWORD nFlag) override {
70     return m_pEnv->FFI_IsCTRLKeyDown(nFlag);
71   }
72   FX_BOOL IsALTKeyDown(FX_DWORD nFlag) override {
73     return m_pEnv->FFI_IsALTKeyDown(nFlag);
74   }
75   FX_BOOL IsINSERTKeyDown(FX_DWORD nFlag) override {
76     return m_pEnv->FFI_IsINSERTKeyDown(nFlag);
77   }
78   FX_SYSTEMTIME GetLocalTime() override;
79   int32_t GetCharSet() override { return m_nCharSet; }
80   void SetCharSet(int32_t nCharSet) override { m_nCharSet = nCharSet; }
81
82  private:
83   CPDFDoc_Environment* m_pEnv;
84   int m_nCharSet;
85 };
86
87 void CFX_SystemHandler::SetCursor(int32_t nCursorType) {
88   m_pEnv->FFI_SetCursor(nCursorType);
89 }
90
91 void CFX_SystemHandler::InvalidateRect(FX_HWND hWnd, FX_RECT rect) {
92   // g_pFormFillApp->FFI_Invalidate();
93   CPDFSDK_Annot* pSDKAnnot = (CPDFSDK_Annot*)hWnd;
94   CPDF_Page* pPage = NULL;
95   CPDFSDK_PageView* pPageView = NULL;
96   pPageView = pSDKAnnot->GetPageView();
97   pPage = pSDKAnnot->GetPDFPage();
98   if (!pPage || !pPageView)
99     return;
100   CPDF_Matrix page2device;
101   pPageView->GetCurrentMatrix(page2device);
102   CPDF_Matrix device2page;
103   device2page.SetReverse(page2device);
104   FX_FLOAT left, top, right, bottom;
105   device2page.Transform((FX_FLOAT)rect.left, (FX_FLOAT)rect.top, left, top);
106   device2page.Transform((FX_FLOAT)rect.right, (FX_FLOAT)rect.bottom, right,
107                         bottom);
108   //  m_pEnv->FFI_DeviceToPage(pPage, rect.left, rect.top, (double*)&left,
109   //  (double*)&top);
110   //  m_pEnv->FFI_DeviceToPage(pPage, rect.right, rect.bottom, (double*)&right,
111   //  (double*)&bottom);
112   CPDF_Rect rcPDF(left, bottom, right, top);
113   rcPDF.Normalize();
114
115   m_pEnv->FFI_Invalidate(pPage, rcPDF.left, rcPDF.top, rcPDF.right,
116                          rcPDF.bottom);
117 }
118 void CFX_SystemHandler::OutputSelectedRect(void* pFormFiller, CPDF_Rect& rect) {
119   CFFL_FormFiller* pFFL = (CFFL_FormFiller*)pFormFiller;
120   if (pFFL) {
121     CPDF_Point leftbottom = CPDF_Point(rect.left, rect.bottom);
122     CPDF_Point righttop = CPDF_Point(rect.right, rect.top);
123     CPDF_Point ptA = pFFL->PWLtoFFL(leftbottom);
124     CPDF_Point ptB = pFFL->PWLtoFFL(righttop);
125
126     CPDFSDK_Annot* pAnnot = pFFL->GetSDKAnnot();
127     ASSERT(pAnnot);
128     CPDF_Page* pPage = pAnnot->GetPDFPage();
129     ASSERT(pPage);
130     m_pEnv->FFI_OutputSelectedRect(pPage, ptA.x, ptB.y, ptB.x, ptA.y);
131   }
132 }
133
134 FX_BOOL CFX_SystemHandler::IsSelectionImplemented() {
135   if (m_pEnv) {
136     FPDF_FORMFILLINFO* pInfo = m_pEnv->GetFormFillInfo();
137     if (pInfo && pInfo->FFI_OutputSelectedRect)
138       return TRUE;
139   }
140   return FALSE;
141 }
142
143 CFX_ByteString CFX_SystemHandler::GetNativeTrueTypeFont(int32_t nCharset) {
144   return "";
145 }
146
147 FX_BOOL CFX_SystemHandler::FindNativeTrueTypeFont(
148     int32_t nCharset,
149     CFX_ByteString sFontFaceName) {
150   CFX_FontMgr* pFontMgr = CFX_GEModule::Get()->GetFontMgr();
151   if (!pFontMgr)
152     return FALSE;
153
154   CFX_FontMapper* pFontMapper = pFontMgr->GetBuiltinMapper();
155   if (!pFontMapper)
156     return FALSE;
157
158   int nSize = pFontMapper->m_InstalledTTFonts.GetSize();
159   if (nSize == 0) {
160     pFontMapper->LoadInstalledFonts();
161     nSize = pFontMapper->m_InstalledTTFonts.GetSize();
162   }
163
164   for (int i = 0; i < nSize; ++i) {
165     if (pFontMapper->m_InstalledTTFonts[i].Compare(sFontFaceName))
166       return TRUE;
167   }
168
169   return FALSE;
170 }
171
172 static int CharSet2CP(int charset) {
173   if (charset == 128)
174     return 932;
175   if (charset == 134)
176     return 936;
177   if (charset == 129)
178     return 949;
179   if (charset == 136)
180     return 950;
181   return 0;
182 }
183 CPDF_Font* CFX_SystemHandler::AddNativeTrueTypeFontToPDF(
184     CPDF_Document* pDoc,
185     CFX_ByteString sFontFaceName,
186     uint8_t nCharset) {
187   if (pDoc) {
188     CFX_Font* pFXFont = new CFX_Font();
189     pFXFont->LoadSubst(sFontFaceName, TRUE, 0, 0, 0, CharSet2CP(nCharset),
190                        FALSE);
191     CPDF_Font* pFont = pDoc->AddFont(pFXFont, nCharset, FALSE);
192     delete pFXFont;
193     return pFont;
194   }
195
196   return NULL;
197 }
198
199 int32_t CFX_SystemHandler::SetTimer(int32_t uElapse,
200                                     TimerCallback lpTimerFunc) {
201   return m_pEnv->FFI_SetTimer(uElapse, lpTimerFunc);
202 }
203 void CFX_SystemHandler::KillTimer(int32_t nID) {
204   m_pEnv->FFI_KillTimer(nID);
205 }
206
207 FX_SYSTEMTIME CFX_SystemHandler::GetLocalTime() {
208   return m_pEnv->FFI_GetLocalTime();
209 }
210
211 CPDFDoc_Environment::CPDFDoc_Environment(CPDF_Document* pDoc,
212                                          FPDF_FORMFILLINFO* pFFinfo)
213     : m_pInfo(pFFinfo),
214       m_pSDKDoc(NULL),
215       m_pPDFDoc(pDoc),
216       m_pSysHandler(new CFX_SystemHandler(this)) {
217 }
218
219 CPDFDoc_Environment::~CPDFDoc_Environment() {
220 }
221
222 int CPDFDoc_Environment::JS_appAlert(const FX_WCHAR* Msg,
223                                      const FX_WCHAR* Title,
224                                      FX_UINT Type,
225                                      FX_UINT Icon) {
226   if (m_pInfo && m_pInfo->m_pJsPlatform && m_pInfo->m_pJsPlatform->app_alert) {
227     CFX_ByteString bsMsg = CFX_WideString(Msg).UTF16LE_Encode();
228     CFX_ByteString bsTitle = CFX_WideString(Title).UTF16LE_Encode();
229     FPDF_WIDESTRING pMsg = (FPDF_WIDESTRING)bsMsg.GetBuffer(bsMsg.GetLength());
230     FPDF_WIDESTRING pTitle =
231         (FPDF_WIDESTRING)bsTitle.GetBuffer(bsTitle.GetLength());
232     int ret = m_pInfo->m_pJsPlatform->app_alert(m_pInfo->m_pJsPlatform, pMsg,
233                                                 pTitle, Type, Icon);
234     bsMsg.ReleaseBuffer();
235     bsTitle.ReleaseBuffer();
236     return ret;
237   }
238   return -1;
239 }
240
241 int CPDFDoc_Environment::JS_appResponse(const FX_WCHAR* Question,
242                                         const FX_WCHAR* Title,
243                                         const FX_WCHAR* Default,
244                                         const FX_WCHAR* cLabel,
245                                         FPDF_BOOL bPassword,
246                                         void* response,
247                                         int length) {
248   if (m_pInfo && m_pInfo->m_pJsPlatform &&
249       m_pInfo->m_pJsPlatform->app_response) {
250     CFX_ByteString bsQuestion = CFX_WideString(Question).UTF16LE_Encode();
251     CFX_ByteString bsTitle = CFX_WideString(Title).UTF16LE_Encode();
252     CFX_ByteString bsDefault = CFX_WideString(Default).UTF16LE_Encode();
253     CFX_ByteString bsLabel = CFX_WideString(cLabel).UTF16LE_Encode();
254     FPDF_WIDESTRING pQuestion =
255         (FPDF_WIDESTRING)bsQuestion.GetBuffer(bsQuestion.GetLength());
256     FPDF_WIDESTRING pTitle =
257         (FPDF_WIDESTRING)bsTitle.GetBuffer(bsTitle.GetLength());
258     FPDF_WIDESTRING pDefault =
259         (FPDF_WIDESTRING)bsDefault.GetBuffer(bsDefault.GetLength());
260     FPDF_WIDESTRING pLabel =
261         (FPDF_WIDESTRING)bsLabel.GetBuffer(bsLabel.GetLength());
262     int ret = m_pInfo->m_pJsPlatform->app_response(
263         m_pInfo->m_pJsPlatform, pQuestion, pTitle, pDefault, pLabel, bPassword,
264         response, length);
265     bsQuestion.ReleaseBuffer();
266     bsTitle.ReleaseBuffer();
267     bsDefault.ReleaseBuffer();
268     bsLabel.ReleaseBuffer();
269     return ret;
270   }
271   return -1;
272 }
273
274 CFX_WideString CPDFDoc_Environment::JS_fieldBrowse() {
275   if (!m_pInfo || !m_pInfo->m_pJsPlatform ||
276       !m_pInfo->m_pJsPlatform->Field_browse) {
277     return L"";
278   }
279
280   const int nRequiredLen =
281       m_pInfo->m_pJsPlatform->Field_browse(m_pInfo->m_pJsPlatform, nullptr, 0);
282   if (nRequiredLen <= 0)
283     return L"";
284
285   nonstd::unique_ptr<char[]> pBuff(new char[nRequiredLen]);
286   memset(pBuff.get(), 0, nRequiredLen);
287   const int nActualLen = m_pInfo->m_pJsPlatform->Field_browse(
288       m_pInfo->m_pJsPlatform, pBuff.get(), nRequiredLen);
289   if (nActualLen <= 0 || nActualLen > nRequiredLen)
290     return L"";
291
292   CFX_ByteString bsRet = CFX_ByteString(pBuff.get(), nActualLen);
293   CFX_WideString wsRet = CFX_WideString::FromLocal(bsRet);
294   return wsRet;
295 }
296
297 CFX_WideString CPDFDoc_Environment::JS_docGetFilePath() {
298   if (!m_pInfo || !m_pInfo->m_pJsPlatform ||
299       !m_pInfo->m_pJsPlatform->Doc_getFilePath) {
300     return L"";
301   }
302
303   const int nRequiredLen = m_pInfo->m_pJsPlatform->Doc_getFilePath(
304       m_pInfo->m_pJsPlatform, nullptr, 0);
305   if (nRequiredLen <= 0)
306     return L"";
307
308   nonstd::unique_ptr<char[]> pBuff(new char[nRequiredLen]);
309   memset(pBuff.get(), 0, nRequiredLen);
310   const int nActualLen = m_pInfo->m_pJsPlatform->Doc_getFilePath(
311       m_pInfo->m_pJsPlatform, pBuff.get(), nRequiredLen);
312   if (nActualLen <= 0 || nActualLen > nRequiredLen)
313     return L"";
314
315   CFX_ByteString bsRet = CFX_ByteString(pBuff.get(), nActualLen);
316   CFX_WideString wsRet = CFX_WideString::FromLocal(bsRet);
317   return wsRet;
318 }
319
320 void CPDFDoc_Environment::JS_docSubmitForm(void* formData,
321                                            int length,
322                                            const FX_WCHAR* URL) {
323   if (m_pInfo && m_pInfo->m_pJsPlatform &&
324       m_pInfo->m_pJsPlatform->Doc_submitForm) {
325     CFX_ByteString bsDestination = CFX_WideString(URL).UTF16LE_Encode();
326     FPDF_WIDESTRING pDestination =
327         (FPDF_WIDESTRING)bsDestination.GetBuffer(bsDestination.GetLength());
328     m_pInfo->m_pJsPlatform->Doc_submitForm(m_pInfo->m_pJsPlatform, formData,
329                                            length, pDestination);
330     bsDestination.ReleaseBuffer();
331   }
332 }
333
334 void CPDFDoc_Environment::JS_docmailForm(void* mailData,
335                                          int length,
336                                          FPDF_BOOL bUI,
337                                          const FX_WCHAR* To,
338                                          const FX_WCHAR* Subject,
339                                          const FX_WCHAR* CC,
340                                          const FX_WCHAR* BCC,
341                                          const FX_WCHAR* Msg) {
342   if (m_pInfo && m_pInfo->m_pJsPlatform && m_pInfo->m_pJsPlatform->Doc_mail) {
343     CFX_ByteString bsTo = CFX_WideString(To).UTF16LE_Encode();
344     CFX_ByteString bsCC = CFX_WideString(Subject).UTF16LE_Encode();
345     CFX_ByteString bsBcc = CFX_WideString(BCC).UTF16LE_Encode();
346     CFX_ByteString bsSubject = CFX_WideString(Subject).UTF16LE_Encode();
347     CFX_ByteString bsMsg = CFX_WideString(Msg).UTF16LE_Encode();
348     FPDF_WIDESTRING pTo = (FPDF_WIDESTRING)bsTo.GetBuffer(bsTo.GetLength());
349     FPDF_WIDESTRING pCC = (FPDF_WIDESTRING)bsCC.GetBuffer(bsCC.GetLength());
350     FPDF_WIDESTRING pBcc = (FPDF_WIDESTRING)bsBcc.GetBuffer(bsBcc.GetLength());
351     FPDF_WIDESTRING pSubject =
352         (FPDF_WIDESTRING)bsSubject.GetBuffer(bsSubject.GetLength());
353     FPDF_WIDESTRING pMsg = (FPDF_WIDESTRING)bsMsg.GetBuffer(bsMsg.GetLength());
354     m_pInfo->m_pJsPlatform->Doc_mail(m_pInfo->m_pJsPlatform, mailData, length,
355                                      bUI, pTo, pSubject, pCC, pBcc, pMsg);
356     bsTo.ReleaseBuffer();
357     bsCC.ReleaseBuffer();
358     bsBcc.ReleaseBuffer();
359     bsSubject.ReleaseBuffer();
360     bsMsg.ReleaseBuffer();
361   }
362 }
363
364 IJS_Runtime* CPDFDoc_Environment::GetJSRuntime() {
365   if (!IsJSInitiated())
366     return NULL;
367   if (!m_pJSRuntime)
368     m_pJSRuntime.reset(IJS_Runtime::Create(this));
369   return m_pJSRuntime.get();
370 }
371
372 CPDFSDK_AnnotHandlerMgr* CPDFDoc_Environment::GetAnnotHandlerMgr() {
373   if (!m_pAnnotHandlerMgr)
374     m_pAnnotHandlerMgr.reset(new CPDFSDK_AnnotHandlerMgr(this));
375   return m_pAnnotHandlerMgr.get();
376 }
377
378 CPDFSDK_ActionHandler* CPDFDoc_Environment::GetActionHander() {
379   if (!m_pActionHandler)
380     m_pActionHandler.reset(new CPDFSDK_ActionHandler());
381   return m_pActionHandler.get();
382 }
383
384 CFFL_IFormFiller* CPDFDoc_Environment::GetIFormFiller() {
385   if (!m_pIFormFiller)
386     m_pIFormFiller.reset(new CFFL_IFormFiller(this));
387   return m_pIFormFiller.get();
388 }
389
390 CPDFSDK_Document::CPDFSDK_Document(CPDF_Document* pDoc,
391                                    CPDFDoc_Environment* pEnv)
392     : m_pDoc(pDoc), m_pFocusAnnot(nullptr), m_pEnv(pEnv), m_bChangeMask(FALSE) {
393 }
394
395 CPDFSDK_Document::~CPDFSDK_Document() {
396   m_bBeingDestroyed = TRUE;
397
398   for (auto& it : m_pageMap)
399     it.second->KillFocusAnnotIfNeeded();
400
401   for (auto& it : m_pageMap)
402     delete it.second;
403   m_pageMap.clear();
404 }
405
406 CPDFSDK_PageView* CPDFSDK_Document::GetPageView(CPDF_Page* pPDFPage,
407                                                 FX_BOOL ReNew) {
408   auto it = m_pageMap.find(pPDFPage);
409   if (it != m_pageMap.end())
410     return it->second;
411
412   if (!ReNew)
413     return nullptr;
414
415   CPDFSDK_PageView* pPageView = new CPDFSDK_PageView(this, pPDFPage);
416   m_pageMap[pPDFPage] = pPageView;
417   // Delay to load all the annotations, to avoid endless loop.
418   pPageView->LoadFXAnnots();
419   return pPageView;
420 }
421
422 CPDFSDK_PageView* CPDFSDK_Document::GetCurrentView() {
423   CPDF_Page* pPage = (CPDF_Page*)m_pEnv->FFI_GetCurrentPage(m_pDoc);
424   return pPage ? GetPageView(pPage, TRUE) : nullptr;
425 }
426
427 CPDFSDK_PageView* CPDFSDK_Document::GetPageView(int nIndex) {
428   CPDF_Page* pTempPage = (CPDF_Page*)m_pEnv->FFI_GetPage(m_pDoc, nIndex);
429   if (!pTempPage)
430     return nullptr;
431
432   auto it = m_pageMap.find(pTempPage);
433   return it->second;
434 }
435
436 void CPDFSDK_Document::ProcJavascriptFun() {
437   CPDF_Document* pPDFDoc = GetDocument();
438   CPDF_DocJSActions docJS(pPDFDoc);
439   int iCount = docJS.CountJSActions();
440   if (iCount < 1)
441     return;
442   for (int i = 0; i < iCount; i++) {
443     CFX_ByteString csJSName;
444     CPDF_Action jsAction = docJS.GetJSAction(i, csJSName);
445     if (m_pEnv->GetActionHander())
446       m_pEnv->GetActionHander()->DoAction_JavaScript(
447           jsAction, CFX_WideString::FromLocal(csJSName), this);
448   }
449 }
450
451 FX_BOOL CPDFSDK_Document::ProcOpenAction() {
452   if (!m_pDoc)
453     return FALSE;
454
455   CPDF_Dictionary* pRoot = m_pDoc->GetRoot();
456   if (!pRoot)
457     return FALSE;
458
459   CPDF_Object* pOpenAction = pRoot->GetDict("OpenAction");
460   if (!pOpenAction)
461     pOpenAction = pRoot->GetArray("OpenAction");
462
463   if (!pOpenAction)
464     return FALSE;
465
466   if (pOpenAction->IsArray())
467     return TRUE;
468
469   if (CPDF_Dictionary* pDict = pOpenAction->AsDictionary()) {
470     CPDF_Action action(pDict);
471     if (m_pEnv->GetActionHander())
472       m_pEnv->GetActionHander()->DoAction_DocOpen(action, this);
473     return TRUE;
474   }
475   return FALSE;
476 }
477
478 CPDF_OCContext* CPDFSDK_Document::GetOCContext() {
479   if (!m_pOccontent)
480     m_pOccontent.reset(new CPDF_OCContext(m_pDoc));
481   return m_pOccontent.get();
482 }
483
484 void CPDFSDK_Document::ReMovePageView(CPDF_Page* pPDFPage) {
485   auto it = m_pageMap.find(pPDFPage);
486   if (it == m_pageMap.end())
487     return;
488
489   CPDFSDK_PageView* pPageView = it->second;
490   if (pPageView->IsLocked())
491     return;
492
493   pPageView->KillFocusAnnotIfNeeded();
494   delete pPageView;
495   m_pageMap.erase(it);
496 }
497
498 CPDF_Page* CPDFSDK_Document::GetPage(int nIndex) {
499   CPDF_Page* pTempPage = (CPDF_Page*)m_pEnv->FFI_GetPage(m_pDoc, nIndex);
500   if (!pTempPage)
501     return NULL;
502   return pTempPage;
503 }
504
505 CPDFSDK_InterForm* CPDFSDK_Document::GetInterForm() {
506   if (!m_pInterForm)
507     m_pInterForm.reset(new CPDFSDK_InterForm(this));
508   return m_pInterForm.get();
509 }
510
511 void CPDFSDK_Document::UpdateAllViews(CPDFSDK_PageView* pSender,
512                                       CPDFSDK_Annot* pAnnot) {
513   for (const auto& it : m_pageMap) {
514     CPDFSDK_PageView* pPageView = it.second;
515     if (pPageView != pSender) {
516       pPageView->UpdateView(pAnnot);
517     }
518   }
519 }
520
521 CPDFSDK_Annot* CPDFSDK_Document::GetFocusAnnot() {
522   return m_pFocusAnnot;
523 }
524
525 FX_BOOL CPDFSDK_Document::SetFocusAnnot(CPDFSDK_Annot* pAnnot, FX_UINT nFlag) {
526   if (m_bBeingDestroyed)
527     return FALSE;
528
529   if (m_pFocusAnnot == pAnnot)
530     return TRUE;
531
532   if (m_pFocusAnnot) {
533     if (!KillFocusAnnot(nFlag))
534       return FALSE;
535   }
536   CPDFSDK_PageView* pPageView = pAnnot->GetPageView();
537   if (pAnnot && pPageView->IsValid()) {
538     CPDFSDK_AnnotHandlerMgr* pAnnotHandler = m_pEnv->GetAnnotHandlerMgr();
539     if (!m_pFocusAnnot) {
540       if (!pAnnotHandler->Annot_OnSetFocus(pAnnot, nFlag))
541         return FALSE;
542       if (!m_pFocusAnnot) {
543         m_pFocusAnnot = pAnnot;
544         return TRUE;
545       }
546     }
547   }
548   return FALSE;
549 }
550
551 FX_BOOL CPDFSDK_Document::KillFocusAnnot(FX_UINT nFlag) {
552   if (m_pFocusAnnot) {
553     CPDFSDK_AnnotHandlerMgr* pAnnotHandler = m_pEnv->GetAnnotHandlerMgr();
554     CPDFSDK_Annot* pFocusAnnot = m_pFocusAnnot;
555     m_pFocusAnnot = nullptr;
556     if (pAnnotHandler->Annot_OnKillFocus(pFocusAnnot, nFlag)) {
557       if (pFocusAnnot->GetType() == FX_BSTRC("Widget")) {
558         CPDFSDK_Widget* pWidget = (CPDFSDK_Widget*)pFocusAnnot;
559         int nFieldType = pWidget->GetFieldType();
560         if (FIELDTYPE_TEXTFIELD == nFieldType ||
561             FIELDTYPE_COMBOBOX == nFieldType) {
562           m_pEnv->FFI_OnSetFieldInputFocus(NULL, NULL, 0, FALSE);
563         }
564       }
565
566       if (!m_pFocusAnnot)
567         return TRUE;
568     } else {
569       m_pFocusAnnot = pFocusAnnot;
570     }
571   }
572   return FALSE;
573 }
574
575 void CPDFSDK_Document::OnCloseDocument() {
576   KillFocusAnnot();
577 }
578
579 FX_BOOL CPDFSDK_Document::GetPermissions(int nFlag) {
580   FX_DWORD dwPermissions = m_pDoc->GetUserPermissions();
581   return dwPermissions & nFlag;
582 }
583
584 IJS_Runtime* CPDFSDK_Document::GetJsRuntime() {
585   ASSERT(m_pEnv != NULL);
586   return m_pEnv->GetJSRuntime();
587 }
588
589 CFX_WideString CPDFSDK_Document::GetPath() {
590   ASSERT(m_pEnv != NULL);
591   return m_pEnv->JS_docGetFilePath();
592 }
593
594 CPDFSDK_PageView::CPDFSDK_PageView(CPDFSDK_Document* pSDKDoc, CPDF_Page* page)
595     : m_page(page),
596       m_pSDKDoc(pSDKDoc),
597       m_CaptureWidget(nullptr),
598       m_bEnterWidget(FALSE),
599       m_bExitWidget(FALSE),
600       m_bOnWidget(FALSE),
601       m_bValid(FALSE),
602       m_bLocked(FALSE),
603       m_bTakeOverPage(FALSE) {
604   CPDFSDK_InterForm* pInterForm = pSDKDoc->GetInterForm();
605   if (pInterForm) {
606     CPDF_InterForm* pPDFInterForm = pInterForm->GetInterForm();
607     pPDFInterForm->FixPageFields(page);
608   }
609   m_page->SetPrivateData((void*)m_page, (void*)this, nullptr);
610 }
611
612 CPDFSDK_PageView::~CPDFSDK_PageView() {
613   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
614   CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
615   for (CPDFSDK_Annot* pAnnot : m_fxAnnotArray)
616     pAnnotHandlerMgr->ReleaseAnnot(pAnnot);
617   m_fxAnnotArray.clear();
618
619   m_pAnnotList.reset();
620
621   m_page->RemovePrivateData((void*)m_page);
622   if (m_bTakeOverPage) {
623     delete m_page;
624   }
625 }
626
627 void CPDFSDK_PageView::PageView_OnDraw(CFX_RenderDevice* pDevice,
628                                        CPDF_Matrix* pUser2Device,
629                                        CPDF_RenderOptions* pOptions) {
630   m_curMatrix = *pUser2Device;
631
632   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
633   CPDFSDK_AnnotIterator annotIterator(this, true);
634   while (CPDFSDK_Annot* pSDKAnnot = annotIterator.Next()) {
635     CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
636     pAnnotHandlerMgr->Annot_OnDraw(this, pSDKAnnot, pDevice, pUser2Device, 0);
637   }
638 }
639
640 CPDF_Annot* CPDFSDK_PageView::GetPDFAnnotAtPoint(FX_FLOAT pageX,
641                                                  FX_FLOAT pageY) {
642   const int nCount = m_pAnnotList->Count();
643   for (int i = 0; i < nCount; ++i) {
644     CPDF_Annot* pAnnot = m_pAnnotList->GetAt(i);
645     CFX_FloatRect annotRect;
646     pAnnot->GetRect(annotRect);
647     if (annotRect.Contains(pageX, pageY))
648       return pAnnot;
649   }
650   return nullptr;
651 }
652
653 CPDF_Annot* CPDFSDK_PageView::GetPDFWidgetAtPoint(FX_FLOAT pageX,
654                                                   FX_FLOAT pageY) {
655   const int nCount = m_pAnnotList->Count();
656   for (int i = 0; i < nCount; ++i) {
657     CPDF_Annot* pAnnot = m_pAnnotList->GetAt(i);
658     if (pAnnot->GetSubType() == "Widget") {
659       CFX_FloatRect annotRect;
660       pAnnot->GetRect(annotRect);
661       if (annotRect.Contains(pageX, pageY))
662         return pAnnot;
663     }
664   }
665   return nullptr;
666 }
667
668 CPDFSDK_Annot* CPDFSDK_PageView::GetFXAnnotAtPoint(FX_FLOAT pageX,
669                                                    FX_FLOAT pageY) {
670   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
671   CPDFSDK_AnnotHandlerMgr* pAnnotMgr = pEnv->GetAnnotHandlerMgr();
672   CPDFSDK_AnnotIterator annotIterator(this, false);
673   while (CPDFSDK_Annot* pSDKAnnot = annotIterator.Next()) {
674     CPDF_Rect rc = pAnnotMgr->Annot_OnGetViewBBox(this, pSDKAnnot);
675     if (rc.Contains(pageX, pageY))
676       return pSDKAnnot;
677   }
678
679   return nullptr;
680 }
681
682 CPDFSDK_Annot* CPDFSDK_PageView::GetFXWidgetAtPoint(FX_FLOAT pageX,
683                                                     FX_FLOAT pageY) {
684   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
685   CPDFSDK_AnnotHandlerMgr* pAnnotMgr = pEnv->GetAnnotHandlerMgr();
686   CPDFSDK_AnnotIterator annotIterator(this, false);
687   while (CPDFSDK_Annot* pSDKAnnot = annotIterator.Next()) {
688     if (pSDKAnnot->GetType() == "Widget") {
689       pAnnotMgr->Annot_OnGetViewBBox(this, pSDKAnnot);
690       CPDF_Point point(pageX, pageY);
691       if (pAnnotMgr->Annot_OnHitTest(this, pSDKAnnot, point))
692         return pSDKAnnot;
693     }
694   }
695
696   return nullptr;
697 }
698
699 void CPDFSDK_PageView::KillFocusAnnotIfNeeded() {
700   // if there is a focused annot on the page, we should kill the focus first.
701   if (CPDFSDK_Annot* focusedAnnot = m_pSDKDoc->GetFocusAnnot()) {
702     auto it =
703         std::find(m_fxAnnotArray.begin(), m_fxAnnotArray.end(), focusedAnnot);
704     if (it != m_fxAnnotArray.end())
705       KillFocusAnnot();
706   }
707 }
708
709 FX_BOOL CPDFSDK_PageView::Annot_HasAppearance(CPDF_Annot* pAnnot) {
710   CPDF_Dictionary* pAnnotDic = pAnnot->GetAnnotDict();
711   if (pAnnotDic)
712     return pAnnotDic->KeyExist("AS");
713   return FALSE;
714 }
715
716 CPDFSDK_Annot* CPDFSDK_PageView::AddAnnot(CPDF_Annot* pPDFAnnot) {
717   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
718   ASSERT(pEnv);
719   CPDFSDK_AnnotHandlerMgr* pAnnotHandler = pEnv->GetAnnotHandlerMgr();
720   CPDFSDK_Annot* pSDKAnnot = pAnnotHandler->NewAnnot(pPDFAnnot, this);
721   if (!pSDKAnnot)
722     return nullptr;
723
724   m_fxAnnotArray.push_back(pSDKAnnot);
725   pAnnotHandler->Annot_OnCreate(pSDKAnnot);
726   return pSDKAnnot;
727 }
728
729 CPDFSDK_Annot* CPDFSDK_PageView::AddAnnot(CPDF_Dictionary* pDict) {
730   return pDict ? AddAnnot(pDict->GetString("Subtype"), pDict) : nullptr;
731 }
732
733 CPDFSDK_Annot* CPDFSDK_PageView::AddAnnot(const FX_CHAR* lpSubType,
734                                           CPDF_Dictionary* pDict) {
735   return NULL;
736 }
737
738 FX_BOOL CPDFSDK_PageView::DeleteAnnot(CPDFSDK_Annot* pAnnot) {
739   return FALSE;
740 }
741
742 CPDF_Document* CPDFSDK_PageView::GetPDFDocument() {
743   if (m_page) {
744     return m_page->m_pDocument;
745   }
746   return NULL;
747 }
748
749 size_t CPDFSDK_PageView::CountAnnots() const {
750   return m_fxAnnotArray.size();
751 }
752
753 CPDFSDK_Annot* CPDFSDK_PageView::GetAnnot(size_t nIndex) {
754   return nIndex < m_fxAnnotArray.size() ? m_fxAnnotArray[nIndex] : nullptr;
755 }
756
757 CPDFSDK_Annot* CPDFSDK_PageView::GetAnnotByDict(CPDF_Dictionary* pDict) {
758   for (CPDFSDK_Annot* pAnnot : m_fxAnnotArray) {
759     if (pAnnot->GetPDFAnnot()->GetAnnotDict() == pDict)
760       return pAnnot;
761   }
762   return nullptr;
763 }
764
765 FX_BOOL CPDFSDK_PageView::OnLButtonDown(const CPDF_Point& point,
766                                         FX_UINT nFlag) {
767   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
768   ASSERT(pEnv);
769   CPDFSDK_Annot* pFXAnnot = GetFXWidgetAtPoint(point.x, point.y);
770   if (!pFXAnnot) {
771     KillFocusAnnot(nFlag);
772     return FALSE;
773   }
774
775   CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
776   FX_BOOL bRet =
777       pAnnotHandlerMgr->Annot_OnLButtonDown(this, pFXAnnot, nFlag, point);
778   if (bRet)
779     SetFocusAnnot(pFXAnnot);
780   return bRet;
781 }
782
783 FX_BOOL CPDFSDK_PageView::OnLButtonUp(const CPDF_Point& point, FX_UINT nFlag) {
784   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
785   ASSERT(pEnv);
786   CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
787   CPDFSDK_Annot* pFXAnnot = GetFXWidgetAtPoint(point.x, point.y);
788   CPDFSDK_Annot* pFocusAnnot = GetFocusAnnot();
789   FX_BOOL bRet = FALSE;
790   if (pFocusAnnot && pFocusAnnot != pFXAnnot) {
791     // Last focus Annot gets a chance to handle the event.
792     bRet = pAnnotHandlerMgr->Annot_OnLButtonUp(this, pFocusAnnot, nFlag, point);
793   }
794   if (pFXAnnot && !bRet)
795     bRet = pAnnotHandlerMgr->Annot_OnLButtonUp(this, pFXAnnot, nFlag, point);
796   return bRet;
797 }
798
799 FX_BOOL CPDFSDK_PageView::OnMouseMove(const CPDF_Point& point, int nFlag) {
800   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
801   CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
802   if (CPDFSDK_Annot* pFXAnnot = GetFXWidgetAtPoint(point.x, point.y)) {
803     if (m_CaptureWidget && m_CaptureWidget != pFXAnnot) {
804       m_bExitWidget = TRUE;
805       m_bEnterWidget = FALSE;
806       pAnnotHandlerMgr->Annot_OnMouseExit(this, m_CaptureWidget, nFlag);
807     }
808     m_CaptureWidget = (CPDFSDK_Widget*)pFXAnnot;
809     m_bOnWidget = TRUE;
810     if (!m_bEnterWidget) {
811       m_bEnterWidget = TRUE;
812       m_bExitWidget = FALSE;
813       pAnnotHandlerMgr->Annot_OnMouseEnter(this, pFXAnnot, nFlag);
814     }
815     pAnnotHandlerMgr->Annot_OnMouseMove(this, pFXAnnot, nFlag, point);
816     return TRUE;
817   }
818   if (m_bOnWidget) {
819     m_bOnWidget = FALSE;
820     m_bExitWidget = TRUE;
821     m_bEnterWidget = FALSE;
822     if (m_CaptureWidget) {
823       pAnnotHandlerMgr->Annot_OnMouseExit(this, m_CaptureWidget, nFlag);
824       m_CaptureWidget = NULL;
825     }
826   }
827   return FALSE;
828 }
829
830 FX_BOOL CPDFSDK_PageView::OnMouseWheel(double deltaX,
831                                        double deltaY,
832                                        const CPDF_Point& point,
833                                        int nFlag) {
834   if (CPDFSDK_Annot* pAnnot = GetFXWidgetAtPoint(point.x, point.y)) {
835     CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
836     CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
837     return pAnnotHandlerMgr->Annot_OnMouseWheel(this, pAnnot, nFlag,
838                                                 (int)deltaY, point);
839   }
840   return FALSE;
841 }
842
843 FX_BOOL CPDFSDK_PageView::OnChar(int nChar, FX_UINT nFlag) {
844   if (CPDFSDK_Annot* pAnnot = GetFocusAnnot()) {
845     CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
846     CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
847     return pAnnotHandlerMgr->Annot_OnChar(pAnnot, nChar, nFlag);
848   }
849
850   return FALSE;
851 }
852
853 FX_BOOL CPDFSDK_PageView::OnKeyDown(int nKeyCode, int nFlag) {
854   if (CPDFSDK_Annot* pAnnot = GetFocusAnnot()) {
855     CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
856     CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
857     return pAnnotHandlerMgr->Annot_OnKeyDown(pAnnot, nKeyCode, nFlag);
858   }
859   return FALSE;
860 }
861
862 FX_BOOL CPDFSDK_PageView::OnKeyUp(int nKeyCode, int nFlag) {
863   return FALSE;
864 }
865
866 void CPDFSDK_PageView::LoadFXAnnots() {
867   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
868
869   FX_BOOL enableAPUpdate = CPDF_InterForm::UpdatingAPEnabled();
870   // Disable the default AP construction.
871   CPDF_InterForm::EnableUpdateAP(FALSE);
872   m_pAnnotList.reset(new CPDF_AnnotList(m_page));
873   CPDF_InterForm::EnableUpdateAP(enableAPUpdate);
874   const int nCount = m_pAnnotList->Count();
875   SetLock(TRUE);
876   for (int i = 0; i < nCount; ++i) {
877     CPDF_Annot* pPDFAnnot = m_pAnnotList->GetAt(i);
878     CPDF_Document* pDoc = GetPDFDocument();
879
880     CheckUnSupportAnnot(pDoc, pPDFAnnot);
881
882     CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
883     CPDFSDK_Annot* pAnnot = pAnnotHandlerMgr->NewAnnot(pPDFAnnot, this);
884     if (!pAnnot)
885       continue;
886     m_fxAnnotArray.push_back(pAnnot);
887
888     pAnnotHandlerMgr->Annot_OnLoad(pAnnot);
889   }
890   SetLock(FALSE);
891 }
892
893 void CPDFSDK_PageView::UpdateRects(CFX_RectArray& rects) {
894   for (int i = 0; i < rects.GetSize(); i++) {
895     CPDF_Rect rc = rects.GetAt(i);
896     CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
897     pEnv->FFI_Invalidate(m_page, rc.left, rc.top, rc.right, rc.bottom);
898   }
899 }
900
901 void CPDFSDK_PageView::UpdateView(CPDFSDK_Annot* pAnnot) {
902   CPDF_Rect rcWindow = pAnnot->GetRect();
903   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
904   pEnv->FFI_Invalidate(m_page, rcWindow.left, rcWindow.top, rcWindow.right,
905                        rcWindow.bottom);
906 }
907
908 int CPDFSDK_PageView::GetPageIndex() {
909   if (m_page) {
910     CPDF_Dictionary* pDic = m_page->m_pFormDict;
911     CPDF_Document* pDoc = m_pSDKDoc->GetDocument();
912     if (pDoc && pDic) {
913       return pDoc->GetPageIndex(pDic->GetObjNum());
914     }
915   }
916   return -1;
917 }
918
919 FX_BOOL CPDFSDK_PageView::IsValidAnnot(CPDF_Annot* p) const {
920   if (!p)
921     return FALSE;
922
923   const int nCount = m_pAnnotList->Count();
924   for (int i = 0; i < nCount; ++i) {
925     if (m_pAnnotList->GetAt(i) == p)
926       return TRUE;
927   }
928   return FALSE;
929 }
930
931 CPDFSDK_Annot* CPDFSDK_PageView::GetFocusAnnot() {
932   CPDFSDK_Annot* pFocusAnnot = m_pSDKDoc->GetFocusAnnot();
933   if (!pFocusAnnot)
934     return nullptr;
935
936   for (CPDFSDK_Annot* pAnnot : m_fxAnnotArray) {
937     if (pAnnot == pFocusAnnot)
938       return pAnnot;
939   }
940   return nullptr;
941 }