2020-03-11 09:17:11 -04:00
|
|
|
/*
|
|
|
|
* database.c
|
|
|
|
* vim: expandtab:ts=4:sts=4:sw=4
|
|
|
|
*
|
|
|
|
* Copyright (C) 2020 Michael Vetter <jubalh@idoru.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.
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
|
2020-03-19 07:10:54 -04:00
|
|
|
#define _GNU_SOURCE 1
|
|
|
|
|
2020-03-23 07:18:25 -04:00
|
|
|
#include <sys/stat.h>
|
2020-03-11 09:17:11 -04:00
|
|
|
#include <sqlite3.h>
|
2020-03-19 07:10:54 -04:00
|
|
|
#include <stdio.h>
|
2020-03-23 07:18:25 -04:00
|
|
|
#include <errno.h>
|
2020-03-11 09:17:11 -04:00
|
|
|
|
|
|
|
#include "log.h"
|
2020-03-18 13:56:22 -04:00
|
|
|
#include "config/files.h"
|
2020-03-11 09:17:11 -04:00
|
|
|
|
2020-03-18 13:56:22 -04:00
|
|
|
static sqlite3 *g_chatlog_database;
|
2020-03-11 09:17:11 -04:00
|
|
|
|
2020-03-27 19:50:32 -04:00
|
|
|
static void _add_to_db(ProfMessage *message, const char * const type, const char * const from_jid, const char * const to_jid);
|
2020-03-27 19:18:13 -04:00
|
|
|
static char* _get_db_filename(ProfAccount *account);
|
|
|
|
|
2020-03-23 07:18:25 -04:00
|
|
|
static char*
|
|
|
|
_get_db_filename(ProfAccount *account)
|
|
|
|
{
|
|
|
|
char *databasedir = files_get_data_path(DIR_DATABASE);
|
|
|
|
GString *basedir = g_string_new(databasedir);
|
|
|
|
free(databasedir);
|
|
|
|
|
|
|
|
g_string_append(basedir, "/");
|
|
|
|
|
|
|
|
gchar *account_dir = str_replace(account->jid, "@", "_at_");
|
|
|
|
g_string_append(basedir, account_dir);
|
|
|
|
free(account_dir);
|
|
|
|
|
|
|
|
int res = g_mkdir_with_parents(basedir->str, S_IRWXU);
|
|
|
|
if (res == -1) {
|
|
|
|
char *errmsg = strerror(errno);
|
|
|
|
if (errmsg) {
|
|
|
|
log_error("DATABASE: error creating directory: %s, %s", basedir->str, errmsg);
|
|
|
|
} else {
|
|
|
|
log_error("DATABASE: creating directory: %s", basedir->str);
|
|
|
|
}
|
|
|
|
g_string_free(basedir, TRUE);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
g_string_append(basedir, "/chatlog.db");
|
|
|
|
char *result = strdup(basedir->str);
|
|
|
|
g_string_free(basedir, TRUE);
|
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2020-03-23 07:33:42 -04:00
|
|
|
gboolean
|
2020-03-23 07:18:25 -04:00
|
|
|
log_database_init(ProfAccount *account)
|
2020-03-11 09:17:11 -04:00
|
|
|
{
|
|
|
|
int ret = sqlite3_initialize();
|
2020-03-27 14:09:29 -04:00
|
|
|
if (ret != SQLITE_OK) {
|
2020-03-11 09:17:11 -04:00
|
|
|
log_error("Error initializing SQLite database: %d", ret);
|
|
|
|
return FALSE;
|
2020-03-27 14:09:29 -04:00
|
|
|
}
|
2020-03-11 09:17:11 -04:00
|
|
|
|
2020-03-23 07:18:25 -04:00
|
|
|
char *filename = _get_db_filename(account);
|
|
|
|
if (!filename) {
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
2020-03-18 13:56:22 -04:00
|
|
|
ret = sqlite3_open(filename, &g_chatlog_database);
|
2020-03-11 09:17:11 -04:00
|
|
|
if (ret != SQLITE_OK) {
|
2020-03-18 13:56:22 -04:00
|
|
|
const char *err_msg = sqlite3_errmsg(g_chatlog_database);
|
2020-03-11 09:17:11 -04:00
|
|
|
log_error("Error opening SQLite database: %s", err_msg);
|
2020-03-18 13:56:22 -04:00
|
|
|
free(filename);
|
2020-03-11 09:17:11 -04:00
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
2020-03-18 13:42:16 -04:00
|
|
|
char *err_msg;
|
2020-03-27 14:09:29 -04:00
|
|
|
// id is the ID of DB the entry
|
|
|
|
// from_jid is the senders jid
|
|
|
|
// to_jid is the receivers jid
|
|
|
|
// from_resource is the senders resource
|
|
|
|
// to_jid is the receivers resource
|
|
|
|
// message is the message text
|
|
|
|
// timestamp the timestamp like "2020/03/24 11:12:14"
|
2020-03-27 14:50:24 -04:00
|
|
|
// type is there to distinguish: message (chat), MUC message (muc), muc pm (mucpm)
|
2020-03-27 14:09:29 -04:00
|
|
|
// stanza_id is the ID from XEP-0359: Unique and Stable Stanza IDs
|
|
|
|
// archive_id is the ID from XEP-0313: Message Archive Management
|
|
|
|
// replace_id is the ID from XEP-0308: Last Message Correction
|
|
|
|
// encryption is to distinguish: none, omemo, otr, pgp
|
|
|
|
// marked_read is 0/1 whether a message has been marked as read via XEP-0333: Chat Markers
|
|
|
|
char *query = "CREATE TABLE IF NOT EXISTS `ChatLogs` ( `id` INTEGER PRIMARY KEY AUTOINCREMENT, `from_jid` TEXT NOT NULL, `to_jid` TEXT NOT NULL, `from_resource` TEXT, `to_resource` TEXT, `message` TEXT, `timestamp` TEXT, `type` TEXT, `stanza_id` TEXT, `archive_id` TEXT, `replace_id` TEXT, `encryption` TEXT, `marked_read` INTEGER)";
|
2020-03-23 10:18:55 -04:00
|
|
|
if( SQLITE_OK != sqlite3_exec(g_chatlog_database, query, NULL, 0, &err_msg)) {
|
2020-03-23 10:21:31 -04:00
|
|
|
goto out;
|
2020-03-23 10:18:55 -04:00
|
|
|
}
|
|
|
|
|
2020-03-27 14:09:29 -04:00
|
|
|
query = "CREATE TABLE IF NOT EXISTS `DbVersion` ( `dv_id` INTEGER PRIMARY KEY, `version` INTEGER UNIQUE)";
|
2020-03-23 10:18:55 -04:00
|
|
|
if( SQLITE_OK != sqlite3_exec(g_chatlog_database, query, NULL, 0, &err_msg)) {
|
2020-03-23 10:21:31 -04:00
|
|
|
goto out;
|
2020-03-23 10:18:55 -04:00
|
|
|
}
|
|
|
|
|
2020-03-27 14:09:29 -04:00
|
|
|
query = "INSERT OR IGNORE INTO `DbVersion` (`version`) VALUES('1')";
|
2020-03-18 13:56:22 -04:00
|
|
|
if( SQLITE_OK != sqlite3_exec(g_chatlog_database, query, NULL, 0, &err_msg)) {
|
2020-03-23 10:21:31 -04:00
|
|
|
goto out;
|
2020-03-18 13:42:16 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
log_debug("Initialized SQLite database: %s", filename);
|
2020-03-18 13:56:22 -04:00
|
|
|
free(filename);
|
2020-03-11 09:17:11 -04:00
|
|
|
return TRUE;
|
2020-03-23 10:21:31 -04:00
|
|
|
|
|
|
|
out:
|
|
|
|
if (err_msg) {
|
|
|
|
log_error("SQLite error: %s", err_msg);
|
|
|
|
sqlite3_free(err_msg);
|
|
|
|
} else {
|
|
|
|
log_error("Unknown SQLite error");
|
|
|
|
}
|
|
|
|
free(filename);
|
|
|
|
return FALSE;
|
2020-03-11 09:17:11 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
log_database_close(void)
|
|
|
|
{
|
2020-03-23 07:23:21 -04:00
|
|
|
if (g_chatlog_database) {
|
|
|
|
sqlite3_close(g_chatlog_database);
|
|
|
|
sqlite3_shutdown();
|
|
|
|
g_chatlog_database = NULL;
|
|
|
|
}
|
2020-03-11 09:17:11 -04:00
|
|
|
}
|
2020-03-19 07:10:54 -04:00
|
|
|
|
2020-03-27 20:16:31 -04:00
|
|
|
static void
|
|
|
|
_log_database_add_incoming(ProfMessage *message, const char * const type)
|
|
|
|
{
|
2020-03-27 19:50:32 -04:00
|
|
|
const char *jid = connection_get_fulljid();
|
|
|
|
Jid *myjid = jid_create(jid);
|
|
|
|
|
2020-03-27 20:16:31 -04:00
|
|
|
_add_to_db(message, type, message->jid->barejid, myjid->barejid);
|
2020-03-27 19:50:32 -04:00
|
|
|
|
|
|
|
jid_destroy(myjid);
|
2020-03-27 19:18:13 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2020-03-27 20:16:31 -04:00
|
|
|
log_database_add_incoming_chat(ProfMessage *message) {
|
|
|
|
_log_database_add_incoming(message, "chat");
|
|
|
|
}
|
2020-03-27 19:50:32 -04:00
|
|
|
|
2020-03-27 20:16:31 -04:00
|
|
|
void
|
|
|
|
log_database_add_incoming_muc(ProfMessage *message) {
|
|
|
|
_log_database_add_incoming(message, "muc");
|
2020-03-27 19:18:13 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
log_database_add_incoming_muc_pm(ProfMessage *message) {
|
2020-03-27 20:16:31 -04:00
|
|
|
_log_database_add_incoming(message, "mucpm");
|
2020-03-27 19:50:32 -04:00
|
|
|
}
|
|
|
|
|
2020-03-27 20:16:31 -04:00
|
|
|
static void
|
|
|
|
_log_database_add_outgoing(const char * const type, const char * const id, const char * const barejid, const char * const message, const char *const replace_id, prof_enc_t enc)
|
2020-03-27 19:50:32 -04:00
|
|
|
{
|
|
|
|
ProfMessage *msg = message_init();
|
|
|
|
|
|
|
|
msg->id = id ? strdup(id) : NULL;
|
|
|
|
msg->jid = jid_create(barejid);
|
|
|
|
msg->plain = message ? strdup(message) : NULL;
|
|
|
|
msg->replace_id = replace_id ? strdup(replace_id) : NULL;
|
|
|
|
msg->timestamp = g_date_time_new_now_local(); //TODO: get from outside. best to have whole ProfMessage from outside
|
2020-03-27 20:16:31 -04:00
|
|
|
msg->enc = enc;
|
2020-03-27 19:50:32 -04:00
|
|
|
|
|
|
|
const char *jid = connection_get_fulljid();
|
|
|
|
Jid *myjid = jid_create(jid);
|
|
|
|
|
2020-03-27 20:16:31 -04:00
|
|
|
_add_to_db(msg, type, myjid->barejid, msg->jid->barejid);
|
2020-03-27 19:50:32 -04:00
|
|
|
|
|
|
|
jid_destroy(myjid);
|
|
|
|
message_free(msg);
|
2020-03-27 19:18:13 -04:00
|
|
|
}
|
|
|
|
|
2020-03-27 20:16:31 -04:00
|
|
|
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)
|
|
|
|
{
|
|
|
|
_log_database_add_outgoing("chat", id, barejid, message, replace_id, enc);
|
|
|
|
}
|
|
|
|
|
|
|
|
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)
|
|
|
|
{
|
|
|
|
_log_database_add_outgoing("muc", id, barejid, message, replace_id, enc);
|
|
|
|
}
|
|
|
|
|
|
|
|
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)
|
|
|
|
{
|
|
|
|
_log_database_add_outgoing("mucpm", id, barejid, message, replace_id, enc);
|
|
|
|
}
|
|
|
|
|
2020-03-27 19:18:13 -04:00
|
|
|
static void
|
2020-03-27 19:50:32 -04:00
|
|
|
_add_to_db(ProfMessage *message, const char * const type, const char * const from_jid, const char * const to_jid)
|
2020-03-27 19:18:13 -04:00
|
|
|
{
|
2020-03-23 07:23:21 -04:00
|
|
|
if (!g_chatlog_database) {
|
|
|
|
log_debug("log_database_add() called but db is not initialized");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2020-03-19 07:10:54 -04:00
|
|
|
char *err_msg;
|
2020-03-27 14:09:29 -04:00
|
|
|
char *query;
|
2020-03-19 07:10:54 -04:00
|
|
|
|
|
|
|
//gchar *date_fmt = g_date_time_format_iso8601(message->timestamp);
|
|
|
|
gchar *date_fmt = g_date_time_format(message->timestamp, "%Y/%m/%d %H:%M:%S");
|
2020-03-27 14:50:24 -04:00
|
|
|
if (asprintf(&query, "INSERT INTO `ChatLogs` (`from_jid`, `from_resource`, `to_jid`, `message`, `timestamp`, `stanza_id`, `replace_id`, `type`) VALUES ('%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s')",
|
2020-03-27 19:50:32 -04:00
|
|
|
from_jid,
|
|
|
|
"test" /*message->jid->resourcepart*/,
|
|
|
|
to_jid,
|
2020-03-27 14:50:24 -04:00
|
|
|
message->plain,
|
|
|
|
date_fmt,
|
|
|
|
message->id ? message->id : "",
|
|
|
|
message->replace_id ? message->replace_id : "",
|
|
|
|
type) == -1) {
|
2020-03-19 07:10:54 -04:00
|
|
|
log_error("log_database_add(): could not allocate memory");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
g_free(date_fmt);
|
|
|
|
|
|
|
|
if( SQLITE_OK != sqlite3_exec(g_chatlog_database, query, NULL, 0, &err_msg)) {
|
|
|
|
if (err_msg) {
|
|
|
|
log_error("SQLite error: %s", err_msg);
|
|
|
|
sqlite3_free(err_msg);
|
|
|
|
} else {
|
|
|
|
log_error("Unknown SQLite error");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
free(query);
|
|
|
|
}
|