jellyflow-runtime 0.1.0

Headless store, rules, schema, profile, and change pipeline for Jellyflow.
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

use crate::rules::ConnectPlan;
use jellyflow_core::core::{Edge, EdgeId, EdgeKind, Graph, PortId};
use jellyflow_core::ops::{EdgeEndpoints, GraphMutationError, GraphMutationPlanner, GraphOp};

use super::capacity::disconnect_for_endpoint_capacity;
use super::endpoints::ConnectionEndpoints;

pub(in crate::rules::connection) fn reject_mutation_error(
    error: GraphMutationError,
) -> ConnectPlan {
    ConnectPlan::reject(error.to_string())
}

pub(in crate::rules::connection) fn ensure_edge_id_available(
    graph: &Graph,
    edge_id: EdgeId,
) -> Result<(), ConnectPlan> {
    if graph.edges.contains_key(&edge_id) {
        return Err(ConnectPlan::reject(format!(
            "edge already exists: {edge_id:?}"
        )));
    }
    Ok(())
}

pub(in crate::rules::connection) fn add_existing_ports_edge_op(
    graph: &Graph,
    edge_id: EdgeId,
    edge: Edge,
) -> Result<GraphOp, ConnectPlan> {
    GraphMutationPlanner::new(graph)
        .add_edge_op(edge_id, edge)
        .map_err(reject_mutation_error)
}

pub(in crate::rules::connection) fn edge_between(kind: EdgeKind, from: PortId, to: PortId) -> Edge {
    Edge {
        kind,
        from,
        to,
        hidden: false,
        selectable: None,
        focusable: None,
        interaction_width: None,
        deletable: None,
        reconnectable: None,
    }
}

pub(in crate::rules::connection) fn edge_like(edge: &Edge, from: PortId, to: PortId) -> Edge {
    Edge {
        kind: edge.kind,
        from,
        to,
        hidden: edge.hidden,
        selectable: edge.selectable,
        focusable: edge.focusable,
        interaction_width: edge.interaction_width,
        deletable: edge.deletable,
        reconnectable: edge.reconnectable,
    }
}

pub(in crate::rules::connection) fn connection_exists(
    graph: &Graph,
    edge_kind: EdgeKind,
    from: PortId,
    to: PortId,
    skip_edge: Option<EdgeId>,
) -> bool {
    graph.edges.iter().any(|(edge_id, edge)| {
        Some(*edge_id) != skip_edge && edge.kind == edge_kind && edge.from == from && edge.to == to
    })
}

pub(in crate::rules::connection) struct ConnectionOpBuilder {
    ops: Vec<GraphOp>,
}

impl ConnectionOpBuilder {
    pub(in crate::rules::connection) fn with_endpoint_capacity_disconnects(
        graph: &Graph,
        endpoints: &ConnectionEndpoints<'_>,
        skip_edge: Option<EdgeId>,
    ) -> Self {
        Self {
            ops: disconnect_for_endpoint_capacity(graph, endpoints, skip_edge),
        }
    }

    pub(in crate::rules::connection) fn push(&mut self, op: GraphOp) {
        self.ops.push(op);
    }

    pub(in crate::rules::connection) fn extend(&mut self, ops: impl IntoIterator<Item = GraphOp>) {
        self.ops.extend(ops);
    }

    pub(in crate::rules::connection) fn push_set_edge_endpoints(
        &mut self,
        id: EdgeId,
        from: EdgeEndpoints,
        to: EdgeEndpoints,
    ) {
        self.push(GraphOp::SetEdgeEndpoints { id, from, to });
    }

    pub(in crate::rules::connection) fn into_ops(self) -> Vec<GraphOp> {
        self.ops
    }
}