5d657efbb5
功能： - SMBConfig: 配置结构体 - SMBManager: 管理API - CLI工具：status/list/create/remove命令 验证： - ✅ status命令JSON输出 - ✅ list命令正确显示 - ✅ create命令生成配置指南 下一步： - 用户手动启用SMB服务（需要sudo） - Windows/macOS客户端测试 - Phase 2: 权限控制优化
89 lines
2.5 KiB
Rust
89 lines
2.5 KiB
Rust
use anyhow::Result;
|
|
use std::path::Path;
|
|
use std::process::Command;
|
|
|
|
use crate::config::SMBConfig;
|
|
|
|
pub struct SMBManager {
|
|
config: SMBConfig,
|
|
}
|
|
|
|
impl SMBManager {
|
|
pub fn new(config: SMBConfig) -> Self {
|
|
SMBManager { config }
|
|
}
|
|
|
|
pub fn check_smb_service() -> Result<bool> {
|
|
let output = Command::new("sharing")
|
|
.arg("-l")
|
|
.output()?;
|
|
|
|
let status = String::from_utf8_lossy(&output.stdout);
|
|
Ok(!status.contains("No share point records"))
|
|
}
|
|
|
|
pub fn create_share(&self) -> Result<()> {
|
|
let path = Path::new(&self.config.path);
|
|
if !path.exists() {
|
|
return Err(anyhow::anyhow!("Path does not exist: {}", self.config.path));
|
|
}
|
|
|
|
eprintln!("Creating SMB share '{}' for path: {}", self.config.share_name, self.config.path);
|
|
|
|
let smb_conf_content = self.config.to_smb_conf();
|
|
eprintln!("Generated smb.conf section:\n{}", smb_conf_content);
|
|
|
|
eprintln!("\nTo enable SMB sharing, run:");
|
|
eprintln!("sudo sharing -a \"{}\" -S \"{}\"", self.config.path, self.config.share_name);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn remove_share(&self) -> Result<()> {
|
|
eprintln!("Removing SMB share '{}'...", self.config.share_name);
|
|
|
|
eprintln!("To remove SMB sharing, run:");
|
|
eprintln!("sudo sharing -r \"{}\"", self.config.share_name);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn list_shares() -> Result<Vec<String>> {
|
|
let output = Command::new("sharing")
|
|
.arg("-l")
|
|
.output()?;
|
|
|
|
let status = String::from_utf8_lossy(&output.stdout);
|
|
|
|
if status.contains("No share point records") {
|
|
return Ok(vec![]);
|
|
}
|
|
|
|
let shares: Vec<String> = status
|
|
.lines()
|
|
.filter(|line| line.contains("name:"))
|
|
.map(|line| {
|
|
line.split(":")
|
|
.nth(1)
|
|
.unwrap_or("")
|
|
.trim()
|
|
.to_string()
|
|
})
|
|
.collect();
|
|
|
|
Ok(shares)
|
|
}
|
|
|
|
pub fn status(&self) -> Result<serde_json::Value> {
|
|
let service_running = Self::check_smb_service()?;
|
|
let shares = Self::list_shares()?;
|
|
|
|
Ok(serde_json::json!({
|
|
"service_running": service_running,
|
|
"share_name": self.config.share_name,
|
|
"path": self.config.path,
|
|
"existing_shares": shares,
|
|
"config": self.config,
|
|
}))
|
|
}
|
|
} |