feat: implementation of M1

This commit is contained in:
2026-04-04 14:18:09 +02:00
parent fd744573dd
commit 2d07ac7866
44 changed files with 12197 additions and 30 deletions

View File

@@ -0,0 +1,210 @@
use rusqlite::{params, Connection};
use crate::db::from_row::{media_from_row, MEDIA_COLUMNS};
use crate::model::Media;
fn tags_to_json(tags: &[String]) -> String {
serde_json::to_string(tags).unwrap_or_else(|_| "[]".into())
}
pub fn insert_media(conn: &Connection, m: &Media) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO media (
id, project_id, filename, original_name, mime_type, size,
width, height, title, alt, caption, author, language,
file_path, sidecar_path, checksum, tags, created_at, updated_at
) VALUES (
?1, ?2, ?3, ?4, ?5, ?6,
?7, ?8, ?9, ?10, ?11, ?12, ?13,
?14, ?15, ?16, ?17, ?18, ?19
)",
params![
m.id,
m.project_id,
m.filename,
m.original_name,
m.mime_type,
m.size,
m.width,
m.height,
m.title,
m.alt,
m.caption,
m.author,
m.language,
m.file_path,
m.sidecar_path,
m.checksum,
tags_to_json(&m.tags),
m.created_at,
m.updated_at,
],
)?;
Ok(())
}
pub fn get_media_by_id(conn: &Connection, id: &str) -> rusqlite::Result<Media> {
conn.query_row(
&format!("SELECT {MEDIA_COLUMNS} FROM media WHERE id = ?1"),
params![id],
media_from_row,
)
}
pub fn list_media_by_project(conn: &Connection, project_id: &str) -> rusqlite::Result<Vec<Media>> {
let mut stmt = conn.prepare(&format!(
"SELECT {MEDIA_COLUMNS} FROM media WHERE project_id = ?1 ORDER BY created_at DESC"
))?;
let rows = stmt.query_map(params![project_id], media_from_row)?;
rows.collect()
}
pub fn update_media(conn: &Connection, m: &Media) -> rusqlite::Result<()> {
conn.execute(
"UPDATE media SET
filename = ?1, original_name = ?2, mime_type = ?3, size = ?4,
width = ?5, height = ?6, title = ?7, alt = ?8, caption = ?9,
author = ?10, language = ?11, file_path = ?12, sidecar_path = ?13,
checksum = ?14, tags = ?15, updated_at = ?16
WHERE id = ?17",
params![
m.filename,
m.original_name,
m.mime_type,
m.size,
m.width,
m.height,
m.title,
m.alt,
m.caption,
m.author,
m.language,
m.file_path,
m.sidecar_path,
m.checksum,
tags_to_json(&m.tags),
m.updated_at,
m.id,
],
)?;
Ok(())
}
pub fn delete_media(conn: &Connection, id: &str) -> rusqlite::Result<()> {
conn.execute("DELETE FROM media WHERE id = ?1", params![id])?;
Ok(())
}
/// Test helper: create a minimal Media value (available to sibling test modules).
#[cfg(test)]
pub fn make_test_media(id: &str, project_id: &str) -> Media {
Media {
id: id.to_string(),
project_id: project_id.to_string(),
filename: format!("{id}.jpg"),
original_name: "photo.jpg".to_string(),
mime_type: "image/jpeg".to_string(),
size: 50000,
width: Some(1920),
height: Some(1080),
title: None,
alt: None,
caption: None,
author: None,
language: None,
file_path: format!("/media/{id}.jpg"),
sidecar_path: format!("/media/{id}.jpg.meta"),
checksum: None,
tags: vec![],
created_at: 1000,
updated_at: 2000,
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::db::queries::project::{insert_project, make_test_project};
use crate::db::Database;
fn setup() -> Database {
let db = Database::open_in_memory().unwrap();
db.migrate().unwrap();
insert_project(db.conn(), &make_test_project("p1", "blog")).unwrap();
db
}
fn make_media_item(id: &str) -> Media {
Media {
id: id.to_string(),
project_id: "p1".to_string(),
filename: "abc.jpg".to_string(),
original_name: "photo.jpg".to_string(),
mime_type: "image/jpeg".to_string(),
size: 50000,
width: Some(1920),
height: Some(1080),
title: Some("Sunset".into()),
alt: Some("A sunset".into()),
caption: None,
author: Some("Bob".into()),
language: Some("en".into()),
file_path: "/media/abc.jpg".to_string(),
sidecar_path: "/media/abc.jpg.meta".to_string(),
checksum: Some("hash1".into()),
tags: vec!["nature".into()],
created_at: 1000,
updated_at: 2000,
}
}
#[test]
fn insert_and_get_by_id() {
let db = setup();
let m = make_media_item("m1");
insert_media(db.conn(), &m).unwrap();
let fetched = get_media_by_id(db.conn(), "m1").unwrap();
assert_eq!(fetched.original_name, "photo.jpg");
assert_eq!(fetched.width, Some(1920));
assert_eq!(fetched.height, Some(1080));
assert_eq!(fetched.tags, vec!["nature"]);
assert_eq!(fetched.size, 50000);
}
#[test]
fn list_by_project() {
let db = setup();
let mut m1 = make_media_item("m1");
m1.created_at = 1000;
let mut m2 = make_media_item("m2");
m2.filename = "def.jpg".into();
m2.created_at = 2000;
insert_media(db.conn(), &m1).unwrap();
insert_media(db.conn(), &m2).unwrap();
let list = list_media_by_project(db.conn(), "p1").unwrap();
assert_eq!(list.len(), 2);
assert_eq!(list[0].id, "m2");
}
#[test]
fn update_media_fields() {
let db = setup();
let mut m = make_media_item("m1");
insert_media(db.conn(), &m).unwrap();
m.title = Some("New Title".into());
m.tags = vec!["updated".into()];
m.updated_at = 9999;
update_media(db.conn(), &m).unwrap();
let fetched = get_media_by_id(db.conn(), "m1").unwrap();
assert_eq!(fetched.title.as_deref(), Some("New Title"));
assert_eq!(fetched.tags, vec!["updated"]);
}
#[test]
fn delete_removes_media() {
let db = setup();
insert_media(db.conn(), &make_media_item("m1")).unwrap();
delete_media(db.conn(), "m1").unwrap();
assert!(get_media_by_id(db.conn(), "m1").is_err());
}
}

View File

@@ -0,0 +1,171 @@
use rusqlite::{params, Connection};
use crate::db::from_row::{media_translation_from_row, MEDIA_TRANSLATION_COLUMNS};
use crate::model::MediaTranslation;
pub fn insert_media_translation(
conn: &Connection,
t: &MediaTranslation,
) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO media_translations (
id, project_id, translation_for, language, title, alt, caption,
created_at, updated_at
) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9)",
params![
t.id,
t.project_id,
t.translation_for,
t.language,
t.title,
t.alt,
t.caption,
t.created_at,
t.updated_at,
],
)?;
Ok(())
}
pub fn get_media_translation_by_media_and_language(
conn: &Connection,
translation_for: &str,
language: &str,
) -> rusqlite::Result<MediaTranslation> {
conn.query_row(
&format!(
"SELECT {MEDIA_TRANSLATION_COLUMNS} FROM media_translations
WHERE translation_for = ?1 AND language = ?2"
),
params![translation_for, language],
media_translation_from_row,
)
}
pub fn list_media_translations_by_media(
conn: &Connection,
translation_for: &str,
) -> rusqlite::Result<Vec<MediaTranslation>> {
let mut stmt = conn.prepare(&format!(
"SELECT {MEDIA_TRANSLATION_COLUMNS} FROM media_translations
WHERE translation_for = ?1 ORDER BY language"
))?;
let rows = stmt.query_map(params![translation_for], media_translation_from_row)?;
rows.collect()
}
pub fn upsert_media_translation(
conn: &Connection,
t: &MediaTranslation,
) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO media_translations (
id, project_id, translation_for, language, title, alt, caption,
created_at, updated_at
) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9)
ON CONFLICT(translation_for, language) DO UPDATE SET
title = excluded.title,
alt = excluded.alt,
caption = excluded.caption,
updated_at = excluded.updated_at",
params![
t.id,
t.project_id,
t.translation_for,
t.language,
t.title,
t.alt,
t.caption,
t.created_at,
t.updated_at,
],
)?;
Ok(())
}
pub fn delete_media_translation(
conn: &Connection,
translation_for: &str,
language: &str,
) -> rusqlite::Result<()> {
conn.execute(
"DELETE FROM media_translations WHERE translation_for = ?1 AND language = ?2",
params![translation_for, language],
)?;
Ok(())
}
#[cfg(test)]
mod tests {
use super::*;
use crate::db::queries::media::{insert_media, make_test_media};
use crate::db::queries::project::{insert_project, make_test_project};
use crate::db::Database;
fn setup() -> Database {
let db = Database::open_in_memory().unwrap();
db.migrate().unwrap();
insert_project(db.conn(), &make_test_project("p1", "blog")).unwrap();
insert_media(db.conn(), &make_test_media("m1", "p1")).unwrap();
db
}
fn make_mt(id: &str, lang: &str) -> MediaTranslation {
MediaTranslation {
id: id.to_string(),
project_id: "p1".to_string(),
translation_for: "m1".to_string(),
language: lang.to_string(),
title: Some(format!("Title {lang}")),
alt: Some(format!("Alt {lang}")),
caption: None,
created_at: 1000,
updated_at: 2000,
}
}
#[test]
fn insert_and_get() {
let db = setup();
insert_media_translation(db.conn(), &make_mt("mt1", "de")).unwrap();
let t = get_media_translation_by_media_and_language(db.conn(), "m1", "de").unwrap();
assert_eq!(t.title.as_deref(), Some("Title de"));
}
#[test]
fn list_by_media() {
let db = setup();
insert_media_translation(db.conn(), &make_mt("mt1", "de")).unwrap();
insert_media_translation(db.conn(), &make_mt("mt2", "fr")).unwrap();
let list = list_media_translations_by_media(db.conn(), "m1").unwrap();
assert_eq!(list.len(), 2);
assert_eq!(list[0].language, "de");
assert_eq!(list[1].language, "fr");
}
#[test]
fn upsert_inserts_then_updates() {
let db = setup();
let mut t = make_mt("mt1", "de");
upsert_media_translation(db.conn(), &t).unwrap();
let fetched = get_media_translation_by_media_and_language(db.conn(), "m1", "de").unwrap();
assert_eq!(fetched.title.as_deref(), Some("Title de"));
t.title = Some("Neuer Titel".into());
t.updated_at = 9999;
upsert_media_translation(db.conn(), &t).unwrap();
let fetched = get_media_translation_by_media_and_language(db.conn(), "m1", "de").unwrap();
assert_eq!(fetched.title.as_deref(), Some("Neuer Titel"));
assert_eq!(fetched.updated_at, 9999);
}
#[test]
fn delete_translation() {
let db = setup();
insert_media_translation(db.conn(), &make_mt("mt1", "de")).unwrap();
delete_media_translation(db.conn(), "m1", "de").unwrap();
assert!(
get_media_translation_by_media_and_language(db.conn(), "m1", "de").is_err()
);
}
}

