foundry_common/

contracts.rs

//! Commonly used contract types and functions.

use crate::{compile::PathOrContractInfo, find_metadata_start, strip_bytecode_placeholders};
use alloy_dyn_abi::JsonAbiExt;
use alloy_json_abi::{Event, Function, JsonAbi};
use alloy_primitives::{Address, B256, Bytes, Selector, address, hex};
use eyre::{OptionExt, Result};
use foundry_compilers::{
    ArtifactId, Project, ProjectCompileOutput,
    artifacts::{
        BytecodeObject, CompactBytecode, CompactContractBytecode, CompactContractBytecodeCow,
        CompactDeployedBytecode, ConfigurableContractArtifact, ContractBytecodeSome, Offsets,
        StorageLayout,
    },
    utils::canonicalized,
};
use std::{
    collections::BTreeMap,
    ops::Deref,
    path::{Path, PathBuf},
    str::FromStr,
    sync::Arc,
};

/// Libraries' runtime code always starts with the following instruction:
/// `PUSH20 0x0000000000000000000000000000000000000000`
///
/// See: <https://docs.soliditylang.org/en/latest/contracts.html#call-protection-for-libraries>
const CALL_PROTECTION_BYTECODE_PREFIX: [u8; 21] =
    hex!("730000000000000000000000000000000000000000");

/// Isolated account used to deploy libraries needed only while executing locally.
///
/// `address(uint160(uint256(keccak256("foundry library deployer"))))`
pub const LIBRARY_DEPLOYER: Address = address!("0x1F95D37F27EA0dEA9C252FC09D5A6eaA97647353");

/// Subset of [CompactBytecode] excluding sourcemaps.
#[expect(missing_docs)]
#[derive(Debug, Clone)]
pub struct BytecodeData {
    pub object: Option<BytecodeObject>,
    pub link_references: BTreeMap<String, BTreeMap<String, Vec<Offsets>>>,
    pub immutable_references: BTreeMap<String, Vec<Offsets>>,
}

impl BytecodeData {
    fn bytes(&self) -> Option<&Bytes> {
        self.object.as_ref().and_then(|b| b.as_bytes())
    }
}

/// Sorts and merges overlapping or adjacent byte ranges.
fn normalize_offsets(offsets: &mut Vec<Offsets>) {
    offsets.sort_by_key(|o| o.start);

    let mut merged: Vec<Offsets> = Vec::with_capacity(offsets.len());
    for offset in offsets.drain(..) {
        if let Some(last) = merged.last_mut() {
            let last_end = last.start as u64 + last.length as u64;
            if offset.start as u64 <= last_end {
                let this_end = offset.start as u64 + offset.length as u64;
                if this_end > last_end {
                    last.length = (this_end - last.start as u64) as u32;
                }
                continue;
            }
        }
        merged.push(offset);
    }
    *offsets = merged;
}

impl From<CompactBytecode> for BytecodeData {
    fn from(bytecode: CompactBytecode) -> Self {
        Self {
            object: Some(bytecode.object),
            link_references: bytecode.link_references,
            immutable_references: BTreeMap::new(),
        }
    }
}

impl From<CompactDeployedBytecode> for BytecodeData {
    fn from(bytecode: CompactDeployedBytecode) -> Self {
        let (object, link_references) = if let Some(compact) = bytecode.bytecode {
            (Some(compact.object), compact.link_references)
        } else {
            (None, BTreeMap::new())
        };
        Self { object, link_references, immutable_references: bytecode.immutable_references }
    }
}

/// Container for commonly used contract data.
#[derive(Debug)]
pub struct ContractData {
    /// Contract name.
    pub name: String,
    /// Contract ABI.
    pub abi: JsonAbi,
    /// Contract creation code.
    pub bytecode: Option<BytecodeData>,
    /// Contract runtime code.
    pub deployed_bytecode: Option<BytecodeData>,
    /// Contract storage layout, if available.
    pub storage_layout: Option<Arc<StorageLayout>>,
}

impl ContractData {
    /// Returns reference to bytes of contract creation code, if present.
    pub fn bytecode(&self) -> Option<&Bytes> {
        self.bytecode.as_ref()?.bytes().filter(|b| !b.is_empty())
    }

    /// Returns reference to bytes of contract deployed code, if present.
    pub fn deployed_bytecode(&self) -> Option<&Bytes> {
        self.deployed_bytecode.as_ref()?.bytes().filter(|b| !b.is_empty())
    }

