profanity/src/database.c

/*
 * database.c
 * vim: expandtab:ts=4:sts=4:sw=4
 *
 * Copyright (C) 2020 - 2025 Michael Vetter <jubalh@iodoru.org>
 *
 * This file is part of Profanity.
 *
 * Profanity 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, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Profanity 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 Profanity.  If not, see <https://www.gnu.org/licenses/>.
 *
 * In addition, as a special exception, the copyright holders give permission to
 * link the code of portions of this program with the OpenSSL library under
 * certain conditions as described in each individual source file, and
 * distribute linked combinations including the two.
 *
 * You must obey the GNU General Public License in all respects for all of the
 * code used other than OpenSSL. If you modify file(s) with this exception, you
 * may extend this exception to your version of the file(s), but you are not
 * obligated to do so. If you do not wish to do so, delete this exception
 * statement from your version. If you delete this exception statement from all
 * source files in the program, then also delete it here.
 *
 */

#include "config.h"

#include <glib.h>
#include <stdlib.h>
#include <string.h>

#include "log.h"
#include "common.h"
#include "config.h"
#include "database.h"
#include "config/preferences.h"
#include "config/accounts.h"
#include "ui/ui.h"
#include "xmpp/xmpp.h"
#include "xmpp/message.h"

db_backend_t* active_db_backend = NULL;

void
integrity_issue_free(integrity_issue_t* issue)
{
    if (issue) {
        g_free(issue->file);
        g_free(issue->message);
        g_free(issue);
    }
}

gboolean
log_database_init(ProfAccount* account)
{
    auto_gchar gchar* pref_dblog = prefs_get_string(PREF_DBLOG);

    // Select backend based on preference
    if (g_strcmp0(pref_dblog, "flatfile") == 0) {
        active_db_backend = db_backend_flatfile();
        log_info("Using flat-file database backend");
    } else {
#ifdef HAVE_SQLITE
        active_db_backend = db_backend_sqlite();
        log_info("Using SQLite database backend");
#else
        log_warning("SQLite not compiled in, falling back to flat-file backend");
        active_db_backend = db_backend_flatfile();
#endif
    }

    if (!active_db_backend) {
        log_error("log_database_init: no backend available");
        return FALSE;
    }

    return active_db_backend->init(account);
}

void
log_database_close(void)
{
    if (active_db_backend) {
        active_db_backend->close();
    }
    active_db_backend = NULL;
}

gboolean
log_database_switch_backend(const char* new_backend)
{
    // Close current backend
    log_database_close();

    // Update preference (user must /save to persist across restarts)
    prefs_set_string(PREF_DBLOG, new_backend);

    // Get current account to reinitialize
    const char* account_name = session_get_account_name();
    if (!account_name) {
        log_error("log_database_switch_backend: no active session");
        return FALSE;
    }

    ProfAccount* account = accounts_get_account(account_name);
    if (!account) {
        log_error("log_database_switch_backend: account '%s' not found", account_name);
        return FALSE;
    }

    gboolean result = log_database_init(account);
    account_free(account);
    return result;
}

void
log_database_add_incoming(ProfMessage* message)
{
    if (active_db_backend && active_db_backend->add_incoming) {
        active_db_backend->add_incoming(message);
    } else {
        log_warning("log_database_add_incoming: no backend or method available");
    }
}

void
log_database_add_outgoing_chat(const char* const id, const char* const barejid, const char* const message, const char* const replace_id, prof_enc_t enc)
{
    if (active_db_backend && active_db_backend->add_outgoing_chat) {
        active_db_backend->add_outgoing_chat(id, barejid, message, replace_id, enc);
    } else {
        log_warning("log_database_add_outgoing_chat: no backend or method available");
    }
}

void
log_database_add_outgoing_muc(const char* const id, const char* const barejid, const char* const message, const char* const replace_id, prof_enc_t enc)
{
    if (active_db_backend && active_db_backend->add_outgoing_muc) {
        active_db_backend->add_outgoing_muc(id, barejid, message, replace_id, enc);
    } else {
        log_warning("log_database_add_outgoing_muc: no backend or method available");
    }
}

void
log_database_add_outgoing_muc_pm(const char* const id, const char* const barejid, const char* const message, const char* const replace_id, prof_enc_t enc)
{
    if (active_db_backend && active_db_backend->add_outgoing_muc_pm) {
        active_db_backend->add_outgoing_muc_pm(id, barejid, message, replace_id, enc);
    } else {
        log_warning("log_database_add_outgoing_muc_pm: no backend or method available");
    }
}

db_history_result_t
log_database_get_previous_chat(const gchar* const contact_barejid, const gchar* start_time, const gchar* end_time, gboolean from_start, gboolean flip, GSList** result)
{
    if (active_db_backend && active_db_backend->get_previous_chat) {
        return active_db_backend->get_previous_chat(contact_barejid, start_time, end_time, from_start, flip, result);
    }
    return DB_RESPONSE_ERROR;
}

ProfMessage*
log_database_get_limits_info(const gchar* const contact_barejid, gboolean is_last)
{
    if (active_db_backend && active_db_backend->get_limits_info) {
        return active_db_backend->get_limits_info(contact_barejid, is_last);
    }
    // Fallback: return an empty message with sane defaults
    ProfMessage* msg = message_init();
    if (is_last) {
        msg->timestamp = g_date_time_new_now_utc();
    }
    return msg;
}

GSList*
log_database_verify_integrity(const gchar* const contact_barejid)
{
    if (active_db_backend && active_db_backend->verify_integrity) {
        return active_db_backend->verify_integrity(contact_barejid);
    }

    GSList* issues = NULL;
    integrity_issue_t* issue = g_malloc0(sizeof(integrity_issue_t));
    issue->level = INTEGRITY_WARNING;
    issue->file = g_strdup("N/A");
    issue->line = 0;
    issue->message = g_strdup("Active backend does not support integrity verification");
    issues = g_slist_append(issues, issue);
    return issues;
}