Add function to document_loader to get PDF as pp::FileRef

pdf/document_loader.h

 // Copyright (c) 2010 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.
 
 #ifndef PDF_DOCUMENT_LOADER_H_
 #define PDF_DOCUMENT_LOADER_H_
 
 #include <list>
 #include <string>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "pdf/chunk_stream.h"
+#include "pdf/file_writer.h"
 #include "ppapi/cpp/url_loader.h"
 #include "ppapi/utility/completion_callback_factory.h"
 
 #define kDefaultRequestSize 32768u
 
+namespace pp {
+class Buffer_Dev;
+}
+
 namespace chrome_pdf {
 
 class DocumentLoader {
  public:
   class Client {
   public:
     // Gets the pp::Instance object.
     virtual pp::Instance* GetPluginInstance() = 0;
     // Creates new URLLoader based on client settings.
     virtual pp::URLLoader CreateURLLoader() = 0;
@@ skipping 21 matching lines @@
 
   // Data availability interface. Return true data avaialble.
   bool IsDataAvailable(uint32 position, uint32 size) const;
 
   // Data availability interface. Return true data avaialble.
   void RequestData(uint32 position, uint32 size);
 
   bool IsDocumentComplete() const;
   uint32 document_size() const { return document_size_; }
 
+  void GetCompleteDocument(
+       pp::CompletionCallbackWithOutput<pp::FileRef> callback);
+
   // Return number of bytes available.
   uint32 GetAvailableData() const;
 
   // Clear pending requests from the queue.
   void ClearPendingRequests();
 
   bool is_partial_document() { return partial_document_; }
 
  private:
   // Called by the completion callback of the document's URLLoader.
@@ skipping 40 matching lines @@
   char buffer_[kDefaultRequestSize];
   uint32 current_pos_;
   uint32 current_chunk_size_;
   uint32 current_chunk_read_;
   uint32 document_size_;
   bool header_request_;
   bool is_multipart_;
   std::string multipart_boundary_;
   uint32 requests_count_;
   std::list<std::vector<unsigned char> > chunk_buffer_;
+  scoped_ptr<FileWriter> file_writer_;
 };
 
 }  // namespace chrome_pdf
 
 #endif  // PDF_DOCUMENT_LOADER_H_