Ade Attwood
3b0d1510e7
This adds dot files into the finder. We are adding the overrides to the `ignore` package, that can be used later to add custom ignore directories that can be passed in as settings. We are also adding a new `ivy_cwd` function to libivy to get the current directory due to the limitations of lua. Fixes-issue: #16
93 lines
2.3 KiB
Rust
93 lines
2.3 KiB
Rust
mod finder;
|
|
mod matcher;
|
|
mod sorter;
|
|
|
|
use std::collections::HashMap;
|
|
use std::ffi::CStr;
|
|
use std::ffi::CString;
|
|
use std::os::raw::{c_char, c_int};
|
|
use std::sync::Mutex;
|
|
|
|
#[macro_use]
|
|
extern crate lazy_static;
|
|
|
|
lazy_static! {
|
|
static ref GLOBAL_FILE_CACHE: Mutex<HashMap<String, Vec<String>>> = Mutex::new(HashMap::new());
|
|
}
|
|
|
|
fn to_string(input: *const c_char) -> String {
|
|
unsafe { CStr::from_ptr(input) }
|
|
.to_str()
|
|
.unwrap()
|
|
.to_string()
|
|
}
|
|
|
|
fn get_files(directory: &String) -> Vec<String> {
|
|
let mut cache = GLOBAL_FILE_CACHE.lock().unwrap();
|
|
if !cache.contains_key(directory) {
|
|
let finder_options = finder::Options {
|
|
directory: directory.clone(),
|
|
};
|
|
cache.insert(directory.clone(), finder::find_files(finder_options));
|
|
}
|
|
|
|
return cache.get(directory).unwrap().to_vec();
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn ivy_init(c_base_dir: *const c_char) {
|
|
let directory = to_string(c_base_dir);
|
|
get_files(&directory);
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn ivy_cwd() -> *const c_char {
|
|
return CString::new(std::env::current_dir().unwrap().to_str().unwrap())
|
|
.unwrap()
|
|
.into_raw();
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn ivy_match(c_pattern: *const c_char, c_text: *const c_char) -> c_int {
|
|
let pattern = to_string(c_pattern);
|
|
let text = to_string(c_text);
|
|
|
|
inner_match(pattern, text)
|
|
}
|
|
|
|
pub fn inner_match(pattern: String, text: String) -> i32 {
|
|
let m = matcher::Matcher::new(pattern);
|
|
|
|
m.score(text.as_str()) as i32
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn ivy_files(c_pattern: *const c_char, c_base_dir: *const c_char) -> *const c_char {
|
|
let pattern = to_string(c_pattern);
|
|
let directory = to_string(c_base_dir);
|
|
|
|
let output = inner_files(pattern, directory);
|
|
|
|
CString::new(output).unwrap().into_raw()
|
|
}
|
|
|
|
pub fn inner_files(pattern: String, base_dir: String) -> String {
|
|
let mut output = String::new();
|
|
|
|
// Bail out early if the pattern is empty; it's never going to find anything
|
|
if pattern.is_empty() {
|
|
return output;
|
|
}
|
|
|
|
let files = get_files(&base_dir);
|
|
|
|
let sorter_options = sorter::Options::new(pattern);
|
|
|
|
let files = sorter::sort_strings(sorter_options, files);
|
|
for file in files.iter() {
|
|
output.push_str(&file.content);
|
|
output.push('\n');
|
|
}
|
|
|
|
output
|
|
}
|