extension_api.rs

 1pub use wit::*;
 2pub type Result<T, E = String> = core::result::Result<T, E>;
 3
 4pub trait Extension: Send + Sync {
 5    fn new() -> Self
 6    where
 7        Self: Sized;
 8
 9    fn language_server_command(
10        &mut self,
11        config: LanguageServerConfig,
12        worktree: &Worktree,
13    ) -> Result<Command>;
14
15    fn language_server_initialization_options(
16        &mut self,
17        _config: LanguageServerConfig,
18        _worktree: &Worktree,
19    ) -> Result<Option<String>> {
20        Ok(None)
21    }
22}
23
24#[macro_export]
25macro_rules! register_extension {
26    ($extension_type:ty) => {
27        #[export_name = "init-extension"]
28        pub extern "C" fn __init_extension() {
29            std::env::set_current_dir(std::env::var("PWD").unwrap()).unwrap();
30            zed_extension_api::register_extension(|| {
31                Box::new(<$extension_type as zed_extension_api::Extension>::new())
32            });
33        }
34    };
35}
36
37#[doc(hidden)]
38pub fn register_extension(build_extension: fn() -> Box<dyn Extension>) {
39    unsafe { EXTENSION = Some((build_extension)()) }
40}
41
42fn extension() -> &'static mut dyn Extension {
43    unsafe { EXTENSION.as_deref_mut().unwrap() }
44}
45
46static mut EXTENSION: Option<Box<dyn Extension>> = None;
47
48#[cfg(target_arch = "wasm32")]
49#[link_section = "zed:api-version"]
50#[doc(hidden)]
51pub static ZED_API_VERSION: [u8; 6] = *include_bytes!(concat!(env!("OUT_DIR"), "/version_bytes"));
52
53mod wit {
54    wit_bindgen::generate!({
55        skip: ["init-extension"],
56        path: "./wit/since_v0.0.4",
57    });
58}
59
60wit::export!(Component);
61
62struct Component;
63
64impl wit::Guest for Component {
65    fn language_server_command(
66        config: wit::LanguageServerConfig,
67        worktree: &wit::Worktree,
68    ) -> Result<wit::Command> {
69        extension().language_server_command(config, worktree)
70    }
71
72    fn language_server_initialization_options(
73        config: LanguageServerConfig,
74        worktree: &Worktree,
75    ) -> Result<Option<String>, String> {
76        extension().language_server_initialization_options(config, worktree)
77    }
78}