sqlite_graphrag/commands/
unlink.rs

1use crate::cli::RelationKind;
2use crate::errors::AppError;
3use crate::i18n::erros;
4use crate::output::{self, OutputFormat};
5use crate::paths::AppPaths;
6use crate::storage::connection::open_rw;
7use crate::storage::entities;
8use serde::Serialize;
9
10#[derive(clap::Args)]
11pub struct UnlinkArgs {
12    /// Entidade de origem. Aceita alias `--source` para compatibilidade com doc bilíngue.
13    #[arg(long, alias = "source")]
14    pub from: String,
15    /// Entidade de destino. Aceita alias `--target` para compatibilidade com doc bilíngue.
16    #[arg(long, alias = "target")]
17    pub to: String,
18    #[arg(long, value_enum)]
19    pub relation: RelationKind,
20    #[arg(long)]
21    pub namespace: Option<String>,
22    #[arg(long, value_enum, default_value = "json")]
23    pub format: OutputFormat,
24    #[arg(long, hide = true, help = "No-op; JSON is always emitted on stdout")]
25    pub json: bool,
26    #[arg(long, env = "SQLITE_GRAPHRAG_DB_PATH")]
27    pub db: Option<String>,
28}
29
30#[derive(Serialize)]
31struct UnlinkResponse {
32    action: String,
33    relationship_id: i64,
34    from_name: String,
35    to_name: String,
36    relation: String,
37    namespace: String,
38    /// Tempo total de execução em milissegundos desde início do handler até serialização.
39    elapsed_ms: u64,
40}
41
42pub fn run(args: UnlinkArgs) -> Result<(), AppError> {
43    let inicio = std::time::Instant::now();
44    let namespace = crate::namespace::resolve_namespace(args.namespace.as_deref())?;
45    let paths = AppPaths::resolve(args.db.as_deref())?;
46
47    if !paths.db.exists() {
48        return Err(AppError::NotFound(erros::banco_nao_encontrado(
49            &paths.db.display().to_string(),
50        )));
51    }
52
53    let relation_str = args.relation.as_str();
54
55    let mut conn = open_rw(&paths.db)?;
56
57    let source_id = entities::find_entity_id(&conn, &namespace, &args.from)?.ok_or_else(|| {
58        AppError::NotFound(erros::entidade_nao_encontrada(&args.from, &namespace))
59    })?;
60    let target_id = entities::find_entity_id(&conn, &namespace, &args.to)?
61        .ok_or_else(|| AppError::NotFound(erros::entidade_nao_encontrada(&args.to, &namespace)))?;
62
63    let rel = entities::find_relationship(&conn, source_id, target_id, relation_str)?.ok_or_else(
64        || {
65            AppError::NotFound(erros::relacionamento_nao_encontrado(
66                &args.from,
67                relation_str,
68                &args.to,
69                &namespace,
70            ))
71        },
72    )?;
73
74    let tx = conn.transaction_with_behavior(rusqlite::TransactionBehavior::Immediate)?;
75    entities::delete_relationship_by_id(&tx, rel.id)?;
76    entities::recalculate_degree(&tx, source_id)?;
77    entities::recalculate_degree(&tx, target_id)?;
78    tx.commit()?;
79
80    let response = UnlinkResponse {
81        action: "deleted".to_string(),
82        relationship_id: rel.id,
83        from_name: args.from.clone(),
84        to_name: args.to.clone(),
85        relation: relation_str.to_string(),
86        namespace: namespace.clone(),
87        elapsed_ms: inicio.elapsed().as_millis() as u64,
88    };
89
90    match args.format {
91        OutputFormat::Json => output::emit_json(&response)?,
92        OutputFormat::Text | OutputFormat::Markdown => {
93            output::emit_text(&format!(
94                "deleted: {} --[{}]--> {} [{}]",
95                response.from_name, response.relation, response.to_name, response.namespace
96            ));
97        }
98    }
99
100    Ok(())
101}
102
103#[cfg(test)]
104mod testes {
105    use super::*;
106    use crate::cli::RelationKind;
107
108    #[test]
109    fn unlink_response_serializa_todos_campos() {
110        let resp = UnlinkResponse {
111            action: "deleted".to_string(),
112            relationship_id: 99,
113            from_name: "entidade-a".to_string(),
114            to_name: "entidade-b".to_string(),
115            relation: "uses".to_string(),
116            namespace: "global".to_string(),
117            elapsed_ms: 5,
118        };
119        let json = serde_json::to_value(&resp).expect("serialização falhou");
120        assert_eq!(json["action"], "deleted");
121        assert_eq!(json["relationship_id"], 99i64);
122        assert_eq!(json["from_name"], "entidade-a");
123        assert_eq!(json["to_name"], "entidade-b");
124        assert_eq!(json["relation"], "uses");
125        assert_eq!(json["namespace"], "global");
126        assert_eq!(json["elapsed_ms"], 5u64);
127    }
128
129    #[test]
130    fn unlink_args_relation_kind_as_str_correto() {
131        assert_eq!(RelationKind::Uses.as_str(), "uses");
132        assert_eq!(RelationKind::DependsOn.as_str(), "depends_on");
133        assert_eq!(RelationKind::AppliesTo.as_str(), "applies_to");
134        assert_eq!(RelationKind::Causes.as_str(), "causes");
135        assert_eq!(RelationKind::Fixes.as_str(), "fixes");
136    }
137
138    #[test]
139    fn unlink_response_action_deve_ser_deleted() {
140        let resp = UnlinkResponse {
141            action: "deleted".to_string(),
142            relationship_id: 1,
143            from_name: "a".to_string(),
144            to_name: "b".to_string(),
145            relation: "related".to_string(),
146            namespace: "global".to_string(),
147            elapsed_ms: 0,
148        };
149        let json = serde_json::to_value(&resp).expect("serialização falhou");
150        assert_eq!(
151            json["action"], "deleted",
152            "ação de unlink deve sempre ser 'deleted'"
153        );
154    }
155
156    #[test]
157    fn unlink_response_relationship_id_positivo() {
158        let resp = UnlinkResponse {
159            action: "deleted".to_string(),
160            relationship_id: 42,
161            from_name: "origem".to_string(),
162            to_name: "destino".to_string(),
163            relation: "supports".to_string(),
164            namespace: "projeto".to_string(),
165            elapsed_ms: 3,
166        };
167        let json = serde_json::to_value(&resp).expect("serialização falhou");
168        assert!(
169            json["relationship_id"].as_i64().unwrap() > 0,
170            "relationship_id deve ser positivo após unlink"
171        );
172    }
173}
sqlite_graphrag/commands/unlink.rs

sqlite_graphrag/commands/
unlink.rs