View File

@@ -0,0 +1,11 @@
pub mod project;
pub mod post;
pub mod post_translation;
pub mod media;
pub mod media_translation;
pub mod tag;
pub mod post_link;
pub mod post_media;
pub mod template;
pub mod script;
pub mod setting;

View File

@@ -0,0 +1,339 @@
use rusqlite::{params, Connection};
use crate::db::from_row::{post_from_row, post_status_to_str, POST_COLUMNS};
use crate::model::{Post, PostStatus};
fn tags_to_json(tags: &[String]) -> String {
serde_json::to_string(tags).unwrap_or_else(|_| "[]".into())
}
pub fn insert_post(conn: &Connection, post: &Post) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO posts (
id, project_id, title, slug, excerpt, content, status, author,
language, do_not_translate, template_slug, file_path, checksum,
tags, categories,
published_title, published_content, published_tags,
published_categories, published_excerpt,
created_at, updated_at, published_at
) VALUES (
?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8,
?9, ?10, ?11, ?12, ?13,
?14, ?15,
?16, ?17, ?18, ?19, ?20,
?21, ?22, ?23
)",
params![
post.id,
post.project_id,
post.title,
post.slug,
post.excerpt,
post.content,
post_status_to_str(&post.status),
post.author,
post.language,
post.do_not_translate as i64,
post.template_slug,
post.file_path,
post.checksum,
tags_to_json(&post.tags),
tags_to_json(&post.categories),
post.published_title,
post.published_content,
post.published_tags,
post.published_categories,
post.published_excerpt,
post.created_at,
post.updated_at,
post.published_at,
],
)?;
Ok(())
}
pub fn get_post_by_id(conn: &Connection, id: &str) -> rusqlite::Result<Post> {
conn.query_row(
&format!("SELECT {POST_COLUMNS} FROM posts WHERE id = ?1"),
params![id],
post_from_row,
)
}
pub fn get_post_by_project_and_slug(
conn: &Connection,
project_id: &str,
slug: &str,
) -> rusqlite::Result<Post> {
conn.query_row(
&format!("SELECT {POST_COLUMNS} FROM posts WHERE project_id = ?1 AND slug = ?2"),
params![project_id, slug],
post_from_row,
)
}
pub fn list_posts_by_project(conn: &Connection, project_id: &str) -> rusqlite::Result<Vec<Post>> {
let mut stmt = conn.prepare(&format!(
"SELECT {POST_COLUMNS} FROM posts WHERE project_id = ?1 ORDER BY created_at DESC"
))?;
let rows = stmt.query_map(params![project_id], post_from_row)?;
rows.collect()
}
pub fn update_post(conn: &Connection, post: &Post) -> rusqlite::Result<()> {
conn.execute(
"UPDATE posts SET
title = ?1, slug = ?2, excerpt = ?3, content = ?4, status = ?5,
author = ?6, language = ?7, do_not_translate = ?8, template_slug = ?9,
file_path = ?10, checksum = ?11, tags = ?12, categories = ?13,
published_title = ?14, published_content = ?15, published_tags = ?16,
published_categories = ?17, published_excerpt = ?18,
updated_at = ?19, published_at = ?20
WHERE id = ?21",
params![
post.title,
post.slug,
post.excerpt,
post.content,
post_status_to_str(&post.status),
post.author,
post.language,
post.do_not_translate as i64,
post.template_slug,
post.file_path,
post.checksum,
tags_to_json(&post.tags),
tags_to_json(&post.categories),
post.published_title,
post.published_content,
post.published_tags,
post.published_categories,
post.published_excerpt,
post.updated_at,
post.published_at,
post.id,
],
)?;
Ok(())
}
pub fn update_post_status(
conn: &Connection,
id: &str,
status: &PostStatus,
updated_at: i64,
) -> rusqlite::Result<()> {
conn.execute(
"UPDATE posts SET status = ?1, updated_at = ?2 WHERE id = ?3",
params![post_status_to_str(status), updated_at, id],
)?;
Ok(())
}
pub fn clear_post_content(conn: &Connection, id: &str, updated_at: i64) -> rusqlite::Result<()> {
conn.execute(
"UPDATE posts SET content = NULL, updated_at = ?1 WHERE id = ?2",
params![updated_at, id],
)?;
Ok(())
}
pub fn set_post_file_path(
conn: &Connection,
id: &str,
file_path: &str,
updated_at: i64,
) -> rusqlite::Result<()> {
conn.execute(
"UPDATE posts SET file_path = ?1, updated_at = ?2 WHERE id = ?3",
params![file_path, updated_at, id],
)?;
Ok(())
}
pub fn set_published_snapshot(
conn: &Connection,
id: &str,
title: &str,
content: &str,
tags: &str,
categories: &str,
excerpt: Option<&str>,
published_at: i64,
updated_at: i64,
) -> rusqlite::Result<()> {
conn.execute(
"UPDATE posts SET
published_title = ?1, published_content = ?2, published_tags = ?3,
published_categories = ?4, published_excerpt = ?5,
published_at = ?6, updated_at = ?7
WHERE id = ?8",
params![title, content, tags, categories, excerpt, published_at, updated_at, id],
)?;
Ok(())
}
pub fn delete_post(conn: &Connection, id: &str) -> rusqlite::Result<()> {
conn.execute("DELETE FROM posts WHERE id = ?1", params![id])?;
Ok(())
}
pub fn count_posts_by_project(conn: &Connection, project_id: &str) -> rusqlite::Result<i64> {
conn.query_row(
"SELECT COUNT(*) FROM posts WHERE project_id = ?1",
params![project_id],
|row| row.get(0),
)
}
#[cfg(test)]
mod tests {
use super::*;
use crate::db::queries::project::{insert_project, make_test_project};
use crate::db::Database;
fn setup() -> Database {
let db = Database::open_in_memory().unwrap();
db.migrate().unwrap();
insert_project(db.conn(), &make_test_project("p1", "blog")).unwrap();
db
}
fn make_post(id: &str, slug: &str) -> Post {
Post {
id: id.to_string(),
project_id: "p1".to_string(),
title: format!("Post {id}"),
slug: slug.to_string(),
excerpt: Some("excerpt".into()),
content: Some("body".into()),
status: PostStatus::Draft,
author: Some("Alice".into()),
language: Some("en".into()),
do_not_translate: false,
template_slug: None,
file_path: format!("posts/{slug}.md"),
checksum: None,
tags: vec!["rust".into()],
categories: vec!["tech".into()],
published_title: None,
published_content: None,
published_tags: None,
published_categories: None,
published_excerpt: None,
created_at: 1000,
updated_at: 2000,
published_at: None,
}
}
#[test]
fn insert_and_get_by_id() {
let db = setup();
let post = make_post("x1", "hello");
insert_post(db.conn(), &post).unwrap();
let fetched = get_post_by_id(db.conn(), "x1").unwrap();
assert_eq!(fetched.title, "Post x1");
assert_eq!(fetched.tags, vec!["rust"]);
assert_eq!(fetched.categories, vec!["tech"]);
assert_eq!(fetched.status, PostStatus::Draft);
assert!(!fetched.do_not_translate);
}
#[test]
fn get_by_project_and_slug() {
let db = setup();
insert_post(db.conn(), &make_post("x1", "hello")).unwrap();
let fetched = get_post_by_project_and_slug(db.conn(), "p1", "hello").unwrap();
assert_eq!(fetched.id, "x1");
}
#[test]
fn list_posts_ordered_by_created_at_desc() {
let db = setup();
let mut p1 = make_post("x1", "first");
p1.created_at = 1000;
let mut p2 = make_post("x2", "second");
p2.created_at = 2000;
insert_post(db.conn(), &p1).unwrap();
insert_post(db.conn(), &p2).unwrap();
let list = list_posts_by_project(db.conn(), "p1").unwrap();
assert_eq!(list.len(), 2);
assert_eq!(list[0].id, "x2");
assert_eq!(list[1].id, "x1");
}
#[test]
fn update_post_fields() {
let db = setup();
let mut post = make_post("x1", "hello");
insert_post(db.conn(), &post).unwrap();
post.title = "Updated".into();
post.tags = vec!["new-tag".into()];
post.updated_at = 9999;
update_post(db.conn(), &post).unwrap();
let fetched = get_post_by_id(db.conn(), "x1").unwrap();
assert_eq!(fetched.title, "Updated");
assert_eq!(fetched.tags, vec!["new-tag"]);
assert_eq!(fetched.updated_at, 9999);
}
#[test]
fn update_status() {
let db = setup();
insert_post(db.conn(), &make_post("x1", "hello")).unwrap();
update_post_status(db.conn(), "x1", &PostStatus::Published, 5000).unwrap();
let fetched = get_post_by_id(db.conn(), "x1").unwrap();
assert_eq!(fetched.status, PostStatus::Published);
assert_eq!(fetched.updated_at, 5000);
}
#[test]
fn clear_content_sets_null() {
let db = setup();
insert_post(db.conn(), &make_post("x1", "hello")).unwrap();
clear_post_content(db.conn(), "x1", 5000).unwrap();
let fetched = get_post_by_id(db.conn(), "x1").unwrap();
assert!(fetched.content.is_none());
}
#[test]
fn set_file_path_updates() {
let db = setup();
insert_post(db.conn(), &make_post("x1", "hello")).unwrap();
set_post_file_path(db.conn(), "x1", "posts/2024/01/hello.md", 5000).unwrap();
let fetched = get_post_by_id(db.conn(), "x1").unwrap();
assert_eq!(fetched.file_path, "posts/2024/01/hello.md");
}
#[test]
fn published_snapshot() {
let db = setup();
insert_post(db.conn(), &make_post("x1", "hello")).unwrap();
set_published_snapshot(
db.conn(), "x1", "Pub Title", "Pub Body",
"[\"rust\"]", "[\"tech\"]", Some("Pub Excerpt"), 3000, 3000,
).unwrap();
let fetched = get_post_by_id(db.conn(), "x1").unwrap();
assert_eq!(fetched.published_title.as_deref(), Some("Pub Title"));
assert_eq!(fetched.published_content.as_deref(), Some("Pub Body"));
assert_eq!(fetched.published_at, Some(3000));
}
#[test]
fn delete_removes_post() {
let db = setup();
insert_post(db.conn(), &make_post("x1", "hello")).unwrap();
delete_post(db.conn(), "x1").unwrap();
assert!(get_post_by_id(db.conn(), "x1").is_err());
}
#[test]
fn count_posts() {
let db = setup();
assert_eq!(count_posts_by_project(db.conn(), "p1").unwrap(), 0);
insert_post(db.conn(), &make_post("x1", "a")).unwrap();
insert_post(db.conn(), &make_post("x2", "b")).unwrap();
assert_eq!(count_posts_by_project(db.conn(), "p1").unwrap(), 2);
}
}

