Skip to main content

flowey_lib_hvlite/
build_prep_steps.rs

1// Copyright (c) Microsoft Corporation.
2// Licensed under the MIT License.
3
4//! Build `prep_steps` binaries
5
6use crate::common::CommonProfile;
7use crate::common::CommonTriple;
8use flowey::node::prelude::*;
9
10#[derive(Serialize, Deserialize)]
11#[serde(untagged)]
12pub enum PrepStepsOutput {
13    LinuxBin {
14        #[serde(rename = "prep_steps")]
15        bin: PathBuf,
16        #[serde(rename = "prep_steps.dbg")]
17        dbg: Option<PathBuf>,
18    },
19    WindowsBin {
20        #[serde(rename = "prep_steps.exe")]
21        exe: PathBuf,
22        #[serde(rename = "prep_steps.pdb")]
23        #[serde(default, skip_serializing_if = "Option::is_none")]
24        pdb: Option<PathBuf>,
25    },
26}
27
28impl Artifact for PrepStepsOutput {}
29
30flowey_request! {
31    pub struct Request {
32        pub target: CommonTriple,
33        pub profile: CommonProfile,
34        pub prep_steps: WriteVar<PrepStepsOutput>,
35    }
36}
37
38new_simple_flow_node!(struct Node);
39
40impl SimpleFlowNode for Node {
41    type Request = Request;
42
43    fn imports(ctx: &mut ImportCtx<'_>) {
44        ctx.import::<crate::run_cargo_build::Node>();
45    }
46
47    fn process_request(request: Self::Request, ctx: &mut NodeCtx<'_>) -> anyhow::Result<()> {
48        let Request {
49            target,
50            profile,
51            prep_steps,
52        } = request;
53
54        let output = ctx.reqv(|v| crate::run_cargo_build::Request {
55            crate_name: "prep_steps".into(),
56            out_name: "prep_steps".into(),
57            crate_type: flowey_lib_common::run_cargo_build::CargoCrateType::Bin,
58            profile: profile.into(),
59            features: Default::default(),
60            target: target.as_triple(),
61            no_split_dbg_info: false,
62            extra_env: None,
63            pre_build_deps: Vec::new(),
64            output: v,
65        });
66
67        ctx.emit_minor_rust_step("report built prep_steps", |ctx| {
68            let prep_steps = prep_steps.claim(ctx);
69            let output = output.claim(ctx);
70            move |rt| {
71                let output = match rt.read(output) {
72                    crate::run_cargo_build::CargoBuildOutput::WindowsBin { exe, pdb } => {
73                        PrepStepsOutput::WindowsBin { exe, pdb }
74                    }
75                    crate::run_cargo_build::CargoBuildOutput::ElfBin { bin, dbg } => {
76                        PrepStepsOutput::LinuxBin { bin, dbg }
77                    }
78                    _ => unreachable!(),
79                };
80
81                rt.write(prep_steps, &output);
82            }
83        });
84
85        Ok(())
86    }
87}