1 // Copyright (c) 2012 The Chromium 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.
5 #include "content/browser/download/save_item.h"
7 #include "base/logging.h"
8 #include "base/strings/string_util.h"
9 #include "content/browser/download/save_file.h"
10 #include "content/browser/download/save_file_manager.h"
11 #include "content/browser/download/save_package.h"
15 // Constructor for SaveItem when creating each saving job.
16 SaveItem::SaveItem(const GURL& url,
17 const Referrer& referrer,
19 SaveFileCreateInfo::SaveFileSource save_source)
26 has_final_name_(false),
28 save_source_(save_source),
33 SaveItem::~SaveItem() {
36 // Set start state for save item.
37 void SaveItem::Start() {
38 DCHECK(state_ == WAIT_START);
42 // If we've received more data than we were expecting (bad server info?),
43 // revert to 'unknown size mode'.
44 void SaveItem::UpdateSize(int64 bytes_so_far) {
45 received_bytes_ = bytes_so_far;
46 if (received_bytes_ >= total_bytes_)
50 // Updates from the file thread may have been posted while this saving job
51 // was being canceled in the UI thread, so we'll accept them unless we're
53 void SaveItem::Update(int64 bytes_so_far) {
54 if (state_ != IN_PROGRESS) {
58 UpdateSize(bytes_so_far);
61 // Cancel this saving item job. If the job is not in progress, ignore
62 // this command. The SavePackage will each in-progress SaveItem's cancel
63 // when canceling whole saving page job.
64 void SaveItem::Cancel() {
65 // If item is in WAIT_START mode, which means no request has been sent.
66 // So we need not to cancel it.
67 if (state_ != IN_PROGRESS) {
68 // Small downloads might be complete before method has a chance to run.
73 Finish(received_bytes_, false);
74 package_->SaveCanceled(this);
77 // Set finish state for a save item
78 void SaveItem::Finish(int64 size, bool is_success) {
79 // When this function is called, the SaveItem should be one of following
81 // a) The data of this SaveItem is finished saving. So it should have
82 // generated final name.
83 // b) Error happened before the start of saving process. So no |save_id_| is
84 // generated for this SaveItem and the |is_success_| should be false.
85 // c) Error happened in the start of saving process, the SaveItem has a save
86 // id, |is_success_| should be false, and the |size| should be 0.
87 DCHECK(has_final_name() || (save_id_ == -1 && !is_success_) ||
88 (save_id_ != -1 && !is_success_ && !size));
90 is_success_ = is_success;
94 // Calculate the percentage of the save item
95 int SaveItem::PercentComplete() const {
104 if (total_bytes_ > 0)
105 percent = static_cast<int>(received_bytes_ * 100.0 / total_bytes_);
115 // Rename the save item with new path.
116 void SaveItem::Rename(const base::FilePath& full_path) {
117 DCHECK(!full_path.empty() && !has_final_name());
118 full_path_ = full_path;
119 file_name_ = full_path_.BaseName();
120 has_final_name_ = true;
123 void SaveItem::SetSaveId(int32 save_id) {
124 DCHECK_EQ(-1, save_id_);
128 void SaveItem::SetTotalBytes(int64 total_bytes) {
129 DCHECK_EQ(0, total_bytes_);
130 total_bytes_ = total_bytes;
133 } // namespace content