wasvy 0.0.9

WASI-powered modding for the Bevy engine
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use anyhow::{Result, bail};
use wasmtime::component::Resource;

use crate::{
    bindings::wasvy::ecs::app::HostQuery,
    host::{WasmHost, WasmQueryResult},
    query::{QueryCursor, QueryId},
    runner::State,
};

pub struct WasmQuery {
    id: QueryId,
    cursor: QueryCursor,
}

impl WasmQuery {
    pub(crate) fn new(id: QueryId) -> Self {
        Self {
            id,
            cursor: QueryCursor::default(),
        }
    }
}

impl HostQuery for WasmHost {
    fn iter(
        &mut self,
        query: Resource<WasmQuery>,
    ) -> std::result::Result<Option<Resource<WasmQueryResult>>, wasmtime::Error> {
        (|| -> Result<_> {
            let State::RunSystem { table, queries, .. } = self.access() else {
                bail!("Query can only be accessed in systems")
            };

            let query = table.get_mut(&query)?;
            let cursor = query.cursor.increment();
            let Some(entity) = cursor.entity(queries, query.id) else {
                // We've reached the end of the results
                return Ok(None);
            };

            let result = WasmQueryResult::new(query.id, entity);
            let result = table.push(result)?;
            Ok(Some(result))
        })()
        .map_err(|err| wasmtime::Error::msg(err.to_string()))
    }

    // Note: this is never guaranteed to be called by the wasi binary
    fn drop(&mut self, query: Resource<WasmQuery>) -> std::result::Result<(), wasmtime::Error> {
        (|| -> Result<()> {
            let _ = self.table().delete(query)?;

            Ok(())
        })()
        .map_err(|err| wasmtime::Error::msg(err.to_string()))
    }
}