turboprop 0.1.2

Fast semantic code search and indexing tool
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

use clap::Parser;
use turboprop::cli::{Cli, Commands};
use turboprop::commands::{
    execute_index_command_cli, execute_mcp_command, execute_search_command_cli,
    handle_model_command, run_benchmark, BenchmarkArgs, SearchCliArgs,
};
use turboprop::config::{CliConfigOverrides, TurboPropConfig};
use turboprop::types::parse_filesize;

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    tracing_subscriber::fmt()
        .with_writer(std::io::stderr)
        .init();

    let cli = Cli::parse();

    match cli.command {
        Commands::Index {
            repo,
            max_filesize,
            model,
            instruction,
            cache_dir,
            verbose,
            worker_threads,
            batch_size,
            watch,
        } => {
            // Load base configuration
            let mut config = TurboPropConfig::load()?;

            // Build CLI overrides
            let mut cli_overrides = CliConfigOverrides::new().with_verbose(verbose);

            // Apply global model options (command-level overrides global)
            let effective_model = model.or(cli.model);
            let effective_instruction = instruction.or(cli.instruction);

            if let Some(model_name) = effective_model {
                cli_overrides = cli_overrides.with_model(model_name);
            }

            if let Some(instruction_text) = effective_instruction {
                cli_overrides = cli_overrides.with_instruction(instruction_text);
            }

            if let Some(cache_dir) = cache_dir {
                cli_overrides = cli_overrides.with_cache_dir(cache_dir);
            }

            if let Some(max_filesize_str) = max_filesize {
                let max_bytes = parse_filesize(&max_filesize_str)
                    .map_err(|e| anyhow::anyhow!("Failed to parse filesize: {}", e))?;
                cli_overrides = cli_overrides.with_max_filesize(max_bytes);
            }

            if let Some(threads) = worker_threads {
                cli_overrides = cli_overrides.with_worker_threads(threads);
            }

            cli_overrides = cli_overrides.with_batch_size(batch_size);

            // Merge CLI overrides with config
            config = config.merge_cli_args(&cli_overrides);

            // Validate configuration
            config.validate()?;

            // Call the enhanced index command with progress tracking
            // Show progress bars by default (true), they work fine with verbose logging
            if watch {
                // Execute the watch mode
                use turboprop::commands::index::execute_watch_command_cli;
                execute_watch_command_cli(&repo, &config, true).await?;
            } else {
                execute_index_command_cli(&repo, &config, true).await?;
            }
        }
        Commands::Search {
            query,
            repo,
            model,
            instruction,
            limit,
            threshold,
            output,
            filetype,
            filter,
        } => {
            // Load base configuration for search command
            let config = TurboPropConfig::load()?;

            // Apply global model options (command-level overrides global)
            let effective_model = model.or(cli.model);
            let effective_instruction = instruction.or(cli.instruction);

            // Apply model and instruction overrides to configuration
            let mut config_overrides = CliConfigOverrides::new();
            if let Some(model_name) = effective_model {
                config_overrides = config_overrides.with_model(model_name);
            }
            if let Some(instruction_text) = effective_instruction {
                config_overrides = config_overrides.with_instruction(instruction_text);
            }

            // TODO: Apply config overrides to search command when supported
            let config =
                if config_overrides.model.is_some() || config_overrides.instruction.is_some() {
                    config.merge_cli_args(&config_overrides)
                } else {
                    config
                };

            // Create CLI args struct
            let args = SearchCliArgs::new(query, repo, limit, threshold, output, filetype, filter);

            // Execute the search command using the new implementation
            execute_search_command_cli(args, &config).await?;
        }
        Commands::Model { action } => {
            handle_model_command(action).await?;
        }
        Commands::Benchmark {
            models,
            text_count,
            iterations,
            sample_file,
            format,
        } => {
            let args = BenchmarkArgs {
                models,
                text_count,
                iterations,
                sample_file,
                format,
            };
            run_benchmark(args).await?;
        }
        Commands::Mcp(args) => {
            // Execute MCP server (it will print setup info after validation)
            execute_mcp_command(args).await?;
        }
    }

    Ok(())
}