starbitdev/zblock
1
0
Fork 0
mirror of https://github.com/WCBROW01/zblock.git synced 2025-12-11 20:18:07 -05:00
Code Issues Packages Projects Releases Wiki Activity

Move insert logic from add command into feed_info.c

Browse Source
This commit is contained in:
Mina Brown
2024-11-23 14:17:15 -05:00
parent 2269ad3385
commit 388306d4a3
3 changed files with 70 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
feed_info.c
View File

@ -15,23 +15,18 @@
#include "config.h"
#include "feed_info.h"
static const char *ZBLOCK_FEED_INFO_ERRORS[] = {
"OK",
"Invalid arguments provided",
"An error was encountered with the feed database",
"The feed already exists",
"The feed does not exist"
};
static_assert(sizeof(ZBLOCK_FEED_INFO_ERRORS) / sizeof(*ZBLOCK_FEED_INFO_ERRORS) == ZBLOCK_FEED_INFO_ERRORCOUNT, "Not all feed info errors implemented");
// returns a string about the result of a feed_info function
const char *zblock_feed_info_strerror(zblock_feed_info_err error) {
static_assert(ZBLOCK_FEED_INFO_ERRORCOUNT == 3, "Not all feed info errors implemented");
switch (error) {
case ZBLOCK_FEED_INFO_OK: {
return "OK";
}
case ZBLOCK_FEED_INFO_NULL: {
return "No feed info was provided";
}
case ZBLOCK_FEED_INFO_POSTGRES: {
return "An error was encountered with the feed database";
}
default: {
return "Unspecified error";
}
}
return error < 0 || error >= ZBLOCK_FEED_INFO_ERRORCOUNT ? "Unspecified error" : ZBLOCK_FEED_INFO_ERRORS[error];
}
// format string for for the time format of pubDate
@ -45,7 +40,56 @@ time_t pubDate_to_time_t(char *s) {
return mktime(&tm);
}
// check if the feed currently exists. the result is in the exists pointer.
zblock_feed_info_err zblock_feed_info_exists(PGconn *conn, const char *url, u64snowflake channel_id, int *exists) {
if (!conn || !exists) return ZBLOCK_FEED_INFO_INVALID_ARGS;
char channel_id_str[21]; // hold a 64-bit int in decimal form
snprintf(channel_id_str, sizeof(channel_id_str), "%ld", channel_id);
const char *const params[] = {url, channel_id_str};
PGresult *res = PQexecParams(conn, "SELECT COUNT(1) FROM feeds WHERE url = $1 AND channel_id = $2", 2, NULL, params, NULL, NULL, 0);
if (PQresultStatus(res) != PGRES_TUPLES_OK) {
PQclear(res);
return ZBLOCK_FEED_INFO_DBERROR;
}
*exists = *(int *) PQgetvalue(res, 0, 0);
PQclear(res);
return ZBLOCK_FEED_INFO_OK;
}
// Insert new feed into the database
zblock_feed_info_err zblock_feed_info_insert(PGconn *conn, zblock_feed_info *feed) {
assert(0 && "not implemented yet");
if (!conn || !feed) return ZBLOCK_FEED_INFO_INVALID_ARGS;
// check if the feed already exists
{
int feed_exists = 0;
zblock_feed_info_err exists_error = zblock_feed_info_exists(conn, feed->url, feed->channel_id, &feed_exists);
if (exists_error) {
return exists_error;
} else if (feed_exists) {
return ZBLOCK_FEED_INFO_EXISTS;
}
}
// I don't want to deal with the extra fuss that is sending these in binary format
char channel_id_str[21];
snprintf(channel_id_str, sizeof(channel_id_str), "%ld", feed->channel_id);
char guild_id_str[21];
snprintf(guild_id_str, sizeof(guild_id_str), "%ld", feed->guild_id);
const char *const insert_params[] = {feed->url, feed->last_pubDate, channel_id_str, feed->title, guild_id_str};
PGresult *insert_res = PQexecParams(conn,
"INSERT INTO feeds (url, last_pubDate, channel_id, title, guild_id) VALUES ($1, $2, $3, $4, $5)",
5, NULL, insert_params, NULL, NULL, 0
);
zblock_feed_info_err result = ZBLOCK_FEED_INFO_OK;
if (PQresultStatus(insert_res) != PGRES_COMMAND_OK) {
result = ZBLOCK_FEED_INFO_DBERROR;
}
PQclear(insert_res);
return result;
}