feat(idx) add auto idx

This commit is contained in:
lumither committed 2024-05-03 21:40:53 +08:00
1 parent 6b311ab255
commit 36825b4950
8 files changed
+115 -17

No files matched your search

+6 -6
View File
@@ -29,30 +29,30 @@ pub async fn refresh(
let post_list: Vec<Post> = md_list
.into_iter()
.map(Post::from_path)
.map(|post| Post::from_path(post).unwrap())
.collect();
for post in post_list {
match query(
r##"
INSERT INTO Post (post_id, title, content, last_update)
VALUES ($1, $2, $3, $4)
INSERT INTO Post (post_id, title, summary, content, last_update)
VALUES ($1, $2, $3, $4, $5)
ON CONFLICT (post_id)
DO UPDATE SET title = excluded.title,
summary = excluded.summary,
content = excluded.content,
last_update = excluded.last_update
"##,
)
.bind(post.post_id)
.bind(post.title)
.bind(post.summary)
.bind(post.content)
.bind(post.last_update)
.execute(&db_connection)
.await
{
Ok(_) => {
dbg!("inserted");
}
Ok(_) => {}
Err(e) => {
return Err((
StatusCode::INTERNAL_SERVER_ERROR,
+1
View File
@@ -12,6 +12,7 @@ use crate::api::post::refresh::refresh;
mod api;
mod model;
mod markdown;
#[tokio::main]
async fn main() -> Result<(), Box<dyn Error>> {
+51
View File
@@ -0,0 +1,51 @@
use std::collections::HashMap;
use regex::Regex;
pub fn parse_meta(md_content: &str) -> HashMap<String, String> {
let mut fields: HashMap<String, String> = HashMap::new();
let match_regex = Regex::new(r"(?s)---\n(.*\n)---\n").unwrap();
if let Some(front_matter) = match_regex.captures(md_content) {
let fm_content = front_matter.get(1).unwrap().as_str();
for line in fm_content.lines() {
if let Some((k, v)) = extract_kv(line) {
fields.insert(k, v);
}
}
}
fields
}
fn extract_kv(line: &str) -> Option<(String, String)> {
let split: Vec<&str> = line.splitn(2, ':').map(|s| s.trim()).collect();
if split.len() == 2 {
let key = split[0].to_string();
let value = split[1].to_string();
Some((key, value))
} else {
None
}
}
#[cfg(test)]
mod meta_test {
use std::error::Error;
use crate::markdown::meta::parse_meta;
#[test]
fn test_parse_meta_front_matter() -> Result<(), Box<dyn Error>> {
let parsed_meta = parse_meta(
r##"
---
title: Hello World!
tags: [ "test", "hello_world" ]
---
"##
);
dbg!(&parsed_meta);
assert_eq!(parsed_meta.get("title"), Some(&"Hello World!".to_string()));
assert_eq!(parsed_meta.get("tags"), Some(&r#"[ "test", "hello_world" ]"#.to_string()));
Ok(())
}
}
+1
View File
@@ -0,0 +1 @@
pub mod meta;
+3
View File
@@ -9,6 +9,9 @@ pub mod post;
pub struct Post {
pub post_id: Uuid,
pub title: String,
pub summary: String,
pub content: String,
pub last_update: NaiveDateTime,
}
+48 -9
View File
@@ -1,16 +1,55 @@
use chrono::Utc;
use std::{env, fs};
use std::error::Error;
use std::process::Command;
use std::str;
use chrono::{NaiveDateTime, Utc};
use uuid::Uuid;
use crate::markdown::meta::parse_meta;
use crate::model::Post;
impl super::Post {
pub fn from_path(path: String) -> Self {
// todo: remove test
Post {
impl Post {
pub fn from_path(path: String) -> Result<Self, Box<dyn Error>> {
let md_content = fs::read_to_string(&path)?;
let meta = parse_meta(&md_content);
let title = match meta.get("title") {
None => "Title Not Found".to_string(),
Some(title) => title.to_owned(),
};
let summary = match meta.get("summary") {
None => "".to_string(),
Some(summary) => summary.to_owned(),
};
let git_output = Command::new("git")
.arg("--no-pager")
.arg("log")
.arg("-1")
.arg("--format=%cI")
.arg("--")
.current_dir(env::var("GIT_WORK_DIR")?)
.arg(&path)
.output()
.expect("Failed to execute git command");
let last_update = if git_output.status.success() {
NaiveDateTime::parse_from_str(
str::from_utf8(&git_output.stdout).unwrap().trim_end(),
"%Y-%m-%dT%H:%M:%S%z",
)
.unwrap_or_else(|_| Utc::now().naive_utc())
} else {
Utc::now().naive_utc()
};
Ok(Post {
post_id: Uuid::new_v4(),
title: "test title".to_string(),
content: "test content".to_string(),
last_update: Utc::now().naive_utc(),
}
title,
summary,
content: md_content,
last_update,
})
}
}