1cfcdfa7ac
This PR overhauls extension registration in order to make it more modular. The `extension` crate now contains an `ExtensionHostProxy` that can be used to register various proxies that the extension host can use to interact with the rest of the system. There are now a number of different proxy traits representing the various pieces of functionality that can be provided by an extension. The respective crates that provide this functionality can implement their corresponding proxy trait in order to register a proxy that the extension host will use to register the bits of functionality provided by the extension. Release Notes: - N/A
76 lines
2.2 KiB
Rust
76 lines
2.2 KiB
Rust
use std::path::PathBuf;
|
|
use std::sync::Arc;
|
|
|
|
use anyhow::Result;
|
|
use async_trait::async_trait;
|
|
use extension::{Extension, ExtensionHostProxy, ExtensionIndexedDocsProviderProxy};
|
|
use gpui::AppContext;
|
|
|
|
use crate::{
|
|
IndexedDocsDatabase, IndexedDocsProvider, IndexedDocsRegistry, PackageName, ProviderId,
|
|
};
|
|
|
|
pub fn init(cx: &mut AppContext) {
|
|
let proxy = ExtensionHostProxy::default_global(cx);
|
|
proxy.register_indexed_docs_provider_proxy(IndexedDocsRegistryProxy {
|
|
indexed_docs_registry: IndexedDocsRegistry::global(cx),
|
|
});
|
|
}
|
|
|
|
struct IndexedDocsRegistryProxy {
|
|
indexed_docs_registry: Arc<IndexedDocsRegistry>,
|
|
}
|
|
|
|
impl ExtensionIndexedDocsProviderProxy for IndexedDocsRegistryProxy {
|
|
fn register_indexed_docs_provider(&self, extension: Arc<dyn Extension>, provider_id: Arc<str>) {
|
|
self.indexed_docs_registry
|
|
.register_provider(Box::new(ExtensionIndexedDocsProvider::new(
|
|
extension,
|
|
ProviderId(provider_id),
|
|
)));
|
|
}
|
|
}
|
|
|
|
pub struct ExtensionIndexedDocsProvider {
|
|
extension: Arc<dyn Extension>,
|
|
id: ProviderId,
|
|
}
|
|
|
|
impl ExtensionIndexedDocsProvider {
|
|
pub fn new(extension: Arc<dyn Extension>, id: ProviderId) -> Self {
|
|
Self { extension, id }
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl IndexedDocsProvider for ExtensionIndexedDocsProvider {
|
|
fn id(&self) -> ProviderId {
|
|
self.id.clone()
|
|
}
|
|
|
|
fn database_path(&self) -> PathBuf {
|
|
let mut database_path = PathBuf::from(self.extension.work_dir().as_ref());
|
|
database_path.push("docs");
|
|
database_path.push(format!("{}.0.mdb", self.id));
|
|
|
|
database_path
|
|
}
|
|
|
|
async fn suggest_packages(&self) -> Result<Vec<PackageName>> {
|
|
let packages = self
|
|
.extension
|
|
.suggest_docs_packages(self.id.0.clone())
|
|
.await?;
|
|
|
|
Ok(packages
|
|
.into_iter()
|
|
.map(|package| PackageName::from(package.as_str()))
|
|
.collect())
|
|
}
|
|
|
|
async fn index(&self, package: PackageName, database: Arc<IndexedDocsDatabase>) -> Result<()> {
|
|
self.extension
|
|
.index_docs(self.id.0.clone(), package.as_ref().into(), database)
|
|
.await
|
|
}
|
|
}
|