From 3afff00b299861cc3ecf9734a670447299f201fe Mon Sep 17 00:00:00 2001
From: "Wyatt J. Miller" <wyatt@wyattjmiller.com>
Date: Sun, 22 Sep 2024 22:37:47 -0400
Subject: [PATCH] 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
---
 backend/task/.gitignore                |  2 ++
 backend/task/src/main.rs               | 12 ++++---
 backend/task/src/tasks/import_posts.rs | 47 ++++++++++++++++++++------
 3 files changed, 46 insertions(+), 15 deletions(-)

diff --git a/backend/task/.gitignore b/backend/task/.gitignore
index 2f7896d..e24581b 100644
--- a/backend/task/.gitignore
+++ b/backend/task/.gitignore
@@ -1 +1,3 @@
 target/
+.env
+app/
diff --git a/backend/task/src/main.rs b/backend/task/src/main.rs
index d4de67d..1ea65df 100644
--- a/backend/task/src/main.rs
+++ b/backend/task/src/main.rs
@@ -1,6 +1,6 @@
 use chrono::{prelude::*, Duration};
 // use once_cell::sync::Lazy;
-use sqlx::{postgres::PgPoolOptions, PgPool, Pool, Postgres};
+use sqlx::{postgres::PgPoolOptions, Pool, Postgres};
 use std::env;
 
 mod tasks;
@@ -83,10 +83,12 @@ impl<'a> TaskManager<'a> {
                 scheduler.add(job_scheduler::Job::new(j.schedule.parse().unwrap(), || {
                     println!("Starting task name: {:?}", j.task_name);
 
-                    match j.task_id {
-                        1 => tasks::import_posts::import_posts("/app", &self.pool),
-                        _ => panic!(),
-                    }
+                    async {
+                        match j.task_id {
+                            1 => tasks::import_posts::import_posts("/app", &self.pool).await,
+                            _ => panic!(),
+                        }
+                    };
                 }));
             })
             .collect::<Vec<_>>();
diff --git a/backend/task/src/tasks/import_posts.rs b/backend/task/src/tasks/import_posts.rs
index d180365..3572174 100644
--- a/backend/task/src/tasks/import_posts.rs
+++ b/backend/task/src/tasks/import_posts.rs
@@ -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,
 }