View File

@@ -0,0 +1,117 @@
use rusqlite::{params, Connection};
use crate::db::from_row::{post_link_from_row, POST_LINK_COLUMNS};
use crate::model::PostLink;
pub fn insert_post_link(conn: &Connection, link: &PostLink) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO post_links (id, source_post_id, target_post_id, link_text, created_at)
VALUES (?1, ?2, ?3, ?4, ?5)",
params![
link.id,
link.source_post_id,
link.target_post_id,
link.link_text,
link.created_at,
],
)?;
Ok(())
}
pub fn delete_links_by_source(conn: &Connection, source_post_id: &str) -> rusqlite::Result<()> {
conn.execute(
"DELETE FROM post_links WHERE source_post_id = ?1",
params![source_post_id],
)?;
Ok(())
}
pub fn list_links_by_source(
conn: &Connection,
source_post_id: &str,
) -> rusqlite::Result<Vec<PostLink>> {
let mut stmt = conn.prepare(&format!(
"SELECT {POST_LINK_COLUMNS} FROM post_links WHERE source_post_id = ?1 ORDER BY created_at"
))?;
let rows = stmt.query_map(params![source_post_id], post_link_from_row)?;
rows.collect()
}
pub fn list_links_by_target(
conn: &Connection,
target_post_id: &str,
) -> rusqlite::Result<Vec<PostLink>> {
let mut stmt = conn.prepare(&format!(
"SELECT {POST_LINK_COLUMNS} FROM post_links WHERE target_post_id = ?1 ORDER BY created_at"
))?;
let rows = stmt.query_map(params![target_post_id], post_link_from_row)?;
rows.collect()
}
#[cfg(test)]
mod tests {
use super::*;
use crate::db::queries::project::{insert_project, make_test_project};
use crate::db::Database;
fn setup() -> Database {
let db = Database::open_in_memory().unwrap();
db.migrate().unwrap();
insert_project(db.conn(), &make_test_project("p1", "blog")).unwrap();
let c = db.conn();
c.execute(
"INSERT INTO posts (id, project_id, title, slug, status, created_at, updated_at)
VALUES ('a', 'p1', 'A', 'a', 'draft', 1000, 1000)",
[],
).unwrap();
c.execute(
"INSERT INTO posts (id, project_id, title, slug, status, created_at, updated_at)
VALUES ('b', 'p1', 'B', 'b', 'draft', 1000, 1000)",
[],
).unwrap();
c.execute(
"INSERT INTO posts (id, project_id, title, slug, status, created_at, updated_at)
VALUES ('c', 'p1', 'C', 'c', 'draft', 1000, 1000)",
[],
).unwrap();
db
}
fn make_link(id: &str, src: &str, tgt: &str) -> PostLink {
PostLink {
id: id.to_string(),
source_post_id: src.to_string(),
target_post_id: tgt.to_string(),
link_text: Some("see also".into()),
created_at: 1000,
}
}
#[test]
fn insert_and_list_by_source() {
let db = setup();
insert_post_link(db.conn(), &make_link("l1", "a", "b")).unwrap();
insert_post_link(db.conn(), &make_link("l2", "a", "c")).unwrap();
let links = list_links_by_source(db.conn(), "a").unwrap();
assert_eq!(links.len(), 2);
}
#[test]
fn list_by_target() {
let db = setup();
insert_post_link(db.conn(), &make_link("l1", "a", "c")).unwrap();
insert_post_link(db.conn(), &make_link("l2", "b", "c")).unwrap();
let links = list_links_by_target(db.conn(), "c").unwrap();
assert_eq!(links.len(), 2);
}
#[test]
fn delete_by_source() {
let db = setup();
insert_post_link(db.conn(), &make_link("l1", "a", "b")).unwrap();
insert_post_link(db.conn(), &make_link("l2", "a", "c")).unwrap();
delete_links_by_source(db.conn(), "a").unwrap();
let links = list_links_by_source(db.conn(), "a").unwrap();
assert!(links.is_empty());
}
}

