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.
4
5 #include "content/browser/download/save_item.h"
6
7 #include "base/check_op.h"
8 #include "base/notreached.h"
9 #include "base/strings/string_util.h"
10 #include "content/browser/download/save_file.h"
11 #include "content/browser/download/save_file_manager.h"
12 #include "content/browser/download/save_package.h"
13 #include "content/public/browser/browser_thread.h"
14
15 namespace content {
16
17 namespace {
18
GetNextSaveItemId()19 SaveItemId GetNextSaveItemId() {
20 static SaveItemId::Generator g_save_item_id_generator;
21 DCHECK_CURRENTLY_ON(BrowserThread::UI);
22 return g_save_item_id_generator.GenerateNextId();
23 }
24
25 } // namespace
26
27 // Constructor for SaveItem when creating each saving job.
SaveItem(const GURL & url,const Referrer & referrer,SavePackage * package,SaveFileCreateInfo::SaveFileSource save_source,int frame_tree_node_id,int container_frame_tree_node_id)28 SaveItem::SaveItem(const GURL& url,
29 const Referrer& referrer,
30 SavePackage* package,
31 SaveFileCreateInfo::SaveFileSource save_source,
32 int frame_tree_node_id,
33 int container_frame_tree_node_id)
34 : save_item_id_(GetNextSaveItemId()),
35 url_(url),
36 referrer_(referrer),
37 frame_tree_node_id_(frame_tree_node_id),
38 container_frame_tree_node_id_(container_frame_tree_node_id),
39 received_bytes_(0),
40 state_(WAIT_START),
41 is_success_(false),
42 save_source_(save_source),
43 package_(package) {
44 DCHECK(package);
45 }
46
~SaveItem()47 SaveItem::~SaveItem() {}
48
49 // Set start state for save item.
Start()50 void SaveItem::Start() {
51 DCHECK_EQ(state_, WAIT_START);
52 state_ = IN_PROGRESS;
53 }
54
UpdateSize(int64_t bytes_so_far)55 void SaveItem::UpdateSize(int64_t bytes_so_far) {
56 received_bytes_ = bytes_so_far;
57 }
58
59 // Updates from the file thread may have been posted while this saving job
60 // was being canceled in the UI thread, so we'll accept them unless we're
61 // complete.
Update(int64_t bytes_so_far)62 void SaveItem::Update(int64_t bytes_so_far) {
63 if (state_ != IN_PROGRESS) {
64 NOTREACHED();
65 return;
66 }
67 UpdateSize(bytes_so_far);
68 }
69
70 // Cancel this saving item job. If the job is not in progress, ignore
71 // this command. The SavePackage will each in-progress SaveItem's cancel
72 // when canceling whole saving page job.
Cancel()73 void SaveItem::Cancel() {
74 // If item is in WAIT_START mode, which means no request has been sent.
75 // So we need not to cancel it.
76 if (state_ != IN_PROGRESS) {
77 // Small downloads might be complete before method has a chance to run.
78 return;
79 }
80 Finish(received_bytes_, false);
81 state_ = CANCELED;
82 package_->SaveCanceled(this);
83 }
84
85 // Set finish state for a save item
Finish(int64_t size,bool is_success)86 void SaveItem::Finish(int64_t size, bool is_success) {
87 DCHECK(has_final_name() || !is_success_);
88 state_ = COMPLETE;
89 is_success_ = is_success;
90 UpdateSize(size);
91 }
92
SetTargetPath(const base::FilePath & full_path)93 void SaveItem::SetTargetPath(const base::FilePath& full_path) {
94 DCHECK(!full_path.empty());
95 DCHECK(!has_final_name());
96 full_path_ = full_path;
97 }
98
99 } // namespace content
100