mojo: Update content handler API

examples/content_handler_demo/content_handler_demo.cc

 // Copyright 2014 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 <stdio.h>
 
+#include "mojo/public/c/system/main.h"
 #include "mojo/public/cpp/application/application_delegate.h"
 #include "mojo/public/cpp/application/application_impl.h"
+#include "mojo/public/cpp/application/application_runner.h"
 #include "mojo/public/cpp/application/interface_factory_impl.h"
 #include "mojo/services/public/interfaces/content_handler/content_handler.mojom.h"
 
 namespace mojo {
 namespace examples {
 
-class ContentHandlerApp;
+class PrintBodyApplication : public InterfaceImpl<Application> {
+ public:
+  PrintBodyApplication(ShellPtr shell, ScopedDataPipeConsumerHandle body)
+      : shell_(shell.Pass()), body_(body.Pass()) {
+    shell_.set_client(this);
+  }
 
-class ContentHandlerImpl : public InterfaceImpl<ContentHandler> {
- public:
-  explicit ContentHandlerImpl(ContentHandlerApp* content_handler_app)
-      : content_handler_app_(content_handler_app) {
+  virtual void Initialize(Array<String> args) override {}

[Aaron Boodman, 2014/10/31 08:24:01]

Nit: add a blank line after this method to match the others.

[qsr, 2014/10/31 12:10:43]

Done.

+  virtual void AcceptConnection(const String& requestor_url,
+                                ServiceProviderPtr service_provider) override {
+    printf("ContentHandler::OnConnect - requestor_url:%s - body follows\n\n",
+           requestor_url.To<std::string>().c_str());
+    PrintResponse(body_.Pass());
+    delete this;
   }
-  virtual ~ContentHandlerImpl() {}
 
  private:
-  virtual void OnConnect(
-      const mojo::String& requestor_url,
-      URLResponsePtr response,
-      InterfaceRequest<ServiceProvider> service_provider) override;
-
-  ContentHandlerApp* content_handler_app_;
-};
-
-class ContentHandlerApp : public ApplicationDelegate {
- public:
-  ContentHandlerApp() : content_handler_factory_(this) {
-  }
-
-  virtual void Initialize(ApplicationImpl* app) override {}
-
-  virtual bool ConfigureIncomingConnection(
-      ApplicationConnection* connection) override {
-    connection->AddService(&content_handler_factory_);
-    return true;
-  }
-
   void PrintResponse(ScopedDataPipeConsumerHandle body) const {
     for (;;) {
       char buf[512];
       uint32_t num_bytes = sizeof(buf);
       MojoResult result = ReadDataRaw(body.get(), buf, &num_bytes,
                                       MOJO_READ_DATA_FLAG_NONE);
       if (result == MOJO_RESULT_SHOULD_WAIT) {
         Wait(body.get(),
              MOJO_HANDLE_SIGNAL_READABLE,
              MOJO_DEADLINE_INDEFINITE);
       } else if (result == MOJO_RESULT_OK) {
         if (fwrite(buf, num_bytes, 1, stdout) != 1) {
           printf("\nUnexpected error writing to file\n");
           break;
         }
       } else {
         break;
       }
 
       printf("\n>>> EOF <<<\n");
     }
   }
 
- private:
-  InterfaceFactoryImplWithContext<ContentHandlerImpl,
-                                  ContentHandlerApp> content_handler_factory_;
+  ShellPtr shell_;
+  ScopedDataPipeConsumerHandle body_;
+
+  MOJO_DISALLOW_COPY_AND_ASSIGN(PrintBodyApplication);
 };
 
-void ContentHandlerImpl::OnConnect(
-    const mojo::String& requestor_url,
-    URLResponsePtr response,
-    InterfaceRequest<ServiceProvider> service_provider) {
-  printf("ContentHandler::OnConnect - requestor_url:%s - body follows\n\n",
-         requestor_url.To<std::string>().c_str());
-  content_handler_app_->PrintResponse(response->body.Pass());
-}
+class ContentHandlerImpl : public InterfaceImpl<ContentHandler> {
+ public:
+  explicit ContentHandlerImpl() {}
+  virtual ~ContentHandlerImpl() {}
+
+ private:
+  virtual void StartApplication(ShellPtr shell,
+                                URLResponsePtr response) override {
+    // The application will delete itself after being connected to.
+    new PrintBodyApplication(shell.Pass(), response->body.Pass());
+  }
+};
+
+class ContentHandlerApp : public ApplicationDelegate {
+ public:
+  ContentHandlerApp() : content_handler_factory_() {}
+
+  virtual void Initialize(ApplicationImpl* app) override {}
+
+  virtual bool ConfigureIncomingConnection(
+      ApplicationConnection* connection) override {
+    connection->AddService(&content_handler_factory_);
+    return true;
+  }
+
+ private:
+  InterfaceFactoryImpl<ContentHandlerImpl> content_handler_factory_;
+
+  MOJO_DISALLOW_COPY_AND_ASSIGN(ContentHandlerApp);
+};
 
 }  // namespace examples
+}  // namespace mojo
 
-// static
-ApplicationDelegate* ApplicationDelegate::Create() {
-  return new examples::ContentHandlerApp();
+MojoResult MojoMain(MojoHandle shell_handle) {
+  mojo::ApplicationRunner runner(new mojo::examples::ContentHandlerApp);
+  return runner.Run(shell_handle);
 }
-
-}  // namespace mojo