chrome/browser/extensions/api/image_writer_private/write_from_url_operation.h

// Copyright 2013 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 CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_WRITE_FROM_URL_OPERATION_H_
#define CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_WRITE_FROM_URL_OPERATION_H_

#include "base/scoped_observer.h"
#include "chrome/browser/extensions/api/image_writer_private/operation.h"
#include "content/public/browser/download_item.h"
#include "net/base/net_errors.h"
#include "url/gurl.h"

namespace content {

class RenderViewHost;

}  // namespace content

namespace extensions {
namespace image_writer {

class OperationManager;

// Encapsulates a write of an image accessed via URL.
class WriteFromUrlOperation : public Operation,
                              public content::DownloadItem::Observer {
 public:
  WriteFromUrlOperation(base::WeakPtr<OperationManager> manager,
                        const ExtensionId& extension_id,
                        content::RenderViewHost* rvh,
                        GURL url,
                        const std::string& hash,
                        bool saveImageAsDownload,
                        const std::string& storage_unit_id);
  virtual void Start() OVERRIDE;
 private:
  virtual ~WriteFromUrlOperation();
  void CreateTempFile();

  void DownloadStart();
  void OnDownloadStarted(content::DownloadItem*, net::Error);
  virtual void OnDownloadUpdated(content::DownloadItem* download) OVERRIDE;
  void DownloadComplete();
  void DownloadCleanUp();

  void VerifyDownloadStart();
  void VerifyDownloadRun();
  void VerifyDownloadCompare(scoped_ptr<std::string> download_hash);
  void VerifyDownloadComplete();

  // Arguments
  content::RenderViewHost* rvh_;
  GURL url_;
  const std::string hash_;
  const bool saveImageAsDownload_;

  // Local state
  scoped_ptr<base::FilePath> tmp_file_;
  bool download_stopped_;
  content::DownloadItem* download_;
  base::FilePath download_path_;
};

} // namespace image_writer
} // namespace extensions

#endif  // CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_WRITE_FROM_URL_OPERATION_H_