move files around
This commit is contained in:
228
src/db.c
Normal file
228
src/db.c
Normal file
@ -0,0 +1,228 @@
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <sqlite3.h>
|
||||
|
||||
void print_usage() {
|
||||
printf("Usage: db <arguments>\n");
|
||||
printf("\tadd <url> [custom_name] \n");
|
||||
printf("\tgenerate <website_name>\n");
|
||||
printf("\tupdate <url> <new_artist_name>\n");
|
||||
printf("\tdelete <url>\n");
|
||||
}
|
||||
|
||||
void exit_nicely(sqlite3 *db, const char *msg) {
|
||||
if (msg != NULL) {
|
||||
fprintf(stderr, "%s\n", msg);
|
||||
}
|
||||
print_usage();
|
||||
sqlite3_close(db);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
static int print_callback(void *unused, int argc, char **argv, char **col_name) {
|
||||
for (size_t i = 0; i < argc; i++) {
|
||||
printf("%s", argv[i] ? argv[i] : "NULL");
|
||||
if (i < argc - 1) {
|
||||
printf(" ");
|
||||
}
|
||||
}
|
||||
printf("\n");
|
||||
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;
|
||||
char *artist_name = malloc(name_length + 1);
|
||||
memcpy(artist_name, url + offset, name_length);
|
||||
artist_name[name_length] = '\0';
|
||||
|
||||
return artist_name;
|
||||
}
|
||||
|
||||
bool add(sqlite3 *db, const char *url) {
|
||||
char *sql;
|
||||
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 false;
|
||||
}
|
||||
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 false;
|
||||
}
|
||||
|
||||
sprintf(sql, "INSERT INTO hentai VALUES ('%s', '%s', '%s');", url, artist_name, website_name);
|
||||
result = sqlite3_exec(db, sql, print_callback, 0, &error_msg);
|
||||
if (result != SQLITE_OK) {
|
||||
fprintf(stderr, "SQL error: %s\n", error_msg);
|
||||
sqlite3_free(error_msg);
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool add_custom_name(sqlite3 *db, const char *url, const char *custom_name) {
|
||||
char *sql;
|
||||
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 false;
|
||||
}
|
||||
|
||||
sprintf(sql, "INSERT INTO hentai VALUES ('%s', '%s', '%s');", url, custom_name, website_name);
|
||||
result = sqlite3_exec(db, sql, print_callback, 0, &error_msg);
|
||||
if (result != SQLITE_OK) {
|
||||
fprintf(stderr, "SQL error: %s\n", error_msg);
|
||||
sqlite3_free(error_msg);
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool generate(sqlite3 *db, const char *website_name) {
|
||||
char *sql;
|
||||
int result;
|
||||
char *error_msg;
|
||||
|
||||
sprintf(sql, "SELECT * FROM hentai WHERE website_name = '%s';", website_name);
|
||||
result = sqlite3_exec(db, sql, print_callback, 0, &error_msg);
|
||||
if (result != SQLITE_OK) {
|
||||
fprintf(stderr, "SQL error: %s\n", error_msg);
|
||||
sqlite3_free(error_msg);
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool update_artist_name(sqlite3 *db, const char *url, const char *new_artist_name) {
|
||||
char *sql;
|
||||
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, print_callback, 0, &error_msg);
|
||||
if (result != SQLITE_OK) {
|
||||
fprintf(stderr, "SQL error: %s\n", error_msg);
|
||||
sqlite3_free(error_msg);
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool delete(sqlite3 *db, const char *url) {
|
||||
char *sql;
|
||||
int result;
|
||||
char *error_msg;
|
||||
|
||||
sprintf(sql, "DELETE FROM hentai WHERE url = '%s';", url);
|
||||
result = sqlite3_exec(db, sql, print_callback, 0, &error_msg);
|
||||
if (result != SQLITE_OK) {
|
||||
fprintf(stderr, "SQL error: %s\n", error_msg);
|
||||
sqlite3_free(error_msg);
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
sqlite3 *db;
|
||||
int result;
|
||||
char *error_msg;
|
||||
char *sql;
|
||||
bool successful;
|
||||
|
||||
result = sqlite3_open("/tmp/Hentai.db", &db);
|
||||
if (result != SQLITE_OK) {
|
||||
exit_nicely(db, sqlite3_errmsg(db));
|
||||
}
|
||||
|
||||
sql = "CREATE TABLE IF NOT EXISTS hentai (url TEXT PRIMARY KEY NOT NULL, artist_name TEXT NOT NULL, website_name TEXT NOT NULL) STRICT;";
|
||||
result = sqlite3_exec(db, sql, NULL, 0, &error_msg);
|
||||
if (result != SQLITE_OK) {
|
||||
fprintf(stderr, "SQL error: %s\n", error_msg);
|
||||
sqlite3_free(error_msg);
|
||||
exit_nicely(db, NULL);
|
||||
}
|
||||
|
||||
if (argc < 2) {
|
||||
exit_nicely(db, "Incorrect amount of arguments");
|
||||
}
|
||||
|
||||
if (strcmp(argv[1], "add") == 0) {
|
||||
if (argc == 3) {
|
||||
successful = add(db, argv[2]);
|
||||
if (!successful) {
|
||||
exit_nicely(db, NULL);
|
||||
}
|
||||
} else if (argc == 4) {
|
||||
successful = add_custom_name(db, argv[2], argv[3]);
|
||||
if (!successful) {
|
||||
exit_nicely(db, NULL);
|
||||
}
|
||||
} else {
|
||||
exit_nicely(db, "Incorrect amount of arguments");
|
||||
}
|
||||
} else if (strcmp(argv[1], "generate") == 0) {
|
||||
if (argc == 3) {
|
||||
successful = generate(db, argv[2]);
|
||||
if (!successful) {
|
||||
exit_nicely(db, NULL);
|
||||
}
|
||||
} else {
|
||||
exit_nicely(db, "Incorrect amount of arguments");
|
||||
}
|
||||
} else if (strcmp(argv[1], "update") == 0) {
|
||||
if (argc == 4) {
|
||||
successful = update_artist_name(db, argv[3], argv[2]);
|
||||
if (!successful) {
|
||||
exit_nicely(db, NULL);
|
||||
}
|
||||
} else {
|
||||
exit_nicely(db, "Incorrect amount of arguments");
|
||||
}
|
||||
|
||||
} else if (strcmp(argv[1], "delete") == 0) {
|
||||
if (argc == 3) {
|
||||
successful = delete(db, argv[2]);
|
||||
if (!successful) {
|
||||
exit_nicely(db, NULL);
|
||||
}
|
||||
} else {
|
||||
exit_nicely(db, "Incorrect amount of arguments");
|
||||
}
|
||||
} else {
|
||||
exit_nicely(db, "Invalid argument");
|
||||
}
|
||||
|
||||
sqlite3_close(db);
|
||||
return 0;
|
||||
}
|
||||
69
src/has_internet.c
Normal file
69
src/has_internet.c
Normal file
@ -0,0 +1,69 @@
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <netdb.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/types.h>
|
||||
#include <unistd.h>
|
||||
|
||||
bool has_internet() {
|
||||
int ret;
|
||||
|
||||
// https://www.man7.org/linux/man-pages/man3/getaddrinfo.3.html
|
||||
struct addrinfo hints;
|
||||
memset(&hints, 0, sizeof(hints));
|
||||
hints.ai_family = AF_INET;
|
||||
hints.ai_socktype = SOCK_STREAM;
|
||||
hints.ai_flags = 0;
|
||||
hints.ai_protocol = 0;
|
||||
struct addrinfo *res, *rp;
|
||||
ret = getaddrinfo("hopeless-cloud.xyz", "80", 0, &res);
|
||||
if (ret != 0) {
|
||||
// fprintf(stderr, "Couldn't getaddrinfo()\n");
|
||||
return false;
|
||||
}
|
||||
int sockfd;
|
||||
for (rp = res; rp != NULL; rp = rp->ai_next) {
|
||||
// https://www.man7.org/linux/man-pages/man2/socket.2.html
|
||||
sockfd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol);
|
||||
if (sockfd == -1) { continue; }
|
||||
|
||||
// https://www.man7.org/linux/man-pages/man2/connect.2.html
|
||||
if (connect(sockfd, rp->ai_addr, rp->ai_addrlen) == 0) { break; }
|
||||
|
||||
close(sockfd);
|
||||
}
|
||||
freeaddrinfo(res);
|
||||
if (rp == NULL) {
|
||||
// fprintf(stderr, "Couldn't connect socket\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
// https://www.man7.org/linux/man-pages/man2/write.2.html
|
||||
char *req_msg = "GET / HTTP/1.1\r\n\r\n";
|
||||
if (write(sockfd, req_msg, strlen(req_msg)) != strlen(req_msg)) {
|
||||
// fprintf(stderr, "Couldn't fully write to socket\n");
|
||||
return false;
|
||||
}
|
||||
|
||||
// https://www.man7.org/linux/man-pages/man2/recv.2.html
|
||||
size_t count = 1000000;
|
||||
char buf[count];
|
||||
if (recv(sockfd, buf, count-1, MSG_WAITALL) == -1 ) {
|
||||
// fprintf(stderr, "Couldn't read from socket\n");
|
||||
return false;
|
||||
}
|
||||
// printf("%s\n", buf);
|
||||
close(sockfd);
|
||||
return true;
|
||||
}
|
||||
|
||||
int main() {
|
||||
if (has_internet()) {
|
||||
// printf("has internet\n");
|
||||
return 0;
|
||||
} else {
|
||||
// printf("no internet\n");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user