    /// Returns the bytecode without placeholders, if present.
    pub fn bytecode_without_placeholders(&self) -> Option<Bytes> {
        strip_bytecode_placeholders(self.bytecode.as_ref()?.object.as_ref()?)
    }

    /// Returns the deployed bytecode without placeholders, if present.
    pub fn deployed_bytecode_without_placeholders(&self) -> Option<Bytes> {
        strip_bytecode_placeholders(self.deployed_bytecode.as_ref()?.object.as_ref()?)
    }
}

/// Builder for creating a `ContractsByArtifact` instance, optionally including storage layouts
/// from project compile output.
pub struct ContractsByArtifactBuilder<'a> {
    /// All compiled artifact bytecodes (borrowed).
    artifacts: BTreeMap<ArtifactId, CompactContractBytecodeCow<'a>>,
    /// Optionally collected storage layouts for matching artifact IDs.
    storage_layouts: BTreeMap<ArtifactId, StorageLayout>,
}

impl<'a> ContractsByArtifactBuilder<'a> {
    /// Creates a new builder from artifacts with present bytecode iterator.
    pub fn new(
        artifacts: impl IntoIterator<Item = (ArtifactId, CompactContractBytecodeCow<'a>)>,
    ) -> Self {
        Self { artifacts: artifacts.into_iter().collect(), storage_layouts: BTreeMap::new() }
    }

    /// Add storage layouts from the given `ProjectCompileOutput` to known artifacts.
    pub fn with_output(self, output: &ProjectCompileOutput, base: &Path) -> Self {
        self.with_storage_layouts(output.artifact_ids().filter_map(|(id, artifact)| {
            artifact
                .storage_layout
                .as_ref()
                .map(|layout| (id.with_stripped_file_prefixes(base), layout.clone()))
        }))
    }

    /// Add storage layouts.
    pub fn with_storage_layouts(
        mut self,
        layouts: impl IntoIterator<Item = (ArtifactId, StorageLayout)>,
    ) -> Self {
        self.storage_layouts.extend(layouts);
        self
    }

    /// Builds `ContractsByArtifact`.
    pub fn build(self) -> ContractsByArtifact {
        let map = self
            .artifacts
            .into_iter()
            .filter_map(|(id, artifact)| {
                let name = id.name.clone();
                let CompactContractBytecodeCow { abi, bytecode, deployed_bytecode } = artifact;

                Some((
                    id.clone(),
                    ContractData {
                        name,
                        abi: abi?.into_owned(),
                        bytecode: bytecode.map(|b| b.into_owned().into()),
                        deployed_bytecode: deployed_bytecode.map(|b| b.into_owned().into()),
                        storage_layout: self.storage_layouts.get(&id).map(|l| Arc::new(l.clone())),
                    },
                ))
            })
            .collect();

        ContractsByArtifact(Arc::new(map))
    }
}

type ArtifactWithContractRef<'a> = (&'a ArtifactId, &'a ContractData);

/// Wrapper type that maps an artifact to a contract ABI and bytecode.
#[derive(Clone, Default, Debug)]
pub struct ContractsByArtifact(Arc<BTreeMap<ArtifactId, ContractData>>);

impl ContractsByArtifact {
    /// Creates a new instance by collecting all artifacts with present bytecode from an iterator.
    pub fn new(artifacts: impl IntoIterator<Item = (ArtifactId, CompactContractBytecode)>) -> Self {
        let map = artifacts
            .into_iter()
            .filter_map(|(id, artifact)| {
                let name = id.name.clone();
                let CompactContractBytecode { abi, bytecode, deployed_bytecode } = artifact;
                Some((
                    id,
                    ContractData {
                        name,
                        abi: abi?,
                        bytecode: bytecode.map(Into::into),
                        deployed_bytecode: deployed_bytecode.map(Into::into),
                        storage_layout: None,
                    },
                ))
            })
            .collect();
        Self(Arc::new(map))
    }

    /// Clears all contracts.
    pub fn clear(&mut self) {
        *self = Self::default();
    }

    /// Finds a contract which has a similar bytecode as `code`.
    pub fn find_by_creation_code(&self, code: &[u8]) -> Option<ArtifactWithContractRef<'_>> {
        self.find_by_code(code, 0.1, true, ContractData::bytecode)
    }