View File

@@ -0,0 +1,144 @@
use rusqlite::{params, Connection};
use crate::db::from_row::{post_media_from_row, POST_MEDIA_COLUMNS};
use crate::model::PostMedia;
pub fn link_media(conn: &Connection, pm: &PostMedia) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO post_media (id, project_id, post_id, media_id, sort_order, created_at)
VALUES (?1, ?2, ?3, ?4, ?5, ?6)",
params![
pm.id,
pm.project_id,
pm.post_id,
pm.media_id,
pm.sort_order,
pm.created_at,
],
)?;
Ok(())
}
pub fn unlink_media(conn: &Connection, post_id: &str, media_id: &str) -> rusqlite::Result<()> {
conn.execute(
"DELETE FROM post_media WHERE post_id = ?1 AND media_id = ?2",
params![post_id, media_id],
)?;
Ok(())
}
pub fn list_post_media_by_post(
conn: &Connection,
post_id: &str,
) -> rusqlite::Result<Vec<PostMedia>> {
let mut stmt = conn.prepare(&format!(
"SELECT {POST_MEDIA_COLUMNS} FROM post_media WHERE post_id = ?1 ORDER BY sort_order"
))?;
let rows = stmt.query_map(params![post_id], post_media_from_row)?;
rows.collect()
}
pub fn list_post_media_by_media(
conn: &Connection,
media_id: &str,
) -> rusqlite::Result<Vec<PostMedia>> {
let mut stmt = conn.prepare(&format!(
"SELECT {POST_MEDIA_COLUMNS} FROM post_media WHERE media_id = ?1 ORDER BY created_at"
))?;
let rows = stmt.query_map(params![media_id], post_media_from_row)?;
rows.collect()
}
pub fn update_sort_order(
conn: &Connection,
post_id: &str,
media_id: &str,
sort_order: i32,
) -> rusqlite::Result<()> {
conn.execute(
"UPDATE post_media SET sort_order = ?1 WHERE post_id = ?2 AND media_id = ?3",
params![sort_order, post_id, media_id],
)?;
Ok(())
}
#[cfg(test)]
mod tests {
use super::*;
use crate::db::queries::media::{insert_media, make_test_media};
use crate::db::queries::project::{insert_project, make_test_project};
use crate::db::Database;
fn setup() -> Database {
let db = Database::open_in_memory().unwrap();
db.migrate().unwrap();
insert_project(db.conn(), &make_test_project("p1", "blog")).unwrap();
db.conn()
.execute(
"INSERT INTO posts (id, project_id, title, slug, status, created_at, updated_at)
VALUES ('post1', 'p1', 'Hello', 'hello', 'draft', 1000, 1000)",
[],
)
.unwrap();
insert_media(db.conn(), &make_test_media("m1", "p1")).unwrap();
insert_media(db.conn(), &make_test_media("m2", "p1")).unwrap();
db
}
fn make_pm(id: &str, media_id: &str, order: i32) -> PostMedia {
PostMedia {
id: id.to_string(),
project_id: "p1".to_string(),
post_id: "post1".to_string(),
media_id: media_id.to_string(),
sort_order: order,
created_at: 1000,
}
}
#[test]
fn link_and_list_by_post() {
let db = setup();
link_media(db.conn(), &make_pm("pm1", "m1", 1)).unwrap();
link_media(db.conn(), &make_pm("pm2", "m2", 0)).unwrap();
let list = list_post_media_by_post(db.conn(), "post1").unwrap();
assert_eq!(list.len(), 2);
assert_eq!(list[0].media_id, "m2"); // sort_order 0 first
assert_eq!(list[1].media_id, "m1");
}
#[test]
fn list_by_media() {
let db = setup();
link_media(db.conn(), &make_pm("pm1", "m1", 0)).unwrap();
let list = list_post_media_by_media(db.conn(), "m1").unwrap();
assert_eq!(list.len(), 1);
assert_eq!(list[0].post_id, "post1");
}
#[test]
fn unlink_removes_association() {
let db = setup();
link_media(db.conn(), &make_pm("pm1", "m1", 0)).unwrap();
unlink_media(db.conn(), "post1", "m1").unwrap();
let list = list_post_media_by_post(db.conn(), "post1").unwrap();
assert!(list.is_empty());
}
#[test]
fn update_sort_order_changes_value() {
let db = setup();
link_media(db.conn(), &make_pm("pm1", "m1", 0)).unwrap();
update_sort_order(db.conn(), "post1", "m1", 10).unwrap();
let list = list_post_media_by_post(db.conn(), "post1").unwrap();
assert_eq!(list[0].sort_order, 10);
}
#[test]
fn duplicate_post_media_rejected() {
let db = setup();
link_media(db.conn(), &make_pm("pm1", "m1", 0)).unwrap();
let result = link_media(db.conn(), &make_pm("pm2", "m1", 1));
assert!(result.is_err());
}
}

View File

