Add functions to collect bookmarks from Pdfium

pdf/pdfium/pdfium_engine.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "pdf/pdfium/pdfium_engine.h"
 
 #include <math.h>
 
 #include "base/json/json_writer.h"
 #include "base/logging.h"
@@ skipping 12 matching lines @@
 #include "ppapi/c/ppb_core.h"
 #include "ppapi/c/private/ppb_pdf.h"
 #include "ppapi/cpp/dev/memory_dev.h"
 #include "ppapi/cpp/input_event.h"
 #include "ppapi/cpp/instance.h"
 #include "ppapi/cpp/module.h"
 #include "ppapi/cpp/private/pdf.h"
 #include "ppapi/cpp/trusted/browser_font_trusted.h"
 #include "ppapi/cpp/url_response_info.h"
 #include "ppapi/cpp/var.h"
+#include "ppapi/cpp/var_dictionary.h"
 #include "third_party/pdfium/fpdfsdk/include/fpdf_ext.h"
 #include "third_party/pdfium/fpdfsdk/include/fpdf_flatten.h"
 #include "third_party/pdfium/fpdfsdk/include/fpdf_searchex.h"
 #include "third_party/pdfium/fpdfsdk/include/fpdf_sysfontinfo.h"
 #include "third_party/pdfium/fpdfsdk/include/fpdf_transformpage.h"
 #include "third_party/pdfium/fpdfsdk/include/fpdfedit.h"
 #include "third_party/pdfium/fpdfsdk/include/fpdfoom.h"
 #include "third_party/pdfium/fpdfsdk/include/fpdfppo.h"
 #include "third_party/pdfium/fpdfsdk/include/fpdfsave.h"
 #include "third_party/pdfium/fpdfsdk/include/pdfwindow/PDFWindow.h"
@@ skipping 2247 matching lines @@
     if (pages_[i]->available()) {
       selection_.push_back(PDFiumRange(pages_[i], 0,
                            pages_[i]->GetCharCount()));
     }
 }
 
 int PDFiumEngine::GetNumberOfPages() {
   return pages_.size();
 }
 
+int NumChildren(FPDF_DOCUMENT doc, FPDF_BOOKMARK bookmark)
+{
+  int num = 0;
+  FPDF_BOOKMARK child = FPDFBookmark_GetFirstChild(doc, bookmark);
+  while (child != NULL) {
+      child = FPDFBookmark_GetNextSibling(doc, child);
+      num++;
+  }
+  return num;
+}
+
+void TraverseBookmarks(FPDF_BOOKMARK bookmark, pp::VarDictionary *dict,

[Sam McNally, 2014/12/16 05:58:34]

Put this in the anonymous namespace above.

Return a pp::VarDictionary instead of taking one to fill. Also, to be
consistent, you should take |doc| before |bookmark|.

[Alexandre Carlton, 2014/12/16 22:14:51]

Done.

+                       FPDF_DOCUMENT doc) {
+  base::string16 title;
+  size_t buffer_size = FPDFBookmark_GetTitle(bookmark, NULL, 0);
+  if (buffer_size > 1) {
+    void *data = WriteInto(&title, buffer_size + 1);
+    FPDFBookmark_GetTitle(bookmark, data, buffer_size);

[Sam McNally, 2014/12/16 05:58:34]

I would pass the result of WriteInto(...) directly into FPDFBookmark_GetTitle.

[Alexandre Carlton, 2014/12/16 22:14:51]

Done.

+  }
+  std::string shortTitle = base::UTF16ToUTF8(title);
+  dict->Set(pp::Var("title"), pp::Var(shortTitle));

[Sam McNally, 2014/12/16 05:58:34]

Remove shortTitle and pass base::UTF16ToUTF8(title).

[Alexandre Carlton, 2014/12/16 22:14:51]

Done.

+
+  FPDF_DEST dest = FPDFBookmark_GetDest(doc, bookmark);
+  int pageIndex = dest ? FPDFDest_GetPageIndex(doc, dest) : -1;

[Sam McNally, 2014/12/16 05:58:34]

How about only setting "page" for bookmarks that have a destination page?

[Alexandre Carlton, 2014/12/16 22:14:51]

Done.

+  dict->Set(pp::Var("page"), pp::Var(pageIndex));
+
+  int length = NumChildren(doc, bookmark);
+  pp::VarArray *children = new pp::VarArray();

[Sam McNally, 2014/12/16 05:58:35]

Put this on the stack. Otherwise, it will leak.

[Alexandre Carlton, 2014/12/16 22:14:51]

Done.

+  children->SetLength(length);

[Sam McNally, 2014/12/16 05:58:34]

You don't need this. VarArray resizes to fit on calls to Set().

[Alexandre Carlton, 2014/12/16 22:14:51]

Done.

+
+  int i = 0;
+  pp::VarDictionary *childDict;

[Sam McNally, 2014/12/16 05:58:34]

Declare this inside the loop (or remove it entirely once this returns the
dictionary instead of taking one). See above about leaking too.

[Alexandre Carlton, 2014/12/16 22:14:51]

Done.

+  bookmark = FPDFBookmark_GetFirstChild(doc, bookmark);

[Sam McNally, 2014/12/16 05:58:35]

Use a different variable for the child bookmarks.

[Alexandre Carlton, 2014/12/16 22:14:51]

Done.

+  while (bookmark != NULL) {

[Sam McNally, 2014/12/16 05:58:34]

I think this would be nicer as a for loop.

[Alexandre Carlton, 2014/12/16 22:14:51]

Done.

+    childDict = new pp::VarDictionary();
+    TraverseBookmarks(bookmark, childDict, doc);
+    children->Set(i, *childDict);
+    bookmark = FPDFBookmark_GetNextSibling(doc, bookmark);
+    i++;
+  }
+  dict->Set(pp::Var("children"), *children);
+}
+
+pp::VarDictionary PDFiumEngine::GetBookmarks() {
+  pp::VarDictionary *dict = new pp::VarDictionary();
+  TraverseBookmarks(NULL, dict, doc_);
+  return *dict;
+}
+
 int PDFiumEngine::GetNamedDestinationPage(const std::string& destination) {
   // Look for the destination.
   FPDF_DEST dest = FPDF_GetNamedDestByName(doc_, destination.c_str());
   if (!dest) {
     // Look for a bookmark with the same name.
     base::string16 destination_wide = base::UTF8ToUTF16(destination);
     FPDF_WIDESTRING destination_pdf_wide =
         reinterpret_cast<FPDF_WIDESTRING>(destination_wide.c_str());
     FPDF_BOOKMARK bookmark = FPDFBookmark_Find(doc_, destination_pdf_wide);
     if (!bookmark)
@@ skipping 1586 matching lines @@
     double* height) {
   FPDF_DOCUMENT doc = FPDF_LoadMemDocument(pdf_buffer, pdf_buffer_size, NULL);
   if (!doc)
     return false;
   bool success = FPDF_GetPageSizeByIndex(doc, page_number, width, height) != 0;
   FPDF_CloseDocument(doc);
   return success;
 }
 
 }  // namespace chrome_pdf