    /// Finds a contract which has a similar deployed bytecode as `code`.
    pub fn find_by_deployed_code(&self, code: &[u8]) -> Option<ArtifactWithContractRef<'_>> {
        self.find_by_code(code, 0.15, false, ContractData::deployed_bytecode)
    }

    /// Finds a contract based on provided bytecode and accepted match score.
    /// If strip constructor args flag is true then removes args from bytecode to compare.
    fn find_by_code(
        &self,
        code: &[u8],
        accepted_score: f64,
        strip_ctor_args: bool,
        get: impl Fn(&ContractData) -> Option<&Bytes>,
    ) -> Option<ArtifactWithContractRef<'_>> {
        self.iter()
            .filter_map(|(id, contract)| {
                if let Some(deployed_bytecode) = get(contract) {
                    let mut code = code;
                    if strip_ctor_args && code.len() > deployed_bytecode.len() {
                        // Try to decode ctor args with contract abi.
                        if let Some(constructor) = contract.abi.constructor() {
                            let constructor_args = &code[deployed_bytecode.len()..];
                            if constructor.abi_decode_input(constructor_args).is_ok() {
                                // If we can decode args with current abi then remove args from
                                // code to compare.
                                code = &code[..deployed_bytecode.len()]
                            }
                        }
                    };

                    let score = bytecode_diff_score(deployed_bytecode.as_ref(), code);
                    (score <= accepted_score).then_some((score, (id, contract)))
                } else {
                    None
                }
            })
            .min_by(|(score1, _), (score2, _)| score1.total_cmp(score2))
            .map(|(_, data)| data)
    }

    /// Finds a contract which deployed bytecode exactly matches the given code. Accounts for link
    /// references and immutables.
    pub fn find_by_deployed_code_exact(&self, code: &[u8]) -> Option<ArtifactWithContractRef<'_>> {
        self.find_by_deployed_code_exact_preferred(code, |_| true)
    }

    /// Same as [`Self::find_by_deployed_code_exact`], preferring artifacts matching `preferred`
    /// only among equally strong matches. Metadata-exact matches always take precedence.
    pub fn find_by_deployed_code_exact_preferred(
        &self,
        code: &[u8],
        preferred: impl Fn(&ArtifactId) -> bool,
    ) -> Option<ArtifactWithContractRef<'_>> {
        self.find_by_deployed_code_exact_inner(code, false, preferred).0
    }

    /// Finds the only contract whose deployed bytecode exactly matches the given code.
    pub fn find_by_deployed_code_exact_unique(
        &self,
        code: &[u8],
    ) -> Option<ArtifactWithContractRef<'_>> {
        self.find_by_deployed_code_exact_inner(code, true, |_| false).0
    }

    /// Finds a deployed-code match, preferring exact metadata matches and rejecting ambiguous
    /// partial metadata matches. A single partial match is accepted.
    pub fn find_by_deployed_code_exact_unambiguous(
        &self,
        code: &[u8],
    ) -> Result<Option<ArtifactWithContractRef<'_>>> {
        let (matched, ambiguous_partial) =
            self.find_by_deployed_code_exact_inner(code, false, |_| true);
        if ambiguous_partial {
            eyre::bail!(
                "Multiple local contracts match the deployed bytecode with different metadata. Specify the contract as <path>:<contract>"
            );
        }
        Ok(matched)
    }

    fn find_by_deployed_code_exact_inner(
        &self,
        code: &[u8],
        unique: bool,
        preferred: impl Fn(&ArtifactId) -> bool,
    ) -> (Option<ArtifactWithContractRef<'_>>, bool) {
        // Immediately return None if the code is empty.
        if code.is_empty() {
            return (None, false);
        }

        let mut partial_match = None;
        let mut exact_match = None;
        let mut ambiguous_partial = false;
        'contracts: for (id, contract) in self.iter() {
            let Some(deployed_bytecode) = &contract.deployed_bytecode else {
                continue;
            };
            let Some(deployed_code) = &deployed_bytecode.object else {
                continue;
            };

            let len = match deployed_code {
                BytecodeObject::Bytecode(bytes) => bytes.len(),
                BytecodeObject::Unlinked(bytes) => bytes.len() / 2,
            };

            // Vyper artifacts contain the complete runtime, but omit the immutable data
            // appended on deployment. Compare that runtime in full, without treating any of
            // its bytes (or the immutable suffix) as Solidity metadata.
            let is_vyper = id.source.extension().is_some_and(|extension| extension == "vy");
            let code = if is_vyper {
                let Some(runtime) = code.get(..len) else { continue };
                runtime
            } else {
                code
            };
            if len == 0 || len != code.len() {
                continue;
            }

            // Collect ignored offsets by chaining link and immutable references.
            let mut ignored = deployed_bytecode
                .immutable_references
                .values()
                .chain(deployed_bytecode.link_references.values().flat_map(|v| v.values()))
                .flatten()
                .cloned()
                .collect::<Vec<_>>();

            // For libraries solidity adds a call protection prefix to the bytecode. We need to
            // ignore it as it includes library address determined at runtime.
            // See https://docs.soliditylang.org/en/latest/contracts.html#call-protection-for-libraries and
            // https://github.com/NomicFoundation/hardhat/blob/af7807cf38842a4f56e7f4b966b806e39631568a/packages/hardhat-verify/src/internal/solc/bytecode.ts#L172
            let has_call_protection = !is_vyper
                && match deployed_code {
                    BytecodeObject::Bytecode(bytes) => {
                        bytes.starts_with(&CALL_PROTECTION_BYTECODE_PREFIX)
                    }
                    BytecodeObject::Unlinked(bytes) => {
                        if let Ok(bytes) =
                            Bytes::from_str(&bytes[..CALL_PROTECTION_BYTECODE_PREFIX.len() * 2])
                        {
                            bytes.starts_with(&CALL_PROTECTION_BYTECODE_PREFIX)
                        } else {
                            false
                        }
                    }
                };

            if has_call_protection {
                ignored.push(Offsets { start: 1, length: 20 });
            }

            let metadata_start = if is_vyper { None } else { find_metadata_start(code) };

            if let Some(metadata) = metadata_start {
                ignored.push(Offsets {
                    start: metadata as u32,
                    length: (code.len() - metadata) as u32,
                });
            }

            // Merge ranges from independent sources before slicing between them.
            normalize_offsets(&mut ignored);

            let mut left = 0;
            for offset in ignored {
                let right = offset.start as usize;

                let matched = match deployed_code {
                    BytecodeObject::Bytecode(bytes) => bytes[left..right] == code[left..right],
                    BytecodeObject::Unlinked(bytes) => {
                        if let Ok(bytes) = Bytes::from_str(&bytes[left * 2..right * 2]) {
                            bytes == code[left..right]
                        } else {
                            false
                        }
                    }
                };

                if !matched {
                    continue 'contracts;
                }

                left = right + offset.length as usize;
            }

            let is_partial = if left < code.len() {
                match deployed_code {
                    BytecodeObject::Bytecode(bytes) => bytes[left..] == code[left..],
                    BytecodeObject::Unlinked(bytes) => {
                        if let Ok(bytes) = Bytes::from_str(&bytes[left * 2..]) {
                            bytes == code[left..]
                        } else {
                            false
                        }
                    }
                }
            } else {
                true
            };

            if !is_partial {
                continue;
            }

            let matches_metadata = metadata_start.is_none_or(|metadata| match deployed_code {
                BytecodeObject::Bytecode(bytes) => bytes[metadata..] == code[metadata..],
                BytecodeObject::Unlinked(bytes) => Bytes::from_str(&bytes[metadata * 2..])
                    .is_ok_and(|bytes| bytes == code[metadata..]),
            });

            if matches_metadata {
                if unique && exact_match.is_some() {
                    return (None, false);
                }
                if exact_match.is_none() || preferred(id) {
                    exact_match = Some((id, contract));
                }
                if !unique && preferred(id) {
                    return (exact_match, false);
                }
            } else {
                ambiguous_partial |= partial_match.is_some();
                if preferred(id) || partial_match.is_none_or(|(id, _)| !preferred(id)) {
                    partial_match = Some((id, contract));
                }
            }
        }

        let ambiguous_partial = exact_match.is_none() && ambiguous_partial;
        (if unique { exact_match } else { exact_match.or(partial_match) }, ambiguous_partial)
    }

    /// Finds a contract which has the same contract name or identifier as `id`. If more than one is
    /// found, return error.
    pub fn find_by_name_or_identifier(
        &self,
        id: &str,
    ) -> Result<Option<ArtifactWithContractRef<'_>>> {
        let mut iter =
            self.iter().filter(|(artifact, _)| artifact.name == id || artifact.identifier() == id);
        let first = iter.next();
        if first.is_some() && iter.next().is_some() {
            eyre::bail!("{id} has more than one implementation.");
        }

        Ok(first)
    }

    /// Finds abi by name or source path
    ///
    /// Returns the abi and the contract name.
    pub fn find_abi_by_name_or_src_path(&self, name_or_path: &str) -> Option<(JsonAbi, String)> {
        self.iter()
            .find(|(artifact, _)| {
                artifact.name == name_or_path || artifact.source == Path::new(name_or_path)
            })
            .map(|(_, contract)| (contract.abi.clone(), contract.name.clone()))
    }

    /// Flattens the contracts into functions, events and errors.
    pub fn flatten(&self) -> (BTreeMap<Selector, Function>, BTreeMap<B256, Event>, JsonAbi) {
        let mut funcs = BTreeMap::new();
        let mut events = BTreeMap::new();
        let mut errors_abi = JsonAbi::new();
        for contract in self.values() {
            for func in contract.abi.functions() {
                funcs.insert(func.selector(), func.clone());
            }
            for event in contract.abi.events() {
                events.insert(event.selector(), event.clone());
            }
            for error in contract.abi.errors() {
                errors_abi.errors.entry(error.name.clone()).or_default().push(error.clone());
            }
        }
        (funcs, events, errors_abi)
    }
}