@@ -0,0 +1,221 @@
use rusqlite::{params, Connection};
use crate::db::from_row::{post_status_to_str, post_translation_from_row, POST_TRANSLATION_COLUMNS};
use crate::model::PostTranslation;
pub fn insert_post_translation(
conn: &Connection,
t: &PostTranslation,
) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO post_translations (
id, project_id, translation_for, language, title, excerpt, content,
status, file_path, checksum, created_at, updated_at, published_at
) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11, ?12, ?13)",
params![
t.id,
t.project_id,
t.translation_for,
t.language,
t.title,
t.excerpt,
t.content,
post_status_to_str(&t.status),
t.file_path,
t.checksum,
t.created_at,
t.updated_at,
t.published_at,
],
)?;
Ok(())
}
pub fn get_post_translation_by_id(
conn: &Connection,
id: &str,
) -> rusqlite::Result<PostTranslation> {
conn.query_row(
&format!("SELECT {POST_TRANSLATION_COLUMNS} FROM post_translations WHERE id = ?1"),
params![id],
post_translation_from_row,
)
}
pub fn get_post_translation_by_post_and_language(
conn: &Connection,
translation_for: &str,
language: &str,
) -> rusqlite::Result<PostTranslation> {
conn.query_row(
&format!(
"SELECT {POST_TRANSLATION_COLUMNS} FROM post_translations
WHERE translation_for = ?1 AND language = ?2"
),
params![translation_for, language],
post_translation_from_row,
)
}
pub fn list_post_translations_by_post(
conn: &Connection,
translation_for: &str,
) -> rusqlite::Result<Vec<PostTranslation>> {
let mut stmt = conn.prepare(&format!(
"SELECT {POST_TRANSLATION_COLUMNS} FROM post_translations
WHERE translation_for = ?1 ORDER BY language"
))?;
let rows = stmt.query_map(params![translation_for], post_translation_from_row)?;
rows.collect()
}
pub fn update_post_translation(
conn: &Connection,
t: &PostTranslation,
) -> rusqlite::Result<()> {
conn.execute(
"UPDATE post_translations SET
title = ?1, excerpt = ?2, content = ?3, status = ?4,
file_path = ?5, checksum = ?6, updated_at = ?7, published_at = ?8
WHERE id = ?9",
params![
t.title,
t.excerpt,
t.content,
post_status_to_str(&t.status),
t.file_path,
t.checksum,
t.updated_at,
t.published_at,
t.id,
],
)?;
Ok(())
}
pub fn delete_post_translation(conn: &Connection, id: &str) -> rusqlite::Result<()> {
conn.execute(
"DELETE FROM post_translations WHERE id = ?1",
params![id],
)?;
Ok(())
}
pub fn delete_all_translations_for_post(
conn: &Connection,
translation_for: &str,
) -> rusqlite::Result<()> {
conn.execute(
"DELETE FROM post_translations WHERE translation_for = ?1",
params![translation_for],
)?;
Ok(())
}
#[cfg(test)]
mod tests {
use super::*;
use crate::db::queries::project::{insert_project, make_test_project};
use crate::db::Database;
use crate::model::PostStatus;
fn setup() -> Database {
let db = Database::open_in_memory().unwrap();
db.migrate().unwrap();
insert_project(db.conn(), &make_test_project("p1", "blog")).unwrap();
// insert a parent post
db.conn()
.execute(
"INSERT INTO posts (id, project_id, title, slug, status, created_at, updated_at)
VALUES ('post1', 'p1', 'Hello', 'hello', 'draft', 1000, 1000)",
[],
)
.unwrap();
db
}
fn make_translation(id: &str, lang: &str) -> PostTranslation {
PostTranslation {
id: id.to_string(),
project_id: "p1".to_string(),
translation_for: "post1".to_string(),
language: lang.to_string(),
title: format!("Title {lang}"),
excerpt: Some("excerpt".into()),
content: Some("body".into()),
status: PostStatus::Draft,
file_path: format!("posts/hello.{lang}.md"),
checksum: None,
created_at: 1000,
updated_at: 2000,
published_at: None,
}
}
#[test]
fn insert_and_get_by_id() {
let db = setup();
insert_post_translation(db.conn(), &make_translation("t1", "de")).unwrap();
let t = get_post_translation_by_id(db.conn(), "t1").unwrap();
assert_eq!(t.language, "de");
assert_eq!(t.title, "Title de");
}
#[test]
fn get_by_post_and_language() {
let db = setup();
insert_post_translation(db.conn(), &make_translation("t1", "de")).unwrap();
let t = get_post_translation_by_post_and_language(db.conn(), "post1", "de").unwrap();
assert_eq!(t.id, "t1");
}
#[test]
fn list_by_post() {
let db = setup();
insert_post_translation(db.conn(), &make_translation("t1", "de")).unwrap();
insert_post_translation(db.conn(), &make_translation("t2", "fr")).unwrap();
let list = list_post_translations_by_post(db.conn(), "post1").unwrap();
assert_eq!(list.len(), 2);
assert_eq!(list[0].language, "de");
assert_eq!(list[1].language, "fr");
}
#[test]
fn update_translation() {
let db = setup();
let mut t = make_translation("t1", "de");
insert_post_translation(db.conn(), &t).unwrap();
t.title = "Neuer Titel".into();
t.updated_at = 9999;
update_post_translation(db.conn(), &t).unwrap();
let fetched = get_post_translation_by_id(db.conn(), "t1").unwrap();
assert_eq!(fetched.title, "Neuer Titel");
assert_eq!(fetched.updated_at, 9999);
}
#[test]
fn delete_single() {
let db = setup();
insert_post_translation(db.conn(), &make_translation("t1", "de")).unwrap();
delete_post_translation(db.conn(), "t1").unwrap();
assert!(get_post_translation_by_id(db.conn(), "t1").is_err());
}
#[test]
fn delete_all_for_post() {
let db = setup();
insert_post_translation(db.conn(), &make_translation("t1", "de")).unwrap();
insert_post_translation(db.conn(), &make_translation("t2", "fr")).unwrap();
delete_all_translations_for_post(db.conn(), "post1").unwrap();
let list = list_post_translations_by_post(db.conn(), "post1").unwrap();
assert!(list.is_empty());
}
#[test]
fn duplicate_language_rejected() {
let db = setup();
insert_post_translation(db.conn(), &make_translation("t1", "de")).unwrap();
let result = insert_post_translation(db.conn(), &make_translation("t2", "de"));
assert!(result.is_err());
}
}

View File

