modified task manager to add markdown, import_posts task

the import_posts task isn't perfect, we need some relation between the
author_id and the author itself
This commit is contained in:
2024-09-22 22:37:47 -04:00
parent 80081b75d5
commit 3afff00b29
3 changed files with 46 additions and 15 deletions

View File

@ -1,35 +1,56 @@
use std::fs;
use std::path;
pub async fn import_posts(dir_path: &str, pool: &sqlx::Pool<sqlx::Postgres>) {
println!("hello from import_posts");
let entries = std::fs::read_dir(dir_path).unwrap();
let entries = fs::read_dir(dir_path).unwrap();
let options = MarkdownOptions {
options: markdown::Constructs::gfm(),
};
for f in entries {
let file = f.unwrap();
let file_path = file.path();
if file_path.is_file() {
let file_name = file.file_name().to_str().unwrap();
let file_name = file.file_name();
let file_name_final = &file_name.to_str().unwrap();
let exists = sqlx::query_as::<_, FilenameExists>(
"SELECT EXISTS(SELECT 1 FROM posts WHERE filename = $1)",
)
.bind(file_name.clone())
.bind(file_name_final)
.fetch_one(pool)
.await
.unwrap()
.filename;
if !exists.is_empty() {
let file_md_contents = std::fs::read_to_string(file).unwrap().as_str();
let content = markdown::to_html(file_md_contents);
println!(
"File does not exist! Inserting: {:?}",
file_path.file_name()
);
let file_md_contents = process_read_file(file_path, &options);
let content = markdown::to_html(&file_md_contents);
sqlx::query_as::<_, InsertPosts>("INSERT INTO posts (title, body) VALUES ($1, $2)")
.bind(file_name.clone())
.fetch_one(pool)
.await
.unwrap();
sqlx::query_as::<_, InsertPosts>(
"INSERT INTO posts (title, body, filename, author_id) VALUES ($1, $2, $3, $4) RETURNING (title, body, filename, author_id)",
)
.bind(String::from("Hello world from Postgres!"))
.bind(content)
.bind(file_name_final)
.bind(1)
.fetch_one(pool)
.await
.unwrap();
}
}
}
}
fn process_read_file<P: AsRef<path::Path>>(path: P, md_opts: &MarkdownOptions) -> String {
let file_contents = fs::read_to_string(path).unwrap();
markdown::to_html(file_contents.as_str())
}
#[derive(Debug, sqlx::FromRow)]
struct FilenameExists {
filename: String,
@ -39,4 +60,10 @@ struct FilenameExists {
struct InsertPosts {
title: String,
body: String,
filename: String,
author_id: i32,
}
struct MarkdownOptions {
options: markdown::Constructs,
}