awsim-cloudfront 0.1.0

AWS CloudFront emulator for AWSim
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
119
120
121
122
123
124
125
126

use awsim_core::AwsError;
use serde_json::{Value, json};

use crate::{
    ids::{new_etag, now_iso8601},
    state::{CloudFrontFunction, CloudFrontState},
};

fn not_found(name: &str) -> AwsError {
    AwsError::not_found(
        "NoSuchFunctionExists",
        format!("The specified function does not exist: {name}"),
    )
}

fn function_to_value(f: &CloudFrontFunction) -> Value {
    json!({
        "Name": f.name,
        "Status": "DEPLOYED",
        "FunctionConfig": {
            "Comment": f.comment,
            "Runtime": f.runtime,
        },
        "FunctionMetadata": {
            "FunctionARN": format!("arn:aws:cloudfront::123456789012:function/{}", f.name),
            "Stage": f.stage,
            "CreatedTime": f.created_at,
            "LastModifiedTime": f.created_at,
        }
    })
}

pub fn create_function(state: &CloudFrontState, input: &Value) -> Result<Value, AwsError> {
    let name = input
        .get("Name")
        .and_then(|v| v.as_str())
        .ok_or_else(|| AwsError::bad_request("InvalidArgument", "Name is required"))?
        .to_string();

    let cfg = input.get("FunctionConfig").unwrap_or(input);
    let comment = cfg
        .get("Comment")
        .and_then(|v| v.as_str())
        .unwrap_or("")
        .to_string();
    let runtime = cfg
        .get("Runtime")
        .and_then(|v| v.as_str())
        .unwrap_or("cloudfront-js-2.0")
        .to_string();

    let etag = new_etag();
    let f = CloudFrontFunction {
        name: name.clone(),
        stage: "DEVELOPMENT".to_string(),
        comment,
        runtime,
        created_at: now_iso8601(),
        etag: etag.clone(),
    };

    let result = function_to_value(&f);
    state.functions.insert(name, f);

    Ok(json!({
        "__xml_root": "FunctionSummary",
        "ETag": etag,
        "Name": result["Name"].clone(),
        "Status": result["Status"].clone(),
        "FunctionConfig": result["FunctionConfig"].clone(),
        "FunctionMetadata": result["FunctionMetadata"].clone(),
    }))
}

pub fn describe_function(state: &CloudFrontState, name: &str) -> Result<Value, AwsError> {
    let f = state.functions.get(name).ok_or_else(|| not_found(name))?;
    let etag = f.etag.clone();
    let result = function_to_value(&f);
    Ok(json!({
        "__xml_root": "FunctionSummary",
        "ETag": etag,
        "Name": result["Name"].clone(),
        "Status": result["Status"].clone(),
        "FunctionConfig": result["FunctionConfig"].clone(),
        "FunctionMetadata": result["FunctionMetadata"].clone(),
    }))
}

pub fn delete_function(state: &CloudFrontState, name: &str) -> Result<Value, AwsError> {
    if state.functions.remove(name).is_none() {
        return Err(not_found(name));
    }
    Ok(json!({}))
}

pub fn list_functions(state: &CloudFrontState) -> Result<Value, AwsError> {
    let items: Vec<Value> = state
        .functions
        .iter()
        .map(|e| function_to_value(e.value()))
        .collect();
    let qty = items.len();
    Ok(json!({
        "FunctionList": {
            "MaxItems": 100,
            "Quantity": qty,
            "Items": { "FunctionSummary": items }
        }
    }))
}

pub fn publish_function(state: &CloudFrontState, name: &str) -> Result<Value, AwsError> {
    let mut f = state
        .functions
        .get_mut(name)
        .ok_or_else(|| not_found(name))?;
    f.stage = "LIVE".to_string();
    let result = function_to_value(&f);
    Ok(json!({
        "__xml_root": "FunctionSummary",
        "Name": result["Name"].clone(),
        "Status": result["Status"].clone(),
        "FunctionConfig": result["FunctionConfig"].clone(),
        "FunctionMetadata": result["FunctionMetadata"].clone(),
    }))
}