app-kit 0.0.2

Application and durable workflow composition for ai-kit
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

use std::collections::BTreeSet;

use serde_json::{Map, Number, Value};

use crate::AppError;
use crate::workflow::StateChange;
use crate::workflow::definition::MAX_COLLECTION_ITEMS;

impl StateChange {
    pub(super) fn apply_values(
        &self,
        before: &Map<String, Value>,
    ) -> Result<Map<String, Value>, AppError> {
        let mut values = if matches!(self, Self::Subset { .. }) {
            Map::new()
        } else {
            before.clone()
        };
        match self {
            Self::Assign { values: entries } => {
                Self::validate_count(entries.len())?;
                for (key, value) in entries {
                    values.insert(key.clone(), value.clone());
                }
            }
            Self::Append { values: entries } => {
                Self::validate_count(entries.len())?;
                for (key, value) in entries {
                    Self::array(&mut values, key)?.push(value.clone());
                }
            }
            Self::Extend { values: entries } => {
                Self::validate_count(entries.len())?;
                for (key, items) in entries {
                    Self::array(&mut values, key)?.extend(items.iter().cloned());
                }
            }
            Self::Increment { values: entries } => {
                Self::validate_count(entries.len())?;
                for (key, value) in entries {
                    let delta = value.as_i64().ok_or_else(|| {
                        AppError::InvalidInput(
                            "state increment requires a signed integer delta".to_owned(),
                        )
                    })?;
                    let prior = match values.get(key) {
                        None => 0,
                        Some(Value::Number(number)) => number.as_i128().ok_or_else(|| {
                            AppError::InvalidInput(
                                "state increment requires an integer value".to_owned(),
                            )
                        })?,
                        Some(_) => {
                            return Err(AppError::InvalidInput(
                                "state increment requires an integer value".to_owned(),
                            ));
                        }
                    };
                    let result = prior
                        .checked_add(i128::from(delta))
                        .and_then(Number::from_i128)
                        .ok_or_else(|| {
                            AppError::InvalidInput(
                                "state increment exceeded the JSON integer range".to_owned(),
                            )
                        })?;
                    values.insert(key.clone(), Value::Number(result));
                }
            }
            Self::Remove { keys } => {
                Self::validate_count(keys.len())?;
                let keys = keys.iter().map(String::as_str).collect::<BTreeSet<_>>();
                values.retain(|key, _| !keys.contains(key.as_str()));
            }
            Self::Wipe { .. } => {
                return Err(AppError::Integration(
                    "wipe must resolve to a removal before mutation".to_owned(),
                ));
            }
            Self::Merge { state } => {
                for (key, value) in state.values() {
                    values.insert(key.clone(), value.clone());
                }
            }
            Self::Subset { keys } => {
                Self::validate_count(keys.len())?;
                for key in keys {
                    if let Some(value) = before.get(key) {
                        values.insert(key.clone(), value.clone());
                    }
                }
            }
        }
        Ok(values)
    }

    fn array<'a>(
        values: &'a mut Map<String, Value>,
        key: &str,
    ) -> Result<&'a mut Vec<Value>, AppError> {
        values
            .entry(key.to_owned())
            .or_insert_with(|| Value::Array(Vec::new()))
            .as_array_mut()
            .ok_or_else(|| {
                AppError::InvalidInput("state array operation requires an array value".to_owned())
            })
    }

    pub(super) fn validate_count(count: usize) -> Result<(), AppError> {
        if count > MAX_COLLECTION_ITEMS {
            return Err(AppError::InvalidInput(
                "state operation exceeds its item bound".to_owned(),
            ));
        }
        Ok(())
    }
}