// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chromeos/ash/components/file_manager/indexing/url_table.h"
namespace ash::file_manager {
namespace {
#define URL_TABLE "url_table"
#define URL_ID "url_id"
#define URL_SPEC "url_spec"
#define URL_INDEX "url_index"
// The statement used to create the URL table.
static constexpr char kCreateUrlTableQuery[] =
// clang-format off
"CREATE TABLE IF NOT EXISTS " URL_TABLE "("
URL_ID " INTEGER PRIMARY KEY AUTOINCREMENT,"
URL_SPEC " TEXT NOT NULL)";
// clang-format on
// The statement used to delete a URL from the database by URL ID.
static constexpr char kDeleteUrlQuery[] =
// clang-format off
"DELETE FROM " URL_TABLE " WHERE " URL_ID "=?";
// clang-format on
// The statement used fetch the ID of the URL.
static constexpr char kGetUrlIdQuery[] =
// clang-format off
"SELECT " URL_ID " FROM " URL_TABLE " WHERE " URL_SPEC "=?";
// clang-format on
// The statement used fetch the ID of the URL.
static constexpr char kGetUrlValueQuery[] =
// clang-format off
"SELECT " URL_SPEC " FROM " URL_TABLE " WHERE " URL_ID "=?";
// clang-format on
// The statement used to insert a new URL into the table.
static constexpr char kInsertUrlQuery[] =
// clang-format off
"INSERT INTO " URL_TABLE "(" URL_SPEC ") VALUES (?) RETURNING " URL_ID;
// clang-format on
// The statement that creates an index on url_spec column.
static constexpr char kCreateUrlIndexQuery[] =
// clang-format off
"CREATE UNIQUE INDEX IF NOT EXISTS " URL_INDEX " ON " URL_TABLE
"(" URL_SPEC ")";
// clang-format on
// The statement that changes URL from one string to another
static constexpr char kChangeUrlSpecQuery[] =
// clang-format off
"UPDATE OR ROLLBACK " URL_TABLE " SET " URL_SPEC "=? WHERE "
URL_SPEC "=? RETURNING " URL_ID;
// clang-format on
} // namespace
UrlTable::UrlTable(sql::Database* db) : TextTable(db, "" URL_TABLE "") {}
UrlTable::~UrlTable() = default;
int64_t UrlTable::DeleteUrl(const GURL& url) {
DCHECK(url.is_valid());
return DeleteValue(url.spec());
}
int64_t UrlTable::GetUrlId(const GURL& url) const {
DCHECK(url.is_valid());
return GetValueId(url.spec());
}
std::optional<std::string> UrlTable::GetUrlSpec(int64_t url_id) const {
return GetValue(url_id);
}
int64_t UrlTable::GetOrCreateUrlId(const GURL& url) {
DCHECK(url.is_valid());
return GetOrCreateValueId(url.spec());
}
int64_t UrlTable::ChangeUrl(const GURL& from, const GURL& to) {
DCHECK(from.is_valid());
DCHECK(to.is_valid());
return ChangeValue(from.spec(), to.spec());
}
std::unique_ptr<sql::Statement> UrlTable::MakeGetValueIdStatement() const {
return std::make_unique<sql::Statement>(
db_->GetCachedStatement(SQL_FROM_HERE, kGetUrlIdQuery));
}
std::unique_ptr<sql::Statement> UrlTable::MakeGetValueStatement() const {
return std::make_unique<sql::Statement>(
db_->GetCachedStatement(SQL_FROM_HERE, kGetUrlValueQuery));
}
std::unique_ptr<sql::Statement> UrlTable::MakeInsertStatement() const {
return std::make_unique<sql::Statement>(
db_->GetCachedStatement(SQL_FROM_HERE, kInsertUrlQuery));
}
std::unique_ptr<sql::Statement> UrlTable::MakeDeleteStatement() const {
return std::make_unique<sql::Statement>(
db_->GetCachedStatement(SQL_FROM_HERE, kDeleteUrlQuery));
}
std::unique_ptr<sql::Statement> UrlTable::MakeCreateTableStatement() const {
return std::make_unique<sql::Statement>(
db_->GetCachedStatement(SQL_FROM_HERE, kCreateUrlTableQuery));
}
std::unique_ptr<sql::Statement> UrlTable::MakeCreateIndexStatement() const {
return std::make_unique<sql::Statement>(
db_->GetCachedStatement(SQL_FROM_HERE, kCreateUrlIndexQuery));
}
std::unique_ptr<sql::Statement> UrlTable::MakeChangeValueStatement() const {
return std::make_unique<sql::Statement>(
db_->GetCachedStatement(SQL_FROM_HERE, kChangeUrlSpecQuery));
}
} // namespace ash::file_manager