Git wants me to commit, I want to sleep. Take me sweet void.
This commit is contained in:
31
src/main.rs
31
src/main.rs
@ -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 {
|
||||
|
Reference in New Issue
Block a user