53a774edcd11bdd5f76ec63ab5a72d04791b2d92
[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_pAnnotHandlerMgr(NULL),
214       m_pActionHandler(NULL),
215       m_pInfo(pFFinfo),
216       m_pSDKDoc(NULL),
217       m_pPDFDoc(pDoc),
218       m_pIFormFiller(NULL) {
219   m_pSysHandler = new CFX_SystemHandler(this);
220 }
221
222 CPDFDoc_Environment::~CPDFDoc_Environment() {
223   delete m_pIFormFiller;
224   m_pIFormFiller = NULL;
225
226   delete m_pSysHandler;
227   m_pSysHandler = NULL;
228
229   delete m_pAnnotHandlerMgr;
230   m_pAnnotHandlerMgr = NULL;
231
232   delete m_pActionHandler;
233   m_pActionHandler = NULL;
234 }
235
236 int CPDFDoc_Environment::JS_appAlert(const FX_WCHAR* Msg,
237                                      const FX_WCHAR* Title,
238                                      FX_UINT Type,
239                                      FX_UINT Icon) {
240   if (m_pInfo && m_pInfo->m_pJsPlatform && m_pInfo->m_pJsPlatform->app_alert) {
241     CFX_ByteString bsMsg = CFX_WideString(Msg).UTF16LE_Encode();
242     CFX_ByteString bsTitle = CFX_WideString(Title).UTF16LE_Encode();
243     FPDF_WIDESTRING pMsg = (FPDF_WIDESTRING)bsMsg.GetBuffer(bsMsg.GetLength());
244     FPDF_WIDESTRING pTitle =
245         (FPDF_WIDESTRING)bsTitle.GetBuffer(bsTitle.GetLength());
246     int ret = m_pInfo->m_pJsPlatform->app_alert(m_pInfo->m_pJsPlatform, pMsg,
247                                                 pTitle, Type, Icon);
248     bsMsg.ReleaseBuffer();
249     bsTitle.ReleaseBuffer();
250     return ret;
251   }
252   return -1;
253 }
254
255 int CPDFDoc_Environment::JS_appResponse(const FX_WCHAR* Question,
256                                         const FX_WCHAR* Title,
257                                         const FX_WCHAR* Default,
258                                         const FX_WCHAR* cLabel,
259                                         FPDF_BOOL bPassword,
260                                         void* response,
261                                         int length) {
262   if (m_pInfo && m_pInfo->m_pJsPlatform &&
263       m_pInfo->m_pJsPlatform->app_response) {
264     CFX_ByteString bsQuestion = CFX_WideString(Question).UTF16LE_Encode();
265     CFX_ByteString bsTitle = CFX_WideString(Title).UTF16LE_Encode();
266     CFX_ByteString bsDefault = CFX_WideString(Default).UTF16LE_Encode();
267     CFX_ByteString bsLabel = CFX_WideString(cLabel).UTF16LE_Encode();
268     FPDF_WIDESTRING pQuestion =
269         (FPDF_WIDESTRING)bsQuestion.GetBuffer(bsQuestion.GetLength());
270     FPDF_WIDESTRING pTitle =
271         (FPDF_WIDESTRING)bsTitle.GetBuffer(bsTitle.GetLength());
272     FPDF_WIDESTRING pDefault =
273         (FPDF_WIDESTRING)bsDefault.GetBuffer(bsDefault.GetLength());
274     FPDF_WIDESTRING pLabel =
275         (FPDF_WIDESTRING)bsLabel.GetBuffer(bsLabel.GetLength());
276     int ret = m_pInfo->m_pJsPlatform->app_response(
277         m_pInfo->m_pJsPlatform, pQuestion, pTitle, pDefault, pLabel, bPassword,
278         response, length);
279     bsQuestion.ReleaseBuffer();
280     bsTitle.ReleaseBuffer();
281     bsDefault.ReleaseBuffer();
282     bsLabel.ReleaseBuffer();
283     return ret;
284   }
285   return -1;
286 }
287
288 CFX_WideString CPDFDoc_Environment::JS_fieldBrowse() {
289   if (!m_pInfo || !m_pInfo->m_pJsPlatform ||
290       !m_pInfo->m_pJsPlatform->Field_browse) {
291     return L"";
292   }
293
294   const int nRequiredLen =
295       m_pInfo->m_pJsPlatform->Field_browse(m_pInfo->m_pJsPlatform, nullptr, 0);
296   if (nRequiredLen <= 0)
297     return L"";
298
299   nonstd::unique_ptr<char[]> pBuff(new char[nRequiredLen]);
300   memset(pBuff.get(), 0, nRequiredLen);
301   const int nActualLen = m_pInfo->m_pJsPlatform->Field_browse(
302       m_pInfo->m_pJsPlatform, pBuff.get(), nRequiredLen);
303   if (nActualLen <= 0 || nActualLen > nRequiredLen)
304     return L"";
305
306   CFX_ByteString bsRet = CFX_ByteString(pBuff.get(), nActualLen);
307   CFX_WideString wsRet = CFX_WideString::FromLocal(bsRet);
308   return wsRet;
309 }
310
311 CFX_WideString CPDFDoc_Environment::JS_docGetFilePath() {
312   if (!m_pInfo || !m_pInfo->m_pJsPlatform ||
313       !m_pInfo->m_pJsPlatform->Doc_getFilePath) {
314     return L"";
315   }
316
317   const int nRequiredLen = m_pInfo->m_pJsPlatform->Doc_getFilePath(
318       m_pInfo->m_pJsPlatform, nullptr, 0);
319   if (nRequiredLen <= 0)
320     return L"";
321
322   nonstd::unique_ptr<char[]> pBuff(new char[nRequiredLen]);
323   memset(pBuff.get(), 0, nRequiredLen);
324   const int nActualLen = m_pInfo->m_pJsPlatform->Doc_getFilePath(
325       m_pInfo->m_pJsPlatform, pBuff.get(), nRequiredLen);
326   if (nActualLen <= 0 || nActualLen > nRequiredLen)
327     return L"";
328
329   CFX_ByteString bsRet = CFX_ByteString(pBuff.get(), nActualLen);
330   CFX_WideString wsRet = CFX_WideString::FromLocal(bsRet);
331   return wsRet;
332 }
333
334 void CPDFDoc_Environment::JS_docSubmitForm(void* formData,
335                                            int length,
336                                            const FX_WCHAR* URL) {
337   if (m_pInfo && m_pInfo->m_pJsPlatform &&
338       m_pInfo->m_pJsPlatform->Doc_submitForm) {
339     CFX_ByteString bsDestination = CFX_WideString(URL).UTF16LE_Encode();
340     FPDF_WIDESTRING pDestination =
341         (FPDF_WIDESTRING)bsDestination.GetBuffer(bsDestination.GetLength());
342     m_pInfo->m_pJsPlatform->Doc_submitForm(m_pInfo->m_pJsPlatform, formData,
343                                            length, pDestination);
344     bsDestination.ReleaseBuffer();
345   }
346 }
347
348 void CPDFDoc_Environment::JS_docmailForm(void* mailData,
349                                          int length,
350                                          FPDF_BOOL bUI,
351                                          const FX_WCHAR* To,
352                                          const FX_WCHAR* Subject,
353                                          const FX_WCHAR* CC,
354                                          const FX_WCHAR* BCC,
355                                          const FX_WCHAR* Msg) {
356   if (m_pInfo && m_pInfo->m_pJsPlatform && m_pInfo->m_pJsPlatform->Doc_mail) {
357     CFX_ByteString bsTo = CFX_WideString(To).UTF16LE_Encode();
358     CFX_ByteString bsCC = CFX_WideString(Subject).UTF16LE_Encode();
359     CFX_ByteString bsBcc = CFX_WideString(BCC).UTF16LE_Encode();
360     CFX_ByteString bsSubject = CFX_WideString(Subject).UTF16LE_Encode();
361     CFX_ByteString bsMsg = CFX_WideString(Msg).UTF16LE_Encode();
362     FPDF_WIDESTRING pTo = (FPDF_WIDESTRING)bsTo.GetBuffer(bsTo.GetLength());
363     FPDF_WIDESTRING pCC = (FPDF_WIDESTRING)bsCC.GetBuffer(bsCC.GetLength());
364     FPDF_WIDESTRING pBcc = (FPDF_WIDESTRING)bsBcc.GetBuffer(bsBcc.GetLength());
365     FPDF_WIDESTRING pSubject =
366         (FPDF_WIDESTRING)bsSubject.GetBuffer(bsSubject.GetLength());
367     FPDF_WIDESTRING pMsg = (FPDF_WIDESTRING)bsMsg.GetBuffer(bsMsg.GetLength());
368     m_pInfo->m_pJsPlatform->Doc_mail(m_pInfo->m_pJsPlatform, mailData, length,
369                                      bUI, pTo, pSubject, pCC, pBcc, pMsg);
370     bsTo.ReleaseBuffer();
371     bsCC.ReleaseBuffer();
372     bsBcc.ReleaseBuffer();
373     bsSubject.ReleaseBuffer();
374     bsMsg.ReleaseBuffer();
375   }
376 }
377
378 IJS_Runtime* CPDFDoc_Environment::GetJSRuntime() {
379   if (!IsJSInitiated())
380     return NULL;
381   if (!m_pJSRuntime)
382     m_pJSRuntime.reset(IJS_Runtime::Create(this));
383   return m_pJSRuntime.get();
384 }
385
386 CPDFSDK_AnnotHandlerMgr* CPDFDoc_Environment::GetAnnotHandlerMgr() {
387   if (!m_pAnnotHandlerMgr)
388     m_pAnnotHandlerMgr = new CPDFSDK_AnnotHandlerMgr(this);
389   return m_pAnnotHandlerMgr;
390 }
391
392 CPDFSDK_ActionHandler* CPDFDoc_Environment::GetActionHander() {
393   if (!m_pActionHandler)
394     m_pActionHandler = new CPDFSDK_ActionHandler();
395   return m_pActionHandler;
396 }
397
398 CFFL_IFormFiller* CPDFDoc_Environment::GetIFormFiller() {
399   if (!m_pIFormFiller)
400     m_pIFormFiller = new CFFL_IFormFiller(this);
401   return m_pIFormFiller;
402 }
403
404 CPDFSDK_Document::CPDFSDK_Document(CPDF_Document* pDoc,
405                                    CPDFDoc_Environment* pEnv)
406     : m_pDoc(pDoc),
407       m_pInterForm(nullptr),
408       m_pFocusAnnot(nullptr),
409       m_pEnv(pEnv),
410       m_pOccontent(nullptr),
411       m_bChangeMask(FALSE) {}
412
413 CPDFSDK_Document::~CPDFSDK_Document() {
414   for (auto& it : m_pageMap)
415     delete it.second;
416   m_pageMap.clear();
417
418   delete m_pInterForm;
419   m_pInterForm = nullptr;
420
421   delete m_pOccontent;
422   m_pOccontent = nullptr;
423 }
424
425 CPDFSDK_PageView* CPDFSDK_Document::GetPageView(CPDF_Page* pPDFPage,
426                                                 FX_BOOL ReNew) {
427   auto it = m_pageMap.find(pPDFPage);
428   if (it != m_pageMap.end())
429     return it->second;
430
431   if (!ReNew)
432     return nullptr;
433
434   CPDFSDK_PageView* pPageView = new CPDFSDK_PageView(this, pPDFPage);
435   m_pageMap[pPDFPage] = pPageView;
436   // Delay to load all the annotations, to avoid endless loop.
437   pPageView->LoadFXAnnots();
438   return pPageView;
439 }
440
441 CPDFSDK_PageView* CPDFSDK_Document::GetCurrentView() {
442   CPDF_Page* pPage = (CPDF_Page*)m_pEnv->FFI_GetCurrentPage(m_pDoc);
443   return pPage ? GetPageView(pPage, TRUE) : nullptr;
444 }
445
446 CPDFSDK_PageView* CPDFSDK_Document::GetPageView(int nIndex) {
447   CPDF_Page* pTempPage = (CPDF_Page*)m_pEnv->FFI_GetPage(m_pDoc, nIndex);
448   if (!pTempPage)
449     return nullptr;
450
451   auto it = m_pageMap.find(pTempPage);
452   return it->second;
453 }
454
455 void CPDFSDK_Document::ProcJavascriptFun() {
456   CPDF_Document* pPDFDoc = GetDocument();
457   CPDF_DocJSActions docJS(pPDFDoc);
458   int iCount = docJS.CountJSActions();
459   if (iCount < 1)
460     return;
461   for (int i = 0; i < iCount; i++) {
462     CFX_ByteString csJSName;
463     CPDF_Action jsAction = docJS.GetJSAction(i, csJSName);
464     if (m_pEnv->GetActionHander())
465       m_pEnv->GetActionHander()->DoAction_JavaScript(
466           jsAction, CFX_WideString::FromLocal(csJSName), this);
467   }
468 }
469
470 FX_BOOL CPDFSDK_Document::ProcOpenAction() {
471   if (!m_pDoc)
472     return FALSE;
473
474   CPDF_Dictionary* pRoot = m_pDoc->GetRoot();
475   if (!pRoot)
476     return FALSE;
477
478   CPDF_Object* pOpenAction = pRoot->GetDict("OpenAction");
479   if (!pOpenAction)
480     pOpenAction = pRoot->GetArray("OpenAction");
481
482   if (!pOpenAction)
483     return FALSE;
484
485   if (pOpenAction->IsArray())
486     return TRUE;
487
488   if (CPDF_Dictionary* pDict = pOpenAction->AsDictionary()) {
489     CPDF_Action action(pDict);
490     if (m_pEnv->GetActionHander())
491       m_pEnv->GetActionHander()->DoAction_DocOpen(action, this);
492     return TRUE;
493   }
494   return FALSE;
495 }
496
497 CPDF_OCContext* CPDFSDK_Document::GetOCContext() {
498   if (!m_pOccontent)
499     m_pOccontent = new CPDF_OCContext(m_pDoc);
500   return m_pOccontent;
501 }
502
503 void CPDFSDK_Document::ReMovePageView(CPDF_Page* pPDFPage) {
504   auto it = m_pageMap.find(pPDFPage);
505   if (it == m_pageMap.end())
506     return;
507
508   CPDFSDK_PageView* pPageView = it->second;
509   if (pPageView->IsLocked())
510     return;
511
512   delete pPageView;
513   m_pageMap.erase(it);
514 }
515
516 CPDF_Page* CPDFSDK_Document::GetPage(int nIndex) {
517   CPDF_Page* pTempPage = (CPDF_Page*)m_pEnv->FFI_GetPage(m_pDoc, nIndex);
518   if (!pTempPage)
519     return NULL;
520   return pTempPage;
521 }
522
523 CPDFSDK_InterForm* CPDFSDK_Document::GetInterForm() {
524   if (!m_pInterForm)
525     m_pInterForm = new CPDFSDK_InterForm(this);
526   return m_pInterForm;
527 }
528
529 void CPDFSDK_Document::UpdateAllViews(CPDFSDK_PageView* pSender,
530                                       CPDFSDK_Annot* pAnnot) {
531   for (const auto& it : m_pageMap) {
532     CPDFSDK_PageView* pPageView = it.second;
533     if (pPageView != pSender) {
534       pPageView->UpdateView(pAnnot);
535     }
536   }
537 }
538
539 CPDFSDK_Annot* CPDFSDK_Document::GetFocusAnnot() {
540   return m_pFocusAnnot;
541 }
542
543 FX_BOOL CPDFSDK_Document::SetFocusAnnot(CPDFSDK_Annot* pAnnot, FX_UINT nFlag) {
544   if (m_pFocusAnnot == pAnnot)
545     return TRUE;
546
547   if (m_pFocusAnnot) {
548     if (!KillFocusAnnot(nFlag))
549       return FALSE;
550   }
551   CPDFSDK_PageView* pPageView = pAnnot->GetPageView();
552   if (pAnnot && pPageView->IsValid()) {
553     CPDFSDK_AnnotHandlerMgr* pAnnotHandler = m_pEnv->GetAnnotHandlerMgr();
554
555     if (pAnnotHandler && !m_pFocusAnnot) {
556       if (!pAnnotHandler->Annot_OnSetFocus(pAnnot, nFlag))
557         return FALSE;
558       if (!m_pFocusAnnot) {
559         m_pFocusAnnot = pAnnot;
560         return TRUE;
561       }
562     }
563   }
564   return FALSE;
565 }
566
567 FX_BOOL CPDFSDK_Document::KillFocusAnnot(FX_UINT nFlag) {
568   if (m_pFocusAnnot) {
569     CPDFSDK_AnnotHandlerMgr* pAnnotHandler = m_pEnv->GetAnnotHandlerMgr();
570     if (pAnnotHandler) {
571       CPDFSDK_Annot* pFocusAnnot = m_pFocusAnnot;
572       m_pFocusAnnot = NULL;
573       if (pAnnotHandler->Annot_OnKillFocus(pFocusAnnot, nFlag)) {
574         if (pFocusAnnot->GetType() == FX_BSTRC("Widget")) {
575           CPDFSDK_Widget* pWidget = (CPDFSDK_Widget*)pFocusAnnot;
576           int nFieldType = pWidget->GetFieldType();
577           if (FIELDTYPE_TEXTFIELD == nFieldType ||
578               FIELDTYPE_COMBOBOX == nFieldType)
579             m_pEnv->FFI_OnSetFieldInputFocus(NULL, NULL, 0, FALSE);
580         }
581
582         if (!m_pFocusAnnot)
583           return TRUE;
584       } else {
585         m_pFocusAnnot = pFocusAnnot;
586       }
587     }
588   }
589   return FALSE;
590 }
591
592 void CPDFSDK_Document::OnCloseDocument() {
593   KillFocusAnnot();
594 }
595
596 FX_BOOL CPDFSDK_Document::GetPermissions(int nFlag) {
597   FX_DWORD dwPermissions = m_pDoc->GetUserPermissions();
598   return dwPermissions & nFlag;
599 }
600
601 IJS_Runtime* CPDFSDK_Document::GetJsRuntime() {
602   ASSERT(m_pEnv != NULL);
603   return m_pEnv->GetJSRuntime();
604 }
605
606 CFX_WideString CPDFSDK_Document::GetPath() {
607   ASSERT(m_pEnv != NULL);
608   return m_pEnv->JS_docGetFilePath();
609 }
610
611 CPDFSDK_PageView::CPDFSDK_PageView(CPDFSDK_Document* pSDKDoc, CPDF_Page* page)
612     : m_page(page),
613       m_pSDKDoc(pSDKDoc),
614       m_CaptureWidget(nullptr),
615       m_bEnterWidget(FALSE),
616       m_bExitWidget(FALSE),
617       m_bOnWidget(FALSE),
618       m_bValid(FALSE),
619       m_bLocked(FALSE),
620       m_bTakeOverPage(FALSE) {
621   CPDFSDK_InterForm* pInterForm = pSDKDoc->GetInterForm();
622   if (pInterForm) {
623     CPDF_InterForm* pPDFInterForm = pInterForm->GetInterForm();
624     pPDFInterForm->FixPageFields(page);
625   }
626   m_page->SetPrivateData((void*)m_page, (void*)this, nullptr);
627 }
628
629 CPDFSDK_PageView::~CPDFSDK_PageView() {
630   // if there is a focused annot on the page, we should kill the focus first.
631   if (CPDFSDK_Annot* focusedAnnot = m_pSDKDoc->GetFocusAnnot()) {
632     auto it =
633         std::find(m_fxAnnotArray.begin(), m_fxAnnotArray.end(), focusedAnnot);
634     if (it != m_fxAnnotArray.end())
635       KillFocusAnnot();
636   }
637
638   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
639   CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
640   for (CPDFSDK_Annot* pAnnot : m_fxAnnotArray)
641     pAnnotHandlerMgr->ReleaseAnnot(pAnnot);
642   m_fxAnnotArray.clear();
643
644   m_pAnnotList.reset();
645
646   m_page->RemovePrivateData((void*)m_page);
647   if (m_bTakeOverPage) {
648     delete m_page;
649   }
650 }
651
652 void CPDFSDK_PageView::PageView_OnDraw(CFX_RenderDevice* pDevice,
653                                        CPDF_Matrix* pUser2Device,
654                                        CPDF_RenderOptions* pOptions) {
655   m_curMatrix = *pUser2Device;
656
657   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
658   CPDFSDK_AnnotIterator annotIterator(this, true);
659   while (CPDFSDK_Annot* pSDKAnnot = annotIterator.Next()) {
660     CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
661     pAnnotHandlerMgr->Annot_OnDraw(this, pSDKAnnot, pDevice, pUser2Device, 0);
662   }
663 }
664
665 CPDF_Annot* CPDFSDK_PageView::GetPDFAnnotAtPoint(FX_FLOAT pageX,
666                                                  FX_FLOAT pageY) {
667   int nCount = CountAnnots();
668   for (int i = 0; i < nCount; i++) {
669     CPDF_Annot* pAnnot = m_pAnnotList->GetAt(i);
670     CFX_FloatRect annotRect;
671     pAnnot->GetRect(annotRect);
672     if (annotRect.Contains(pageX, pageY))
673       return pAnnot;
674   }
675   return nullptr;
676 }
677
678 CPDF_Annot* CPDFSDK_PageView::GetPDFWidgetAtPoint(FX_FLOAT pageX,
679                                                   FX_FLOAT pageY) {
680   int nCount = CountAnnots();
681   for (int i = 0; i < nCount; ++i) {
682     CPDF_Annot* pAnnot = m_pAnnotList->GetAt(i);
683     if (pAnnot->GetSubType() == "Widget") {
684       CFX_FloatRect annotRect;
685       pAnnot->GetRect(annotRect);
686       if (annotRect.Contains(pageX, pageY))
687         return pAnnot;
688     }
689   }
690   return nullptr;
691 }
692
693 CPDFSDK_Annot* CPDFSDK_PageView::GetFXAnnotAtPoint(FX_FLOAT pageX,
694                                                    FX_FLOAT pageY) {
695   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
696   CPDFSDK_AnnotHandlerMgr* pAnnotMgr = pEnv->GetAnnotHandlerMgr();
697   CPDFSDK_AnnotIterator annotIterator(this, false);
698   while (CPDFSDK_Annot* pSDKAnnot = annotIterator.Next()) {
699     CPDF_Rect rc = pAnnotMgr->Annot_OnGetViewBBox(this, pSDKAnnot);
700     if (rc.Contains(pageX, pageY))
701       return pSDKAnnot;
702   }
703
704   return nullptr;
705 }
706
707 CPDFSDK_Annot* CPDFSDK_PageView::GetFXWidgetAtPoint(FX_FLOAT pageX,
708                                                     FX_FLOAT pageY) {
709   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
710   CPDFSDK_AnnotHandlerMgr* pAnnotMgr = pEnv->GetAnnotHandlerMgr();
711   CPDFSDK_AnnotIterator annotIterator(this, false);
712   while (CPDFSDK_Annot* pSDKAnnot = annotIterator.Next()) {
713     if (pSDKAnnot->GetType() == "Widget") {
714       pAnnotMgr->Annot_OnGetViewBBox(this, pSDKAnnot);
715       CPDF_Point point(pageX, pageY);
716       if (pAnnotMgr->Annot_OnHitTest(this, pSDKAnnot, point))
717         return pSDKAnnot;
718     }
719   }
720
721   return nullptr;
722 }
723
724 FX_BOOL CPDFSDK_PageView::Annot_HasAppearance(CPDF_Annot* pAnnot) {
725   CPDF_Dictionary* pAnnotDic = pAnnot->GetAnnotDict();
726   if (pAnnotDic)
727     return pAnnotDic->KeyExist("AS");
728   return FALSE;
729 }
730
731 CPDFSDK_Annot* CPDFSDK_PageView::AddAnnot(CPDF_Annot* pPDFAnnot) {
732   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
733   ASSERT(pEnv);
734   CPDFSDK_AnnotHandlerMgr* pAnnotHandler = pEnv->GetAnnotHandlerMgr();
735   if (!pAnnotHandler)
736     return nullptr;
737
738   CPDFSDK_Annot* pSDKAnnot = pAnnotHandler->NewAnnot(pPDFAnnot, this);
739   if (!pSDKAnnot)
740     return nullptr;
741
742   m_fxAnnotArray.push_back(pSDKAnnot);
743   pAnnotHandler->Annot_OnCreate(pSDKAnnot);
744   return pSDKAnnot;
745 }
746
747 CPDFSDK_Annot* CPDFSDK_PageView::AddAnnot(CPDF_Dictionary* pDict) {
748   return pDict ? AddAnnot(pDict->GetString("Subtype"), pDict) : nullptr;
749 }
750
751 CPDFSDK_Annot* CPDFSDK_PageView::AddAnnot(const FX_CHAR* lpSubType,
752                                           CPDF_Dictionary* pDict) {
753   return NULL;
754 }
755
756 FX_BOOL CPDFSDK_PageView::DeleteAnnot(CPDFSDK_Annot* pAnnot) {
757   return FALSE;
758 }
759
760 CPDF_Document* CPDFSDK_PageView::GetPDFDocument() {
761   if (m_page) {
762     return m_page->m_pDocument;
763   }
764   return NULL;
765 }
766
767 int CPDFSDK_PageView::CountAnnots() const {
768   return m_pAnnotList->Count();
769 }
770
771 CPDFSDK_Annot* CPDFSDK_PageView::GetAnnot(size_t nIndex) {
772   return nIndex < m_fxAnnotArray.size() ? m_fxAnnotArray[nIndex] : nullptr;
773 }
774
775 CPDFSDK_Annot* CPDFSDK_PageView::GetAnnotByDict(CPDF_Dictionary* pDict) {
776   for (CPDFSDK_Annot* pAnnot : m_fxAnnotArray) {
777     if (pAnnot->GetPDFAnnot()->GetAnnotDict() == pDict)
778       return pAnnot;
779   }
780   return nullptr;
781 }
782
783 FX_BOOL CPDFSDK_PageView::OnLButtonDown(const CPDF_Point& point,
784                                         FX_UINT nFlag) {
785   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
786   ASSERT(pEnv);
787   CPDFSDK_Annot* pFXAnnot = GetFXWidgetAtPoint(point.x, point.y);
788   if (!pFXAnnot) {
789     KillFocusAnnot(nFlag);
790   } else {
791     CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
792     ASSERT(pAnnotHandlerMgr);
793
794     FX_BOOL bRet =
795         pAnnotHandlerMgr->Annot_OnLButtonDown(this, pFXAnnot, nFlag, point);
796     if (bRet) {
797       SetFocusAnnot(pFXAnnot);
798     }
799     return bRet;
800   }
801   return FALSE;
802 }
803
804 FX_BOOL CPDFSDK_PageView::OnLButtonUp(const CPDF_Point& point, FX_UINT nFlag) {
805   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
806   ASSERT(pEnv);
807   CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
808   ASSERT(pAnnotHandlerMgr);
809   CPDFSDK_Annot* pFXAnnot = GetFXWidgetAtPoint(point.x, point.y);
810   CPDFSDK_Annot* pFocusAnnot = GetFocusAnnot();
811   FX_BOOL bRet = FALSE;
812   if (pFocusAnnot && pFocusAnnot != pFXAnnot) {
813     // Last focus Annot gets a chance to handle the event.
814     bRet = pAnnotHandlerMgr->Annot_OnLButtonUp(this, pFocusAnnot, nFlag, point);
815   }
816   if (pFXAnnot && !bRet) {
817     bRet = pAnnotHandlerMgr->Annot_OnLButtonUp(this, pFXAnnot, nFlag, point);
818     return bRet;
819   }
820   return bRet;
821 }
822
823 FX_BOOL CPDFSDK_PageView::OnMouseMove(const CPDF_Point& point, int nFlag) {
824   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
825   CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
826   ASSERT(pAnnotHandlerMgr);
827   if (CPDFSDK_Annot* pFXAnnot = GetFXWidgetAtPoint(point.x, point.y)) {
828     if (m_CaptureWidget && m_CaptureWidget != pFXAnnot) {
829       m_bExitWidget = TRUE;
830       m_bEnterWidget = FALSE;
831       pAnnotHandlerMgr->Annot_OnMouseExit(this, m_CaptureWidget, nFlag);
832     }
833     m_CaptureWidget = (CPDFSDK_Widget*)pFXAnnot;
834     m_bOnWidget = TRUE;
835     if (!m_bEnterWidget) {
836       m_bEnterWidget = TRUE;
837       m_bExitWidget = FALSE;
838       pAnnotHandlerMgr->Annot_OnMouseEnter(this, pFXAnnot, nFlag);
839     }
840     pAnnotHandlerMgr->Annot_OnMouseMove(this, pFXAnnot, nFlag, point);
841     return TRUE;
842   }
843   if (m_bOnWidget) {
844     m_bOnWidget = FALSE;
845     m_bExitWidget = TRUE;
846     m_bEnterWidget = FALSE;
847     if (m_CaptureWidget) {
848       pAnnotHandlerMgr->Annot_OnMouseExit(this, m_CaptureWidget, nFlag);
849       m_CaptureWidget = NULL;
850     }
851   }
852   return FALSE;
853 }
854
855 FX_BOOL CPDFSDK_PageView::OnMouseWheel(double deltaX,
856                                        double deltaY,
857                                        const CPDF_Point& point,
858                                        int nFlag) {
859   if (CPDFSDK_Annot* pAnnot = GetFXWidgetAtPoint(point.x, point.y)) {
860     CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
861     CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
862     ASSERT(pAnnotHandlerMgr);
863     return pAnnotHandlerMgr->Annot_OnMouseWheel(this, pAnnot, nFlag,
864                                                 (int)deltaY, point);
865   }
866   return FALSE;
867 }
868
869 FX_BOOL CPDFSDK_PageView::OnChar(int nChar, FX_UINT nFlag) {
870   if (CPDFSDK_Annot* pAnnot = GetFocusAnnot()) {
871     CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
872     CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
873     ASSERT(pAnnotHandlerMgr);
874     return pAnnotHandlerMgr->Annot_OnChar(pAnnot, nChar, nFlag);
875   }
876
877   return FALSE;
878 }
879
880 FX_BOOL CPDFSDK_PageView::OnKeyDown(int nKeyCode, int nFlag) {
881   if (CPDFSDK_Annot* pAnnot = GetFocusAnnot()) {
882     CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
883     CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
884     ASSERT(pAnnotHandlerMgr);
885     return pAnnotHandlerMgr->Annot_OnKeyDown(pAnnot, nKeyCode, nFlag);
886   }
887   return FALSE;
888 }
889
890 FX_BOOL CPDFSDK_PageView::OnKeyUp(int nKeyCode, int nFlag) {
891   return FALSE;
892 }
893
894 void CPDFSDK_PageView::LoadFXAnnots() {
895   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
896
897   FX_BOOL enableAPUpdate = CPDF_InterForm::UpdatingAPEnabled();
898   // Disable the default AP construction.
899   CPDF_InterForm::EnableUpdateAP(FALSE);
900   m_pAnnotList.reset(new CPDF_AnnotList(m_page));
901   CPDF_InterForm::EnableUpdateAP(enableAPUpdate);
902   int nCount = CountAnnots();
903   SetLock(TRUE);
904   for (int i = 0; i < nCount; ++i) {
905     CPDF_Annot* pPDFAnnot = m_pAnnotList->GetAt(i);
906     CPDF_Document* pDoc = GetPDFDocument();
907
908     CheckUnSupportAnnot(pDoc, pPDFAnnot);
909
910     CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr = pEnv->GetAnnotHandlerMgr();
911     ASSERT(pAnnotHandlerMgr != NULL);
912
913     if (pAnnotHandlerMgr) {
914       CPDFSDK_Annot* pAnnot = pAnnotHandlerMgr->NewAnnot(pPDFAnnot, this);
915       if (!pAnnot)
916         continue;
917       m_fxAnnotArray.push_back(pAnnot);
918
919       pAnnotHandlerMgr->Annot_OnLoad(pAnnot);
920     }
921   }
922   SetLock(FALSE);
923 }
924
925 void CPDFSDK_PageView::UpdateRects(CFX_RectArray& rects) {
926   for (int i = 0; i < rects.GetSize(); i++) {
927     CPDF_Rect rc = rects.GetAt(i);
928     CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
929     pEnv->FFI_Invalidate(m_page, rc.left, rc.top, rc.right, rc.bottom);
930   }
931 }
932
933 void CPDFSDK_PageView::UpdateView(CPDFSDK_Annot* pAnnot) {
934   CPDF_Rect rcWindow = pAnnot->GetRect();
935   CPDFDoc_Environment* pEnv = m_pSDKDoc->GetEnv();
936   pEnv->FFI_Invalidate(m_page, rcWindow.left, rcWindow.top, rcWindow.right,
937                        rcWindow.bottom);
938 }
939
940 int CPDFSDK_PageView::GetPageIndex() {
941   if (m_page) {
942     CPDF_Dictionary* pDic = m_page->m_pFormDict;
943     CPDF_Document* pDoc = m_pSDKDoc->GetDocument();
944     if (pDoc && pDic) {
945       return pDoc->GetPageIndex(pDic->GetObjNum());
946     }
947   }
948   return -1;
949 }
950
951 FX_BOOL CPDFSDK_PageView::IsValidAnnot(CPDF_Annot* p) const {
952   if (!p)
953     return FALSE;
954
955   int nCount = CountAnnots();
956   for (int i = 0; i < nCount; ++i) {
957     if (m_pAnnotList->GetAt(i) == p)
958       return TRUE;
959   }
960   return FALSE;
961 }
962
963 CPDFSDK_Annot* CPDFSDK_PageView::GetFocusAnnot() {
964   CPDFSDK_Annot* pFocusAnnot = m_pSDKDoc->GetFocusAnnot();
965   if (!pFocusAnnot)
966     return nullptr;
967
968   for (CPDFSDK_Annot* pAnnot : m_fxAnnotArray) {
969     if (pAnnot == pFocusAnnot)
970       return pAnnot;
971   }
972   return nullptr;
973 }