@@ -0,0 +1,215 @@
use rusqlite::{params, Connection};
use crate::db::from_row::{project_from_row, PROJECT_COLUMNS};
use crate::model::Project;
pub fn insert_project(conn: &Connection, project: &Project) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO projects (id, name, slug, description, data_path, is_active, created_at, updated_at)
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8)",
params![
project.id,
project.name,
project.slug,
project.description,
project.data_path,
project.is_active as i64,
project.created_at,
project.updated_at,
],
)?;
Ok(())
}
pub fn get_project_by_id(conn: &Connection, id: &str) -> rusqlite::Result<Project> {
conn.query_row(
&format!("SELECT {PROJECT_COLUMNS} FROM projects WHERE id = ?1"),
params![id],
project_from_row,
)
}
pub fn get_project_by_slug(conn: &Connection, slug: &str) -> rusqlite::Result<Project> {
conn.query_row(
&format!("SELECT {PROJECT_COLUMNS} FROM projects WHERE slug = ?1"),
params![slug],
project_from_row,
)
}
pub fn get_active_project(conn: &Connection) -> rusqlite::Result<Project> {
conn.query_row(
&format!("SELECT {PROJECT_COLUMNS} FROM projects WHERE is_active = 1 LIMIT 1"),
[],
project_from_row,
)
}
pub fn set_active_project(conn: &Connection, id: &str) -> rusqlite::Result<()> {
conn.execute("UPDATE projects SET is_active = 0 WHERE is_active = 1", [])?;
conn.execute(
"UPDATE projects SET is_active = 1 WHERE id = ?1",
params![id],
)?;
Ok(())
}
pub fn list_projects(conn: &Connection) -> rusqlite::Result<Vec<Project>> {
let mut stmt = conn.prepare(&format!(
"SELECT {PROJECT_COLUMNS} FROM projects ORDER BY name"
))?;
let rows = stmt.query_map([], project_from_row)?;
rows.collect()
}
pub fn update_project(conn: &Connection, project: &Project) -> rusqlite::Result<()> {
conn.execute(
"UPDATE projects SET name = ?1, slug = ?2, description = ?3, data_path = ?4,
is_active = ?5, updated_at = ?6
WHERE id = ?7",
params![
project.name,
project.slug,
project.description,
project.data_path,
project.is_active as i64,
project.updated_at,
project.id,
],
)?;
Ok(())
}
pub fn delete_project(conn: &Connection, id: &str) -> rusqlite::Result<()> {
conn.execute("DELETE FROM projects WHERE id = ?1", params![id])?;
Ok(())
}
/// Test helper: create a minimal Project value (available to sibling test modules).
#[cfg(test)]
pub fn make_test_project(id: &str, slug: &str) -> Project {
Project {
id: id.to_string(),
name: format!("Project {id}"),
slug: slug.to_string(),
description: Some("A test project".into()),
data_path: Some("/data".into()),
is_active: false,
created_at: 1000,
updated_at: 2000,
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::db::Database;
fn setup() -> Database {
let db = Database::open_in_memory().unwrap();
db.migrate().unwrap();
db
}
fn make_project(id: &str, slug: &str) -> Project {
Project {
id: id.to_string(),
name: format!("Project {id}"),
slug: slug.to_string(),
description: Some("A test project".into()),
data_path: Some("/data".into()),
is_active: false,
created_at: 1000,
updated_at: 2000,
}
}
#[test]
fn insert_and_get_by_id() {
let db = setup();
let p = make_project("p1", "blog");
insert_project(db.conn(), &p).unwrap();
let fetched = get_project_by_id(db.conn(), "p1").unwrap();
assert_eq!(fetched.id, "p1");
assert_eq!(fetched.name, "Project p1");
assert_eq!(fetched.slug, "blog");
assert_eq!(fetched.description.as_deref(), Some("A test project"));
assert_eq!(fetched.data_path.as_deref(), Some("/data"));
assert!(!fetched.is_active);
assert_eq!(fetched.created_at, 1000);
assert_eq!(fetched.updated_at, 2000);
}
#[test]
fn get_by_slug() {
let db = setup();
insert_project(db.conn(), &make_project("p1", "my-blog")).unwrap();
let fetched = get_project_by_slug(db.conn(), "my-blog").unwrap();
assert_eq!(fetched.id, "p1");
}
#[test]
fn active_project_flow() {
let db = setup();
insert_project(db.conn(), &make_project("p1", "blog1")).unwrap();
insert_project(db.conn(), &make_project("p2", "blog2")).unwrap();
set_active_project(db.conn(), "p1").unwrap();
let active = get_active_project(db.conn()).unwrap();
assert_eq!(active.id, "p1");
set_active_project(db.conn(), "p2").unwrap();
let active = get_active_project(db.conn()).unwrap();
assert_eq!(active.id, "p2");
let p1 = get_project_by_id(db.conn(), "p1").unwrap();
assert!(!p1.is_active);
}
#[test]
fn list_projects_ordered() {
let db = setup();
let mut p2 = make_project("p2", "zebra");
p2.name = "Zebra".into();
let mut p1 = make_project("p1", "alpha");
p1.name = "Alpha".into();
insert_project(db.conn(), &p2).unwrap();
insert_project(db.conn(), &p1).unwrap();
let list = list_projects(db.conn()).unwrap();
assert_eq!(list.len(), 2);
assert_eq!(list[0].name, "Alpha");
assert_eq!(list[1].name, "Zebra");
}
#[test]
fn update_project_fields() {
let db = setup();
let mut p = make_project("p1", "blog");
insert_project(db.conn(), &p).unwrap();
p.name = "Updated".into();
p.description = None;
p.updated_at = 9999;
update_project(db.conn(), &p).unwrap();
let fetched = get_project_by_id(db.conn(), "p1").unwrap();
assert_eq!(fetched.name, "Updated");
assert!(fetched.description.is_none());
assert_eq!(fetched.updated_at, 9999);
}
#[test]
fn delete_project_removes_row() {
let db = setup();
insert_project(db.conn(), &make_project("p1", "blog")).unwrap();
delete_project(db.conn(), "p1").unwrap();
let result = get_project_by_id(db.conn(), "p1");
assert!(result.is_err());
}
#[test]
fn duplicate_slug_rejected() {
let db = setup();
insert_project(db.conn(), &make_project("p1", "blog")).unwrap();
let result = insert_project(db.conn(), &make_project("p2", "blog"));
assert!(result.is_err());
}
}

View File

@@ -0,0 +1,194 @@
use rusqlite::{params, Connection};
use crate::db::from_row::{
script_from_row, script_kind_to_str, script_status_to_str, SCRIPT_COLUMNS,
};
use crate::model::Script;
pub fn insert_script(conn: &Connection, s: &Script) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO scripts (
id, project_id, slug, title, kind, entrypoint, enabled, version,
file_path, status, content, created_at, updated_at
) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11, ?12, ?13)",
params![
s.id,
s.project_id,
s.slug,
s.title,
script_kind_to_str(&s.kind),
s.entrypoint,
s.enabled as i64,
s.version,
s.file_path,
script_status_to_str(&s.status),
s.content,
s.created_at,
s.updated_at,
],
)?;
Ok(())
}
pub fn get_script_by_id(conn: &Connection, id: &str) -> rusqlite::Result<Script> {
conn.query_row(
&format!("SELECT {SCRIPT_COLUMNS} FROM scripts WHERE id = ?1"),
params![id],
script_from_row,
)
}
pub fn get_script_by_slug(
conn: &Connection,
project_id: &str,
slug: &str,
) -> rusqlite::Result<Script> {
conn.query_row(
&format!(
"SELECT {SCRIPT_COLUMNS} FROM scripts WHERE project_id = ?1 AND slug = ?2"
),
params![project_id, slug],
script_from_row,
)
}
pub fn list_scripts_by_project(
conn: &Connection,
project_id: &str,
) -> rusqlite::Result<Vec<Script>> {
let mut stmt = conn.prepare(&format!(
"SELECT {SCRIPT_COLUMNS} FROM scripts WHERE project_id = ?1 ORDER BY title"
))?;
let rows = stmt.query_map(params![project_id], script_from_row)?;
rows.collect()
}
pub fn update_script(conn: &Connection, s: &Script) -> rusqlite::Result<()> {
conn.execute(
"UPDATE scripts SET
slug = ?1, title = ?2, kind = ?3, entrypoint = ?4, enabled = ?5,
version = ?6, file_path = ?7, status = ?8, content = ?9, updated_at = ?10
WHERE id = ?11",
params![
s.slug,
s.title,
script_kind_to_str(&s.kind),
s.entrypoint,
s.enabled as i64,
s.version,
s.file_path,
script_status_to_str(&s.status),
s.content,
s.updated_at,
s.id,
],
)?;
Ok(())
}
pub fn delete_script(conn: &Connection, id: &str) -> rusqlite::Result<()> {
conn.execute("DELETE FROM scripts WHERE id = ?1", params![id])?;
Ok(())
}
#[cfg(test)]
mod tests {
use super::*;
use crate::db::queries::project::{insert_project, make_test_project};
use crate::db::Database;
use crate::model::{ScriptKind, ScriptStatus};
fn setup() -> Database {
let db = Database::open_in_memory().unwrap();
db.migrate().unwrap();
insert_project(db.conn(), &make_test_project("p1", "blog")).unwrap();
db
}
fn make_script(id: &str, slug: &str) -> Script {
Script {
id: id.to_string(),
project_id: "p1".to_string(),
slug: slug.to_string(),
title: format!("Script {slug}"),
kind: ScriptKind::Macro,
entrypoint: "render".to_string(),
enabled: true,
version: 1,
file_path: format!("scripts/{slug}.lua"),
status: ScriptStatus::Published,
content: Some("return html".into()),
created_at: 1000,
updated_at: 2000,
}
}
#[test]
fn insert_and_get_by_id() {
let db = setup();
insert_script(db.conn(), &make_script("s1", "gallery")).unwrap();
let s = get_script_by_id(db.conn(), "s1").unwrap();
assert_eq!(s.slug, "gallery");
assert_eq!(s.kind, ScriptKind::Macro);
assert_eq!(s.entrypoint, "render");
assert!(s.enabled);
}
#[test]
fn get_by_slug() {
let db = setup();
insert_script(db.conn(), &make_script("s1", "gallery")).unwrap();
let s = get_script_by_slug(db.conn(), "p1", "gallery").unwrap();
assert_eq!(s.id, "s1");
}
#[test]
fn list_by_project() {
let db = setup();
let mut s1 = make_script("s1", "zebra");
s1.title = "Zebra".into();
let mut s2 = make_script("s2", "alpha");
s2.title = "Alpha".into();
insert_script(db.conn(), &s1).unwrap();
insert_script(db.conn(), &s2).unwrap();
let list = list_scripts_by_project(db.conn(), "p1").unwrap();
assert_eq!(list.len(), 2);
assert_eq!(list[0].title, "Alpha");
}
#[test]
fn update_script_fields() {
let db = setup();
let mut s = make_script("s1", "gallery");
insert_script(db.conn(), &s).unwrap();
s.kind = ScriptKind::Utility;
s.enabled = false;
s.version = 3;
s.status = ScriptStatus::Draft;
s.content = None;
s.updated_at = 9999;
update_script(db.conn(), &s).unwrap();
let fetched = get_script_by_id(db.conn(), "s1").unwrap();
assert_eq!(fetched.kind, ScriptKind::Utility);
assert!(!fetched.enabled);
assert_eq!(fetched.version, 3);
assert_eq!(fetched.status, ScriptStatus::Draft);
assert!(fetched.content.is_none());
}
#[test]
fn delete_removes_script() {
let db = setup();
insert_script(db.conn(), &make_script("s1", "gallery")).unwrap();
delete_script(db.conn(), "s1").unwrap();
assert!(get_script_by_id(db.conn(), "s1").is_err());
}
#[test]
fn duplicate_slug_rejected() {
let db = setup();
insert_script(db.conn(), &make_script("s1", "gallery")).unwrap();
let result = insert_script(db.conn(), &make_script("s2", "gallery"));
assert!(result.is_err());
}
}

