0afe1e4e67
* Test commands for proper names and search terms Assert that the `Command.name()` is equal to `Signature.name` Check that search terms are not just substrings of the command name as they would not help finding the command. * Clean up search terms Remove redundant terms that just replicate the command name. Try to eliminate substring between search terms, clean up where necessary.
56 lines
1.5 KiB
Rust
56 lines
1.5 KiB
Rust
use super::super::SQLiteDatabase;
|
|
use nu_engine::CallExt;
|
|
use nu_protocol::{
|
|
ast::Call,
|
|
engine::{Command, EngineState, Stack},
|
|
Category, Example, IntoPipelineData, PipelineData, ShellError, Signature, Spanned, SyntaxShape,
|
|
Type,
|
|
};
|
|
use std::path::PathBuf;
|
|
|
|
#[derive(Clone)]
|
|
pub struct OpenDb;
|
|
|
|
impl Command for OpenDb {
|
|
fn name(&self) -> &str {
|
|
"open-db"
|
|
}
|
|
|
|
fn signature(&self) -> Signature {
|
|
Signature::build(self.name())
|
|
.required("query", SyntaxShape::Filepath, "SQLite file to be opened")
|
|
.input_type(Type::Any)
|
|
.output_type(Type::Custom("database".into()))
|
|
.category(Category::Custom("database".into()))
|
|
}
|
|
|
|
fn usage(&self) -> &str {
|
|
"Open a database"
|
|
}
|
|
|
|
fn search_terms(&self) -> Vec<&str> {
|
|
vec!["database"]
|
|
}
|
|
|
|
fn examples(&self) -> Vec<Example> {
|
|
vec![Example {
|
|
description: "Creates a connection to a sqlite database based on the file name",
|
|
example: r#"open-db file.sqlite"#,
|
|
result: None,
|
|
}]
|
|
}
|
|
|
|
fn run(
|
|
&self,
|
|
engine_state: &EngineState,
|
|
stack: &mut Stack,
|
|
call: &Call,
|
|
_input: PipelineData,
|
|
) -> Result<PipelineData, ShellError> {
|
|
let path: Spanned<PathBuf> = call.req(engine_state, stack, 0)?;
|
|
|
|
SQLiteDatabase::try_from_path(path.item.as_path(), path.span)
|
|
.map(|db| db.into_value(call.head).into_pipeline_data())
|
|
}
|
|
}
|