489 lines
20 KiB
Rust
489 lines
20 KiB
Rust
//! 项目域 Repo:ProjectRepo / BranchRepo / ReleaseRepo / WorkflowRepo / NodeExecutionRepo
|
|
|
|
use std::sync::Arc;
|
|
|
|
use rusqlite::{params, Connection, OptionalExtension, Row};
|
|
use tokio::sync::Mutex;
|
|
|
|
use df_types::error::Result;
|
|
use df_types::types::ProjectStatus;
|
|
|
|
use crate::db::Database;
|
|
use crate::models::{
|
|
BranchRecord, NodeExecutionRecord, ProjectRecord, ReleaseRecord, WorkflowRecord,
|
|
};
|
|
|
|
use super::impl_repo;
|
|
use super::{normalize_stored_path, now_millis_str, storage_err, validate_column_name};
|
|
|
|
// ============================================================
|
|
// 项目查询入参(F-260621-02 P2/P3 查询维度补全)
|
|
// ============================================================
|
|
|
|
/// 项目列表查询条件(可选字段,全 None = 当前 list_active 全量行为)。
|
|
///
|
|
/// 对齐方案 docs/02-架构设计/专项设计/查询能力补全方案-2026-06-21.md §4.1 统一查询结构。
|
|
/// 项目表无 status 生命周期(P1 跳过),仅软删 deleted_at(由 list_by_query 固定排除)。
|
|
/// 故本结构不含 status,只含 keyword/order_by/limit/offset 四维。
|
|
///
|
|
/// 序列化 derive:命令层经 serde 从前端 IPC 接收(Option 字段缺省 None,旧调用方不传等价全量)。
|
|
#[derive(Debug, Default, Clone, serde::Serialize, serde::Deserialize)]
|
|
pub struct ProjectQuery {
|
|
/// 关键词:name/description LIKE %kw%(trim 后空字符串视为无关键词)
|
|
#[serde(default)]
|
|
pub keyword: Option<String>,
|
|
/// 排序字段(白名单校验,格式 "col" / "col asc" / "col desc",默认 created_at desc)
|
|
#[serde(default)]
|
|
pub order_by: Option<String>,
|
|
/// 返回上限(钳制 [0,200],None = 不加 LIMIT 全量)
|
|
#[serde(default)]
|
|
pub limit: Option<u32>,
|
|
/// 偏移量(None = 0,配合 limit 分页)
|
|
#[serde(default)]
|
|
pub offset: Option<u32>,
|
|
}
|
|
|
|
/// 解析 order_by 入参为 "col DIR" SQL 片段(列名走白名单校验防注入)。
|
|
///
|
|
/// 接受 "col" / "col asc" / "col desc"(DIR 大小写不敏感)。col 走 `validate_column_name`
|
|
/// 校验(列名不可参数化,必须拼字符串,白名单是唯一防注入手段,对齐 impl_repo! 宏)。
|
|
/// 非法列名返回 Err;合法但无 DIR 默认 DESC(与 list_active 一致)。
|
|
fn build_order_clause(order_by: Option<&str>) -> Result<String> {
|
|
let Some(raw) = order_by else {
|
|
return Ok("created_at DESC".to_string());
|
|
};
|
|
let raw = raw.trim();
|
|
if raw.is_empty() {
|
|
return Ok("created_at DESC".to_string());
|
|
}
|
|
// 拆 "col [dir]",最多 2 段。
|
|
let parts: Vec<&str> = raw.split_whitespace().collect();
|
|
let col = parts[0];
|
|
let dir = parts.get(1).map(|s| s.to_ascii_uppercase());
|
|
// 白名单校验列名(防 SQL 注入:列名拼字符串前必须校验)。
|
|
validate_column_name(col, "projects")?;
|
|
match dir.as_deref() {
|
|
None | Some("DESC") => Ok(format!("{col} DESC")),
|
|
Some("ASC") => Ok(format!("{col} ASC")),
|
|
// 其它非法方向词:拒绝(防 "; DROP TABLE" 之类拼串绕过)。
|
|
Some(other) => Err(df_types::error::Error::Storage(format!(
|
|
"非法排序方向: {other}(仅支持 asc/desc)"
|
|
))),
|
|
}
|
|
}
|
|
|
|
// ============================================================
|
|
// from_row 辅助函数
|
|
// ============================================================
|
|
|
|
fn project_from_row(row: &Row<'_>) -> std::result::Result<ProjectRecord, rusqlite::Error> {
|
|
Ok(ProjectRecord {
|
|
id: row.get("id")?,
|
|
name: row.get("name")?,
|
|
description: row.get("description")?,
|
|
status: {
|
|
let s: String = row.get("status")?;
|
|
ProjectStatus::from_db_str(&s).unwrap_or_default()
|
|
},
|
|
idea_id: row.get("idea_id")?,
|
|
path: row.get("path")?,
|
|
stack: row.get("stack")?,
|
|
created_at: row.get("created_at")?,
|
|
updated_at: row.get("updated_at")?,
|
|
})
|
|
}
|
|
|
|
fn branch_from_row(row: &Row<'_>) -> std::result::Result<BranchRecord, rusqlite::Error> {
|
|
Ok(BranchRecord {
|
|
id: row.get("id")?,
|
|
project_id: row.get("project_id")?,
|
|
task_id: row.get("task_id")?,
|
|
name: row.get("name")?,
|
|
base: row.get("base")?,
|
|
status: row.get("status")?,
|
|
created_at: row.get("created_at")?,
|
|
updated_at: row.get("updated_at")?,
|
|
merged_at: row.get("merged_at")?,
|
|
})
|
|
}
|
|
|
|
fn release_from_row(row: &Row<'_>) -> std::result::Result<ReleaseRecord, rusqlite::Error> {
|
|
Ok(ReleaseRecord {
|
|
id: row.get("id")?,
|
|
project_id: row.get("project_id")?,
|
|
version: row.get("version")?,
|
|
status: row.get("status")?,
|
|
task_ids: row.get("task_ids")?,
|
|
changelog: row.get("changelog")?,
|
|
created_at: row.get("created_at")?,
|
|
released_at: row.get("released_at")?,
|
|
})
|
|
}
|
|
|
|
fn workflow_from_row(row: &Row<'_>) -> std::result::Result<WorkflowRecord, rusqlite::Error> {
|
|
Ok(WorkflowRecord {
|
|
id: row.get("id")?,
|
|
name: row.get("name")?,
|
|
dag_json: row.get("dag_json")?,
|
|
status: row.get("status")?,
|
|
triggered_by: row.get("triggered_by")?,
|
|
project_id: row.get("project_id")?,
|
|
task_id: row.get("task_id")?,
|
|
created_at: row.get("created_at")?,
|
|
completed_at: row.get("completed_at")?,
|
|
})
|
|
}
|
|
|
|
fn node_execution_from_row(
|
|
row: &Row<'_>,
|
|
) -> std::result::Result<NodeExecutionRecord, rusqlite::Error> {
|
|
Ok(NodeExecutionRecord {
|
|
id: row.get("id")?,
|
|
workflow_id: row.get("workflow_id")?,
|
|
node_id: row.get("node_id")?,
|
|
node_type: row.get("node_type")?,
|
|
status: row.get("status")?,
|
|
input_json: row.get("input_json")?,
|
|
output_json: row.get("output_json")?,
|
|
error_message: row.get("error_message")?,
|
|
started_at: row.get("started_at")?,
|
|
completed_at: row.get("completed_at")?,
|
|
})
|
|
}
|
|
|
|
// ============================================================
|
|
// Repo 实现
|
|
// ============================================================
|
|
|
|
impl_repo!(
|
|
/// 项目表 CRUD
|
|
ProjectRepo,
|
|
ProjectRecord,
|
|
"projects",
|
|
from_row => |row| project_from_row(row),
|
|
insert => |conn, rec| {
|
|
conn.execute(
|
|
"INSERT INTO projects (id, name, description, status, idea_id, path, stack, created_at, updated_at)
|
|
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9)",
|
|
params![
|
|
rec.id, rec.name, rec.description, rec.status.as_str(), rec.idea_id,
|
|
rec.path, rec.stack, rec.created_at, rec.updated_at
|
|
],
|
|
)
|
|
},
|
|
update => |conn, rec| {
|
|
conn.execute(
|
|
"UPDATE projects SET name = ?1, description = ?2, status = ?3, idea_id = ?4, path = ?5, stack = ?6, updated_at = ?7 WHERE id = ?8",
|
|
params![
|
|
rec.name, rec.description, rec.status.as_str(), rec.idea_id,
|
|
rec.path, rec.stack, rec.updated_at, rec.id
|
|
],
|
|
)
|
|
}
|
|
);
|
|
|
|
impl ProjectRepo {
|
|
/// 列出未删除项目(deleted_at IS NULL)
|
|
pub async fn list_active(&self) -> Result<Vec<ProjectRecord>> {
|
|
let conn = self.conn.clone();
|
|
tokio::task::spawn_blocking(move || {
|
|
let guard = conn.blocking_lock();
|
|
let mut stmt = guard
|
|
.prepare("SELECT id, name, description, status, idea_id, path, stack, created_at, updated_at FROM projects WHERE deleted_at IS NULL ORDER BY created_at DESC")
|
|
.map_err(storage_err)?;
|
|
let rows = stmt
|
|
.query_map([], |row| project_from_row(row))
|
|
.map_err(storage_err)?;
|
|
let mut results = Vec::new();
|
|
for r in rows {
|
|
results.push(r.map_err(storage_err)?);
|
|
}
|
|
Ok(results)
|
|
})
|
|
.await
|
|
.map_err(storage_err)?
|
|
}
|
|
|
|
/// 按条件查询未删除项目(F-260621-02 P2/P3:关键词搜索 + 排序 + 分页)。
|
|
///
|
|
/// 复用 `KnowledgeRepo::search` 动态 WHERE 拼接模式:按可选字段 if-let 拼 SQL 子句 +
|
|
/// 分支化参数绑定。始终排除 `deleted_at IS NOT NULL`(软删),与 `list_active` 行为对齐。
|
|
///
|
|
/// - `keyword`:name/description `LIKE %kw%`(对齐知识库 title/content LIKE)。
|
|
/// - `order_by`:走白名单 `validate_column_name` 防 SQL 注入(列名不可参数化,必须拼字符串,
|
|
/// 白名单是唯一防注入手段)。默认 `created_at DESC`(与 list_active 等价)。
|
|
/// - `limit`:钳制上限 200(对齐 `KnowledgeEventsRepo::list_recent`),防恶意/失误传超大值。
|
|
/// - `offset`:偏移量,配合 limit 做分页。
|
|
///
|
|
/// 向后兼容:全字段 None 时 SQL 退化为
|
|
/// `SELECT ... WHERE deleted_at IS NULL ORDER BY created_at DESC`
|
|
/// 与 `list_active` 完全等价,零破坏。
|
|
pub async fn list_by_query(&self, q: ProjectQuery) -> Result<Vec<ProjectRecord>> {
|
|
// order_by 白名单校验(列名拼字符串前必须校验,防注入)。
|
|
// 校验后再拼方向:允许 "col" / "col asc" / "col desc"(大小写不敏感)。
|
|
let order_clause = build_order_clause(q.order_by.as_deref())?;
|
|
|
|
let conn = self.conn.clone();
|
|
// limit 钳制:None 不加 LIMIT(全量,对齐 list_active);Some 钳到 [0,200]。
|
|
let limit_i: Option<i64> = q.limit.map(|n| n.min(200) as i64);
|
|
let offset_i: i64 = q.offset.unwrap_or(0) as i64;
|
|
|
|
tokio::task::spawn_blocking(move || {
|
|
let guard = conn.blocking_lock();
|
|
// 动态拼 WHERE + 参数。WHERE 1=1 起点 + deleted_at IS NULL 固定基础条件,
|
|
// 后续可选条件 AND 拼接(对齐 KnowledgeRepo::search 的 if-let 分支模式)。
|
|
let mut sql = String::from(
|
|
"SELECT id, name, description, status, idea_id, path, stack, created_at, updated_at \
|
|
FROM projects WHERE deleted_at IS NULL",
|
|
);
|
|
// keyword 用 ?N 占位两次(name LIKE ? AND description LIKE ? 同 pattern),
|
|
// 对齐 KnowledgeRepo::search 的 `params![pattern, pattern, ...]`。
|
|
let mut params_vec: Vec<Box<dyn rusqlite::ToSql>> = Vec::new();
|
|
if let Some(kw) = &q.keyword {
|
|
let trimmed = kw.trim();
|
|
if !trimmed.is_empty() {
|
|
sql.push_str(" AND (name LIKE ? OR description LIKE ?)");
|
|
let pattern = format!("%{trimmed}%");
|
|
params_vec.push(Box::new(pattern.clone()));
|
|
params_vec.push(Box::new(pattern));
|
|
}
|
|
}
|
|
sql.push_str(" ORDER BY ");
|
|
sql.push_str(&order_clause);
|
|
// 分页:有 limit 才加 LIMIT/OFFSET(全量场景不加,等价 list_active)。
|
|
if let Some(lim) = limit_i {
|
|
sql.push_str(" LIMIT ? OFFSET ?");
|
|
params_vec.push(Box::new(lim));
|
|
params_vec.push(Box::new(offset_i));
|
|
}
|
|
let mut stmt = guard.prepare(&sql).map_err(storage_err)?;
|
|
let param_refs: Vec<&dyn rusqlite::ToSql> =
|
|
params_vec.iter().map(|p| p.as_ref()).collect();
|
|
let rows = stmt
|
|
.query_map(param_refs.as_slice(), |row| project_from_row(row))
|
|
.map_err(storage_err)?;
|
|
let mut results = Vec::new();
|
|
for r in rows {
|
|
results.push(r.map_err(storage_err)?);
|
|
}
|
|
Ok(results)
|
|
})
|
|
.await
|
|
.map_err(storage_err)?
|
|
}
|
|
|
|
/// 软删:标记 deleted_at(进回收站,可恢复)。仅作用于未删项目,返回是否命中。
|
|
pub async fn soft_delete(&self, id: &str) -> Result<bool> {
|
|
let conn = self.conn.clone();
|
|
let id = id.to_owned();
|
|
let now = now_millis_str();
|
|
tokio::task::spawn_blocking(move || {
|
|
let guard = conn.blocking_lock();
|
|
let affected = guard
|
|
.execute(
|
|
"UPDATE projects SET deleted_at = ?1, updated_at = ?1 WHERE id = ?2 AND deleted_at IS NULL",
|
|
params![now, id],
|
|
)
|
|
.map_err(storage_err)?;
|
|
Ok(affected > 0)
|
|
})
|
|
.await
|
|
.map_err(storage_err)?
|
|
}
|
|
|
|
/// 恢复:清 deleted_at(从回收站还原)
|
|
pub async fn restore(&self, id: &str) -> Result<bool> {
|
|
let conn = self.conn.clone();
|
|
let id = id.to_owned();
|
|
let now = now_millis_str();
|
|
tokio::task::spawn_blocking(move || {
|
|
let guard = conn.blocking_lock();
|
|
let affected = guard
|
|
.execute(
|
|
"UPDATE projects SET deleted_at = NULL, updated_at = ?1 WHERE id = ?2 AND deleted_at IS NOT NULL",
|
|
params![now, id],
|
|
)
|
|
.map_err(storage_err)?;
|
|
Ok(affected > 0)
|
|
})
|
|
.await
|
|
.map_err(storage_err)?
|
|
}
|
|
|
|
/// 查找已绑定该规范化路径的项目(排除 exclude_id 自身)。无冲突返回 None。
|
|
///
|
|
/// DRY(R-PD-11):统一 project.rs::find_binding_conflict 与 tool_registry.rs::bind_dir_to_project
|
|
/// 的「列项目→排除自身→按规范化路径比较」逻辑。本 crate 不依赖 df-project,故 norm_path 须由
|
|
/// 调用方先经 `df_project::scan::normalize_path`(内含 canonicalize,防 `C:\a\b` vs `C:/a/b/` 绕过)。
|
|
pub async fn find_path_conflict(
|
|
&self,
|
|
norm_path: &str,
|
|
exclude_id: Option<&str>,
|
|
) -> Result<Option<ProjectRecord>> {
|
|
let projects = self.list_active().await?;
|
|
Ok(projects.into_iter().find(|p| {
|
|
let excluded = exclude_id.is_some_and(|eid| p.id == eid);
|
|
!excluded && p.path.as_ref().is_some_and(|pp| {
|
|
// 路径规范化由调用方完成目标侧;这里为已存项目路径补规范化
|
|
// (历史路径写法可能不规范,统一规范化比较避免误判/漏判)
|
|
normalize_stored_path(pp) == norm_path
|
|
})
|
|
}))
|
|
}
|
|
|
|
/// 列出回收站(deleted_at IS NOT NULL),按更新时间(≈删除时间)降序
|
|
pub async fn list_deleted(&self) -> Result<Vec<ProjectRecord>> {
|
|
let conn = self.conn.clone();
|
|
tokio::task::spawn_blocking(move || {
|
|
let guard = conn.blocking_lock();
|
|
let mut stmt = guard
|
|
.prepare("SELECT id, name, description, status, idea_id, path, stack, created_at, updated_at FROM projects WHERE deleted_at IS NOT NULL ORDER BY updated_at DESC")
|
|
.map_err(storage_err)?;
|
|
let rows = stmt
|
|
.query_map([], |row| project_from_row(row))
|
|
.map_err(storage_err)?;
|
|
let mut results = Vec::new();
|
|
for r in rows {
|
|
results.push(r.map_err(storage_err)?);
|
|
}
|
|
Ok(results)
|
|
})
|
|
.await
|
|
.map_err(storage_err)?
|
|
}
|
|
|
|
/// 彻底删除:事务级联删 branches→releases→tasks→projects(不可恢复)
|
|
///
|
|
/// SQLite 已开 PRAGMA foreign_keys=ON 但表无 ON DELETE CASCADE,ALTER 改不了 FK 约束,
|
|
/// 故应用层级联:子表先于父表删,单事务保证一致性。
|
|
pub async fn purge_with_descendants(&self, id: &str) -> Result<bool> {
|
|
let conn = self.conn.clone();
|
|
let id = id.to_owned();
|
|
tokio::task::spawn_blocking(move || {
|
|
let mut guard = conn.blocking_lock();
|
|
let tx = guard.transaction().map_err(storage_err)?;
|
|
tx.execute("DELETE FROM branches WHERE project_id = ?1", params![id])
|
|
.map_err(storage_err)?;
|
|
tx.execute("DELETE FROM releases WHERE project_id = ?1", params![id])
|
|
.map_err(storage_err)?;
|
|
tx.execute("DELETE FROM tasks WHERE project_id = ?1", params![id])
|
|
.map_err(storage_err)?;
|
|
let affected = tx
|
|
.execute("DELETE FROM projects WHERE id = ?1", params![id])
|
|
.map_err(storage_err)?;
|
|
tx.commit().map_err(storage_err)?;
|
|
Ok(affected > 0)
|
|
})
|
|
.await
|
|
.map_err(storage_err)?
|
|
}
|
|
}
|
|
|
|
impl_repo!(
|
|
/// 分支表 CRUD
|
|
BranchRepo,
|
|
BranchRecord,
|
|
"branches",
|
|
from_row => |row| branch_from_row(row),
|
|
insert => |conn, rec| {
|
|
conn.execute(
|
|
"INSERT INTO branches (id, project_id, task_id, name, base, status, created_at, updated_at, merged_at)
|
|
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9)",
|
|
params![
|
|
rec.id, rec.project_id, rec.task_id, rec.name, rec.base,
|
|
rec.status, rec.created_at, rec.updated_at, rec.merged_at
|
|
],
|
|
)
|
|
},
|
|
update => |conn, rec| {
|
|
conn.execute(
|
|
"UPDATE branches SET project_id = ?1, task_id = ?2, name = ?3, base = ?4, status = ?5, updated_at = ?6, merged_at = ?7 WHERE id = ?8",
|
|
params![
|
|
rec.project_id, rec.task_id, rec.name, rec.base,
|
|
rec.status, rec.updated_at, rec.merged_at, rec.id
|
|
],
|
|
)
|
|
}
|
|
);
|
|
|
|
impl_repo!(
|
|
/// 发布表 CRUD
|
|
ReleaseRepo,
|
|
ReleaseRecord,
|
|
"releases",
|
|
from_row => |row| release_from_row(row),
|
|
insert => |conn, rec| {
|
|
conn.execute(
|
|
"INSERT INTO releases (id, project_id, version, status, task_ids, changelog, created_at, released_at)
|
|
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8)",
|
|
params![
|
|
rec.id, rec.project_id, rec.version, rec.status, rec.task_ids,
|
|
rec.changelog, rec.created_at, rec.released_at
|
|
],
|
|
)
|
|
},
|
|
update => |conn, rec| {
|
|
conn.execute(
|
|
"UPDATE releases SET project_id = ?1, version = ?2, status = ?3, task_ids = ?4, changelog = ?5, released_at = ?6 WHERE id = ?7",
|
|
params![
|
|
rec.project_id, rec.version, rec.status, rec.task_ids,
|
|
rec.changelog, rec.released_at, rec.id
|
|
],
|
|
)
|
|
}
|
|
);
|
|
|
|
impl_repo!(
|
|
/// 工作流执行表 CRUD
|
|
WorkflowRepo,
|
|
WorkflowRecord,
|
|
"workflow_executions",
|
|
from_row => |row| workflow_from_row(row),
|
|
insert => |conn, rec| {
|
|
conn.execute(
|
|
"INSERT INTO workflow_executions (id, name, dag_json, status, triggered_by, project_id, task_id, created_at, completed_at)
|
|
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9)",
|
|
params![
|
|
rec.id, rec.name, rec.dag_json, rec.status, rec.triggered_by,
|
|
rec.project_id, rec.task_id, rec.created_at, rec.completed_at
|
|
],
|
|
)
|
|
},
|
|
update => |conn, rec| {
|
|
conn.execute(
|
|
"UPDATE workflow_executions SET name = ?1, dag_json = ?2, status = ?3, triggered_by = ?4, project_id = ?5, task_id = ?6, completed_at = ?7 WHERE id = ?8",
|
|
params![
|
|
rec.name, rec.dag_json, rec.status, rec.triggered_by,
|
|
rec.project_id, rec.task_id, rec.completed_at, rec.id
|
|
],
|
|
)
|
|
}
|
|
);
|
|
|
|
impl_repo!(
|
|
/// 节点执行表 CRUD
|
|
NodeExecutionRepo,
|
|
NodeExecutionRecord,
|
|
"node_executions",
|
|
from_row => |row| node_execution_from_row(row),
|
|
insert => |conn, rec| {
|
|
conn.execute(
|
|
"INSERT INTO node_executions (id, workflow_id, node_id, node_type, status, input_json, output_json, error_message, started_at, completed_at)
|
|
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10)",
|
|
params![
|
|
rec.id, rec.workflow_id, rec.node_id, rec.node_type, rec.status,
|
|
rec.input_json, rec.output_json, rec.error_message, rec.started_at, rec.completed_at
|
|
],
|
|
)
|
|
},
|
|
update => |conn, rec| {
|
|
conn.execute(
|
|
"UPDATE node_executions SET workflow_id = ?1, node_id = ?2, node_type = ?3, status = ?4, input_json = ?5, output_json = ?6, error_message = ?7, started_at = ?8, completed_at = ?9 WHERE id = ?10",
|
|
params![
|
|
rec.workflow_id, rec.node_id, rec.node_type, rec.status,
|
|
rec.input_json, rec.output_json, rec.error_message, rec.started_at, rec.completed_at, rec.id
|
|
],
|
|
)
|
|
}
|
|
);
|