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

This commit is contained in:
2024-11-02 01:11:29 +01:00
parent d41707ff88
commit 7d8c27047a
2 changed files with 384 additions and 375 deletions

View File

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