// Copyright 2015 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.

#ifndef EXTENSIONS_BROWSER_API_MIME_HANDLER_PRIVATE_MIME_HANDLER_PRIVATE_H_
#define EXTENSIONS_BROWSER_API_MIME_HANDLER_PRIVATE_MIME_HANDLER_PRIVATE_H_

#include "base/memory/weak_ptr.h"
#include "extensions/common/api/mime_handler.mojom.h"

namespace extensions {
class StreamContainer;
class MimeHandlerServiceImplTest;

class MimeHandlerServiceImpl
    : public mojo::InterfaceImpl<mime_handler::MimeHandlerService> {
 public:
  static void Create(
      base::WeakPtr<StreamContainer> stream_container,
      mojo::InterfaceRequest<mime_handler::MimeHandlerService> request);

 private:
  friend class MimeHandlerServiceImplTest;

  explicit MimeHandlerServiceImpl(
      base::WeakPtr<StreamContainer> stream_container);
  ~MimeHandlerServiceImpl() override;

  // mime_handler::MimeHandlerService overrides.
  void GetStreamInfo(const mojo::Callback<void(mime_handler::StreamInfoPtr)>&
                         callback) override;
  void AbortStream(const mojo::Callback<void()>& callback) override;

  // Invoked by the callback used to abort |stream_|.
  void OnStreamClosed(const mojo::Callback<void()>& callback);

  // A handle to the stream being handled by the MimeHandlerViewGuest.
  base::WeakPtr<StreamContainer> stream_;

  base::WeakPtrFactory<MimeHandlerServiceImpl> weak_factory_;
};

}  // namespace extensions

namespace mojo {

template <>
struct TypeConverter<extensions::mime_handler::StreamInfoPtr,
                     extensions::StreamContainer> {
  static extensions::mime_handler::StreamInfoPtr Convert(
      const extensions::StreamContainer& stream);
};

}  // namespace mojo

#endif  // EXTENSIONS_BROWSER_API_MIME_HANDLER_PRIVATE_MIME_HANDLER_PRIVATE_H_