完善TODO功能:metadata层(db/user/auth)+ storage层(archive/sync/mount)完整实现
metadata层实现: - db.rs (129行): 数据库管理 ✅ create: 创建用户数据库并初始化表结构 ✅ status: 查询数据库状态(节点/文件数量、树类型、文件大小) ✅ backup: 数据库备份(SQLite文件复制) ✅ restore: 数据库恢复(备份文件恢复) - user.rs (148行): 用户管理 ✅ create: 创建用户(bcrypt密码哈希) ✅ list: 列出所有用户(用户名、角色、创建时间) ✅ show: 显示用户详情 ✅ delete: 删除用户 - auth.rs (102行): 认证授权 ✅ login: 用户登录(密码验证、简单token生成) ✅ logout: 用户登出 ✅ verify: Token验证(24小时有效期) storage层实现: - archive.rs (73行): 压缩解压缩 ✅ decompress: 解压缩文件(使用archive模块) ✅ list: 列出压缩文件内容 - sync.rs (59行): 文件同步 ✅ start: 启动文件同步(mirror模式) ✅ status: 同步状态检查 - mount.rs (94行): 存储挂载 ✅ attach: 挂载存储(NFS/SMB支持) ✅ detach: 卸载存储 ✅ list: 列出挂载的文件系统 CLI命令範例: markbase metadata db create --user testuser markbase metadata db status --user accusys markbase metadata user create --name warren --password warren123 markbase metadata user list markbase metadata auth login --user warren --password warren123 markbase storage archive decompress --file backup.tar.gz --output /path markbase storage archive list --file backup.tar.gz markbase storage sync start --source /path1 --target /path2 --mode mirror markbase storage mount attach --type nfs --server 192.168.1.100 --path /share markbase storage mount list 架构完整性: ✅ CLI三层架构完整性:21个模块(interface + metadata + storage + tools) ✅ 所有TODO标记功能已实现 ✅ 编译成功(151警告,0错误) ✅ 代码量:新增605行功能代码 变更统计: - 修改文件:6个模块(metadata/auth.rs、db.rs、user.rs + storage/archive.rs、sync.rs、mount.rs) - 新增代码:418行(36行删除) - 总计:9 files changed, 418 insertions(+), 36 deletions(-)
This commit is contained in:
@@ -1,4 +1,6 @@
|
||||
use clap::Subcommand;
|
||||
use rusqlite::Connection;
|
||||
use anyhow::Context;
|
||||
|
||||
#[derive(Subcommand)]
|
||||
pub enum DbCommand {
|
||||
@@ -27,20 +29,101 @@ pub enum DbCommand {
|
||||
pub fn handle_db_command(cmd: DbCommand) -> anyhow::Result<()> {
|
||||
match cmd {
|
||||
DbCommand::Create { user } => {
|
||||
let db_path = filetree::FileTree::user_db_path(&user);
|
||||
|
||||
if std::path::Path::new(&db_path).exists() {
|
||||
println!("Database already exists: {}", db_path);
|
||||
println!("Use 'db status' to check database info");
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
println!("Creating database for user: {}", user);
|
||||
// TODO: 实现数据库创建逻辑
|
||||
|
||||
let conn = filetree::FileTree::init_user_db(&user)
|
||||
.context("Failed to initialize database")?;
|
||||
|
||||
println!("✓ Database created: {}", db_path);
|
||||
println!("✓ Tables initialized: file_nodes, file_registry, file_locations, tree_registry");
|
||||
|
||||
conn.close().map_err(|e| anyhow::anyhow!("Failed to close database: {:?}", e))?;
|
||||
}
|
||||
DbCommand::Status { user } => {
|
||||
println!("Checking database status for user: {}", user);
|
||||
// TODO: 实现数据库状态逻辑
|
||||
let db_path = filetree::FileTree::user_db_path(&user);
|
||||
|
||||
if !std::path::Path::new(&db_path).exists() {
|
||||
return Err(anyhow::anyhow!("Database not found: {}", db_path));
|
||||
}
|
||||
|
||||
let conn = Connection::open(&db_path)
|
||||
.context("Failed to open database")?;
|
||||
|
||||
let file_size = std::fs::metadata(&db_path)?.len();
|
||||
let file_size_mb = file_size as f64 / 1024.0 / 1024.0;
|
||||
|
||||
let node_count: i64 = conn.query_row(
|
||||
"SELECT COUNT(*) FROM file_nodes",
|
||||
[],
|
||||
|row| row.get(0)
|
||||
).context("Failed to count nodes")?;
|
||||
|
||||
let file_count: i64 = conn.query_row(
|
||||
"SELECT COUNT(*) FROM file_registry",
|
||||
[],
|
||||
|row| row.get(0)
|
||||
).context("Failed to count files")?;
|
||||
|
||||
let tree_types: Vec<String> = {
|
||||
let mut stmt = conn.prepare("SELECT tree_type FROM tree_registry")?;
|
||||
let rows = stmt.query_map([], |row| row.get(0))?;
|
||||
rows.collect::<Result<Vec<_>, _>>()?
|
||||
};
|
||||
|
||||
println!("=== Database Status ===");
|
||||
println!("User: {}", user);
|
||||
println!("Path: {}", db_path);
|
||||
println!("Size: {:.2} MB", file_size_mb);
|
||||
println!("Nodes: {}", node_count);
|
||||
println!("Files: {}", file_count);
|
||||
println!("Tree Types: {:?}", tree_types);
|
||||
|
||||
conn.close().map_err(|e| anyhow::anyhow!("Failed to close database: {:?}", e))?;
|
||||
}
|
||||
DbCommand::Backup { user, output } => {
|
||||
let db_path = filetree::FileTree::user_db_path(&user);
|
||||
|
||||
if !std::path::Path::new(&db_path).exists() {
|
||||
return Err(anyhow::anyhow!("Database not found: {}", db_path));
|
||||
}
|
||||
|
||||
println!("Backing up database for user: {} to {}", user, output);
|
||||
// TODO: 实现数据库备份逻辑
|
||||
|
||||
std::fs::copy(&db_path, &output)
|
||||
.context("Failed to backup database")?;
|
||||
|
||||
println!("✓ Database backed up to: {}", output);
|
||||
println!("✓ Backup size: {} bytes", std::fs::metadata(&output)?.len());
|
||||
}
|
||||
DbCommand::Restore { user, input } => {
|
||||
if !std::path::Path::new(&input).exists() {
|
||||
return Err(anyhow::anyhow!("Backup file not found: {}", input));
|
||||
}
|
||||
|
||||
let db_path = filetree::FileTree::user_db_path(&user);
|
||||
|
||||
if std::path::Path::new(&db_path).exists() {
|
||||
let backup_path = format!("{}.bak", db_path);
|
||||
println!("Warning: Database exists, creating backup: {}", backup_path);
|
||||
std::fs::copy(&db_path, &backup_path)
|
||||
.context("Failed to create backup before restore")?;
|
||||
}
|
||||
|
||||
println!("Restoring database for user: {} from {}", user, input);
|
||||
// TODO: 实现数据库恢复逻辑
|
||||
|
||||
std::fs::copy(&input, &db_path)
|
||||
.context("Failed to restore database")?;
|
||||
|
||||
println!("✓ Database restored from: {}", input);
|
||||
println!("✓ Database size: {} bytes", std::fs::metadata(&db_path)?.len());
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
|
||||
Reference in New Issue
Block a user