[Privet FS] Allow files to be able to be copied in to the privet filesystem

chrome/browser/local_discovery/privet_http_impl.cc

 // 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.
 
 #include "chrome/browser/local_discovery/privet_http_impl.h"
 
 #include <algorithm>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 42 matching lines @@
 const char kPrivetTypeFile[] = "file";
 
 const char kPrivetKeyJobID[] = "job_id";
 
 const int kPrivetCancelationTimeoutSeconds = 3;
 
 const int kPrivetLocalPrintMaxRetries = 2;
 
 const int kPrivetLocalPrintDefaultTimeout = 5;
 
+const char kFileTypeOctetStream[] = "application/octet-stream";
+
 GURL CreatePrivetURL(const std::string& path) {
   GURL url(kUrlPlaceHolder);
   GURL::Replacements replacements;
   replacements.SetPathStr(path);
   return url.ReplaceComponents(replacements);
 }
 
 GURL CreatePrivetRegisterURL(const std::string& action,
                              const std::string& user) {
   GURL url = CreatePrivetURL(kPrivetRegisterPath);
@@ skipping 307 matching lines @@
 }
 
 void PrivetJSONOperationImpl::AppendQueryParameter(const std::string& name,
                                                    const std::string& value) {
   url_ = net::AppendQueryParameter(url_, name, value);
 }
 
 PrivetDataReadOperationImpl::PrivetDataReadOperationImpl(
     PrivetHTTPClientImpl* privet_client,
     const std::string& path,
+    net::URLFetcher::RequestType request_type,
     const PrivetDataReadOperation::ResultCallback& callback)
     : privet_client_(privet_client),
       url_(CreatePrivetURL(path)),
+      request_type_(request_type),
       callback_(callback),
       has_range_(false),
-      save_to_file_(false) {}
+      save_to_file_(false),
+      has_upload_file_(false) {}
 
 PrivetDataReadOperationImpl::~PrivetDataReadOperationImpl() {
 }
 
 void PrivetDataReadOperationImpl::Start() {
-  url_fetcher_ =
-      privet_client_->CreateURLFetcher(url_, net::URLFetcher::GET, this);
+  url_fetcher_ = privet_client_->CreateURLFetcher(url_, request_type_, this);
   url_fetcher_->DoNotRetryOnTransientError();
 
   if (has_range_) {
     url_fetcher_->SetByteRange(range_start_, range_end_);
   }
 
   if (save_to_file_) {
     url_fetcher_->SaveResponseToFile();
   }
 
+  if (has_upload_file_) {
+    DCHECK(request_type_ == net::URLFetcher::PUT);
+    url_fetcher_->SetUploadFilePath(kFileTypeOctetStream, upload_file_);
+  }
+
   url_fetcher_->Start();
 }
 
 void PrivetDataReadOperationImpl::SetDataRange(int range_start, int range_end) {
   has_range_ = true;
   range_start_ = range_start;
   range_end_ = range_end;
 }
 
 void PrivetDataReadOperationImpl::SaveDataToFile() { save_to_file_ = true; }
 
+void PrivetDataReadOperationImpl::SetUploadFile(
+    const base::FilePath& upload_file) {
+  has_upload_file_ = true;
+  upload_file_ = upload_file;
+}
+
 PrivetHTTPClient* PrivetDataReadOperationImpl::GetHTTPClient() {
   return privet_client_;
 }
 
 void PrivetDataReadOperationImpl::OnError(
     PrivetURLFetcher* fetcher,
     PrivetURLFetcher::ErrorType error) {
   callback_.Run(RESPONSE_TYPE_ERROR, std::string(), base::FilePath());
 }
 
@@ skipping 406 matching lines @@
   operation->AppendQueryParameter(kPrivetStorageParamPath, path);
   return operation.PassAs<PrivetJSONOperation>();
 }
 
 scoped_ptr<PrivetDataReadOperation>
 PrivetHTTPClientImpl::CreateStorageReadOperation(
     const std::string& path,
     const PrivetDataReadOperation::ResultCallback& callback) {
   scoped_ptr<PrivetDataReadOperationImpl> operation(
       new PrivetDataReadOperationImpl(
-          this, kPrivetStorageContentPath, callback));
+          this, kPrivetStorageContentPath, net::URLFetcher::GET, callback));
 
   operation->AppendQueryParameter(kPrivetStorageParamPath, path);
   return operation.PassAs<PrivetDataReadOperation>();
 }
 
 scoped_ptr<PrivetJSONOperation>
 PrivetHTTPClientImpl::CreateStorageDeleteOperation(
     const std::string& path,
     const PrivetJSONOperation::ResultCallback& callback) {
   scoped_ptr<PrivetJSONOperationImpl> operation(new PrivetJSONOperationImpl(
@@ skipping 39 matching lines @@
       this, kPrivetStorageCreatePath, net::URLFetcher::POST, callback));
 
   operation->AppendQueryParameter(kPrivetStorageParamPath, path);
   operation->AppendQueryParameter(kPrivetStorageParamName, name);
   operation->AppendQueryParameter(kPrivetStorageParamType,
                                   is_dir ? kPrivetTypeDir : kPrivetTypeFile);
 
   return operation.PassAs<PrivetJSONOperation>();
 }
 
+scoped_ptr<PrivetDataReadOperation>
+PrivetHTTPClientImpl::CreateStorageOverwriteOperation(
+    const std::string& path,
+    const PrivetDataReadOperation::ResultCallback& callback) {
+  scoped_ptr<PrivetDataReadOperationImpl> operation(
+      new PrivetDataReadOperationImpl(
+          this, kPrivetStorageContentPath, net::URLFetcher::PUT, callback));
+
+  operation->AppendQueryParameter(kPrivetStorageParamPath, path);
+  return operation.PassAs<PrivetDataReadOperation>();
+}
+
 const std::string& PrivetHTTPClientImpl::GetName() {
   return name_;
 }
 
 scoped_ptr<PrivetURLFetcher> PrivetHTTPClientImpl::CreateURLFetcher(
     const GURL& url, net::URLFetcher::RequestType request_type,
     PrivetURLFetcher::Delegate* delegate) const {
   GURL::Replacements replacements;
   replacements.SetHostStr(host_port_.host());
   std::string port(base::IntToString(host_port_.port()));  // Keep string alive.
@@ skipping 31 matching lines @@
   TokenCallbackVector token_callbacks;
   token_callbacks_.swap(token_callbacks);
 
   for (TokenCallbackVector::iterator i = token_callbacks.begin();
        i != token_callbacks.end(); i++) {
     i->Run(token);
   }
 }
 
 }  // namespace local_discovery