1 files changed, 167 insertions, 15 deletions

diff --git a/src/data.cpp b/src/data.cpp
index d6c4c5f..4960afc 100644
--- a/src/data.cpp
+++ b/src/data.cpp
@@ -6,6 +6,7 @@
 #include <mw/database.hpp>
 #include <mw/error.hpp>
 #include <mw/utils.hpp>
+#include <utility>
 
 #include "data.hpp"
 
@@ -39,6 +40,18 @@ LinkItem linkFromTuple(const LinkTuple& t)
 
 } // namespace
 
+std::string LinkItem::visibilityToStr(Visibility v)
+{
+    switch(v)
+    {
+    case PUBLIC:
+        return "public";
+    case PRIVATE:
+        return "private";
+    }
+    std::unreachable();
+}
+
 mw::E<std::unique_ptr<DataSourceSQLite>>
 DataSourceSQLite::fromFile(const std::string& db_file)
 {
@@ -53,7 +66,7 @@ DataSourceSQLite::fromFile(const std::string& db_file)
     DO_OR_RETURN(data_source->setSchemaVersion(1));
     DO_OR_RETURN(data_source->db->execute(
         "CREATE TABLE IF NOT EXISTS Users "
-        "(id INTEGER PRIMARY KEY, openid_uid TEXT, name TEXT);"));
+        "(id INTEGER PRIMARY KEY, openid_uid TEXT UNIQUE, name TEXT UNIQUE);"));
     DO_OR_RETURN(data_source->db->execute(
         "CREATE TABLE IF NOT EXISTS LinkItems "
         "(id INTEGER PRIMARY KEY, owner_id INTEGER NOT NULL,"
@@ -80,25 +93,138 @@ mw::E<void> DataSourceSQLite::setSchemaVersion(int64_t v) const
     return db->execute(std::format("PRAGMA user_version = {};", v));
 }
 
-mw::E<std::vector<LinkItem>>
-DataSourceSQLite::items(std::optional<int64_t> parent)
+mw::E<std::optional<User>>
+DataSourceSQLite::userByOpenIDUID(const std::string& uid) const
+{
+    if(db == nullptr)
+    {
+        return std::unexpected(mw::runtimeError("Database is not connected."));
+    }
+    ASSIGN_OR_RETURN(mw::SQLiteStatement sql, db->statementFromStr(
+        "SELECT id, openid_uid, name FROM Users WHERE openid_uid = ?;"));
+    DO_OR_RETURN(sql.bind(uid));
+    ASSIGN_OR_RETURN(auto users, (db->eval<int64_t, std::string, std::string>(
+        std::move(sql))));
+    if(users.empty())
+    {
+        return std::nullopt;
+    }
+    if(users.size() > 1)
+    {
+        return std::unexpected(mw::runtimeError("Found duplicated users"));
+    }
+    User u;
+    u.id = std::get<0>(users[0]);
+    u.openid_uid = std::get<1>(users[0]);
+    u.name = std::get<2>(users[0]);
+    return u;
+}
+
+mw::E<int64_t> DataSourceSQLite::addUser(User&& u) const
 {
+    if(db == nullptr)
+    {
+        return std::unexpected(mw::runtimeError("Database is not connected."));
+    }
+    ASSIGN_OR_RETURN(mw::SQLiteStatement sql, db->statementFromStr(
+        "INSERT INTO Users (id, openid_uid, name) VALUES (NULL, ?, ?)"));
+    DO_OR_RETURN(sql.bind(u.openid_uid, u.name));
+    DO_OR_RETURN(db->execute(std::move(sql)));
+    return db->lastInsertRowID();
+}
+
+mw::E<std::optional<User>> DataSourceSQLite::userByID(const int64_t id) const
+{
+    if(db == nullptr)
+    {
+        return std::unexpected(mw::runtimeError("Database is not connected."));
+    }
+    ASSIGN_OR_RETURN(mw::SQLiteStatement sql, db->statementFromStr(
+        "SELECT id, openid_uid, name FROM Users WHERE id = ?;"));
+    DO_OR_RETURN(sql.bind(id));
+    ASSIGN_OR_RETURN(auto users, (db->eval<int64_t, std::string, std::string>(
+        std::move(sql))));
+    if(users.empty())
+    {
+        return std::nullopt;
+    }
+    if(users.size() > 1)
+    {
+        return std::unexpected(mw::runtimeError("Found duplicated users"));
+    }
+    User u;
+    u.id = std::get<0>(users[0]);
+    u.openid_uid = std::get<1>(users[0]);
+    u.name = std::get<2>(users[0]);
+    return u;
+}
+
+mw::E<std::optional<User>> DataSourceSQLite::userByName(const std::string& name)
+    const
+{
+    if(db == nullptr)
+    {
+        return std::unexpected(mw::runtimeError("Database is not connected."));
+    }
+    ASSIGN_OR_RETURN(mw::SQLiteStatement sql, db->statementFromStr(
+        "SELECT id, openid_uid, name FROM Users WHERE name = ?;"));
+    DO_OR_RETURN(sql.bind(name));
+    ASSIGN_OR_RETURN(auto users, (db->eval<int64_t, std::string, std::string>(
+        std::move(sql))));
+    if(users.empty())
+    {
+        return std::nullopt;
+    }
+    if(users.size() > 1)
+    {
+        return std::unexpected(mw::runtimeError("Found duplicated users"));
+    }
+    User u;
+    u.id = std::get<0>(users[0]);
+    u.openid_uid = std::get<1>(users[0]);
+    u.name = std::get<2>(users[0]);
+    return u;
+}
+
+mw::E<std::optional<LinkItem>> DataSourceSQLite::itemByID(int64_t id) const
+{
+    if(db == nullptr)
+    {
+        return std::unexpected(mw::runtimeError("Database is not connected."));
+    }
     std::vector<LinkTuple> links;
-    if(!parent.has_value())
+    // Querying root-level links.
+    ASSIGN_OR_RETURN(mw::SQLiteStatement stat, db->statementFromStr(
+        "SELECT id, owner_id, parent_id, name, url, description, visibility,"
+        " time from LinkItems WHERE id = ?;"));
+    DO_OR_RETURN(stat.bind(id));
+    ASSIGN_OR_RETURN(links, (db->eval<LINK_TUPLE_TYPES>(std::move(stat))));
+    if(links.empty())
     {
-        // Querying root-level links.
-        ASSIGN_OR_RETURN(links, (db->eval<LINK_TUPLE_TYPES>(
-            "SELECT id, owner_id, parent_id, name, url, description, visibility,"
-            " time from LinkItems WHERE parent_id IS NULL;")));
+        return std::nullopt;
     }
-    else
+    if(links.size() > 1)
+    {
+        return std::unexpected(mw::runtimeError("Duplicate item ID"));
+    }
+
+    return linkFromTuple(links[0]);
+}
+
+mw::E<std::vector<LinkItem>> DataSourceSQLite::itemsByParent(int64_t parent)
+    const
+{
+    if(db == nullptr)
     {
-        ASSIGN_OR_RETURN(mw::SQLiteStatement stat, db->statementFromStr(
-            "SELECT id, owner_id, parent_id, name, url, description, visibility,"
-            " time from LinkItems WHERE parent_id = ?;"));
-        DO_OR_RETURN(stat.bind<int64_t>(*parent));
-        ASSIGN_OR_RETURN(links, (db->eval<LINK_TUPLE_TYPES>(std::move(stat))));
+        return std::unexpected(mw::runtimeError("Database is not connected."));
     }
+    std::vector<LinkTuple> links;
+    // Querying root-level links.
+    ASSIGN_OR_RETURN(mw::SQLiteStatement stat, db->statementFromStr(
+        "SELECT id, owner_id, parent_id, name, url, description, visibility,"
+        " time from LinkItems WHERE parent_id = ?;"));
+    DO_OR_RETURN(stat.bind(parent));
+    ASSIGN_OR_RETURN(links, (db->eval<LINK_TUPLE_TYPES>(std::move(stat))));
     std::vector<LinkItem> result;
     for(const LinkTuple& t : links)
     {
@@ -107,8 +233,34 @@ DataSourceSQLite::items(std::optional<int64_t> parent)
     return result;
 }
 
-mw::E<int64_t> DataSourceSQLite::addLink(LinkItem&& link)
+mw::E<std::vector<LinkItem>>
+DataSourceSQLite::itemsTopLevelByUser(int64_t user_id) const
 {
+    if(db == nullptr)
+    {
+        return std::unexpected(mw::runtimeError("Database is not connected."));
+    }
+    std::vector<LinkTuple> links;
+    // Querying root-level links.
+    ASSIGN_OR_RETURN(mw::SQLiteStatement stat, db->statementFromStr(
+        "SELECT id, owner_id, parent_id, name, url, description, visibility,"
+        " time from LinkItems WHERE parent_id IS NULL AND owner_id = ?;"));
+    DO_OR_RETURN(stat.bind(user_id));
+    ASSIGN_OR_RETURN(links, (db->eval<LINK_TUPLE_TYPES>(std::move(stat))));
+    std::vector<LinkItem> result;
+    for(const LinkTuple& t : links)
+    {
+        result.push_back(linkFromTuple(t));
+    }
+    return result;
+}
+
+mw::E<int64_t> DataSourceSQLite::addLink(LinkItem&& link) const
+{
+    if(db == nullptr)
+    {
+        return std::unexpected(mw::runtimeError("Database is not connected."));
+    }
     ASSIGN_OR_RETURN(mw::SQLiteStatement sql, db->statementFromStr(
         "INSERT INTO LinkItems (id, owner_id, parent_id, name, url,"
         " description, visibility, time) VALUES (?, ?, ?, ?, ?, ?, ?, ?)"));