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