src/mojo/examples/content_handler_demo/content_handler_demo.cc

   1 // Copyright 2014 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 <stdio.h>
   6
   7 #include "mojo/public/cpp/application/application_delegate.h"
   8 #include "mojo/public/cpp/application/application_impl.h"
   9 #include "mojo/public/cpp/application/interface_factory_impl.h"
  10 #include "mojo/services/public/interfaces/content_handler/content_handler.mojom.h"
  11
  12 namespace mojo {
  13 namespace examples {
  14
  15 class ContentHandlerApp;
  16
  17 class ContentHandlerImpl : public InterfaceImpl<ContentHandler> {
  18  public:
  19   explicit ContentHandlerImpl(ContentHandlerApp* content_handler_app)
  20       : content_handler_app_(content_handler_app) {
  21   }
  22   virtual ~ContentHandlerImpl() {}
  23
  24  private:
  25   virtual void OnConnect(const mojo::String& url,
  26                          URLResponsePtr response,
  27                          InterfaceRequest<ServiceProvider> service_provider)
  28       MOJO_OVERRIDE;
  29
  30   ContentHandlerApp* content_handler_app_;
  31 };
  32
  33 class ContentHandlerApp : public ApplicationDelegate {
  34  public:
  35   ContentHandlerApp() : content_handler_factory_(this) {
  36   }
  37
  38   virtual void Initialize(ApplicationImpl* app) MOJO_OVERRIDE {
  39   }
  40
  41   virtual bool ConfigureIncomingConnection(ApplicationConnection* connection)
  42       MOJO_OVERRIDE {
  43     connection->AddService(&content_handler_factory_);
  44     return true;
  45   }
  46
  47   void PrintResponse(ScopedDataPipeConsumerHandle body) const {
  48     for (;;) {
  49       char buf[512];
  50       uint32_t num_bytes = sizeof(buf);
  51       MojoResult result = ReadDataRaw(body.get(), buf, &num_bytes,
  52                                       MOJO_READ_DATA_FLAG_NONE);
  53       if (result == MOJO_RESULT_SHOULD_WAIT) {
  54         Wait(body.get(),
  55              MOJO_HANDLE_SIGNAL_READABLE,
  56              MOJO_DEADLINE_INDEFINITE);
  57       } else if (result == MOJO_RESULT_OK) {
  58         if (fwrite(buf, num_bytes, 1, stdout) != 1) {
  59           printf("\nUnexpected error writing to file\n");
  60           break;
  61         }
  62       } else {
  63         break;
  64       }
  65
  66       printf("\n>>> EOF <<<\n");
  67     }
  68   }
  69
  70  private:
  71   InterfaceFactoryImplWithContext<ContentHandlerImpl,
  72                                   ContentHandlerApp> content_handler_factory_;
  73 };
  74
  75 void ContentHandlerImpl::OnConnect(
  76     const mojo::String& url,
  77     URLResponsePtr response,
  78     InterfaceRequest<ServiceProvider> service_provider) {
  79   printf("ContentHandler::OnConnect - url:%s - body follows\n\n",
  80          url.To<std::string>().c_str());
  81   content_handler_app_->PrintResponse(response->body.Pass());
  82 }
  83
  84 }  // namespace examples
  85
  86 // static
  87 ApplicationDelegate* ApplicationDelegate::Create() {
  88   return new examples::ContentHandlerApp();
  89 }
  90
  91 }  // namespace mojo