impl From<ProjectCompileOutput> for ContractsByArtifact {
    fn from(value: ProjectCompileOutput) -> Self {
        Self::new(value.into_artifacts().map(|(id, ar)| {
            (
                id,
                CompactContractBytecode {
                    abi: ar.abi,
                    bytecode: ar.bytecode,
                    deployed_bytecode: ar.deployed_bytecode,
                },
            )
        }))
    }
}

impl Deref for ContractsByArtifact {
    type Target = BTreeMap<ArtifactId, ContractData>;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

/// Wrapper type that maps an address to a contract identifier and contract ABI.
pub type ContractsByAddress = BTreeMap<Address, (String, JsonAbi)>;

/// Very simple fuzzy matching of contract bytecode.
///
/// Returns a value between `0.0` (identical) and `1.0` (completely different).
/// Returns whether `creation_code` is exactly this contract's linked creation bytecode followed by
/// a complete, canonically encoded constructor argument tuple.
pub fn matches_contract_creation(contract: &ContractData, creation_code: &[u8]) -> bool {
    let Some(bytecode) = contract.bytecode() else { return false };
    let Some(arguments) = creation_code.strip_prefix(bytecode.as_ref()) else { return false };
    match contract.abi.constructor() {
        Some(constructor) => constructor
            .abi_decode_input(arguments)
            .ok()
            .and_then(|values| constructor.abi_encode_input(&values).ok())
            .is_some_and(|encoded| encoded == arguments),
        None => arguments.is_empty(),
    }
}

pub fn bytecode_diff_score<'a>(mut a: &'a [u8], mut b: &'a [u8]) -> f64 {
    // Make sure `a` is the longer one.
    if a.len() < b.len() {
        std::mem::swap(&mut a, &mut b);
    }

