flowey_lib_hvlite/
download_release_igvm_files_from_gh.rs

1// Copyright (c) Microsoft Corporation.
2// Licensed under the MIT License.
3
4use flowey::node::prelude::*;
5
6#[derive(Serialize, Deserialize)]
7pub struct ReleaseOutput {
8    // Individual artifact paths (may reside in different directories)
9    pub openhcl_direct: Option<PathBuf>,
10    pub openhcl: Option<PathBuf>,
11    pub openhcl_aarch64: Option<PathBuf>,
12}
13
14impl Artifact for ReleaseOutput {}
15
16#[derive(Serialize, Deserialize, Clone, Debug, Copy)]
17pub enum OpenhclReleaseVersion {
18    Release2411,
19    Release2505,
20    Release2511,
21}
22
23impl OpenhclReleaseVersion {
24    pub fn branch_name(&self) -> String {
25        match self {
26            OpenhclReleaseVersion::Release2411 => "release/2411".to_string(),
27            OpenhclReleaseVersion::Release2505 => "release/2505".to_string(),
28            OpenhclReleaseVersion::Release2511 => "release/1.7.2511".to_string(),
29        }
30    }
31
32    pub const ALL: [OpenhclReleaseVersion; 3] = [
33        OpenhclReleaseVersion::Release2411,
34        OpenhclReleaseVersion::Release2505,
35        OpenhclReleaseVersion::Release2511,
36    ];
37
38    pub fn latest() -> Self {
39        *Self::ALL.last().unwrap()
40    }
41}
42
43impl std::fmt::Display for OpenhclReleaseVersion {
44    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
45        let s = match self {
46            OpenhclReleaseVersion::Release2411 => "release-2411",
47            OpenhclReleaseVersion::Release2505 => "release-2505",
48            OpenhclReleaseVersion::Release2511 => "release-2511",
49        };
50        f.write_str(s)
51    }
52}
53
54pub mod resolve {
55    use super::OpenhclReleaseVersion;
56    use super::ReleaseOutput;
57    use crate::run_cargo_build::common::CommonArch;
58    use flowey::node::prelude::*;
59
60    flowey_request! {
61        pub struct Request {
62            pub release_igvm_files: WriteVar<ReleaseOutput>,
63            pub release_version: OpenhclReleaseVersion,
64            pub arch: CommonArch,
65        }
66    }
67
68    new_simple_flow_node!(struct Node);
69
70    impl SimpleFlowNode for Node {
71        type Request = Request;
72
73        fn imports(ctx: &mut ImportCtx<'_>) {
74            ctx.import::<flowey_lib_common::download_gh_artifact::Node>();
75            ctx.import::<flowey_lib_common::gh_latest_completed_workflow_id::Node>();
76        }
77
78        fn process_request(request: Self::Request, ctx: &mut NodeCtx<'_>) -> anyhow::Result<()> {
79            let branch_name: ReadVar<String> =
80                ReadVar::from_static(request.release_version.branch_name());
81
82            let run_id =
83                ctx.reqv(
84                    |v| flowey_lib_common::gh_latest_completed_workflow_id::Request {
85                        repo: "microsoft/openvmm".into(),
86                        branch: branch_name.clone(),
87                        pipeline_name: "openvmm-ci.yaml".into(),
88                        gh_workflow_id: v,
89                    },
90                );
91            let output = request.release_igvm_files;
92
93            let arch_str = match request.arch {
94                CommonArch::X86_64 => "x64",
95                CommonArch::Aarch64 => "aarch64",
96            };
97
98            let downloaded_artifact =
99                ctx.reqv(|v| flowey_lib_common::download_gh_artifact::Request {
100                    repo_owner: "microsoft".into(),
101                    repo_name: "openvmm".into(),
102                    file_name: format!("{arch_str}-openhcl-igvm"),
103                    path: v,
104                    run_id: run_id.clone(),
105                });
106            let arch = request.arch;
107
108            ctx.emit_rust_step("write to directory variables", |ctx| {
109                let downloaded_artifact = downloaded_artifact.claim(ctx);
110                let write_release_output = output.claim(ctx);
111
112                move |rt| {
113                    let mut openhcl_direct = None;
114                    let mut openhcl = None;
115                    let mut openhcl_aarch64 = None;
116
117                    match arch {
118                        CommonArch::X86_64 => {
119                            // x64 build contains both openhcl.bin and openhcl-direct.bin
120                            let x64_dir = rt.read(downloaded_artifact).join("x64-openhcl-igvm");
121                            openhcl_direct = Some(x64_dir.join("openhcl-direct.bin"));
122                            openhcl = Some(x64_dir.join("openhcl.bin"));
123                        }
124                        CommonArch::Aarch64 => {
125                            let aarch64_dir =
126                                rt.read(downloaded_artifact).join("aarch64-openhcl-igvm");
127                            openhcl_aarch64 = Some(aarch64_dir.join("openhcl-aarch64.bin"));
128                        }
129                    }
130
131                    rt.write_not_secret(
132                        write_release_output,
133                        &ReleaseOutput {
134                            openhcl_direct,
135                            openhcl,
136                            openhcl_aarch64,
137                        },
138                    );
139
140                    Ok(())
141                }
142            });
143
144            Ok(())
145        }
146    }
147}
flowey_lib_hvlite/download_release_igvm_files_from_gh.rs

flowey_lib_hvlite/
download_release_igvm_files_from_gh.rs
