revive/crates/solidity/src/missing_libraries.rs

//! The missing Solidity libraries.

use std::collections::BTreeMap;
use std::collections::HashSet;

use revive_solc_json_interface::SolcStandardJsonOutput;

use crate::solc::version::Version as SolcVersion;
use crate::ResolcVersion;

/// The missing Solidity libraries.
pub struct MissingLibraries {
    /// The missing libraries.
    pub contract_libraries: BTreeMap<String, HashSet<String>>,
}

impl MissingLibraries {
    /// A shortcut constructor.
    pub fn new(contract_libraries: BTreeMap<String, HashSet<String>>) -> Self {
        Self { contract_libraries }
    }

    /// Writes the missing libraries to the standard JSON.
    pub fn write_to_standard_json(
        mut self,
        standard_json: &mut SolcStandardJsonOutput,
        solc_version: &SolcVersion,
    ) -> anyhow::Result<()> {
        let contracts = match standard_json.contracts.as_mut() {
            Some(contracts) => contracts,
            None => return Ok(()),
        };

        for (path, contracts) in contracts.iter_mut() {
            for (name, contract) in contracts.iter_mut() {
                let full_name = format!("{path}:{name}");
                let missing_libraries = self.contract_libraries.remove(full_name.as_str());

                if let Some(missing_libraries) = missing_libraries {
                    contract.missing_libraries = Some(missing_libraries);
                }
            }
        }

        standard_json.version = Some(solc_version.default.to_string());
        standard_json.long_version = Some(solc_version.long.to_owned());
        standard_json.revive_version = Some(ResolcVersion::default().long);

        Ok(())
    }
}