mindflow 0.1.2

A command line client for Mindflow
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

use clap::{Parser, ArgAction};

use crate::resolve_handling::resolve::resolve;
use crate::requests::query::request_query;
use crate::resolve_handling::generate_index::generate_index;
use crate::utils::response::handle_response_text;


#[derive(Parser)]
pub(crate) struct Query {
    #[arg(index = 1)]
    pub(crate) query: String,
    #[arg(index = 2)]
    pub(crate) references: Vec<String>,
    #[arg(short = 'i', long = "index", action = ArgAction::SetTrue, value_name = "Specifies whether you want to create an index for the files you query on.")]
    pub(crate) index: bool,
    #[arg(short = 'p', long = "prompt", action = ArgAction::SetTrue, value_name = "Get prompt to enter into ChatGPT.")]
    pub(crate) return_prompt: bool,
    #[arg(short = 's', long = "skip-clipboard", action = ArgAction::SetTrue, value_name = "Copy response to clipboard.")]
    pub(crate) skip_clipboard: bool,
}

impl Query {
    pub(crate) async fn execute(&mut self) {
        // create query request handler
        let client = reqwest::Client::new();

        // Get resolved references and their hashes
        let all_resolved = resolve(&self.references).await;
        let resolved_hashes = all_resolved.iter().filter_map(|resolved| 
            resolved.text_hash()
        ).collect::<Vec<String>>();

        // Generate index in Mindflow server if specified.
        if self.index {
            generate_index(all_resolved).await;
        }

        // Send query to Mindflow server.
        let request_query_response = request_query(&client, self.query.clone(), resolved_hashes, true).await;
        handle_response_text(request_query_response.text, self.skip_clipboard);
    }
}