    // Account for different lengths.
    let mut n_different_bytes = a.len() - b.len();

    // If the difference is more than 32 bytes and more than 10% of the total length,
    // we assume the bytecodes are completely different.
    // This is a simple heuristic to avoid checking every byte when the lengths are very different.
    // 32 is chosen to be a reasonable minimum as it's the size of metadata hashes and one EVM word.
    if n_different_bytes > 32 && n_different_bytes * 10 > a.len() {
        return 1.0;
    }

    // Count different bytes.
    // SAFETY: `a` is longer than `b`.
    n_different_bytes += unsafe { count_different_bytes(a, b) };

    n_different_bytes as f64 / a.len() as f64
}

/// Returns the amount of different bytes between two slices.
///
/// # Safety
///
/// `a` must be at least as long as `b`.
const unsafe fn count_different_bytes(a: &[u8], b: &[u8]) -> usize {
    // This could've been written as `std::iter::zip(a, b).filter(|(x, y)| x != y).count()`,
    // however this function is very hot, and has been written to be as primitive as
    // possible for lower optimization levels.

    let a_ptr = a.as_ptr();
    let b_ptr = b.as_ptr();
    let len = b.len();

    let mut sum = 0;
    let mut i = 0;
    while i < len {
        // SAFETY: `a` is at least as long as `b`, and `i` is in bound of `b`.
        sum += unsafe { *a_ptr.add(i) != *b_ptr.add(i) } as usize;
        i += 1;
    }
    sum
}

/// Returns contract name for a given contract identifier.
///
/// Artifact/Contract identifier can take the following form:
/// `<artifact file name>:<contract name>`, the `artifact file name` is the name of the json file of
/// the contract's artifact and the contract name is the name of the solidity contract, like
/// `SafeTransferLibTest.json:SafeTransferLibTest`
///
/// This returns the `contract name` part
///
/// # Example
///
/// ```
/// use foundry_common::*;
/// assert_eq!(
///     "SafeTransferLibTest",
///     get_contract_name("SafeTransferLibTest.json:SafeTransferLibTest")
/// );
/// ```
pub fn get_contract_name(id: &str) -> &str {
    id.rsplit(':').next().unwrap_or(id)
}

