mirror of https://github.com/helix-editor/helix
work in progress adding of steel
mattwparas
2 years ago
parent
f2ccc03332
commit
ddca44e02b
@ -0,0 +1,327 @@
|
|||||||
|
use fuzzy_matcher::FuzzyMatcher;
|
||||||
|
use helix_core::{graphemes, Tendril};
|
||||||
|
use helix_view::{document::Mode, Editor};
|
||||||
|
use once_cell::sync::Lazy;
|
||||||
|
use steel::{
|
||||||
|
gc::unsafe_erased_pointers::CustomReference,
|
||||||
|
rvals::{IntoSteelVal, SteelString},
|
||||||
|
steel_vm::register_fn::RegisterFn,
|
||||||
|
};
|
||||||
|
|
||||||
|
use std::{
|
||||||
|
borrow::Cow,
|
||||||
|
collections::{HashMap, VecDeque},
|
||||||
|
sync::Mutex,
|
||||||
|
};
|
||||||
|
use std::{
|
||||||
|
collections::HashSet,
|
||||||
|
sync::{Arc, RwLock},
|
||||||
|
};
|
||||||
|
|
||||||
|
use steel::{rvals::Custom, steel_vm::builtin::BuiltInModule};
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
compositor::{self, Compositor},
|
||||||
|
job::{self, Callback},
|
||||||
|
keymap::{merge_keys, Keymap},
|
||||||
|
ui::{self, Popup, PromptEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::{
|
||||||
|
insert::{insert_char, insert_string},
|
||||||
|
Context, MappableCommand, TYPABLE_COMMAND_LIST,
|
||||||
|
};
|
||||||
|
|
||||||
|
thread_local! {
|
||||||
|
pub static ENGINE: std::rc::Rc<std::cell::RefCell<steel::steel_vm::engine::Engine>> = configure_engine();
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn initialize_engine() {
|
||||||
|
ENGINE.with(|x| x.borrow().globals().first().copied());
|
||||||
|
}
|
||||||
|
|
||||||
|
pub static KEYBINDING_QUEUE: Lazy<SharedKeyBindingsEventQueue> =
|
||||||
|
Lazy::new(|| SharedKeyBindingsEventQueue::new());
|
||||||
|
|
||||||
|
pub static EXPORTED_IDENTIFIERS: Lazy<ExportedIdentifiers> =
|
||||||
|
Lazy::new(|| ExportedIdentifiers::default());
|
||||||
|
|
||||||
|
/// In order to send events from the engine back to the configuration, we can created a shared
|
||||||
|
/// queue that the engine and the config push and pull from. Alternatively, we could use a channel
|
||||||
|
/// directly, however this was easy enough to set up.
|
||||||
|
pub struct SharedKeyBindingsEventQueue {
|
||||||
|
raw_bindings: Arc<Mutex<VecDeque<String>>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl SharedKeyBindingsEventQueue {
|
||||||
|
pub fn new() -> Self {
|
||||||
|
Self {
|
||||||
|
raw_bindings: std::sync::Arc::new(std::sync::Mutex::new(VecDeque::new())),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn merge(other_as_json: String) {
|
||||||
|
KEYBINDING_QUEUE
|
||||||
|
.raw_bindings
|
||||||
|
.lock()
|
||||||
|
.unwrap()
|
||||||
|
.push_back(other_as_json);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get() -> Option<HashMap<Mode, Keymap>> {
|
||||||
|
let mut guard = KEYBINDING_QUEUE.raw_bindings.lock().unwrap();
|
||||||
|
|
||||||
|
if let Some(initial) = guard.pop_front() {
|
||||||
|
let mut initial = serde_json::from_str(&initial).unwrap();
|
||||||
|
|
||||||
|
while let Some(remaining_event) = guard.pop_front() {
|
||||||
|
let bindings = serde_json::from_str(&remaining_event).unwrap();
|
||||||
|
|
||||||
|
merge_keys(&mut initial, bindings);
|
||||||
|
}
|
||||||
|
|
||||||
|
return Some(initial);
|
||||||
|
}
|
||||||
|
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Custom for PromptEvent {}
|
||||||
|
|
||||||
|
impl<'a> CustomReference for Context<'a> {}
|
||||||
|
|
||||||
|
fn configure_engine() -> std::rc::Rc<std::cell::RefCell<steel::steel_vm::engine::Engine>> {
|
||||||
|
let mut engine = steel::steel_vm::engine::Engine::new();
|
||||||
|
|
||||||
|
let mut module = BuiltInModule::new("helix/core/keybindings".to_string());
|
||||||
|
module.register_fn("set-keybindings!", SharedKeyBindingsEventQueue::merge);
|
||||||
|
|
||||||
|
engine.register_module(module);
|
||||||
|
|
||||||
|
let mut module = BuiltInModule::new("helix/core/typable".to_string());
|
||||||
|
|
||||||
|
module.register_value(
|
||||||
|
"PromptEvent::Validate",
|
||||||
|
PromptEvent::Validate.into_steelval().unwrap(),
|
||||||
|
);
|
||||||
|
|
||||||
|
// Register everything in the typable command list. Now these are all available
|
||||||
|
for command in TYPABLE_COMMAND_LIST {
|
||||||
|
let func = |cx: &mut Context, args: &[Cow<str>], event: PromptEvent| {
|
||||||
|
let mut cx = compositor::Context {
|
||||||
|
editor: cx.editor,
|
||||||
|
scroll: None,
|
||||||
|
jobs: cx.jobs,
|
||||||
|
};
|
||||||
|
|
||||||
|
(command.fun)(&mut cx, args, event)
|
||||||
|
};
|
||||||
|
|
||||||
|
module.register_fn(command.name, func);
|
||||||
|
}
|
||||||
|
|
||||||
|
engine.register_module(module);
|
||||||
|
|
||||||
|
let mut module = BuiltInModule::new("helix/core/static".to_string());
|
||||||
|
|
||||||
|
// Register everything in the static command list as well
|
||||||
|
// These just accept the context, no arguments
|
||||||
|
for command in MappableCommand::STATIC_COMMAND_LIST {
|
||||||
|
if let MappableCommand::Static { name, fun, .. } = command {
|
||||||
|
module.register_fn(name, fun);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.register_fn("insert_char", insert_char);
|
||||||
|
module.register_fn("insert_string", insert_string);
|
||||||
|
module.register_fn("current_selection", get_selection);
|
||||||
|
module.register_fn("current-highlighted-text!", get_highlighted_text);
|
||||||
|
module.register_fn("run-in-engine!", run_in_engine);
|
||||||
|
module.register_fn("get-helix-scm-path", get_helix_scm_path);
|
||||||
|
|
||||||
|
engine.register_module(module);
|
||||||
|
|
||||||
|
let helix_module_path = helix_loader::helix_module_file();
|
||||||
|
|
||||||
|
engine
|
||||||
|
.run(&format!(
|
||||||
|
r#"(require "{}")"#,
|
||||||
|
helix_module_path.to_str().unwrap()
|
||||||
|
))
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
// __module-mangler/home/matt/Documents/steel/cogs/logging/log.scm
|
||||||
|
|
||||||
|
// TODO: Use the helix.scm file located in the configuration directory instead
|
||||||
|
// let mut working_directory = std::env::current_dir().unwrap();
|
||||||
|
|
||||||
|
// working_directory.push("helix.scm");
|
||||||
|
|
||||||
|
// working_directory = working_directory.canonicalize().unwrap();
|
||||||
|
|
||||||
|
let helix_path =
|
||||||
|
"__module-mangler".to_string() + helix_module_path.as_os_str().to_str().unwrap();
|
||||||
|
|
||||||
|
let module = engine.extract_value(&helix_path).unwrap();
|
||||||
|
|
||||||
|
if let steel::rvals::SteelVal::HashMapV(m) = module {
|
||||||
|
let exported = m
|
||||||
|
.iter()
|
||||||
|
.filter(|(_, v)| v.is_function())
|
||||||
|
.map(|(k, _)| {
|
||||||
|
if let steel::rvals::SteelVal::SymbolV(s) = k {
|
||||||
|
s.to_string()
|
||||||
|
} else {
|
||||||
|
panic!("Found a non symbol!")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.collect::<HashSet<_>>();
|
||||||
|
|
||||||
|
*EXPORTED_IDENTIFIERS.identifiers.write().unwrap() = exported;
|
||||||
|
|
||||||
|
let docs = m
|
||||||
|
.iter()
|
||||||
|
.filter_map(|(k, v)| {
|
||||||
|
if let steel::rvals::SteelVal::SymbolV(s) = k {
|
||||||
|
if s.ends_with("__doc__") {
|
||||||
|
if let steel::rvals::SteelVal::StringV(d) = v {
|
||||||
|
return Some((
|
||||||
|
s.strip_suffix("__doc__").unwrap().to_string(),
|
||||||
|
d.to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
None
|
||||||
|
})
|
||||||
|
.collect::<HashMap<_, _>>();
|
||||||
|
|
||||||
|
*EXPORTED_IDENTIFIERS.docs.write().unwrap() = docs;
|
||||||
|
} else {
|
||||||
|
panic!("Unable to parse exported identifiers from helix module!")
|
||||||
|
}
|
||||||
|
|
||||||
|
std::rc::Rc::new(std::cell::RefCell::new(engine))
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Default, Debug)]
|
||||||
|
pub struct ExportedIdentifiers {
|
||||||
|
identifiers: Arc<RwLock<HashSet<String>>>,
|
||||||
|
docs: Arc<RwLock<HashMap<String, String>>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ExportedIdentifiers {
|
||||||
|
pub(crate) fn fuzzy_match<'a>(
|
||||||
|
fuzzy_matcher: &'a fuzzy_matcher::skim::SkimMatcherV2,
|
||||||
|
input: &'a str,
|
||||||
|
) -> Vec<(String, i64)> {
|
||||||
|
EXPORTED_IDENTIFIERS
|
||||||
|
.identifiers
|
||||||
|
.read()
|
||||||
|
.unwrap()
|
||||||
|
.iter()
|
||||||
|
.filter_map(|name| {
|
||||||
|
fuzzy_matcher
|
||||||
|
.fuzzy_match(name, input)
|
||||||
|
.map(|score| (name, score))
|
||||||
|
})
|
||||||
|
.map(|x| (x.0.to_string(), x.1))
|
||||||
|
.collect::<Vec<_>>()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn is_exported(ident: &str) -> bool {
|
||||||
|
EXPORTED_IDENTIFIERS
|
||||||
|
.identifiers
|
||||||
|
.read()
|
||||||
|
.unwrap()
|
||||||
|
.contains(ident)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn engine_get_doc(ident: &str) -> Option<String> {
|
||||||
|
EXPORTED_IDENTIFIERS.get_doc(ident)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_doc(&self, ident: &str) -> Option<String> {
|
||||||
|
self.docs.read().unwrap().get(ident).cloned()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_highlighted_text(cx: &mut Context) -> String {
|
||||||
|
let (view, doc) = current_ref!(cx.editor);
|
||||||
|
let text = doc.text().slice(..);
|
||||||
|
doc.selection(view.id).primary().slice(text).to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_selection(cx: &mut Context) -> String {
|
||||||
|
let (view, doc) = current_ref!(cx.editor);
|
||||||
|
let text = doc.text().slice(..);
|
||||||
|
|
||||||
|
let grapheme_start = doc.selection(view.id).primary().cursor(text);
|
||||||
|
let grapheme_end = graphemes::next_grapheme_boundary(text, grapheme_start);
|
||||||
|
|
||||||
|
if grapheme_start == grapheme_end {
|
||||||
|
return "".into();
|
||||||
|
}
|
||||||
|
|
||||||
|
let grapheme = text.slice(grapheme_start..grapheme_end).to_string();
|
||||||
|
|
||||||
|
let printable = grapheme.chars().fold(String::new(), |mut s, c| {
|
||||||
|
match c {
|
||||||
|
'\0' => s.push_str("\\0"),
|
||||||
|
'\t' => s.push_str("\\t"),
|
||||||
|
'\n' => s.push_str("\\n"),
|
||||||
|
'\r' => s.push_str("\\r"),
|
||||||
|
_ => s.push(c),
|
||||||
|
}
|
||||||
|
|
||||||
|
s
|
||||||
|
});
|
||||||
|
|
||||||
|
printable
|
||||||
|
}
|
||||||
|
|
||||||
|
fn run_in_engine(cx: &mut Context, arg: String) -> anyhow::Result<()> {
|
||||||
|
let callback = async move {
|
||||||
|
let output = ENGINE
|
||||||
|
.with(|x| x.borrow_mut().run(&arg))
|
||||||
|
.map(|x| format!("{:?}", x));
|
||||||
|
|
||||||
|
let (output, success) = match output {
|
||||||
|
Ok(v) => (Tendril::from(v), true),
|
||||||
|
Err(e) => (Tendril::from(e.to_string()), false),
|
||||||
|
};
|
||||||
|
|
||||||
|
let call: job::Callback = Callback::EditorCompositor(Box::new(
|
||||||
|
move |editor: &mut Editor, compositor: &mut Compositor| {
|
||||||
|
if !output.is_empty() {
|
||||||
|
let contents = ui::Markdown::new(
|
||||||
|
format!("```\n{}\n```", output),
|
||||||
|
editor.syn_loader.clone(),
|
||||||
|
);
|
||||||
|
let popup = Popup::new("engine", contents).position(Some(
|
||||||
|
helix_core::Position::new(editor.cursor().0.unwrap_or_default().row, 2),
|
||||||
|
));
|
||||||
|
compositor.replace_or_push("engine", popup);
|
||||||
|
}
|
||||||
|
if success {
|
||||||
|
editor.set_status("Command succeeded");
|
||||||
|
} else {
|
||||||
|
editor.set_error("Command failed");
|
||||||
|
}
|
||||||
|
},
|
||||||
|
));
|
||||||
|
Ok(call)
|
||||||
|
};
|
||||||
|
cx.jobs.callback(callback);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_helix_scm_path() -> String {
|
||||||
|
helix_loader::helix_module_file()
|
||||||
|
.to_str()
|
||||||
|
.unwrap()
|
||||||
|
.to_string()
|
||||||
|
}
|
||||||
@ -0,0 +1,178 @@
|
|||||||
|
(require-builtin helix/core/typable as helix.)
|
||||||
|
(require-builtin helix/core/static as helix.static.)
|
||||||
|
(require-builtin helix/core/keybindings as helix.keybindings.)
|
||||||
|
|
||||||
|
|
||||||
|
(provide
|
||||||
|
set-theme-dracula
|
||||||
|
set-theme-dracula__doc__
|
||||||
|
set-theme-custom
|
||||||
|
set-theme-custom__doc__
|
||||||
|
theme-then-vsplit
|
||||||
|
theme-then-vsplit__doc__
|
||||||
|
custom-undo
|
||||||
|
custom-undo__doc__
|
||||||
|
lam
|
||||||
|
lam__doc__
|
||||||
|
delete-word-forward
|
||||||
|
insert-string-at-selection
|
||||||
|
highlight-to-matching-paren
|
||||||
|
highlight-to-matching-paren__doc__
|
||||||
|
delete-sexpr
|
||||||
|
delete-sexpr__doc__
|
||||||
|
run-expr
|
||||||
|
run-highlight
|
||||||
|
make-minor-mode!
|
||||||
|
git-status
|
||||||
|
reload-helix-scm
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
;;@doc
|
||||||
|
;; Sets the theme to be the dracula theme
|
||||||
|
(define (set-theme-dracula cx)
|
||||||
|
(helix.theme cx (list "dracula") helix.PromptEvent::Validate))
|
||||||
|
|
||||||
|
;;@doc
|
||||||
|
;; Sets the theme to be the theme passed in
|
||||||
|
(define (set-theme-custom cx entered-theme)
|
||||||
|
(helix.theme cx (list entered-theme) helix.PromptEvent::Validate))
|
||||||
|
|
||||||
|
;;@doc
|
||||||
|
;; Switch theme to the entered theme, then split the current file into
|
||||||
|
;; a vsplit
|
||||||
|
(define (theme-then-vsplit cx entered-theme)
|
||||||
|
(set-theme-custom cx entered-theme)
|
||||||
|
(helix.vsplit cx '() helix.PromptEvent::Validate))
|
||||||
|
|
||||||
|
;;@doc
|
||||||
|
;; Perform an undo
|
||||||
|
(define (custom-undo cx)
|
||||||
|
(helix.static.undo cx))
|
||||||
|
|
||||||
|
;;@doc
|
||||||
|
;; Insert a lambda
|
||||||
|
(define (lam cx)
|
||||||
|
(helix.static.insert_char cx #\λ)
|
||||||
|
(helix.static.insert_mode cx))
|
||||||
|
|
||||||
|
;;@doc
|
||||||
|
;; Insert the string at the selection and go back into insert mode
|
||||||
|
(define (insert-string-at-selection cx str)
|
||||||
|
(helix.static.insert_string cx str)
|
||||||
|
(helix.static.insert_mode cx))
|
||||||
|
|
||||||
|
;;@doc
|
||||||
|
;; Delete the word forward
|
||||||
|
(define (delete-word-forward cx)
|
||||||
|
(helix.static.delete_word_forward cx))
|
||||||
|
|
||||||
|
|
||||||
|
;;@doc
|
||||||
|
;; Registers a minor mode with the registered modifer and key map
|
||||||
|
;;
|
||||||
|
;; Examples:
|
||||||
|
;; ```scheme
|
||||||
|
;; (make-minor-mode! "+"
|
||||||
|
;; (hash "P" ":lam"))
|
||||||
|
;; ```
|
||||||
|
(define (make-minor-mode! modifier bindings)
|
||||||
|
(~> (hash "normal" (hash modifier bindings))
|
||||||
|
(value->jsexpr-string)
|
||||||
|
(helix.keybindings.set-keybindings!)))
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
(define-syntax minor-mode!
|
||||||
|
(syntax-rules (=>)
|
||||||
|
[(minor-mode! modifier (key => function))
|
||||||
|
(make-minor-mode! modifier (minor-mode-cruncher (key => function)))]
|
||||||
|
|
||||||
|
[(minor-mode! modifier (key => (function ...)))
|
||||||
|
(make-minor-mode! modifier (minor-mode-cruncher (key => (function ...))))]
|
||||||
|
|
||||||
|
[(minor-mode! modifier (key => function) remaining ...)
|
||||||
|
(make-minor-mode! modifier (minor-mode-cruncher (key => function) remaining ...))]
|
||||||
|
|
||||||
|
[(minor-mode! modifier (key => (function ...)) remaining ...)
|
||||||
|
(make-minor-mode! modifier (minor-mode-cruncher (key => function) ... remaining ...))]
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
))
|
||||||
|
|
||||||
|
(define-syntax minor-mode-cruncher
|
||||||
|
(syntax-rules (=>)
|
||||||
|
|
||||||
|
|
||||||
|
[(minor-mode-cruncher (key => (function ...)))
|
||||||
|
(hash key (map (lambda (x)
|
||||||
|
(string-append ":" (symbol->string x)))
|
||||||
|
(quote (function ...))))]
|
||||||
|
|
||||||
|
[(minor-mode-cruncher (key => function))
|
||||||
|
(hash key (string-append ":" (symbol->string (quote function))))]
|
||||||
|
|
||||||
|
[(minor-mode-cruncher (key => (function ...)) remaining ...)
|
||||||
|
(hash-insert
|
||||||
|
(minor-mode-cruncher remaining ...)
|
||||||
|
key (map (lambda (x)
|
||||||
|
(string-append ":" (symbol->string x)))
|
||||||
|
(quote (function ...))))]
|
||||||
|
|
||||||
|
[(minor-mode-cruncher (key => function) remaining ...)
|
||||||
|
(hash-insert
|
||||||
|
(minor-mode-cruncher remaining ...)
|
||||||
|
key
|
||||||
|
(string-append ":" (symbol->string (quote function))))]
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
))
|
||||||
|
|
||||||
|
;;@doc
|
||||||
|
;; Highlight to the matching paren
|
||||||
|
(define (highlight-to-matching-paren cx)
|
||||||
|
(helix.static.select_mode cx)
|
||||||
|
(helix.static.match_brackets cx))
|
||||||
|
|
||||||
|
(define (run-expr cx)
|
||||||
|
(define current-selection (helix.static.current_selection cx))
|
||||||
|
(when (or (equal? "(" current-selection)
|
||||||
|
(equal? ")" current-selection))
|
||||||
|
(highlight-to-matching-paren cx)
|
||||||
|
(helix.static.run-in-engine! cx (helix.static.current-highlighted-text! cx))
|
||||||
|
(helix.static.normal_mode cx)))
|
||||||
|
|
||||||
|
|
||||||
|
(define (run-highlight cx)
|
||||||
|
(helix.static.run-in-engine! cx (helix.static.current-highlighted-text! cx)))
|
||||||
|
|
||||||
|
;;@doc
|
||||||
|
;; Delete the s-expression matching this bracket
|
||||||
|
;; If the current selection is not on a bracket, this is a no-op
|
||||||
|
(define (delete-sexpr cx)
|
||||||
|
(define current-selection (helix.static.current_selection cx))
|
||||||
|
(when (or (equal? "(" current-selection)
|
||||||
|
(equal? ")" current-selection))
|
||||||
|
(highlight-to-matching-paren cx)
|
||||||
|
(helix.static.delete_selection cx)))
|
||||||
|
|
||||||
|
; (minor-mode! "+" ("l" => lam)
|
||||||
|
; ("q" => (set-theme-dracula lam)))
|
||||||
|
|
||||||
|
(minor-mode! "P" ("l" => lam)
|
||||||
|
("p" => highlight-to-matching-paren)
|
||||||
|
("d" => delete-sexpr)
|
||||||
|
("r" => run-expr))
|
||||||
|
|
||||||
|
(make-minor-mode! "+" (hash "l" ":lam"))
|
||||||
|
|
||||||
|
(define (git-status cx)
|
||||||
|
(helix.run-shell-command cx '("git" "status") helix.PromptEvent::Validate))
|
||||||
|
|
||||||
|
(minor-mode! "G" ("s" => git-status))
|
||||||
|
|
||||||
|
(define (reload-helix-scm cx)
|
||||||
|
(helix.static.run-in-engine! cx
|
||||||
|
(string-append "(require \"" (helix.static.get-helix.scm-path) "\")")))
|
||||||
Loading…
Reference in New Issue