OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/chromeos/extensions/file_manager/private_api_tasks.h" | 5 #include "chrome/browser/chromeos/extensions/file_manager/private_api_tasks.h" |
6 | 6 |
7 #include <set> | 7 #include <set> |
8 #include <string> | 8 #include <string> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
11 #include "chrome/browser/chromeos/drive/file_system_util.h" | 11 #include "chrome/browser/chromeos/drive/file_system_util.h" |
| 12 #include "chrome/browser/chromeos/file_manager/file_tasks.h" |
12 #include "chrome/browser/chromeos/file_manager/fileapi_util.h" | 13 #include "chrome/browser/chromeos/file_manager/fileapi_util.h" |
13 #include "chrome/browser/chromeos/file_manager/mime_util.h" | 14 #include "chrome/browser/chromeos/file_manager/mime_util.h" |
14 #include "chrome/browser/chromeos/fileapi/file_system_backend.h" | 15 #include "chrome/browser/chromeos/fileapi/file_system_backend.h" |
15 #include "chrome/browser/profiles/profile.h" | 16 #include "chrome/browser/profiles/profile.h" |
16 #include "content/public/browser/browser_thread.h" | |
17 #include "net/base/filename_util.h" | |
18 #include "net/base/mime_sniffer.h" | |
19 #include "webkit/browser/fileapi/file_system_context.h" | 17 #include "webkit/browser/fileapi/file_system_context.h" |
20 #include "webkit/browser/fileapi/file_system_url.h" | 18 #include "webkit/browser/fileapi/file_system_url.h" |
21 | 19 |
22 using content::BrowserThread; | |
23 using extensions::app_file_handler_util::PathAndMimeTypeSet; | |
24 using fileapi::FileSystemURL; | 20 using fileapi::FileSystemURL; |
25 | 21 |
26 namespace extensions { | 22 namespace extensions { |
27 namespace { | 23 namespace { |
28 | 24 |
29 // Error messages. | 25 // Error messages. |
30 const char kInvalidFileUrl[] = "Invalid file URL"; | 26 const char kInvalidFileUrl[] = "Invalid file URL"; |
31 | 27 |
32 // Make a set of unique filename suffixes out of the list of file URLs. | 28 // Make a set of unique filename suffixes out of the list of file URLs. |
33 std::set<std::string> GetUniqueSuffixes( | 29 std::set<std::string> GetUniqueSuffixes( |
(...skipping 17 matching lines...) Expand all Loading... |
51 std::set<std::string> mime_types; | 47 std::set<std::string> mime_types; |
52 for (size_t i = 0; i < mime_type_list.size(); ++i) { | 48 for (size_t i = 0; i < mime_type_list.size(); ++i) { |
53 const std::string mime_type = mime_type_list[i]; | 49 const std::string mime_type = mime_type_list[i]; |
54 // We'll skip empty MIME types and existing MIME types. | 50 // We'll skip empty MIME types and existing MIME types. |
55 if (!mime_type.empty()) | 51 if (!mime_type.empty()) |
56 mime_types.insert(mime_type); | 52 mime_types.insert(mime_type); |
57 } | 53 } |
58 return mime_types; | 54 return mime_types; |
59 } | 55 } |
60 | 56 |
61 void SniffMimeType(PathAndMimeTypeSet* path_mime_set, | |
62 std::vector<GURL>* file_urls) { | |
63 PathAndMimeTypeSet sniffed_path_mime_set; | |
64 std::vector<char> content(net::kMaxBytesToSniff); | |
65 | |
66 // For each files, sniff its MIME type if it is empty | |
67 for (PathAndMimeTypeSet::iterator it = path_mime_set->begin(); | |
68 it != path_mime_set->end(); | |
69 ++it) { | |
70 const base::FilePath& file_path = it->first; | |
71 std::string mime_type = it->second; | |
72 // Note: sniff MIME type only for local files. | |
73 if (mime_type.empty() && !drive::util::IsUnderDriveMountPoint(file_path)) { | |
74 int bytes_read = base::ReadFile(file_path, &content[0], content.size()); | |
75 if (bytes_read >= 0) { | |
76 net::SniffMimeType(&content[0], | |
77 bytes_read, | |
78 net::FilePathToFileURL(file_path), | |
79 std::string(), // type_hint (passes no hint) | |
80 &mime_type); | |
81 } | |
82 } | |
83 sniffed_path_mime_set.insert(std::make_pair(file_path, mime_type)); | |
84 } | |
85 path_mime_set->swap(sniffed_path_mime_set); | |
86 } | |
87 | |
88 } // namespace | 57 } // namespace |
89 | 58 |
90 bool FileBrowserPrivateExecuteTaskFunction::RunImpl() { | 59 bool FileBrowserPrivateExecuteTaskFunction::RunImpl() { |
91 using extensions::api::file_browser_private::ExecuteTask::Params; | 60 using extensions::api::file_browser_private::ExecuteTask::Params; |
92 using extensions::api::file_browser_private::ExecuteTask::Results::Create; | 61 using extensions::api::file_browser_private::ExecuteTask::Results::Create; |
93 const scoped_ptr<Params> params(Params::Create(*args_)); | 62 const scoped_ptr<Params> params(Params::Create(*args_)); |
94 EXTENSION_FUNCTION_VALIDATE(params); | 63 EXTENSION_FUNCTION_VALIDATE(params); |
95 | 64 |
96 file_manager::file_tasks::TaskDescriptor task; | 65 file_manager::file_tasks::TaskDescriptor task; |
97 if (!file_manager::file_tasks::ParseTaskID(params->task_id, &task)) { | 66 if (!file_manager::file_tasks::ParseTaskID(params->task_id, &task)) { |
(...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
158 if (params->mime_types.size() != params->file_urls.size() && | 127 if (params->mime_types.size() != params->file_urls.size() && |
159 params->mime_types.size() != 0) | 128 params->mime_types.size() != 0) |
160 return false; | 129 return false; |
161 | 130 |
162 const scoped_refptr<fileapi::FileSystemContext> file_system_context = | 131 const scoped_refptr<fileapi::FileSystemContext> file_system_context = |
163 file_manager::util::GetFileSystemContextForRenderViewHost( | 132 file_manager::util::GetFileSystemContextForRenderViewHost( |
164 GetProfile(), render_view_host()); | 133 GetProfile(), render_view_host()); |
165 | 134 |
166 // Collect all the URLs, convert them to GURLs, and crack all the urls into | 135 // Collect all the URLs, convert them to GURLs, and crack all the urls into |
167 // file paths. | 136 // file paths. |
168 scoped_ptr<PathAndMimeTypeSet> path_mime_set(new PathAndMimeTypeSet); | 137 extensions::app_file_handler_util::PathAndMimeTypeSet path_mime_set; |
169 scoped_ptr<std::vector<GURL> > file_urls(new std::vector<GURL>); | 138 std::vector<GURL> file_urls; |
170 for (size_t i = 0; i < params->file_urls.size(); ++i) { | 139 for (size_t i = 0; i < params->file_urls.size(); ++i) { |
171 std::string mime_type; | 140 std::string mime_type; |
172 if (params->mime_types.size() != 0) | 141 if (params->mime_types.size() != 0) |
173 mime_type = params->mime_types[i]; | 142 mime_type = params->mime_types[i]; |
174 | 143 |
175 const GURL file_url(params->file_urls[i]); | 144 const GURL file_url(params->file_urls[i]); |
176 fileapi::FileSystemURL file_system_url( | 145 fileapi::FileSystemURL file_system_url( |
177 file_system_context->CrackURL(file_url)); | 146 file_system_context->CrackURL(file_url)); |
178 if (!chromeos::FileSystemBackend::CanHandleURL(file_system_url)) | 147 if (!chromeos::FileSystemBackend::CanHandleURL(file_system_url)) |
179 continue; | 148 continue; |
180 const base::FilePath file_path = file_system_url.path(); | 149 const base::FilePath file_path = file_system_url.path(); |
181 | 150 |
182 file_urls->push_back(file_url); | 151 file_urls.push_back(file_url); |
183 | 152 |
184 // If MIME type is not provided, guess it from the file path. | 153 // If MIME type is not provided, guess it from the file path. |
185 if (mime_type.empty()) | 154 if (mime_type.empty()) |
186 mime_type = file_manager::util::GetMimeTypeForPath(file_path); | 155 mime_type = file_manager::util::GetMimeTypeForPath(file_path); |
187 | 156 |
188 path_mime_set->insert(std::make_pair(file_path, mime_type)); | 157 path_mime_set.insert(std::make_pair(file_path, mime_type)); |
189 } | 158 } |
190 | 159 |
191 // In case the MIME type of some files are empty, | |
192 // try to sniff their MIME type by their content. | |
193 PathAndMimeTypeSet* path_mime_set_ptr = path_mime_set.get(); | |
194 std::vector<GURL>* file_urls_ptr = file_urls.get(); | |
195 | |
196 BrowserThread::PostBlockingPoolTaskAndReply( | |
197 FROM_HERE, | |
198 base::Bind(&SniffMimeType, path_mime_set_ptr, file_urls_ptr), | |
199 base::Bind( | |
200 &FileBrowserPrivateGetFileTasksFunction::OnSniffingMimeTypeCompleted, | |
201 this, | |
202 base::Passed(&path_mime_set), | |
203 base::Passed(&file_urls))); | |
204 return true; | |
205 } | |
206 | |
207 void FileBrowserPrivateGetFileTasksFunction::OnSniffingMimeTypeCompleted( | |
208 scoped_ptr<PathAndMimeTypeSet> path_mime_set, | |
209 scoped_ptr<std::vector<GURL> > file_urls) { | |
210 std::vector<file_manager::file_tasks::FullTaskDescriptor> tasks; | 160 std::vector<file_manager::file_tasks::FullTaskDescriptor> tasks; |
211 file_manager::file_tasks::FindAllTypesOfTasks( | 161 file_manager::file_tasks::FindAllTypesOfTasks( |
212 GetProfile(), | 162 GetProfile(), |
213 drive::util::GetDriveAppRegistryByProfile(GetProfile()), | 163 drive::util::GetDriveAppRegistryByProfile(GetProfile()), |
214 *path_mime_set, | 164 path_mime_set, |
215 *file_urls, | 165 file_urls, |
216 &tasks); | 166 &tasks); |
217 | 167 |
218 // Convert the tasks into JSON compatible objects. | 168 // Convert the tasks into JSON compatible objects. |
219 using api::file_browser_private::FileTask; | 169 using api::file_browser_private::FileTask; |
220 std::vector<linked_ptr<FileTask> > results; | 170 std::vector<linked_ptr<FileTask> > results; |
221 for (size_t i = 0; i < tasks.size(); ++i) { | 171 for (size_t i = 0; i < tasks.size(); ++i) { |
222 const file_manager::file_tasks::FullTaskDescriptor& task = tasks[i]; | 172 const file_manager::file_tasks::FullTaskDescriptor& task = tasks[i]; |
223 const linked_ptr<FileTask> converted(new FileTask); | 173 const linked_ptr<FileTask> converted(new FileTask); |
224 converted->task_id = file_manager::file_tasks::TaskDescriptorToId( | 174 converted->task_id = file_manager::file_tasks::TaskDescriptorToId( |
225 task.task_descriptor()); | 175 task.task_descriptor()); |
226 if (!task.icon_url().is_empty()) | 176 if (!task.icon_url().is_empty()) |
227 converted->icon_url = task.icon_url().spec(); | 177 converted->icon_url = task.icon_url().spec(); |
228 converted->title = task.task_title(); | 178 converted->title = task.task_title(); |
229 converted->is_default = task.is_default(); | 179 converted->is_default = task.is_default(); |
230 results.push_back(converted); | 180 results.push_back(converted); |
231 } | 181 } |
232 results_ = extensions::api::file_browser_private::GetFileTasks::Results:: | 182 results_ = extensions::api::file_browser_private::GetFileTasks::Results:: |
233 Create(results); | 183 Create(results); |
234 SendResponse(true); | 184 SendResponse(true); |
| 185 return true; |
235 } | 186 } |
236 | 187 |
237 bool FileBrowserPrivateSetDefaultTaskFunction::RunImpl() { | 188 bool FileBrowserPrivateSetDefaultTaskFunction::RunImpl() { |
238 using extensions::api::file_browser_private::SetDefaultTask::Params; | 189 using extensions::api::file_browser_private::SetDefaultTask::Params; |
239 const scoped_ptr<Params> params(Params::Create(*args_)); | 190 const scoped_ptr<Params> params(Params::Create(*args_)); |
240 EXTENSION_FUNCTION_VALIDATE(params); | 191 EXTENSION_FUNCTION_VALIDATE(params); |
241 | 192 |
242 const scoped_refptr<fileapi::FileSystemContext> file_system_context = | 193 const scoped_refptr<fileapi::FileSystemContext> file_system_context = |
243 file_manager::util::GetFileSystemContextForRenderViewHost( | 194 file_manager::util::GetFileSystemContextForRenderViewHost( |
244 GetProfile(), render_view_host()); | 195 GetProfile(), render_view_host()); |
(...skipping 19 matching lines...) Expand all Loading... |
264 SetResult(new base::FundamentalValue(true)); | 215 SetResult(new base::FundamentalValue(true)); |
265 return true; | 216 return true; |
266 } | 217 } |
267 | 218 |
268 file_manager::file_tasks::UpdateDefaultTask( | 219 file_manager::file_tasks::UpdateDefaultTask( |
269 GetProfile()->GetPrefs(), params->task_id, suffixes, mime_types); | 220 GetProfile()->GetPrefs(), params->task_id, suffixes, mime_types); |
270 return true; | 221 return true; |
271 } | 222 } |
272 | 223 |
273 } // namespace extensions | 224 } // namespace extensions |
OLD | NEW |