259 lines
7.3 KiB
C
259 lines
7.3 KiB
C
#include "db.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sqlite3.h>
|
|
|
|
sqlite3 *db;
|
|
|
|
int h_db_open(const char *db_name) {
|
|
int ret;
|
|
char *err_msg;
|
|
|
|
if (db != NULL) {
|
|
fprintf(stderr, "There is already a sqlite db open\n");
|
|
return 1;
|
|
}
|
|
|
|
ret = sqlite3_open(db_name, &db);
|
|
if (ret != SQLITE_OK) {
|
|
fprintf(stderr, "%s\n", sqlite3_errmsg(db));
|
|
h_db_close();
|
|
return 1;
|
|
}
|
|
|
|
char *sql = "CREATE TABLE IF NOT EXISTS hentai (url TEXT PRIMARY KEY NOT NULL, artist_name TEXT NOT NULL, website_name TEXT NOT NULL) STRICT;";
|
|
ret = sqlite3_exec(db, sql, NULL, 0, &err_msg);
|
|
if (ret != SQLITE_OK) {
|
|
fprintf(stderr, "SQL error: %s\n", err_msg);
|
|
sqlite3_free(err_msg);
|
|
h_db_close();
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void h_db_close() {
|
|
sqlite3_close(db);
|
|
}
|
|
|
|
static int fprintf_callback(void *fptr, int argc, char **argv, char **col_name) {
|
|
if (fptr == NULL) {
|
|
fprintf(stderr, "fptr is NULL\n");
|
|
return 1;
|
|
}
|
|
|
|
size_t i;
|
|
for (i = 0; i < argc; i++) {
|
|
fprintf(fptr, "%s", argv[i] ? argv[i] : "NULL");
|
|
if (i < argc - 1) {
|
|
fprintf(fptr, " ");
|
|
}
|
|
}
|
|
fprintf(fptr, "\n");
|
|
return 0;
|
|
}
|
|
|
|
static int db_row_callback(void *db_row, int argc, char **argv, char **col_name) {
|
|
snprintf(db_row, strlen(argv[0]) * 2, "%s %s %s", argv[0], argv[1], argv[2]);
|
|
return 0;
|
|
}
|
|
|
|
char* get_website_from_url(const char *url) {
|
|
if (strstr(url, "rule34.xxx") != NULL) {
|
|
return "Rule34";
|
|
} else if (strstr(url, "iwara.tv") != NULL) {
|
|
return "Iwara";
|
|
} else {
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
char* get_artist_from_url(const char *url, const char *website_name) {
|
|
size_t offset = 0;
|
|
if (strcmp(website_name, "Rule34") == 0) {
|
|
offset = 51;
|
|
} else if (strcmp(website_name, "Gelbooru") == 0) {
|
|
offset = 53;
|
|
} else if (strcmp(website_name, "ATF") == 0) {
|
|
offset = 42;
|
|
} else {
|
|
return NULL;
|
|
}
|
|
|
|
size_t name_length = strlen(url) - offset;
|
|
size_t char_count = name_length;
|
|
char *tmp_name = malloc(name_length + 1);
|
|
memcpy(tmp_name, url + offset, name_length);
|
|
tmp_name[name_length] = '\0';
|
|
|
|
size_t i;
|
|
size_t j = 0;
|
|
for (i = 0; i < strlen(tmp_name); i++) {
|
|
if (tmp_name[i] == '%') {
|
|
char url_encoded[4];
|
|
url_encoded[0] = tmp_name[i];
|
|
url_encoded[1] = tmp_name[i + 1];
|
|
url_encoded[2] = tmp_name[i + 2];
|
|
url_encoded[3] = '\0';
|
|
if (strstr("%20%21%22%23%24%26%27%28%29%2A%2C%2E%2F%3B%3C%3E%3F%5B%5C%5D%5E%60%7B%7C%7D%7E", url_encoded) != NULL) {
|
|
tmp_name[i] = ' ';
|
|
char_count -= 1;
|
|
} else if (strcmp(url_encoded, "%25") == 0) {
|
|
tmp_name[i] = '%';
|
|
} else if (strcmp(url_encoded, "%2B") == 0) {
|
|
tmp_name[i] = '+';
|
|
} else if (strcmp(url_encoded, "%2D") == 0) {
|
|
tmp_name[i] = '-';
|
|
} else if (strcmp(url_encoded, "%2E") == 0) {
|
|
tmp_name[i] = '.';
|
|
} else if (strcmp(url_encoded, "%2F") == 0) {
|
|
tmp_name[i] = '/';
|
|
} else if (strcmp(url_encoded, "%3A") == 0) {
|
|
tmp_name[i] = ':';
|
|
} else if (strcmp(url_encoded, "%3D") == 0) {
|
|
tmp_name[i] = '=';
|
|
} else if (strcmp(url_encoded, "%40") == 0) {
|
|
tmp_name[i] = '@';
|
|
} else if (strcmp(url_encoded, "%5F") == 0) {
|
|
tmp_name[i] = '_';
|
|
} else {
|
|
return NULL;
|
|
}
|
|
tmp_name[i += 1] = ' ';
|
|
tmp_name[i += 1] = ' ';
|
|
char_count -= 2;
|
|
}
|
|
}
|
|
|
|
char *artist_name = malloc(char_count + 1);
|
|
memset(artist_name, 0, char_count + 1);
|
|
for (i = 0; i < name_length; i++) {
|
|
if (tmp_name[i] == ' ') {
|
|
continue;
|
|
}
|
|
artist_name[j] = tmp_name[i];
|
|
j++;
|
|
}
|
|
artist_name[char_count] = '\0';
|
|
|
|
j = strlen(artist_name) - strlen("artist");
|
|
if (strcmp(artist_name + j, "artist") == 0) {
|
|
for (; j < strlen(artist_name); j++) {
|
|
artist_name[j] = '\0';
|
|
}
|
|
}
|
|
|
|
j = strlen(artist_name) - 1;
|
|
while (artist_name[j] == '_') {
|
|
artist_name[j] = '\0';
|
|
j--;
|
|
}
|
|
|
|
if (strlen(artist_name) < 2) {
|
|
return NULL;
|
|
}
|
|
|
|
return artist_name;
|
|
}
|
|
|
|
int h_db_add(char *db_row, const char *url) {
|
|
char *sql = malloc(10000);
|
|
int result;
|
|
char *error_msg;
|
|
|
|
char *website_name = get_website_from_url(url);
|
|
if (website_name == NULL) {
|
|
fprintf(stderr, "Couldn't get website_name from url %s\n", url);
|
|
return 1;
|
|
}
|
|
char *artist_name = get_artist_from_url(url, website_name);
|
|
if (artist_name == NULL) {
|
|
fprintf(stderr, "Couldn't get artist_name from url %s\n", url);
|
|
return 1;
|
|
}
|
|
|
|
sprintf(sql, "INSERT INTO hentai VALUES ('%s', '%s', '%s'); SELECT * from hentai WHERE url = '%s';", url, artist_name, website_name, url);
|
|
result = sqlite3_exec(db, sql, db_row_callback, db_row, &error_msg);
|
|
free(sql);
|
|
free(artist_name);
|
|
if (result != SQLITE_OK) {
|
|
fprintf(stderr, "SQL error: %s\n", error_msg);
|
|
sqlite3_free(error_msg);
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int h_db_add_custom_name(char *db_row, const char *url, const char *custom_name) {
|
|
char *sql = malloc(10000);
|
|
int result;
|
|
char *error_msg;
|
|
|
|
char *website_name = get_website_from_url(url);
|
|
if (website_name == NULL) {
|
|
fprintf(stderr, "Couldn't get website_name from url %s\n", url);
|
|
return 1;
|
|
}
|
|
|
|
sprintf(sql, "INSERT INTO hentai VALUES ('%s', '%s', '%s'); SELECT * from hentai WHERE url = '%s';", url, custom_name, website_name, url);
|
|
result = sqlite3_exec(db, sql, db_row_callback, db_row, &error_msg);
|
|
free(sql);
|
|
if (result != SQLITE_OK) {
|
|
fprintf(stderr, "SQL error: %s\n", error_msg);
|
|
sqlite3_free(error_msg);
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int h_db_generate_file(FILE *fptr, const char *website_name) {
|
|
char *sql = malloc(10000);
|
|
int result;
|
|
char *error_msg;
|
|
|
|
sprintf(sql, "SELECT * FROM hentai WHERE website_name = '%s' ORDER BY url ASC;", website_name);
|
|
result = sqlite3_exec(db, sql, fprintf_callback, fptr, &error_msg);
|
|
free(sql);
|
|
if (result != SQLITE_OK) {
|
|
fprintf(stderr, "SQL error: %s\n", error_msg);
|
|
sqlite3_free(error_msg);
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int h_db_update_artist_name(const char *url, const char *new_artist_name) {
|
|
char *sql = malloc(10000);
|
|
int result;
|
|
char *error_msg;
|
|
|
|
sprintf(sql, "UPDATE hentai set artist_name = '%s' WHERE url = '%s';", new_artist_name, url);
|
|
result = sqlite3_exec(db, sql, fprintf_callback, stdout, &error_msg);
|
|
free(sql);
|
|
if (result != SQLITE_OK) {
|
|
fprintf(stderr, "SQL error: %s\n", error_msg);
|
|
sqlite3_free(error_msg);
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int h_db_delete(const char *url) {
|
|
char *sql = malloc(10000);
|
|
int result;
|
|
char *error_msg;
|
|
|
|
sprintf(sql, "DELETE FROM hentai WHERE url = '%s';", url);
|
|
result = sqlite3_exec(db, sql, fprintf_callback, stdout, &error_msg);
|
|
free(sql);
|
|
if (result != SQLITE_OK) {
|
|
fprintf(stderr, "SQL error: %s\n", error_msg);
|
|
sqlite3_free(error_msg);
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|