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 crate::WasmPlugin;
use proto_core::{
    async_trait, is_offline, is_semantic_version, remove_v_prefix, Describable, ProtoError,
    Resolvable, Tool, VersionManifest, VersionManifestEntry,
};
use proto_pdk_api::{
    LoadVersionsInput, LoadVersionsOutput, ResolveVersionInput, ResolveVersionOutput,
};
use tracing::debug;

#[async_trait]
impl Resolvable<'_> for WasmPlugin {
    fn get_resolved_version(&self) -> &str {
        match self.version.as_ref() {
            Some(version) => version,
            None => "latest",
        }
    }

    async fn load_version_manifest(&self) -> Result<VersionManifest, ProtoError> {
        let mut available: LoadVersionsOutput = self
            .container
            .cache_func_with(
                "load_versions",
                LoadVersionsInput {
                    env: self.get_environment()?,
                    initial: self.get_resolved_version().to_owned(),
                },
            )
            .map_err(|e| ProtoError::Message(e.to_string()))?;

        available.versions.sort_by(|a, d| d.cmp(a));
        available.canary_versions.sort_by(|a, d| d.cmp(a));

        let mut manifest = VersionManifest::default();

        for (alias, version) in available.aliases {
            manifest.aliases.insert(alias, version.to_string());
        }

        manifest.aliases.insert(
            "latest".into(),
            available
                .latest
                .unwrap_or_else(|| available.versions[0].clone())
                .to_string(),
        );

        for version in available.versions {
            manifest.versions.insert(
                version.to_string(),
                VersionManifestEntry {
                    alias: None,
                    version: version.to_string(),
                },
            );
        }

        manifest.inherit_aliases(&self.get_manifest()?.aliases);

        Ok(manifest)
    }

    async fn resolve_version(&mut self, initial_version: &str) -> Result<String, ProtoError> {
        if self.get_resolved_version() != "latest" {
            return Ok(self.get_resolved_version().to_owned());
        }

        let initial_version = remove_v_prefix(initial_version).to_lowercase();

        self.set_version(&initial_version);

        // If offline but we have a fully qualified semantic version,
        // exit early and assume the version is legitimate
        if is_semantic_version(&initial_version) && is_offline() {
            return Ok(initial_version);
        }

        debug!(
            tool = self.get_id(),
            initial_version = initial_version,
            "Resolving a semantic version for \"{}\"",
            initial_version
        );

        let manifest = self.load_version_manifest().await?;
        let mut version = "";

        if self.container.has_func("resolve_version") {
            let resolved: ResolveVersionOutput = self
                .container
                .call_func_with(
                    "resolve_version",
                    ResolveVersionInput {
                        initial: initial_version.to_owned(),
                        env: self.get_environment()?,
                    },
                )
                .map_err(|e| ProtoError::Message(e.to_string()))?;

            if let Some(candidate) = resolved.candidate {
                debug!(
                    tool = self.get_id(),
                    candidate = &candidate,
                    "Received a candidate version or alias to use instead",
                );

                version = manifest.find_version(candidate)?;
            }
        }

        if version.is_empty() {
            version = manifest.find_version(&initial_version)?;
        }

        debug!(tool = self.get_id(), version, "Resolved to {}", version);

        self.set_version(version);

        Ok(version.to_owned())
    }

    fn set_version(&mut self, version: &str) {
        self.version = Some(version.to_owned());
    }
}