nils-memo-cli 0.3.7

CLI crate for nils-memo-cli in the nils-cli workspace.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

use serde_json::json;

use crate::cli::{OutputMode, SearchField as CliSearchField};
use crate::errors::AppError;
use crate::output::{emit_json_results_with_meta, format_item_id, text};
use crate::storage::Storage;
use crate::storage::repository::QueryState;
use crate::storage::search;

pub fn run(
    storage: &Storage,
    output_mode: OutputMode,
    state: QueryState,
    query: &str,
    fields: &[CliSearchField],
    match_mode: search::SearchMatchMode,
    limit: usize,
) -> Result<(), AppError> {
    let query = query.trim();
    if query.is_empty() {
        return Err(AppError::usage("search requires a non-empty query"));
    }

    let search_fields = map_search_fields(fields);
    let rows = storage.with_connection(|conn| {
        search::search_items(conn, query, state, &search_fields, match_mode, limit)
    })?;

    if output_mode.is_json() {
        let results = rows
            .iter()
            .map(|row| {
                json!({
                    "item_id": format_item_id(row.item_id),
                    "created_at": row.created_at,
                    "score": row.score,
                    "matched_fields": row.matched_fields,
                    "preview": row.preview,
                    "content_type": row.content_type,
                    "validation_status": row.validation_status,
                })
            })
            .collect::<Vec<_>>();
        return emit_json_results_with_meta(
            "memo-cli.search.v1",
            "memo-cli search",
            results,
            None,
            Some(json!({
                "query": query,
                "limit": limit,
                "state": query_state_label(state),
                "fields": search_field_labels(&search_fields),
                "match": search_match_mode_label(match_mode),
            })),
        );
    }

    text::print_search(&rows);

    Ok(())
}

fn query_state_label(state: QueryState) -> &'static str {
    match state {
        QueryState::All => "all",
        QueryState::Pending => "pending",
        QueryState::Enriched => "enriched",
    }
}

fn map_search_fields(fields: &[CliSearchField]) -> Vec<search::SearchField> {
    let mut out = Vec::new();
    let source = if fields.is_empty() {
        &[
            CliSearchField::Raw,
            CliSearchField::Derived,
            CliSearchField::Tags,
        ][..]
    } else {
        fields
    };

    for field in source {
        let mapped = match field {
            CliSearchField::Raw => search::SearchField::Raw,
            CliSearchField::Derived => search::SearchField::Derived,
            CliSearchField::Tags => search::SearchField::Tags,
        };
        if !out.contains(&mapped) {
            out.push(mapped);
        }
    }

    out
}

fn search_field_labels(fields: &[search::SearchField]) -> Vec<&'static str> {
    fields.iter().map(|field| field.label()).collect()
}

fn search_match_mode_label(mode: search::SearchMatchMode) -> &'static str {
    match mode {
        search::SearchMatchMode::Fts => "fts",
        search::SearchMatchMode::Prefix => "prefix",
        search::SearchMatchMode::Contains => "contains",
    }
}