// 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.

#ifndef CHROME_BROWSER_DOWNLOAD_DOWNLOAD_ITEM_MODEL_H_
#define CHROME_BROWSER_DOWNLOAD_DOWNLOAD_ITEM_MODEL_H_
#pragma once

#include <string>

#include "base/basictypes.h"
#include "base/compiler_specific.h"
#include "base/string16.h"

class SavePackage;

namespace content {
class DownloadItem;
}

namespace gfx {
class Font;
}

// This class is an abstraction for common UI tasks associated with a download.
class BaseDownloadItemModel {
 public:
  explicit BaseDownloadItemModel(content::DownloadItem* download)
      : download_(download) { }
  virtual ~BaseDownloadItemModel() { }

  // Cancel the task corresponding to the item.
  virtual void CancelTask() = 0;

  // Get the status text to display.
  virtual string16 GetStatusText() = 0;

  // Get the warning text to display for a dangerous download. The |base_width|
  // is the maximum width of an embedded filename (if there is one). The metrics
  // for the filename will be based on |font|. Should only be called if
  // IsDangerous() is true.
  virtual string16 GetWarningText(const gfx::Font& font, int base_width) = 0;

  // Get the caption text for a button for confirming a dangerous download
  // warning.
  virtual string16 GetWarningConfirmButtonText() = 0;

  // Is this considered a malicious download? Implies IsDangerous().
  virtual bool IsMalicious() = 0;

  // Is this considered a dangerous download?
  virtual bool IsDangerous() = 0;

  content::DownloadItem* download() { return download_; }

 protected:
  content::DownloadItem* download_;
};

// Concrete implementation of BaseDownloadItemModel.
class DownloadItemModel : public BaseDownloadItemModel {
 public:
  explicit DownloadItemModel(content::DownloadItem* download);
  virtual ~DownloadItemModel() { }

  // BaseDownloadItemModel
  virtual void CancelTask() OVERRIDE;
  virtual string16 GetStatusText() OVERRIDE;
  virtual string16 GetWarningText(const gfx::Font& font,
                                  int base_width) OVERRIDE;
  virtual string16 GetWarningConfirmButtonText() OVERRIDE;
  virtual bool IsMalicious() OVERRIDE;
  virtual bool IsDangerous() OVERRIDE;

 private:
  DISALLOW_COPY_AND_ASSIGN(DownloadItemModel);
};

#endif  // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_ITEM_MODEL_H_