remwharead/src/lib/sqlite.cpp

/*  This file is part of remwharead.
 *  Copyright © 2019 tastytea <tastytea@tastytea.de>
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, version 3.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <exception>
#include <iostream>
#include <algorithm>
#include <basedir.h>
#include <sqlite/execute.hpp>
#include <sqlite/query.hpp>
#include "time.hpp"
#include "sqlite.hpp"

namespace remwharead
{
    using std::cerr;
    using std::endl;

    Database::Database()
        : _connected(false)
    {
        try
        {
            xdgHandle xdg;
            xdgInitHandle(&xdg);
            _dbpath = xdgDataHome(&xdg) / fs::path("remwharead");
            xdgWipeHandle(&xdg);

            if (!fs::exists(_dbpath))
            {
                fs::create_directories(_dbpath);
            }
            _dbpath /= "database.sqlite";

            _con = std::make_unique<sqlite::connection>(_dbpath);
            sqlite::execute(*_con, "CREATE TABLE IF NOT EXISTS remwharead("
                            "uri TEXT, archive_uri TEXT, datetime TEXT, "
                            "tags TEXT, title TEXT, description TEXT, "
                            "fulltext TEXT);", true);

            _connected = true;
        }
        catch (std::exception &e)
        {
            cerr << "Error in " << __func__ << ": " << e.what() << endl;
        }
    }

    Database::operator bool() const
    {
        return _connected;
    }

    bool operator ==(const Database::entry &a, const Database::entry &b)
    {
        return (a.datetime == b.datetime);
    }

    const string Database::entry::fulltext_oneline() const
    {
        string oneline = fulltext;
        size_t pos = 0;
        while ((pos = oneline.find('\n', pos)) != std::string::npos)
        {
            oneline.replace(pos, 1, "\\n");
        }
        return oneline;
    }

    void Database::store(const Database::entry &data) const
    {
        try
        {
            const string strdatetime = timepoint_to_string(data.datetime, true);
            string strtags;
            for (const string &tag : data.tags)
            {
                strtags += tag;
                if (tag != *(data.tags.rbegin()))
                {
                    strtags += ",";
                }
            }

            sqlite::execute ins(*_con, "INSERT INTO remwharead "
                                "VALUES(?, ?, ?, ?, ?, ?, ?);");
            ins % data.uri % data.archive_uri % strdatetime % strtags
                % data.title % data.description % data.fulltext;
            ins();
        }
        catch (std::exception &e)
        {
            cerr << "Error in " << __func__ << ": " << e.what() << endl;
        }
    }

    const vector<Database::entry> Database::retrieve(
        const time_point &start, const time_point &end) const
    {
        try
        {
            const string query = "SELECT * FROM remwharead WHERE datetime "
                "BETWEEN '" + timepoint_to_string(start, true)
                +  "' AND '" + timepoint_to_string(end, true)
                + "' ORDER BY datetime DESC;";

            sqlite::query q(*_con, query);
            sqlite::result_type res = q.get_result();
            vector<entry> entries;

            while(res->next_row())
            {
                vector<string> tags;
                const string strtags = res->get_string(3);
                size_t pos = 0;
                while (pos != std::string::npos)
                {
                    const size_t newpos = strtags.find(',', pos);
                    tags.push_back(strtags.substr(pos, newpos - pos));
                    pos = newpos;
                    if (pos != std::string::npos)
                    {
                        ++pos;
                    }
                }
                entries.push_back
                    ({
                        res->get_string(0),
                        res->get_string(1),
                        string_to_timepoint(res->get_string(2), true),
                        tags,
                        res->get_string(4),
                        res->get_string(5),
                        res->get_string(6)
                    });
            }

            return entries;
        }
        catch (std::exception &e)
        {
            cerr << "Error in " << __func__ << ": " << e.what() << endl;
        }

        return {};
    }
}