autoschematic-core 0.14.2

Core shared functionality for Autoschematic: workflow engine, state management, and Git integrations
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

use std::{path::Path, sync::Arc};

use crate::{
    config::AutoschematicConfig, connector::FilterResponse, connector_cache::ConnectorCache, error::AutoschematicError,
    keystore::KeyStore,
};

pub enum CheckDriftResult {
    /// Neither the desired (local) resource nor the current (remote) resource exist.
    NeitherExist,
    /// The address provided doesn't correspond to a resource file for any loaded connector
    InvalidAddress,
    /// Both exist, and have drifted; Connector::eq(...) returned false.
    NotEqual {
        current: Option<Vec<u8>>,
        desired: Option<Vec<u8>>,
    },
    /// Both exist, and Connector::eq(...) returned true.
    Equal,
}

pub async fn check_drift(
    autoschematic_config: &AutoschematicConfig,
    connector_cache: &ConnectorCache,
    keystore: Option<Arc<dyn KeyStore>>,
    prefix: &Path,
    virt_addr: &Path,
) -> Result<CheckDriftResult, AutoschematicError> {
    let Some(prefix_name) = prefix.to_str() else {
        return Ok(CheckDriftResult::InvalidAddress);
    };

    let Some(prefix_def) = autoschematic_config.prefixes.get(prefix_name) else {
        return Ok(CheckDriftResult::InvalidAddress);
    };

    let desired_path = prefix.join(virt_addr);

    let desired_state = if desired_path.is_file() {
        Some(tokio::fs::read_to_string(desired_path).await?)
    } else {
        None
    };

    let mut current_state = None;

    for connector_def in &prefix_def.connectors {
        let (connector, _inbox) = connector_cache
            .get_or_spawn_connector(autoschematic_config, prefix_name, connector_def, keystore.clone(), true)
            .await?;

        if connector.filter(virt_addr).await? == FilterResponse::Resource {
            match connector.addr_virt_to_phy(virt_addr).await? {
                crate::connector::VirtToPhyResponse::NotPresent => {
                    current_state = None;
                }
                crate::connector::VirtToPhyResponse::Deferred(_read_outputs) => {
                    current_state = None;
                }
                crate::connector::VirtToPhyResponse::Present(phy_addr) => {
                    if let Some(body) = connector.get(&phy_addr).await? {
                        // return Ok(Some(body.resource_definition));
                        // TODO we also need to check if outputs have drifted!
                        current_state = Some(body.resource_definition);
                    }
                }
                crate::connector::VirtToPhyResponse::Null(virt_addr) => {
                    if let Some(body) = connector.get(&virt_addr).await? {
                        current_state = Some(body.resource_definition);
                        // return Ok(Some(body.resource_definition));
                    }
                }
            }

            match (current_state, desired_state) {
                (None, None) => return Ok(CheckDriftResult::NeitherExist),
                (None, Some(b)) => {
                    return Ok(CheckDriftResult::NotEqual {
                        current: None,
                        desired: Some(b.into()),
                    });
                }
                (Some(a), None) => {
                    return Ok(CheckDriftResult::NotEqual {
                        current: Some(a),
                        desired: None,
                    });
                }
                (Some(a), Some(b)) => match connector.eq(virt_addr, &a, b.as_bytes()).await? {
                    true => return Ok(CheckDriftResult::Equal),
                    false => {
                        return Ok(CheckDriftResult::NotEqual {
                            current: Some(a),
                            desired: Some(b.into()),
                        });
                    }
                },
            }
        }
    }

    Ok(CheckDriftResult::InvalidAddress)
}