View File

@@ -0,0 +1,83 @@
use rusqlite::{params, Connection};
use crate::db::from_row::{setting_from_row, SETTING_COLUMNS};
use crate::model::Setting;
pub fn get_setting_by_key(conn: &Connection, key: &str) -> rusqlite::Result<Setting> {
conn.query_row(
&format!("SELECT {SETTING_COLUMNS} FROM settings WHERE key = ?1"),
params![key],
setting_from_row,
)
}
pub fn set_setting_value(
conn: &Connection,
key: &str,
value: &str,
updated_at: i64,
) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO settings (key, value, updated_at) VALUES (?1, ?2, ?3)
ON CONFLICT(key) DO UPDATE SET value = excluded.value, updated_at = excluded.updated_at",
params![key, value, updated_at],
)?;
Ok(())
}
pub fn list_all_settings(conn: &Connection) -> rusqlite::Result<Vec<Setting>> {
let mut stmt = conn.prepare(&format!(
"SELECT {SETTING_COLUMNS} FROM settings ORDER BY key"
))?;
let rows = stmt.query_map([], setting_from_row)?;
rows.collect()
}
#[cfg(test)]
mod tests {
use super::*;
use crate::db::Database;
fn setup() -> Database {
let db = Database::open_in_memory().unwrap();
db.migrate().unwrap();
db
}
#[test]
fn set_and_get() {
let db = setup();
set_setting_value(db.conn(), "theme", "dark", 1000).unwrap();
let s = get_setting_by_key(db.conn(), "theme").unwrap();
assert_eq!(s.key, "theme");
assert_eq!(s.value, "dark");
assert_eq!(s.updated_at, 1000);
}
#[test]
fn upsert_updates_existing() {
let db = setup();
set_setting_value(db.conn(), "theme", "dark", 1000).unwrap();
set_setting_value(db.conn(), "theme", "light", 2000).unwrap();
let s = get_setting_by_key(db.conn(), "theme").unwrap();
assert_eq!(s.value, "light");
assert_eq!(s.updated_at, 2000);
}
#[test]
fn list_all() {
let db = setup();
set_setting_value(db.conn(), "b_key", "val_b", 1000).unwrap();
set_setting_value(db.conn(), "a_key", "val_a", 1000).unwrap();
let list = list_all_settings(db.conn()).unwrap();
assert_eq!(list.len(), 2);
assert_eq!(list[0].key, "a_key");
assert_eq!(list[1].key, "b_key");
}
#[test]
fn get_nonexistent_returns_error() {
let db = setup();
assert!(get_setting_by_key(db.conn(), "nope").is_err());
}
}

View File

@@ -0,0 +1,157 @@
use rusqlite::{params, Connection};
use crate::db::from_row::{tag_from_row, TAG_COLUMNS};
use crate::model::Tag;
pub fn insert_tag(conn: &Connection, tag: &Tag) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO tags (id, project_id, name, color, post_template_slug, created_at, updated_at)
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7)",
params![
tag.id,
tag.project_id,
tag.name,
tag.color,
tag.post_template_slug,
tag.created_at,
tag.updated_at,
],
)?;
Ok(())
}
pub fn get_tag_by_id(conn: &Connection, id: &str) -> rusqlite::Result<Tag> {
conn.query_row(
&format!("SELECT {TAG_COLUMNS} FROM tags WHERE id = ?1"),
params![id],
tag_from_row,
)
}
pub fn get_tag_by_project_and_name(
conn: &Connection,
project_id: &str,
name: &str,
) -> rusqlite::Result<Tag> {
conn.query_row(
&format!(
"SELECT {TAG_COLUMNS} FROM tags WHERE project_id = ?1 AND LOWER(name) = LOWER(?2)"
),
params![project_id, name],
tag_from_row,
)
}
pub fn list_tags_by_project(conn: &Connection, project_id: &str) -> rusqlite::Result<Vec<Tag>> {
let mut stmt = conn.prepare(&format!(
"SELECT {TAG_COLUMNS} FROM tags WHERE project_id = ?1 ORDER BY name"
))?;
let rows = stmt.query_map(params![project_id], tag_from_row)?;
rows.collect()
}
pub fn update_tag(conn: &Connection, tag: &Tag) -> rusqlite::Result<()> {
conn.execute(
"UPDATE tags SET name = ?1, color = ?2, post_template_slug = ?3, updated_at = ?4
WHERE id = ?5",
params![
tag.name,
tag.color,
tag.post_template_slug,
tag.updated_at,
tag.id,
],
)?;
Ok(())
}
pub fn delete_tag(conn: &Connection, id: &str) -> rusqlite::Result<()> {
conn.execute("DELETE FROM tags WHERE id = ?1", params![id])?;
Ok(())
}
#[cfg(test)]
mod tests {
use super::*;
use crate::db::queries::project::{insert_project, make_test_project};
use crate::db::Database;
fn setup() -> Database {
let db = Database::open_in_memory().unwrap();
db.migrate().unwrap();
insert_project(db.conn(), &make_test_project("p1", "blog")).unwrap();
db
}
fn make_tag(id: &str, name: &str) -> Tag {
Tag {
id: id.to_string(),
project_id: "p1".to_string(),
name: name.to_string(),
color: Some("#ff0000".into()),
post_template_slug: None,
created_at: 1000,
updated_at: 2000,
}
}
#[test]
fn insert_and_get_by_id() {
let db = setup();
insert_tag(db.conn(), &make_tag("t1", "rust")).unwrap();
let tag = get_tag_by_id(db.conn(), "t1").unwrap();
assert_eq!(tag.name, "rust");
assert_eq!(tag.color.as_deref(), Some("#ff0000"));
}
#[test]
fn get_by_project_and_name_case_insensitive() {
let db = setup();
insert_tag(db.conn(), &make_tag("t1", "Rust")).unwrap();
let tag = get_tag_by_project_and_name(db.conn(), "p1", "rust").unwrap();
assert_eq!(tag.id, "t1");
let tag = get_tag_by_project_and_name(db.conn(), "p1", "RUST").unwrap();
assert_eq!(tag.id, "t1");
}
#[test]
fn list_ordered_by_name() {
let db = setup();
insert_tag(db.conn(), &make_tag("t1", "zebra")).unwrap();
insert_tag(db.conn(), &make_tag("t2", "alpha")).unwrap();
let list = list_tags_by_project(db.conn(), "p1").unwrap();
assert_eq!(list.len(), 2);
assert_eq!(list[0].name, "alpha");
assert_eq!(list[1].name, "zebra");
}
#[test]
fn update_tag_fields() {
let db = setup();
let mut tag = make_tag("t1", "rust");
insert_tag(db.conn(), &tag).unwrap();
tag.name = "go".into();
tag.color = None;
tag.updated_at = 9999;
update_tag(db.conn(), &tag).unwrap();
let fetched = get_tag_by_id(db.conn(), "t1").unwrap();
assert_eq!(fetched.name, "go");
assert!(fetched.color.is_none());
}
#[test]
fn delete_tag_removes_row() {
let db = setup();
insert_tag(db.conn(), &make_tag("t1", "rust")).unwrap();
delete_tag(db.conn(), "t1").unwrap();
assert!(get_tag_by_id(db.conn(), "t1").is_err());
}
#[test]
fn duplicate_name_rejected() {
let db = setup();
insert_tag(db.conn(), &make_tag("t1", "rust")).unwrap();
let result = insert_tag(db.conn(), &make_tag("t2", "rust"));
assert!(result.is_err());
}
}

