chookchat/server/src/db.h

#ifndef DB_H
#define DB_H

#include <cstdint>
#include <optional>
#include <string>
#include <sqlite3.h>
#include <vector>

#include "post.h"

void sanitize(std::string& str);

std::string generateSecureToken(size_t numBytes);

struct User {
    uint64_t id = 0;
    std::string name = "";
    std::string passwordHash = "";

    User(uint64_t id, std::string namein, const std::string& passwordHash) :
        id(id), passwordHash(passwordHash) {
            sanitize(namein);
            name = namein;
        }
};

class Database {

    sqlite3* db;

    public:
    Database() = delete;
    Database(const std::string& path);

    ~Database();

    std::optional<Post> getPost(uint64_t id);
    std::vector<Post> getUserPosts(uint64_t userId);
    std::vector<Post> getTopPosts(uint64_t amount);
    void addPost(const Post& post);
    void addLike(uint64_t postId, uint64_t userId);

    std::optional<User> getUser(uint64_t id);
    std::optional<User> getUserByName(const std::string& name);
    std::optional<User> getUserByToken(const std::string& token);
    std::optional<std::string> createNewToken(uint64_t id);

    // this will modify the user to have their user ID
    void addUser(User& user);

};

#endif