Initial commit, mostly just copied from shrt.

1 files changed, 73 insertions, 0 deletions

diff --git a/src/data.hpp b/src/data.hpp
new file mode 100644
index 0000000..dc64e6f
--- /dev/null
+++ b/src/data.hpp
@@ -0,0 +1,73 @@
+#pragma once
+
+#include <memory>
+#include <string>
+#include <optional>
+#include <vector>
+
+#include <mw/database.hpp>
+#include <mw/error.hpp>
+#include <mw/utils.hpp>
+
+struct LinkItem
+{
+    enum Visibility {PUBLIC, PRIVATE};
+
+    int64_t id;
+    int64_t owner_id;
+    // Top-level items don’t have parents.
+    std::optional<int64_t> parent_id;
+    std::string name;
+    // If this is empty, it’s a parent.
+    std::string url;
+    std::string description;
+    Visibility visibility;
+    mw::Time time;
+};
+
+struct User
+{
+    int64_t id;
+    std::string openid_uid;
+    std::string name;
+};
+
+class DataSourceInterface
+{
+public:
+    virtual ~DataSourceInterface() = default;
+
+    // The schema version is the version of the database. It starts
+    // from 1. Every time the schema change, someone should increase
+    // this number by 1, manually, by hand. The intended use is to
+    // help with database migration.
+    virtual mw::E<int64_t> getSchemaVersion() const = 0;
+
+    virtual std::vector<LinkItem> items(std::optional<int64_t> parent) = 0;
+
+protected:
+    virtual mw::E<void> setSchemaVersion(int64_t v) const = 0;
+};
+
+class DataSourceSQLite : public DataSourceInterface
+{
+public:
+    explicit DataSourceSQLite(std::unique_ptr<mw::SQLite> conn)
+            : db(std::move(conn)) {}
+    ~DataSourceSQLite() override = default;
+
+    static mw::E<std::unique_ptr<DataSourceSQLite>>
+    fromFile(const std::string& db_file);
+    static mw::E<std::unique_ptr<DataSourceSQLite>> newFromMemory();
+
+    mw::E<int64_t> getSchemaVersion() const override;
+
+    // Do not use.
+    DataSourceSQLite() = default;
+
+protected:
+    mw::E<void> setSchemaVersion(int64_t v) const override;
+
+private:
+    std::unique_ptr<mw::SQLite> db;
+};