Implement items() and addLink() in data source.

1 files changed, 83 insertions, 5 deletions

diff --git a/src/data.cpp b/src/data.cpp
index 78cb500..d6c4c5f 100644
--- a/src/data.cpp
+++ b/src/data.cpp
@@ -1,3 +1,4 @@
+#include <format>
 #include <memory>
 #include <string>
 #include <expected>
@@ -8,8 +9,33 @@
 
 #include "data.hpp"
 
+// id, owner_id, parent_id, name, url, desc, visibility, time.
+#define LINK_TUPLE_TYPES int64_t,int64_t,int64_t,std::string,std::string,std::string,int,int64_t
+
 namespace
 {
+using LinkTuple = std::tuple<LINK_TUPLE_TYPES>;
+LinkItem linkFromTuple(const LinkTuple& t)
+{
+    LinkItem l;
+    l.id = std::get<0>(t);
+    l.owner_id = std::get<1>(t);
+    int64_t p = std::get<2>(t);
+    if(p == 0)
+    {
+        l.parent_id = std::nullopt;
+    }
+    else
+    {
+        l.parent_id = p;
+    }
+    l.name = std::get<3>(t);
+    l.url = std::get<4>(t);
+    l.description = std::get<5>(t);
+    l.visibility = static_cast<LinkItem::Visibility>(std::get<6>(t));
+    l.time = mw::secondsToTime(std::get<7>(t));
+    return l;
+}
 
 } // namespace
 
@@ -18,7 +44,7 @@ DataSourceSQLite::fromFile(const std::string& db_file)
 {
     auto data_source = std::make_unique<DataSourceSQLite>();
     ASSIGN_OR_RETURN(data_source->db, mw::SQLite::connectFile(db_file));
-
+    DO_OR_RETURN(data_source->db->execute("PRAGMA foreign_keys = ON;"));
     // Perform schema upgrade here.
     //
     // data_source->upgradeSchema1To2();
@@ -30,11 +56,12 @@ DataSourceSQLite::fromFile(const std::string& db_file)
         "(id INTEGER PRIMARY KEY, openid_uid TEXT, name TEXT);"));
     DO_OR_RETURN(data_source->db->execute(
         "CREATE TABLE IF NOT EXISTS LinkItems "
-        "(id INTEGER PRIMARY KEY,"
-        " FOREIGN KEY(owner_id) REFERENCES Users(id) NOT NULL,"
-        " FOREIGN KEY(parent_id) REFERENCES LinkItems(id),"
+        "(id INTEGER PRIMARY KEY, owner_id INTEGER NOT NULL,"
+        " parent_id INTEGER,"
         " name TEXT NOT NULL, url TEXT, description TEXT,"
-        " visibility INTEGER NOT NULL, time INTEGER NOT NULL);"));
+        " visibility INTEGER NOT NULL, time INTEGER NOT NULL,"
+        " FOREIGN KEY(owner_id) REFERENCES Users(id),"
+        " FOREIGN KEY(parent_id) REFERENCES LinkItems(id));"));
     return data_source;
 }
 
@@ -52,3 +79,54 @@ 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)
+{
+    std::vector<LinkTuple> links;
+    if(!parent.has_value())
+    {
+        // 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;")));
+    }
+    else
+    {
+        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))));
+    }
+    std::vector<LinkItem> result;
+    for(const LinkTuple& t : links)
+    {
+        result.push_back(linkFromTuple(t));
+    }
+    return result;
+}
+
+mw::E<int64_t> DataSourceSQLite::addLink(LinkItem&& link)
+{
+    ASSIGN_OR_RETURN(mw::SQLiteStatement sql, db->statementFromStr(
+        "INSERT INTO LinkItems (id, owner_id, parent_id, name, url,"
+        " description, visibility, time) VALUES (?, ?, ?, ?, ?, ?, ?, ?)"));
+    if(link.parent_id.has_value())
+    {
+        DO_OR_RETURN(sql.bind(
+            std::nullopt, link.owner_id, *link.parent_id, link.name, link.url,
+            link.description, static_cast<int>(link.visibility),
+            mw::timeToSeconds(mw::Clock::now())));
+    }
+    else
+    {
+        DO_OR_RETURN(sql.bind(
+            std::nullopt, link.owner_id, std::nullopt, link.name, link.url,
+            link.description, static_cast<int>(link.visibility),
+            mw::timeToSeconds(mw::Clock::now())));
+    }
+
+    DO_OR_RETURN(db->execute(std::move(sql)));
+    return db->lastInsertRowID();
+}