/// This returns the `file name` part, See [`get_contract_name`]
///
/// # Example
///
/// ```
/// use foundry_common::*;
/// assert_eq!(
///     "SafeTransferLibTest.json",
///     get_file_name("SafeTransferLibTest.json:SafeTransferLibTest")
/// );
/// ```
pub fn get_file_name(id: &str) -> &str {
    id.split(':').next().unwrap_or(id)
}

/// Helper function to convert CompactContractBytecode ~> ContractBytecodeSome
pub fn compact_to_contract(contract: CompactContractBytecode) -> Result<ContractBytecodeSome> {
    Ok(ContractBytecodeSome {
        abi: contract.abi.ok_or_else(|| eyre::eyre!("No contract abi"))?,
        bytecode: contract.bytecode.ok_or_else(|| eyre::eyre!("No contract bytecode"))?.into(),
        deployed_bytecode: contract
            .deployed_bytecode
            .ok_or_else(|| eyre::eyre!("No contract deployed bytecode"))?
            .into(),
    })
}

/// Returns the canonicalized target path for the given identifier.
pub fn find_target_path(project: &Project, identifier: &PathOrContractInfo) -> Result<PathBuf> {
    match identifier {
        PathOrContractInfo::Path(path) => Ok(canonicalized(project.root().join(path))),
        PathOrContractInfo::ContractInfo(info) => {
            if let Some(path) = info.path.as_ref() {
                let path = canonicalized(project.root().join(path));
                if !path.is_file() {
                    eyre::bail!(
                        "Could not find source file for contract `{}` at {}",
                        info.name,
                        path.strip_prefix(project.root()).unwrap_or(&path).display()
                    );
                }
                return Ok(path);
            }
            // If ContractInfo.path hasn't been provided we try to find the contract using the name.
            // This will fail if projects have multiple contracts with the same name. In that case,
            // path must be specified.
            let path = project.find_contract_path(&info.name)?;
            Ok(path)
        }
    }
}

