mirror of
https://github.com/aljazceru/turso.git
synced 2025-12-24 19:44:21 +01:00
fbc3d0dbc3
In the hopes of doing a good job at teaching people what Turso can do, I am adding built-in manual pages. When the CLI starts, it picks a feature at random, and tells the user that the feature exists: ``` Turso v0.2.0-pre.8 Enter ".help" for usage hints. Did you know that Turso supports Change Data Capture? Type .manual cdc to learn more. This software is ALPHA, only use for development, testing, and experimentation. Connected to a transient in-memory database. Use ".open FILENAME" to reopen on a persistent database ``` There is a lot we can do to make this feature world class: - we can automatically compile examples during compile time like rust-doc, to make sure examples used in the manuals always work - we can implement scrolling and navigation - we can document a lot more features But for now, this is a start!
94 lines
2.6 KiB
Rust
94 lines
2.6 KiB
Rust
#![allow(clippy::arc_with_non_send_sync)]
|
|
mod app;
|
|
mod commands;
|
|
mod config;
|
|
mod helper;
|
|
mod input;
|
|
mod manual;
|
|
mod mcp_server;
|
|
mod opcodes_dictionary;
|
|
|
|
use config::CONFIG_DIR;
|
|
use mcp_server::TursoMcpServer;
|
|
use rustyline::{error::ReadlineError, Config, Editor};
|
|
use std::{
|
|
path::PathBuf,
|
|
sync::{atomic::Ordering, LazyLock},
|
|
};
|
|
|
|
#[cfg(not(target_family = "wasm"))]
|
|
#[global_allocator]
|
|
static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
|
|
|
|
fn rustyline_config() -> Config {
|
|
Config::builder()
|
|
.completion_type(rustyline::CompletionType::List)
|
|
.auto_add_history(true)
|
|
.build()
|
|
}
|
|
|
|
pub static HOME_DIR: LazyLock<PathBuf> =
|
|
LazyLock::new(|| dirs::home_dir().expect("Could not determine home directory"));
|
|
|
|
pub static HISTORY_FILE: LazyLock<PathBuf> = LazyLock::new(|| HOME_DIR.join(".limbo_history"));
|
|
|
|
fn run_mcp_server(app: app::Limbo) -> anyhow::Result<()> {
|
|
let conn = app.get_connection();
|
|
let interrupt_count = app.get_interrupt_count();
|
|
let mcp_server = TursoMcpServer::new(conn, interrupt_count);
|
|
|
|
mcp_server.run()
|
|
}
|
|
|
|
fn main() -> anyhow::Result<()> {
|
|
let (mut app, _guard) = app::Limbo::new()?;
|
|
|
|
if app.is_mcp_mode() {
|
|
return run_mcp_server(app);
|
|
}
|
|
|
|
if std::io::IsTerminal::is_terminal(&std::io::stdin()) {
|
|
let mut rl = Editor::with_config(rustyline_config())?;
|
|
if HISTORY_FILE.exists() {
|
|
rl.load_history(HISTORY_FILE.as_path())?;
|
|
}
|
|
let config_file = CONFIG_DIR.join("limbo.toml");
|
|
|
|
let config = config::Config::from_config_file(config_file);
|
|
tracing::info!("Configuration: {:?}", config);
|
|
app = app.with_config(config);
|
|
|
|
app = app.with_readline(rl);
|
|
} else {
|
|
tracing::debug!("not in tty");
|
|
}
|
|
|
|
loop {
|
|
match app.readline() {
|
|
Ok(_) => app.consume(false),
|
|
Err(ReadlineError::Interrupted) => {
|
|
// At prompt, increment interrupt count
|
|
if app.interrupt_count.fetch_add(1, Ordering::SeqCst) >= 1 {
|
|
eprintln!("Interrupted. Exiting...");
|
|
let _ = app.close_conn();
|
|
break;
|
|
}
|
|
println!("Use .quit to exit or press Ctrl-C again to force quit.");
|
|
app.reset_input();
|
|
continue;
|
|
}
|
|
Err(ReadlineError::Eof) => {
|
|
// consume remaining input before exit
|
|
app.consume(true);
|
|
let _ = app.close_conn();
|
|
break;
|
|
}
|
|
Err(err) => {
|
|
let _ = app.close_conn();
|
|
anyhow::bail!(err)
|
|
}
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|