radix-transactions 1.3.1

use super::*;
use crate::internal_prelude::*;

//=================================================================================
// NOTE:
// This isn't actually embedded as a model - it's just a useful model which we use
// in eg the manifest builder
//=================================================================================

/// Can be built with a [`ManifestV2Builder`]
#[derive(Debug, Clone, Default, PartialEq, Eq, ManifestSbor, ScryptoDescribe)]
pub struct TransactionManifestV2 {
    pub instructions: Vec<InstructionV2>,
    pub blobs: IndexMap<Hash, Vec<u8>>,
    pub children: IndexSet<ChildSubintentSpecifier>,
    pub object_names: ManifestObjectNames,
}

impl ReadableManifestBase for TransactionManifestV2 {
    fn is_subintent(&self) -> bool {
        false
    }

    fn get_blobs(&self) -> impl Iterator<Item = (&Hash, &Vec<u8>)> {
        self.blobs.iter()
    }

    fn get_child_subintent_hashes(
        &self,
    ) -> impl ExactSizeIterator<Item = &ChildSubintentSpecifier> {
        self.children.iter()
    }

    fn get_known_object_names_ref(&self) -> ManifestObjectNamesRef<'_> {
        self.object_names.as_ref()
    }
}

impl TypedReadableManifest for TransactionManifestV2 {
    type Instruction = InstructionV2;

    fn get_typed_instructions(&self) -> &[Self::Instruction] {
        &self.instructions
    }
}

impl BuildableManifest for TransactionManifestV2 {
    fn add_instruction(&mut self, instruction: Self::Instruction) {
        self.instructions.push(instruction)
    }

    fn add_blob(&mut self, hash: Hash, content: Vec<u8>) {
        self.blobs.insert(hash, content);
    }

    fn set_names(&mut self, names: KnownManifestObjectNames) {
        self.object_names = names.into()
    }

    fn add_child_subintent(&mut self, hash: SubintentHash) -> Result<(), ManifestBuildError> {
        if !self.children.insert(ChildSubintentSpecifier { hash }) {
            return Err(ManifestBuildError::DuplicateChildSubintentHash);
        }
        Ok(())
    }

    fn default_test_execution_config_type(&self) -> DefaultTestExecutionConfigType {
        DefaultTestExecutionConfigType::Test
    }

    fn into_executable_with_proofs(
        self,
        nonce: u32,
        initial_proofs: BTreeSet<NonFungibleGlobalId>,
        validator: &TransactionValidator,
    ) -> Result<ExecutableTransaction, String> {
        TestTransaction::new_v2_builder(nonce)
            .finish_with_root_intent(self, initial_proofs)
            .into_executable(validator)
            .map_err(|err| format!("Could not prepare: {err:?}"))
    }
}

impl BuildableManifestSupportingChildren for TransactionManifestV2 {}

impl TransactionManifestV2 {
    pub fn empty() -> Self {
        Self {
            instructions: Default::default(),
            blobs: Default::default(),
            children: Default::default(),
            object_names: ManifestObjectNames::Unknown,
        }
    }

    pub fn from_intent_core(intent: &IntentCoreV2) -> Self {
        Self {
            instructions: intent.instructions.to_vec(),
            blobs: intent.blobs.clone().into(),
            children: intent.children.children.clone(),
            object_names: ManifestObjectNames::Unknown,
        }
    }

    pub fn to_intent_core(self, header: IntentHeaderV2, message: MessageV2) -> IntentCoreV2 {
        IntentCoreV2 {
            header,
            blobs: self.blobs.into(),
            message,
            instructions: self.instructions.into(),
            children: ChildSubintentSpecifiersV2 {
                children: self.children,
            },
        }
    }

    pub fn for_intent(self) -> (InstructionsV2, BlobsV1, ChildSubintentSpecifiersV2) {
        (
            self.instructions.into(),
            self.blobs.into(),
            ChildSubintentSpecifiersV2 {
                children: self.children,
            },
        )
    }

    pub fn for_intent_with_names(
        self,
    ) -> (
        InstructionsV2,
        BlobsV1,
        ChildSubintentSpecifiersV2,
        ManifestObjectNames,
    ) {
        (
            self.instructions.into(),
            self.blobs.into(),
            ChildSubintentSpecifiersV2 {
                children: self.children,
            },
            self.object_names,
        )
    }
}