/// Returns the target artifact given the path and name.
pub fn find_matching_contract_artifact(
    output: &mut ProjectCompileOutput,
    target_path: &Path,
    target_name: Option<&str>,
) -> eyre::Result<ConfigurableContractArtifact> {
    if let Some(name) = target_name {
        if let Some(artifact) = output.remove(target_path, name) {
            return Ok(artifact);
        }

        let target_path = canonicalized(target_path);
        let matching_source = output.artifact_ids().find_map(|(id, _artifact)| {
            (id.name == name && canonicalized(&id.source) == target_path).then(|| id.source.clone())
        });

        matching_source
            .and_then(|source| output.remove(&source, name))
            .ok_or_eyre(format!("Could not find artifact `{name}` in the compiled artifacts"))
    } else {
        let possible_targets = output
            .artifact_ids()
            .filter(|(id, _artifact)| id.source == target_path)
            .collect::<Vec<_>>();

        if possible_targets.is_empty() {
            eyre::bail!(
                "Could not find artifact linked to source `{target_path:?}` in the compiled artifacts"
            );
        }

        let (target_id, target_artifact) = possible_targets[0].clone();
        if possible_targets.len() == 1 {
            return Ok(target_artifact.clone());
        }

        // If all artifact_ids in `possible_targets` have the same name (without ".", indicates
        // additional compiler profiles), it means that there are multiple contracts in the
        // same file.
        if !target_id.name.contains('.')
            && possible_targets.iter().any(|(id, _)| id.name != target_id.name)
        {
            eyre::bail!(
                "Multiple contracts found in the same file, please specify the target <path>:<contract> or <contract>"
            );
        }

        // Otherwise, we're dealing with additional compiler profiles wherein `id.source` is the
        // same but `id.path` is different.
        let artifact = possible_targets
            .iter()
            .find_map(|(id, artifact)| (id.profile == "default").then_some(*artifact))
            .unwrap_or(target_artifact);

        Ok(artifact.clone())
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use alloy_dyn_abi::DynSolValue;
    use alloy_primitives::U256;
    use semver::Version;

    fn deployed_artifact(name: &str, code: Bytes) -> (ArtifactId, CompactContractBytecode) {
        deployed_artifact_with_immutables(name, code, Default::default())
    }

    fn deployed_artifact_with_immutables(
        name: &str,
        code: Bytes,
        immutable_references: BTreeMap<String, Vec<Offsets>>,
    ) -> (ArtifactId, CompactContractBytecode) {
        (
            ArtifactId {
                path: format!("out/{name}.json").into(),
                name: name.to_owned(),
                source: format!("src/{name}.sol").into(),
                version: Version::new(0, 8, 30),
                build_id: String::new(),
                profile: "default".to_owned(),
            },
            CompactContractBytecode {
                abi: Some(Default::default()),
                bytecode: None,
                deployed_bytecode: Some(CompactDeployedBytecode {
                    bytecode: Some(CompactBytecode {
                        object: BytecodeObject::Bytecode(code),
                        source_map: None,
                        link_references: Default::default(),
                    }),
                    immutable_references,
                }),
            },
        )
    }

    #[test]
    fn exact_creation_match_requires_canonical_constructor_suffix() {
        let abi = JsonAbi::parse(["constructor(uint256 value)"]).unwrap();
        let bytecode = Bytes::from_static(&[0x60, 0x00]);
        let contract = ContractData {
            name: "C".to_owned(),
            abi,
            bytecode: Some(BytecodeData {
                object: Some(BytecodeObject::Bytecode(bytecode.clone())),
                link_references: BTreeMap::new(),
                immutable_references: BTreeMap::new(),
            }),
            deployed_bytecode: None,
            storage_layout: None,
        };
        let arguments = contract
            .abi
            .constructor()
            .unwrap()
            .abi_encode_input(&[DynSolValue::Uint(U256::ONE, 256)])
            .unwrap();
        let creation = [bytecode.as_ref(), &arguments].concat();

        assert!(matches_contract_creation(&contract, &creation));
        assert!(!matches_contract_creation(&contract, &creation[..creation.len() - 1]));
        assert!(!matches_contract_creation(&contract, &[creation, vec![0]].concat()));
    }

    #[test]
    fn bytecode_diffing() {
        assert_eq!(bytecode_diff_score(b"a", b"a"), 0.0);
        assert_eq!(bytecode_diff_score(b"a", b"b"), 1.0);

        let a_100 = &b"a".repeat(100)[..];
        assert_eq!(bytecode_diff_score(a_100, &b"b".repeat(100)), 1.0);
        assert_eq!(bytecode_diff_score(a_100, &b"b".repeat(99)), 1.0);
        assert_eq!(bytecode_diff_score(a_100, &b"b".repeat(101)), 1.0);
        assert_eq!(bytecode_diff_score(a_100, &b"b".repeat(120)), 1.0);
        assert_eq!(bytecode_diff_score(a_100, &b"b".repeat(1000)), 1.0);

        let a_99 = &b"a".repeat(99)[..];
        assert!(bytecode_diff_score(a_100, a_99) <= 0.01);
    }

    #[test]
    fn find_by_deployed_code_exact_with_empty_deployed() {
        let contracts = ContractsByArtifact::new(vec![]);

        assert!(contracts.find_by_deployed_code_exact(&[]).is_none());
    }

    #[test]
    fn find_by_deployed_code_exact_unique_rejects_ambiguity() {
        let code = Bytes::from_static(&[0x60, 0x00]);
        let contracts = ContractsByArtifact::new([
            deployed_artifact("A", code.clone()),
            deployed_artifact("B", code.clone()),
        ]);

        assert!(contracts.find_by_deployed_code_exact_unique(&code).is_none());

        let contracts = ContractsByArtifact::new([deployed_artifact("A", code.clone())]);
        assert_eq!(contracts.find_by_deployed_code_exact_unique(&code).unwrap().0.name, "A");
    }

    #[test]
    fn find_by_deployed_code_exact_unique_rejects_partial_metadata_match() {
        let artifact_code = Bytes::from_static(&[0x60, 0x00, 0xa0, 0x00, 0x01]);
        let deployed_code = Bytes::from_static(&[0x60, 0x00, 0xf6, 0x00, 0x01]);
        let contracts = ContractsByArtifact::new([deployed_artifact("A", artifact_code)]);

        assert!(contracts.find_by_deployed_code_exact(&deployed_code).is_some());
        assert!(contracts.find_by_deployed_code_exact_unique(&deployed_code).is_none());
    }

    #[test]
    fn find_by_deployed_code_exact_unambiguous_preserves_single_partial() {
        let artifact_code = Bytes::from_static(&[0x60, 0x00, 0xa0, 0x00, 0x01]);
        let deployed_code = Bytes::from_static(&[0x60, 0x00, 0xf6, 0x00, 0x01]);
        let contracts = ContractsByArtifact::new([deployed_artifact("A", artifact_code)]);

        assert_eq!(
            contracts
                .find_by_deployed_code_exact_unambiguous(&deployed_code)
                .unwrap()
                .unwrap()
                .0
                .name,
            "A"
        );
        assert!(contracts.find_by_deployed_code_exact_unambiguous(&[0x00]).unwrap().is_none());
    }

    #[test]
    fn find_by_deployed_code_exact_unambiguous_prefers_exact_over_ambiguous_partial() {
        let partial_code = Bytes::from_static(&[0x60, 0x00, 0xa0, 0x00, 0x01]);
        let deployed_code = Bytes::from_static(&[0x60, 0x00, 0xf6, 0x00, 0x01]);
        for exact_name in ["A", "Z"] {
            let contracts = ContractsByArtifact::new([
                deployed_artifact("B", partial_code.clone()),
                deployed_artifact("C", partial_code.clone()),
            ]);
            assert!(contracts.find_by_deployed_code_exact(&deployed_code).is_some());
            assert!(contracts.find_by_deployed_code_exact_unique(&deployed_code).is_none());
            assert_eq!(
                contracts
                    .find_by_deployed_code_exact_unambiguous(&deployed_code)
                    .unwrap_err()
                    .to_string(),
                "Multiple local contracts match the deployed bytecode with different metadata. Specify the contract as <path>:<contract>"
            );

            let contracts = ContractsByArtifact::new([
                deployed_artifact(exact_name, deployed_code.clone()),
                deployed_artifact("B", partial_code.clone()),
                deployed_artifact("C", partial_code.clone()),
            ]);
            assert_eq!(
                contracts
                    .find_by_deployed_code_exact_unambiguous(&deployed_code)
                    .unwrap()
                    .unwrap()
                    .0
                    .name,
                exact_name
            );
        }
    }

    /// Tests an immutable reference within a library call-protection prefix.
    #[test]
    fn find_by_deployed_code_exact_handles_overlapping_ignored_ranges() {
        // Call-protection prefix covering [1, 21).
        let mut code = vec![0x73u8];
        code.extend(std::iter::repeat_n(0u8, 20));
        // Bytes matched after the ignored range.
        code.extend_from_slice(&[0xAA, 0xBB, 0xCC, 0xDD]);
        let code = Bytes::from(code);

        // Immutable reference [5, 8) lies within [1, 21).
        let immutable_references =
            BTreeMap::from([("someImmutable".to_string(), vec![Offsets { start: 5, length: 3 }])]);

        let contracts = ContractsByArtifact::new([deployed_artifact_with_immutables(
            "LibWithEarlyImmutable",
            code.clone(),
            immutable_references,
        )]);

        // The overlap must not panic or prevent a match.
        assert!(contracts.find_by_deployed_code_exact(&code).is_some());
    }

    #[test]
    fn normalize_offsets_merges_overlaps_and_adjacency() {
        // Contained overlap.
        let mut offsets = vec![Offsets { start: 1, length: 20 }, Offsets { start: 5, length: 3 }];
        normalize_offsets(&mut offsets);
        assert_eq!(offsets, vec![Offsets { start: 1, length: 20 }]);

        // Extending overlap.
        let mut offsets = vec![Offsets { start: 1, length: 20 }, Offsets { start: 15, length: 15 }];
        normalize_offsets(&mut offsets);
        assert_eq!(offsets, vec![Offsets { start: 1, length: 29 }]);

        // Adjacent ranges.
        let mut offsets = vec![Offsets { start: 1, length: 20 }, Offsets { start: 21, length: 4 }];
        normalize_offsets(&mut offsets);
        assert_eq!(offsets, vec![Offsets { start: 1, length: 24 }]);

        // Disjoint ranges.
        let mut offsets = vec![Offsets { start: 1, length: 5 }, Offsets { start: 10, length: 5 }];
        normalize_offsets(&mut offsets);
        assert_eq!(
            offsets,
            vec![Offsets { start: 1, length: 5 }, Offsets { start: 10, length: 5 }]
        );

        // Unsorted ranges.
        let mut offsets = vec![Offsets { start: 10, length: 5 }, Offsets { start: 1, length: 5 }];
        normalize_offsets(&mut offsets);
        assert_eq!(
            offsets,
            vec![Offsets { start: 1, length: 5 }, Offsets { start: 10, length: 5 }]
        );
    }
}