/*
 * Copyright (C) 2011 Google Inc. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *
 *     * Redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above
 * copyright notice, this list of conditions and the following disclaimer
 * in the documentation and/or other materials provided with the
 * distribution.
 *     * Neither the name of Google Inc. nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "modules/webdatabase/DatabaseTracker.h"

#include "core/dom/Document.h"
#include "core/dom/ExecutionContext.h"
#include "core/dom/ExecutionContextTask.h"
#include "core/dom/TaskRunnerHelper.h"
#include "modules/webdatabase/Database.h"
#include "modules/webdatabase/DatabaseClient.h"
#include "modules/webdatabase/DatabaseContext.h"
#include "modules/webdatabase/QuotaTracker.h"
#include "modules/webdatabase/sqlite/SQLiteFileSystem.h"
#include "platform/weborigin/SecurityOrigin.h"
#include "platform/weborigin/SecurityOriginHash.h"
#include "public/platform/Platform.h"
#include "public/platform/WebDatabaseObserver.h"
#include "public/platform/WebSecurityOrigin.h"
#include "public/platform/WebTraceLocation.h"
#include "wtf/Assertions.h"
#include "wtf/Functional.h"
#include "wtf/PtrUtil.h"
#include "wtf/StdLibExtras.h"

namespace blink {

static void databaseClosed(Database* database) {
  if (Platform::current()->databaseObserver()) {
    Platform::current()->databaseObserver()->databaseClosed(
        WebSecurityOrigin(database->getSecurityOrigin()),
        database->stringIdentifier());
  }
}

DatabaseTracker& DatabaseTracker::tracker() {
  DEFINE_THREAD_SAFE_STATIC_LOCAL(DatabaseTracker, tracker,
                                  new DatabaseTracker);
  return tracker;
}

DatabaseTracker::DatabaseTracker() {
  SQLiteFileSystem::registerSQLiteVFS();
}

bool DatabaseTracker::canEstablishDatabase(DatabaseContext* databaseContext,
                                           const String& name,
                                           const String& displayName,
                                           unsigned estimatedSize,
                                           DatabaseError& error) {
  ExecutionContext* executionContext = databaseContext->getExecutionContext();
  bool success =
      DatabaseClient::from(executionContext)
          ->allowDatabase(executionContext, name, displayName, estimatedSize);
  if (!success)
    error = DatabaseError::GenericSecurityError;
  return success;
}

String DatabaseTracker::fullPathForDatabase(SecurityOrigin* origin,
                                            const String& name,
                                            bool) {
  return String(Platform::current()->databaseCreateOriginIdentifier(
             WebSecurityOrigin(origin))) +
         "/" + name + "#";
}

void DatabaseTracker::addOpenDatabase(Database* database) {
  MutexLocker openDatabaseMapLock(m_openDatabaseMapGuard);
  if (!m_openDatabaseMap)
    m_openDatabaseMap = WTF::wrapUnique(new DatabaseOriginMap);

  String originString = database->getSecurityOrigin()->toRawString();
  DatabaseNameMap* nameMap = m_openDatabaseMap->get(originString);
  if (!nameMap) {
    nameMap = new DatabaseNameMap();
    m_openDatabaseMap->set(originString, nameMap);
  }

  String name(database->stringIdentifier());
  DatabaseSet* databaseSet = nameMap->get(name);
  if (!databaseSet) {
    databaseSet = new DatabaseSet();
    nameMap->set(name, databaseSet);
  }

  databaseSet->add(database);
}

void DatabaseTracker::removeOpenDatabase(Database* database) {
  {
    MutexLocker openDatabaseMapLock(m_openDatabaseMapGuard);
    String originString = database->getSecurityOrigin()->toRawString();
    ASSERT(m_openDatabaseMap);
    DatabaseNameMap* nameMap = m_openDatabaseMap->get(originString);
    if (!nameMap)
      return;

    String name(database->stringIdentifier());
    DatabaseSet* databaseSet = nameMap->get(name);
    if (!databaseSet)
      return;

    DatabaseSet::iterator found = databaseSet->find(database);
    if (found == databaseSet->end())
      return;

    databaseSet->remove(found);
    if (databaseSet->isEmpty()) {
      nameMap->remove(name);
      delete databaseSet;
      if (nameMap->isEmpty()) {
        m_openDatabaseMap->remove(originString);
        delete nameMap;
      }
    }
  }
  databaseClosed(database);
}

void DatabaseTracker::prepareToOpenDatabase(Database* database) {
  ASSERT(
      database->getDatabaseContext()->getExecutionContext()->isContextThread());
  if (Platform::current()->databaseObserver()) {
    Platform::current()->databaseObserver()->databaseOpened(
        WebSecurityOrigin(database->getSecurityOrigin()),
        database->stringIdentifier(), database->displayName(),
        database->estimatedSize());
  }
}

void DatabaseTracker::failedToOpenDatabase(Database* database) {
  databaseClosed(database);
}

unsigned long long DatabaseTracker::getMaxSizeForDatabase(
    const Database* database) {
  unsigned long long spaceAvailable = 0;
  unsigned long long databaseSize = 0;
  QuotaTracker::instance().getDatabaseSizeAndSpaceAvailableToOrigin(
      database->getSecurityOrigin(), database->stringIdentifier(),
      &databaseSize, &spaceAvailable);
  return databaseSize + spaceAvailable;
}

void DatabaseTracker::closeDatabasesImmediately(SecurityOrigin* origin,
                                                const String& name) {
  String originString = origin->toRawString();
  MutexLocker openDatabaseMapLock(m_openDatabaseMapGuard);
  if (!m_openDatabaseMap)
    return;

  DatabaseNameMap* nameMap = m_openDatabaseMap->get(originString);
  if (!nameMap)
    return;

  DatabaseSet* databaseSet = nameMap->get(name);
  if (!databaseSet)
    return;

  // We have to call closeImmediately() on the context thread.
  for (DatabaseSet::iterator it = databaseSet->begin();
       it != databaseSet->end(); ++it)
    (*it)->getDatabaseContext()->getExecutionContext()->postTask(
        TaskType::DatabaseAccess, BLINK_FROM_HERE,
        createCrossThreadTask(&DatabaseTracker::closeOneDatabaseImmediately,
                              crossThreadUnretained(this), originString, name,
                              *it));
}

void DatabaseTracker::forEachOpenDatabaseInPage(
    Page* page,
    std::unique_ptr<DatabaseCallback> callback) {
  MutexLocker openDatabaseMapLock(m_openDatabaseMapGuard);
  if (!m_openDatabaseMap)
    return;
  for (auto& originMap : *m_openDatabaseMap) {
    for (auto& nameDatabaseSet : *originMap.value) {
      for (Database* database : *nameDatabaseSet.value) {
        ExecutionContext* context = database->getExecutionContext();
        ASSERT(context->isDocument());
        if (toDocument(context)->frame()->page() == page)
          (*callback)(database);
      }
    }
  }
}

void DatabaseTracker::closeOneDatabaseImmediately(const String& originString,
                                                  const String& name,
                                                  Database* database) {
  // First we have to confirm the 'database' is still in our collection.
  {
    MutexLocker openDatabaseMapLock(m_openDatabaseMapGuard);
    if (!m_openDatabaseMap)
      return;

    DatabaseNameMap* nameMap = m_openDatabaseMap->get(originString);
    if (!nameMap)
      return;

    DatabaseSet* databaseSet = nameMap->get(name);
    if (!databaseSet)
      return;

    DatabaseSet::iterator found = databaseSet->find(database);
    if (found == databaseSet->end())
      return;
  }

  // And we have to call closeImmediately() without our collection lock being
  // held.
  database->closeImmediately();
}

}  // namespace blink