// 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 "content/browser/push_messaging_message_filter.h"

#include <string>

#include "base/bind.h"
#include "base/strings/string_number_conversions.h"
#include "content/browser/renderer_host/render_process_host_impl.h"
#include "content/browser/service_worker/service_worker_context_wrapper.h"
#include "content/common/push_messaging_messages.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/push_messaging_service.h"

namespace content {

PushMessagingMessageFilter::PushMessagingMessageFilter(
    int render_process_id,
    ServiceWorkerContextWrapper* service_worker_context)
    : BrowserMessageFilter(PushMessagingMsgStart),
      render_process_id_(render_process_id),
      service_worker_context_(service_worker_context),
      service_(NULL),
      weak_factory_(this) {
}

PushMessagingMessageFilter::~PushMessagingMessageFilter() {}

bool PushMessagingMessageFilter::OnMessageReceived(
    const IPC::Message& message) {
  bool handled = true;
  IPC_BEGIN_MESSAGE_MAP(PushMessagingMessageFilter, message)
    IPC_MESSAGE_HANDLER(PushMessagingHostMsg_Register, OnRegister)
    IPC_MESSAGE_UNHANDLED(handled = false)
  IPC_END_MESSAGE_MAP()
  return handled;
}

void PushMessagingMessageFilter::OnRegister(int render_frame_id,
                                            int callbacks_id,
                                            const std::string& sender_id,
                                            bool user_gesture,
                                            int service_worker_provider_id) {
  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
  // TODO(mvanouwerkerk): Validate arguments?
  ServiceWorkerProviderHost* service_worker_host =
      service_worker_context_->context()->GetProviderHost(
          render_process_id_, service_worker_provider_id);
  if (!service_worker_host || !service_worker_host->active_version()) {
    Send(new PushMessagingMsg_RegisterError(
        render_frame_id,
        callbacks_id,
        PUSH_MESSAGING_STATUS_REGISTRATION_FAILED_NO_SERVICE_WORKER));
    return;
  }
  BrowserThread::PostTask(
      BrowserThread::UI,
      FROM_HERE,
      base::Bind(&PushMessagingMessageFilter::DoRegister,
                 weak_factory_.GetWeakPtr(),
                 render_frame_id,
                 callbacks_id,
                 sender_id,
                 user_gesture,
                 service_worker_host->active_version()->scope().GetOrigin(),
                 service_worker_host->active_version()->registration_id()));
}

void PushMessagingMessageFilter::DoRegister(
    int render_frame_id,
    int callbacks_id,
    const std::string& sender_id,
    bool user_gesture,
    const GURL& origin,
    int64 service_worker_registration_id) {
  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  if (!service()) {
    Send(new PushMessagingMsg_RegisterError(
        render_frame_id,
        callbacks_id,
        PUSH_MESSAGING_STATUS_REGISTRATION_FAILED_SERVICE_NOT_AVAILABLE));
    return;
  }
  service()->Register(origin,
                      service_worker_registration_id,
                      sender_id,
                      render_process_id_,
                      render_frame_id,
                      user_gesture,
                      base::Bind(&PushMessagingMessageFilter::DidRegister,
                                 weak_factory_.GetWeakPtr(),
                                 render_frame_id,
                                 callbacks_id));
}

void PushMessagingMessageFilter::DidRegister(
    int render_frame_id,
    int callbacks_id,
    const GURL& push_endpoint,
    const std::string& push_registration_id,
    PushMessagingStatus status) {
  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  if (status == PUSH_MESSAGING_STATUS_OK) {
    Send(new PushMessagingMsg_RegisterSuccess(
        render_frame_id, callbacks_id, push_endpoint, push_registration_id));
  } else {
    Send(new PushMessagingMsg_RegisterError(
        render_frame_id, callbacks_id, status));
  }
}

PushMessagingService* PushMessagingMessageFilter::service() {
  if (!service_) {
    RenderProcessHost* process_host =
        RenderProcessHost::FromID(render_process_id_);
    if (!process_host)
      return NULL;
    service_ = process_host->GetBrowserContext()->GetPushMessagingService();
  }
  return service_;
}

}  // namespace content