/*
 * Copyright 2019 Google
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "Firestore/core/src/api/query_listener_registration.h"

#include <utility>

#include "Firestore/core/src/core/event_listener.h"
#include "Firestore/core/src/core/firestore_client.h"
#include "Firestore/core/src/core/query_listener.h"
#include "Firestore/core/src/core/view_snapshot.h"
#include "absl/synchronization/mutex.h"

namespace firebase {
namespace firestore {
namespace api {

QueryListenerRegistration::QueryListenerRegistration(
    std::shared_ptr<core::FirestoreClient> client,
    std::shared_ptr<core::AsyncEventListener<core::ViewSnapshot>>
        async_listener,
    std::shared_ptr<core::QueryListener> query_listener)
    : client_(std::move(client)),
      async_listener_(std::move(async_listener)),
      query_listener_(std::move(query_listener)) {
}

void QueryListenerRegistration::Remove() {
  absl::MutexLock lock(&mutex_);

  auto async_listener = async_listener_.lock();
  if (async_listener) {
    async_listener->Mute();
    async_listener_.reset();
  }

  auto query_listener = query_listener_.lock();
  if (query_listener) {
    client_->RemoveListener(query_listener);
    query_listener_.reset();
  }

  client_.reset();
}

}  // namespace api
}  // namespace firestore
}  // namespace firebase
