1use std::path::PathBuf;
 2use std::sync::Arc;
 3
 4use anyhow::{anyhow, Result};
 5use async_trait::async_trait;
 6use futures::FutureExt;
 7use indexed_docs::{IndexedDocsDatabase, IndexedDocsProvider, PackageName, ProviderId};
 8use wasmtime_wasi::WasiView;
 9
10use crate::wasm_host::{WasmExtension, WasmHost};
11
12pub struct ExtensionIndexedDocsProvider {
13    pub(crate) extension: WasmExtension,
14    pub(crate) host: Arc<WasmHost>,
15    pub(crate) id: ProviderId,
16}
17
18#[async_trait]
19impl IndexedDocsProvider for ExtensionIndexedDocsProvider {
20    fn id(&self) -> ProviderId {
21        self.id.clone()
22    }
23
24    fn database_path(&self) -> PathBuf {
25        let mut database_path = self.host.work_dir.clone();
26        database_path.push(self.extension.manifest.id.as_ref());
27        database_path.push("docs");
28        database_path.push(format!("{}.0.mdb", self.id));
29
30        database_path
31    }
32
33    async fn index(&self, package: PackageName, database: Arc<IndexedDocsDatabase>) -> Result<()> {
34        self.extension
35            .call({
36                let id = self.id.clone();
37                |extension, store| {
38                    async move {
39                        let database_resource = store.data_mut().table().push(database)?;
40                        extension
41                            .call_index_docs(
42                                store,
43                                id.as_ref(),
44                                package.as_ref(),
45                                database_resource,
46                            )
47                            .await?
48                            .map_err(|err| anyhow!("{err:?}"))?;
49
50                        anyhow::Ok(())
51                    }
52                    .boxed()
53                }
54            })
55            .await
56    }
57}