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
// Copyright (c) The Diem Core Contributors
// SPDX-License-Identifier: Apache-2.0

use crate::{
    compilation::compiled_package::CompiledPackage, resolution::resolution_graph::ResolvedGraph,
    source_package::parsed_manifest::PackageName,
};
use anyhow::Result;
use petgraph::algo::toposort;
use std::{collections::BTreeMap, io::Write};

#[derive(Debug, Clone)]
pub struct BuildPlan {
    root: PackageName,
    sorted_deps: Vec<PackageName>,
    resolution_graph: ResolvedGraph,
}

impl BuildPlan {
    pub fn create(resolution_graph: ResolvedGraph) -> Result<Self> {
        let mut sorted_deps = match toposort(&resolution_graph.graph, None) {
            Ok(nodes) => nodes,
            Err(err) => {
                // Is a DAG after resolution otherwise an error should be raised from that.
                anyhow::bail!("IPE: Cyclic dependency found after resolution {:?}", err)
            }
        };

        sorted_deps.reverse();

        Ok(Self {
            root: resolution_graph.root_package.package.name,
            sorted_deps,
            resolution_graph,
        })
    }

    pub fn compile<W: Write>(&self, writer: &mut W) -> Result<CompiledPackage> {
        let package_root = &self.resolution_graph.package_table[&self.root];
        let project_root = &package_root.package_path;
        let mut compiled: BTreeMap<PackageName, CompiledPackage> = BTreeMap::new();
        for package_ident in &self.sorted_deps {
            let resolved_package = self.resolution_graph.get_package(package_ident);
            let dependencies: Vec<_> = resolved_package
                .transitive_dependencies(&self.resolution_graph)
                .into_iter()
                .map(|package_name| compiled.get(&package_name).unwrap().clone())
                .collect();
            let compiled_package = CompiledPackage::build(
                writer,
                project_root,
                resolved_package.clone(),
                dependencies,
                &self.resolution_graph,
            )?;
            compiled.insert(*package_ident, compiled_package);
        }
        Ok(compiled
            .remove(&package_root.source_package.package.name)
            .unwrap())
    }
}