Index: trunk/src/mojo/shell/dynamic_service_loader.cc |
=================================================================== |
--- trunk/src/mojo/shell/dynamic_service_loader.cc (revision 288042) |
+++ trunk/src/mojo/shell/dynamic_service_loader.cc (working copy) |
@@ -20,148 +20,159 @@ |
namespace mojo { |
namespace shell { |
- |
namespace { |
-void RunLibraryComplete(DynamicServiceRunner* runner, |
- const base::FilePath& temp_file) { |
- delete runner; |
- if (!temp_file.empty()) |
- base::DeleteFile(temp_file, false); |
-} |
+class Loader { |
+ public: |
+ explicit Loader(scoped_ptr<DynamicServiceRunner> runner) |
+ : runner_(runner.Pass()) { |
+ } |
-} // namespace |
+ virtual void Start(const GURL& url, |
+ ScopedMessagePipeHandle service_handle, |
+ Context* context) = 0; |
-DynamicServiceLoader::DynamicServiceLoader( |
- Context* context, |
- scoped_ptr<DynamicServiceRunnerFactory> runner_factory) |
- : context_(context), |
- runner_factory_(runner_factory.Pass()), |
- weak_ptr_factory_(this) { |
-} |
+ void StartService(const base::FilePath& path, |
+ ScopedMessagePipeHandle service_handle, |
+ bool path_is_valid) { |
+ if (path_is_valid) { |
+ runner_->Start(path, service_handle.Pass(), |
+ base::Bind(&Loader::AppCompleted, base::Unretained(this))); |
+ } else { |
+ AppCompleted(); |
+ } |
+ } |
-DynamicServiceLoader::~DynamicServiceLoader() { |
-} |
+ protected: |
+ virtual ~Loader() {} |
-void DynamicServiceLoader::RegisterContentHandler( |
- const std::string& mime_type, |
- const GURL& content_handler_url) { |
- mime_type_to_url_[mime_type] = content_handler_url; |
-} |
- |
-void DynamicServiceLoader::Load(ServiceManager* manager, |
- const GURL& url, |
- scoped_refptr<LoadCallbacks> callbacks) { |
- GURL resolved_url; |
- if (url.SchemeIs("mojo")) { |
- resolved_url = context_->mojo_url_resolver()->Resolve(url); |
- } else { |
- resolved_url = url; |
+ private: |
+ void AppCompleted() { |
+ delete this; |
} |
- if (resolved_url.SchemeIsFile()) |
- LoadLocalService(resolved_url, callbacks); |
- else |
- LoadNetworkService(resolved_url, callbacks); |
-} |
+ scoped_ptr<DynamicServiceRunner> runner_; |
+}; |
-void DynamicServiceLoader::LoadLocalService( |
- const GURL& resolved_url, |
- scoped_refptr<LoadCallbacks> callbacks) { |
- base::FilePath path; |
- net::FileURLToFilePath(resolved_url, &path); |
- const bool kDeleteFileAfter = false; |
+// For loading services via file:// URLs. |
+class LocalLoader : public Loader { |
+ public: |
+ explicit LocalLoader(scoped_ptr<DynamicServiceRunner> runner) |
+ : Loader(runner.Pass()) { |
+ } |
- // Async for consistency with network case. |
- base::MessageLoop::current()->PostTask( |
- FROM_HERE, |
- base::Bind(&DynamicServiceLoader::RunLibrary, |
- weak_ptr_factory_.GetWeakPtr(), |
- path, |
- callbacks, |
- kDeleteFileAfter, |
- base::PathExists(path))); |
-} |
+ virtual void Start(const GURL& url, |
+ ScopedMessagePipeHandle service_handle, |
+ Context* context) OVERRIDE { |
+ base::FilePath path; |
+ net::FileURLToFilePath(url, &path); |
-void DynamicServiceLoader::LoadNetworkService( |
- const GURL& resolved_url, |
- scoped_refptr<LoadCallbacks> callbacks) { |
- if (!network_service_) { |
- context_->service_manager()->ConnectToService( |
- GURL("mojo:mojo_network_service"), |
- &network_service_); |
+ // Complete asynchronously for consistency with NetworkServiceLoader. |
+ base::MessageLoop::current()->PostTask( |
+ FROM_HERE, |
+ base::Bind(&Loader::StartService, |
+ base::Unretained(this), |
+ path, |
+ base::Passed(&service_handle), |
+ base::PathExists(path))); |
} |
- if (!url_loader_) |
- network_service_->CreateURLLoader(Get(&url_loader_)); |
+}; |
- URLRequestPtr request(URLRequest::New()); |
- request->url = String::From(resolved_url); |
- request->auto_follow_redirects = true; |
- |
- if (base::CommandLine::ForCurrentProcess()->HasSwitch( |
- switches::kDisableCache)) { |
- request->bypass_cache = true; |
+// For loading services via the network stack. |
+class NetworkLoader : public Loader { |
+ public: |
+ explicit NetworkLoader(scoped_ptr<DynamicServiceRunner> runner, |
+ NetworkService* network_service) |
+ : Loader(runner.Pass()) { |
+ network_service->CreateURLLoader(Get(&url_loader_)); |
} |
- url_loader_->Start( |
- request.Pass(), |
- base::Bind(&DynamicServiceLoader::OnLoadNetworkServiceComplete, |
- weak_ptr_factory_.GetWeakPtr(), |
- callbacks)); |
-} |
+ virtual void Start(const GURL& url, |
+ ScopedMessagePipeHandle service_handle, |
+ Context* context) OVERRIDE { |
+ service_handle_ = service_handle.Pass(); |
+ context_ = context; |
-void DynamicServiceLoader::OnLoadNetworkServiceComplete( |
- scoped_refptr<LoadCallbacks> callbacks, URLResponsePtr response) { |
- if (response->error) { |
- LOG(ERROR) << "Error (" << response->error->code << ": " |
- << response->error->description << ") while fetching " |
- << response->url; |
+ URLRequestPtr request(URLRequest::New()); |
+ request->url = String::From(url); |
+ request->auto_follow_redirects = true; |
+ |
+ if (base::CommandLine::ForCurrentProcess()->HasSwitch( |
+ switches::kDisableCache)) { |
+ request->bypass_cache = true; |
+ } |
+ |
+ url_loader_->Start(request.Pass(), |
+ base::Bind(&NetworkLoader::OnReceivedResponse, |
+ base::Unretained(this))); |
} |
- MimeTypeToURLMap::iterator iter = |
- mime_type_to_url_.find(response->mime_type); |
- if (iter != mime_type_to_url_.end()) { |
- callbacks->LoadWithContentHandler(iter->second, response.Pass()); |
- return; |
+ private: |
+ virtual ~NetworkLoader() { |
+ if (!file_.empty()) |
+ base::DeleteFile(file_, false); |
} |
- base::FilePath file; |
- base::CreateTemporaryFile(&file); |
+ void OnReceivedResponse(URLResponsePtr response) { |
+ if (response->error) { |
+ LOG(ERROR) << "Error (" << response->error->code << ": " |
+ << response->error->description << ") while fetching " |
+ << response->url; |
+ } |
- const bool kDeleteFileAfter = true; |
- common::CopyToFile(response->body.Pass(), |
- file, |
- context_->task_runners()->blocking_pool(), |
- base::Bind(&DynamicServiceLoader::RunLibrary, |
- weak_ptr_factory_.GetWeakPtr(), |
- file, |
- callbacks, |
- kDeleteFileAfter)); |
+ base::CreateTemporaryFile(&file_); |
+ common::CopyToFile(response->body.Pass(), |
+ file_, |
+ context_->task_runners()->blocking_pool(), |
+ base::Bind(&Loader::StartService, |
+ base::Unretained(this), |
+ file_, |
+ base::Passed(&service_handle_))); |
+ } |
+ |
+ Context* context_; |
+ NetworkServicePtr network_service_; |
+ URLLoaderPtr url_loader_; |
+ ScopedMessagePipeHandle service_handle_; |
+ base::FilePath file_; |
+}; |
+ |
+} // namespace |
+ |
+DynamicServiceLoader::DynamicServiceLoader( |
+ Context* context, |
+ scoped_ptr<DynamicServiceRunnerFactory> runner_factory) |
+ : context_(context), |
+ runner_factory_(runner_factory.Pass()) { |
} |
-void DynamicServiceLoader::RunLibrary(const base::FilePath& path, |
- scoped_refptr<LoadCallbacks> callbacks, |
- bool delete_file_after, |
- bool path_exists) { |
- // TODO(aa): We need to create a runner, even if we're not going to use it, |
- // because it getting destroyed is what causes shell to shut down. If we don't |
- // create this, in the case where shell never successfully creates even one |
- // app, then shell will never shut down, because no runners are ever |
- // destroyed. |
- scoped_ptr<DynamicServiceRunner> runner(runner_factory_->Create(context_)); |
- if (!path_exists) |
- return; |
+DynamicServiceLoader::~DynamicServiceLoader() { |
+} |
- ScopedMessagePipeHandle shell_handle = callbacks->RegisterApplication(); |
- if (!shell_handle.is_valid()) |
- return; |
+void DynamicServiceLoader::LoadService(ServiceManager* manager, |
+ const GURL& url, |
+ ScopedMessagePipeHandle shell_handle) { |
+ scoped_ptr<DynamicServiceRunner> runner = runner_factory_->Create(context_); |
- DynamicServiceRunner* runner_raw = runner.release(); |
- runner_raw->Start(path, |
- shell_handle.Pass(), |
- base::Bind(&RunLibraryComplete, |
- base::Unretained(runner_raw), |
- delete_file_after ? path : base::FilePath())); |
+ GURL resolved_url; |
+ if (url.SchemeIs("mojo")) { |
+ resolved_url = context_->mojo_url_resolver()->Resolve(url); |
+ } else { |
+ resolved_url = url; |
+ } |
+ |
+ Loader* loader; |
+ if (resolved_url.SchemeIsFile()) { |
+ loader = new LocalLoader(runner.Pass()); |
+ } else { |
+ if (!network_service_) { |
+ context_->service_manager()->ConnectToService( |
+ GURL("mojo:mojo_network_service"), |
+ &network_service_); |
+ } |
+ loader = new NetworkLoader(runner.Pass(), network_service_.get()); |
+ } |
+ loader->Start(resolved_url, shell_handle.Pass(), context_); |
} |
void DynamicServiceLoader::OnServiceError(ServiceManager* manager, |