AustrianToast/HDB
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Git wants me to commit, I want to sleep. Take me sweet void.

Browse Source
This commit is contained in:
AustrianToast
2024-11-02 01:11:29 +01:00
parent d41707ff88
commit 7d8c27047a
2 changed files with 384 additions and 375 deletions
Download Patch File Download Diff File Expand all files Collapse all files

728
Rule34.links
View File

File diff suppressed because it is too large Load Diff

31
src/main.rs
View File

@ -1,13 +1,15 @@
use postgres::{Client, NoTls}; use postgres::{Client, NoTls};
use std::{ use std::{
env, fs, env::args,
fs::File,
io::{BufRead, BufReader, Write}, io::{BufRead, BufReader, Write},
}; };
fn generate_file(website_name: &str) { fn generate_file(website_name: &str) {
let mut client = Client::connect("host=192.168.0.10 port=28945 user=hentai password=h99nqaNPhpfbuuhCDwQXLpZAnoVTjSQP7taoqmQhpzc2rPLVC4JUAKxAHfuuhuU9", NoTls).unwrap(); let mut client = Client::connect("host=192.168.0.10 port=28945 user=hentai password=h99nqaNPhpfbuuhCDwQXLpZAnoVTjSQP7taoqmQhpzc2rPLVC4JUAKxAHfuuhuU9", NoTls).unwrap();
let sqlstmt = format!("SELECT website_exists('{}');", website_name); let output = client
let output = client.query_one(&sqlstmt, &[]).unwrap(); .query_one("SELECT website_exists($1);", &[&website_name])
.unwrap();
let result: bool = output.get(0); let result: bool = output.get(0);
if result == false { if result == false {
@ -15,7 +17,7 @@ fn generate_file(website_name: &str) {
} }
let filename = format!("{website_name}_generated.txt"); let filename = format!("{website_name}_generated.txt");
let mut file = match fs::File::create(filename) { let mut file = match File::create(filename) {
Ok(file) => file, Ok(file) => file,
Err(error) => { Err(error) => {
let message = error.to_string(); let message = error.to_string();
@ -30,21 +32,28 @@ fn generate_file(website_name: &str) {
let url: String = row.get(0); let url: String = row.get(0);
let folder_path: String = row.get(1); let folder_path: String = row.get(1);
write!(file, "{} # {}", url, folder_path).unwrap(); write!(file, "{} # {}\n", url, folder_path).unwrap();
} }
} }
fn import_file(filename: &str) { fn import_file(filename: &str) {
let file = fs::File::open(filename).unwrap(); let file = File::open(filename).unwrap();
let mut reader = BufReader::new(file); let reader_iter = BufReader::new(file).lines();
let mut line = String::new(); let mut client = Client::connect("host=192.168.0.10 port=28945 user=hentai password=h99nqaNPhpfbuuhCDwQXLpZAnoVTjSQP7taoqmQhpzc2rPLVC4JUAKxAHfuuhuU9", NoTls).unwrap();
reader.read_line(&mut line).unwrap();
dbg!(line); for line in reader_iter {
let line = line.unwrap();
if !line.contains("#") {
let output = client.query_one("SELECT insert_url($1)", &[&line]).unwrap();
let result: i32 = output.get(0);
dbg!(result);
}
}
} }
fn main() { fn main() {
let args: Vec<String> = env::args().collect(); let args: Vec<String> = args().collect();
dbg!(&args); dbg!(&args);
if args.len() != 3 { if args.len() != 3 {