// 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 "chrome/browser/net/quota_policy_channel_id_store.h"

#include <list>

#include "base/basictypes.h"
#include "base/bind.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/logging.h"
#include "base/metrics/histogram.h"
#include "base/strings/string_util.h"
#include "base/threading/thread.h"
#include "base/threading/thread_restrictions.h"
#include "net/cookies/cookie_util.h"
#include "net/extras/sqlite/sqlite_channel_id_store.h"
#include "storage/browser/quota/special_storage_policy.h"
#include "url/gurl.h"

QuotaPolicyChannelIDStore::QuotaPolicyChannelIDStore(
    const base::FilePath& path,
    const scoped_refptr<base::SequencedTaskRunner>& background_task_runner,
    const scoped_refptr<storage::SpecialStoragePolicy>& special_storage_policy)
    : special_storage_policy_(special_storage_policy),
      persistent_store_(
          new net::SQLiteChannelIDStore(path, background_task_runner)) {
  DCHECK(background_task_runner.get());
}

QuotaPolicyChannelIDStore::~QuotaPolicyChannelIDStore() {
  if (!special_storage_policy_.get() ||
      !special_storage_policy_->HasSessionOnlyOrigins()) {
    return;
  }
  std::list<std::string> session_only_server_identifiers;
  for (std::set<std::string>::iterator it = server_identifiers_.begin();
       it != server_identifiers_.end();
       ++it) {
    GURL url(net::cookie_util::CookieOriginToURL(*it, true));
    if (special_storage_policy_->IsStorageSessionOnly(url))
      session_only_server_identifiers.push_back(*it);
  }
  persistent_store_->DeleteAllInList(session_only_server_identifiers);
}

void QuotaPolicyChannelIDStore::Load(const LoadedCallback& loaded_callback) {
  persistent_store_->Load(
      base::Bind(&QuotaPolicyChannelIDStore::OnLoad, this, loaded_callback));
}

void QuotaPolicyChannelIDStore::AddChannelID(
    const net::DefaultChannelIDStore::ChannelID& channel_id) {
  server_identifiers_.insert(channel_id.server_identifier());
  persistent_store_->AddChannelID(channel_id);
}

void QuotaPolicyChannelIDStore::DeleteChannelID(
    const net::DefaultChannelIDStore::ChannelID& channel_id) {
  server_identifiers_.erase(channel_id.server_identifier());
  persistent_store_->DeleteChannelID(channel_id);
}

void QuotaPolicyChannelIDStore::SetForceKeepSessionState() {
  special_storage_policy_ = NULL;
}

void QuotaPolicyChannelIDStore::OnLoad(
    const LoadedCallback& loaded_callback,
    scoped_ptr<ChannelIDVector> channel_ids) {
  for (ChannelIDVector::const_iterator channel_id = channel_ids->begin();
       channel_id != channel_ids->end();
       ++channel_id) {
    server_identifiers_.insert((*channel_id)->server_identifier());
  }
  loaded_callback.Run(channel_ids.Pass());
}