View File

@@ -0,0 +1,190 @@
use rusqlite::{params, Connection};
use crate::db::from_row::{
template_from_row, template_kind_to_str, template_status_to_str, TEMPLATE_COLUMNS,
};
use crate::model::Template;
pub fn insert_template(conn: &Connection, t: &Template) -> rusqlite::Result<()> {
conn.execute(
"INSERT INTO templates (
id, project_id, slug, title, kind, enabled, version,
file_path, status, content, created_at, updated_at
) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11, ?12)",
params![
t.id,
t.project_id,
t.slug,
t.title,
template_kind_to_str(&t.kind),
t.enabled as i64,
t.version,
t.file_path,
template_status_to_str(&t.status),
t.content,
t.created_at,
t.updated_at,
],
)?;
Ok(())
}
pub fn get_template_by_id(conn: &Connection, id: &str) -> rusqlite::Result<Template> {
conn.query_row(
&format!("SELECT {TEMPLATE_COLUMNS} FROM templates WHERE id = ?1"),
params![id],
template_from_row,
)
}
pub fn get_template_by_slug(
conn: &Connection,
project_id: &str,
slug: &str,
) -> rusqlite::Result<Template> {
conn.query_row(
&format!(
"SELECT {TEMPLATE_COLUMNS} FROM templates WHERE project_id = ?1 AND slug = ?2"
),
params![project_id, slug],
template_from_row,
)
}
pub fn list_templates_by_project(
conn: &Connection,
project_id: &str,
) -> rusqlite::Result<Vec<Template>> {
let mut stmt = conn.prepare(&format!(
"SELECT {TEMPLATE_COLUMNS} FROM templates WHERE project_id = ?1 ORDER BY title"
))?;
let rows = stmt.query_map(params![project_id], template_from_row)?;
rows.collect()
}
pub fn update_template(conn: &Connection, t: &Template) -> rusqlite::Result<()> {
conn.execute(
"UPDATE templates SET
slug = ?1, title = ?2, kind = ?3, enabled = ?4, version = ?5,
file_path = ?6, status = ?7, content = ?8, updated_at = ?9
WHERE id = ?10",
params![
t.slug,
t.title,
template_kind_to_str(&t.kind),
t.enabled as i64,
t.version,
t.file_path,
template_status_to_str(&t.status),
t.content,
t.updated_at,
t.id,
],
)?;
Ok(())
}
pub fn delete_template(conn: &Connection, id: &str) -> rusqlite::Result<()> {
conn.execute("DELETE FROM templates WHERE id = ?1", params![id])?;
Ok(())
}
#[cfg(test)]
mod tests {
use super::*;
use crate::db::queries::project::{insert_project, make_test_project};
use crate::db::Database;
use crate::model::{TemplateKind, TemplateStatus};
fn setup() -> Database {
let db = Database::open_in_memory().unwrap();
db.migrate().unwrap();
insert_project(db.conn(), &make_test_project("p1", "blog")).unwrap();
db
}
fn make_tpl(id: &str, slug: &str) -> Template {
Template {
id: id.to_string(),
project_id: "p1".to_string(),
slug: slug.to_string(),
title: format!("Template {slug}"),
kind: TemplateKind::Post,
enabled: true,
version: 1,
file_path: format!("templates/{slug}.liquid"),
status: TemplateStatus::Published,
content: Some("html".into()),
created_at: 1000,
updated_at: 2000,
}
}
#[test]
fn insert_and_get_by_id() {
let db = setup();
insert_template(db.conn(), &make_tpl("t1", "default")).unwrap();
let t = get_template_by_id(db.conn(), "t1").unwrap();
assert_eq!(t.slug, "default");
assert_eq!(t.kind, TemplateKind::Post);
assert!(t.enabled);
}
#[test]
fn get_by_slug() {
let db = setup();
insert_template(db.conn(), &make_tpl("t1", "default")).unwrap();
let t = get_template_by_slug(db.conn(), "p1", "default").unwrap();
assert_eq!(t.id, "t1");
}
#[test]
fn list_by_project() {
let db = setup();
let mut t1 = make_tpl("t1", "zebra");
t1.title = "Zebra".into();
let mut t2 = make_tpl("t2", "alpha");
t2.title = "Alpha".into();
insert_template(db.conn(), &t1).unwrap();
insert_template(db.conn(), &t2).unwrap();
let list = list_templates_by_project(db.conn(), "p1").unwrap();
assert_eq!(list.len(), 2);
assert_eq!(list[0].title, "Alpha");
}
#[test]
fn update_template_fields() {
let db = setup();
let mut t = make_tpl("t1", "default");
insert_template(db.conn(), &t).unwrap();
t.kind = TemplateKind::List;
t.enabled = false;
t.version = 5;
t.status = TemplateStatus::Draft;
t.content = None;
t.updated_at = 9999;
update_template(db.conn(), &t).unwrap();
let fetched = get_template_by_id(db.conn(), "t1").unwrap();
assert_eq!(fetched.kind, TemplateKind::List);
assert!(!fetched.enabled);
assert_eq!(fetched.version, 5);
assert_eq!(fetched.status, TemplateStatus::Draft);
assert!(fetched.content.is_none());
}
#[test]
fn delete_removes_template() {
let db = setup();
insert_template(db.conn(), &make_tpl("t1", "default")).unwrap();
delete_template(db.conn(), "t1").unwrap();
assert!(get_template_by_id(db.conn(), "t1").is_err());
}
#[test]
fn duplicate_slug_rejected() {
let db = setup();
insert_template(db.conn(), &make_tpl("t1", "default")).unwrap();
let result = insert_template(db.conn(), &make_tpl("t2", "default"));
assert!(result.is_err());
}
}