1
use std::collections::HashMap;
2
use std::sync::{Arc, Mutex};
3

            
4
use finance::commodity::{Commodity, CommodityBuilder};
5
use rust_i18n::t;
6
use wasmtime::{Engine, Linker, Module, Store};
7

            
8
use crate::error::HookError;
9
use crate::wrappers::{
10
    DataStore, wrap_wasm_str_list_get, wrap_wasm_str_map, wrap_wasm_str_transform,
11
};
12

            
13
56
async fn update(
14
56
    commodity: Commodity,
15
56
    tags: Arc<Mutex<HashMap<String, String>>>,
16
56
    script: &[u8],
17
56
) -> Result<Commodity, HookError> {
18
56
    let engine = Engine::default();
19
56
    let mut linker = Linker::new(&engine);
20
56
    let mut builder = CommodityBuilder::new();
21
56
    builder.id(commodity.id);
22

            
23
56
    let store_data = Arc::new(DataStore {
24
56
        data: Mutex::new((builder, tags.clone())),
25
56
        memory: Mutex::new(None),
26
56
    });
27

            
28
56
    let mut store = Store::new(&engine, store_data.clone());
29

            
30
56
    let module = Module::new(store.engine(), script).map_err(|e| {
31
        tracing::error!("{}: {}", t!("Module creation error"), e);
32
        HookError::WASM(e)
33
    })?;
34

            
35
56
    let get_tag_list = wrap_wasm_str_list_get(&mut store, |data| {
36
56
        let tags = data
37
56
            .1
38
56
            .lock()
39
56
            .map_err(|_| anyhow::anyhow!("Failed to lock tags"))?;
40
56
        tracing::info!("{:?}", tags.keys().cloned());
41
56
        Ok(tags.keys().cloned().collect())
42
56
    });
43

            
44
56
    linker.define(&store, "env", "get_tag_list", get_tag_list)?;
45

            
46
84
    let update_tag = wrap_wasm_str_map(&mut store, |data, key, value| {
47
84
        let mut tags = data
48
84
            .1
49
84
            .lock()
50
84
            .map_err(|_| anyhow::anyhow!("Failed to lock tags"))?;
51
84
        tracing::info!("Inserting {key:?} {value:?}");
52
84
        tags.insert(key, value);
53
84
        Ok(())
54
84
    });
55

            
56
56
    linker.define(&store, "env", "update_tag", update_tag)?;
57

            
58
56
    let get_tag = wrap_wasm_str_transform(&mut store, |data, key| {
59
        // Lock the tags HashMap
60
28
        let tags = data
61
28
            .1
62
28
            .lock()
63
28
            .map_err(|_| anyhow::anyhow!("Failed to lock tags HashMap"))?;
64

            
65
28
        tracing::info!("Found value: {:?}", tags.get(&key).cloned());
66
        // Fetch the value corresponding to the key
67
28
        tags.get(&key)
68
28
            .cloned()
69
28
            .ok_or_else(|| anyhow::anyhow!("Key not found in tags: {key}"))
70
28
    });
71

            
72
56
    linker.define(&store, "env", "get_tag", get_tag)?;
73

            
74
56
    let instance = linker.instantiate(&mut store, &module)?;
75
56
    if let Some(memory) = instance.get_memory(&mut store, "memory") {
76
56
        store_data.memory.lock()?.replace(memory);
77
    } else {
78
        tracing::error!("{}", t!("WASM memory access problem"));
79
        return Err(HookError::WASMMem);
80
    }
81

            
82
56
    let process = instance.get_typed_func::<(), ()>(&mut store, "process")?;
83
56
    process.call(&mut store, ())?;
84

            
85
56
    tracing::info!("{:?}", store_data.clone().data.lock()?.1);
86

            
87
56
    let result = store_data.clone().data.lock()?.0.build()?;
88
56
    Ok(result)
89
56
}
90

            
91
56
pub async fn apply_commodity_hook(
92
56
    commodity: Commodity,
93
56
    tags: Arc<Mutex<HashMap<String, String>>>,
94
56
    hook: &[u8],
95
56
) -> Result<Commodity, HookError> {
96
56
    let updated = update(commodity, tags, hook).await?;
97
56
    Ok(updated)
98
56
}