thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-21 15:12:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(rust): Fix skim's panic handler

Browse Source
This commit is contained in:
Maximilian Roos 2024-02-29 21:26:39 -08:00
parent 891689a415
commit 1df6f4cd5c
No known key found for this signature in database
GPG Key ID: 2A9413063820C8CD
1 changed files with 9 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
rust/skim/src/lib.rs
View File

@ -1,7 +1,7 @@
use skim::prelude::*;
use term::terminfo::TermInfo;
use cxx::{CxxString, CxxVector}; use cxx::{CxxString, CxxVector};
use skim::prelude::*;
use std::panic; use std::panic;
use term::terminfo::TermInfo;
#[cxx::bridge] #[cxx::bridge]
mod ffi { mod ffi {
@ -16,7 +16,7 @@ struct Item {
} }
impl Item { impl Item {
fn new(text: String) -> Self { fn new(text: String) -> Self {
return Self{ Self {
// Text that will be printed by skim, and will be used for matching. // Text that will be printed by skim, and will be used for matching.
// //
// Text that will be shown should not contains new lines since in this case skim may // Text that will be shown should not contains new lines since in this case skim may
@ -24,16 +24,16 @@ impl Item {
text_no_newlines: text.replace("\n", " "), text_no_newlines: text.replace("\n", " "),
// This will be used when the match had been selected. // This will be used when the match had been selected.
orig_text: text, orig_text: text,
}; }
} }
} }
impl SkimItem for Item { impl SkimItem for Item {
fn text(&self) -> Cow<str> { fn text(&self) -> Cow<str> {
return Cow::Borrowed(&self.text_no_newlines); Cow::Borrowed(&self.text_no_newlines)
} }
fn output(&self) -> Cow<str> { fn output(&self) -> Cow<str> {
return Cow::Borrowed(&self.orig_text); Cow::Borrowed(&self.orig_text)
} }
} }
@ -88,14 +88,11 @@ fn skim_impl(prefix: &CxxString, words: &CxxVector<CxxString>) -> Result<String,
if output.selected_items.is_empty() { if output.selected_items.is_empty() {
return Err("No items had been selected".to_string()); return Err("No items had been selected".to_string());
} }
return Ok(output.selected_items[0].output().to_string()); Ok(output.selected_items[0].output().to_string())
} }
fn skim(prefix: &CxxString, words: &CxxVector<CxxString>) -> Result<String, String> { fn skim(prefix: &CxxString, words: &CxxVector<CxxString>) -> Result<String, String> {
let ret = panic::catch_unwind(|| { match panic::catch_unwind(|| skim_impl(prefix, words)) {
return skim_impl(prefix, words);
});
return match ret {
Err(err) => { Err(err) => {
let e = if let Some(s) = err.downcast_ref::<String>() { let e = if let Some(s) = err.downcast_ref::<String>() {
format!("{}", s) format!("{}", s)
@ -105,7 +102,7 @@ fn skim(prefix: &CxxString, words: &CxxVector<CxxString>) -> Result<String, Stri
format!("Unknown panic type: {:?}", err.type_id()) format!("Unknown panic type: {:?}", err.type_id())
}; };
Err(format!("Rust panic: {:?}", e)) Err(format!("Rust panic: {:?}", e))
}, }
Ok(res) => res, Ok(res) => res,
} }
} }