chore: clean up tx in info
This commit is contained in:
rvcas
parent
d5f8b749ae
commit
564cd41626
|
|
@ -1,7 +1,7 @@
|
||||||
use std::{collections::HashMap, path::PathBuf};
|
use std::{collections::HashMap, path::PathBuf};
|
||||||
|
|
||||||
use clap::{Parser, Subcommand};
|
use clap::{Parser, Subcommand};
|
||||||
use serde::{de, Deserialize};
|
use serde::Deserialize;
|
||||||
|
|
||||||
/// Cardano smart contract toolchain
|
/// Cardano smart contract toolchain
|
||||||
#[derive(Parser)]
|
#[derive(Parser)]
|
||||||
|
|
|
||||||
|
|
@ -1,20 +1,18 @@
|
||||||
use std::{
|
use std::{
|
||||||
fmt::Write as _,
|
fmt::Write as _,
|
||||||
fs::{self, File},
|
fs::{self, File},
|
||||||
io::BufReader, collections::HashMap, thread::LocalKey,
|
io::BufReader,
|
||||||
};
|
};
|
||||||
|
|
||||||
use pallas_addresses::Address;
|
|
||||||
use pallas_codec::{minicbor::bytes::ByteVec, utils::{MaybeIndefArray, KeyValuePairs}};
|
|
||||||
use pallas_primitives::babbage::{BigInt, Constr};
|
|
||||||
use pallas_traverse::{Era, MultiEraTx};
|
use pallas_traverse::{Era, MultiEraTx};
|
||||||
use uplc::{
|
use uplc::{
|
||||||
ast::{Constant, DeBruijn, FakeNamedDeBruijn, Name, NamedDeBruijn, Program, Term},
|
ast::{DeBruijn, FakeNamedDeBruijn, Name, NamedDeBruijn, Program, Term},
|
||||||
machine::cost_model::ExBudget,
|
machine::cost_model::ExBudget,
|
||||||
parser, PlutusData,
|
parser,
|
||||||
};
|
};
|
||||||
|
|
||||||
mod args;
|
mod args;
|
||||||
|
mod utils;
|
||||||
|
|
||||||
use args::{Args, TxCommand, UplcCommand};
|
use args::{Args, TxCommand, UplcCommand};
|
||||||
|
|
||||||
|
|
@ -69,131 +67,8 @@ fn main() -> anyhow::Result<()> {
|
||||||
let file = File::open(&resolved_inputs)?;
|
let file = File::open(&resolved_inputs)?;
|
||||||
let reader = BufReader::new(file);
|
let reader = BufReader::new(file);
|
||||||
let resolved_inputs: Vec<ResolvedInput> = serde_json::from_reader(reader)?;
|
let resolved_inputs: Vec<ResolvedInput> = serde_json::from_reader(reader)?;
|
||||||
let tx_in_info: Vec<PlutusData> = resolved_inputs
|
|
||||||
.iter()
|
|
||||||
.map(|resolved_input| {
|
|
||||||
let tx_out_ref = PlutusData::Constr(Constr {
|
|
||||||
tag: 0,
|
|
||||||
any_constructor: None,
|
|
||||||
fields: MaybeIndefArray::Indef(vec![
|
|
||||||
PlutusData::BoundedBytes(
|
|
||||||
hex::decode(resolved_input.input.tx_hash.clone())
|
|
||||||
.unwrap()
|
|
||||||
.into(),
|
|
||||||
),
|
|
||||||
PlutusData::BigInt(BigInt::Int(
|
|
||||||
resolved_input.input.index.into(),
|
|
||||||
)),
|
|
||||||
]),
|
|
||||||
});
|
|
||||||
|
|
||||||
let address =
|
let tx_in_info = utils::get_tx_in_info(&resolved_inputs)?;
|
||||||
Address::from_bech32(&resolved_input.ouput.address).unwrap();
|
|
||||||
|
|
||||||
let payment_tag = match address.typeid() % 2 {
|
|
||||||
0 => 0,
|
|
||||||
1 => 1,
|
|
||||||
_ => unreachable!(),
|
|
||||||
};
|
|
||||||
let stake_tag = match address.typeid() {
|
|
||||||
0 | 1 => Some(0),
|
|
||||||
2 | 3 => Some(1),
|
|
||||||
_ => None,
|
|
||||||
};
|
|
||||||
|
|
||||||
let (payment_part, stake_part) = match address {
|
|
||||||
Address::Shelley(s) => {
|
|
||||||
(s.payment().to_vec(), s.delegation().to_vec())
|
|
||||||
}
|
|
||||||
_ => unreachable!(),
|
|
||||||
};
|
|
||||||
|
|
||||||
let lovelace = resolved_input.ouput.value.0;
|
|
||||||
|
|
||||||
let mut assets = resolved_input.ouput.value.1.clone();
|
|
||||||
|
|
||||||
assets.insert("".to_string(), vec![("".to_string(), lovelace)].into_iter().collect());
|
|
||||||
|
|
||||||
let tx_out = PlutusData::Constr(Constr {
|
|
||||||
tag: 0,
|
|
||||||
any_constructor: None,
|
|
||||||
fields: MaybeIndefArray::Indef(vec![
|
|
||||||
// txOutAddress
|
|
||||||
PlutusData::Constr(Constr {
|
|
||||||
tag: 0,
|
|
||||||
any_constructor: None,
|
|
||||||
fields: MaybeIndefArray::Indef(vec![
|
|
||||||
// addressCredential
|
|
||||||
PlutusData::Constr(Constr {
|
|
||||||
tag: payment_tag,
|
|
||||||
any_constructor: None,
|
|
||||||
fields: MaybeIndefArray::Indef(vec![
|
|
||||||
PlutusData::BoundedBytes(
|
|
||||||
payment_part.into(),
|
|
||||||
),
|
|
||||||
]),
|
|
||||||
}),
|
|
||||||
// addressStakingCredential
|
|
||||||
PlutusData::Constr(Constr {
|
|
||||||
tag: if stake_tag.is_some() { 0 } else { 1 },
|
|
||||||
any_constructor: None,
|
|
||||||
fields: MaybeIndefArray::Indef(
|
|
||||||
if stake_tag.is_some() {
|
|
||||||
vec![
|
|
||||||
// StakingCredential
|
|
||||||
PlutusData::Constr(Constr {
|
|
||||||
tag: 0,
|
|
||||||
any_constructor: None,
|
|
||||||
fields: MaybeIndefArray::Indef(vec![
|
|
||||||
// StakingHash
|
|
||||||
PlutusData::Constr(Constr {
|
|
||||||
tag: stake_tag.unwrap(),
|
|
||||||
any_constructor: None,
|
|
||||||
fields: MaybeIndefArray::Indef(vec![
|
|
||||||
PlutusData::BoundedBytes(
|
|
||||||
stake_part.into(),
|
|
||||||
),
|
|
||||||
]),
|
|
||||||
}),
|
|
||||||
]),
|
|
||||||
}),
|
|
||||||
]
|
|
||||||
|
|
||||||
} else {
|
|
||||||
vec![]
|
|
||||||
},
|
|
||||||
),
|
|
||||||
}),
|
|
||||||
]),
|
|
||||||
}),
|
|
||||||
|
|
||||||
// txOutValue
|
|
||||||
PlutusData::Map(KeyValuePairs::Def(
|
|
||||||
assets.iter().map(|val| {
|
|
||||||
let currency_symbol = PlutusData::BoundedBytes(hex::decode(val.0).unwrap().into());
|
|
||||||
let token_map = PlutusData::Map(KeyValuePairs::Def(
|
|
||||||
val.1.iter().map(|token| {
|
|
||||||
( PlutusData::BoundedBytes(token.0.as_bytes().to_vec().into()), PlutusData::BigInt(BigInt::Int((*token.1).into())))
|
|
||||||
}).collect()
|
|
||||||
|
|
||||||
));
|
|
||||||
(currency_symbol, token_map)
|
|
||||||
}).collect()
|
|
||||||
) ),
|
|
||||||
|
|
||||||
|
|
||||||
]),
|
|
||||||
});
|
|
||||||
PlutusData::Constr(Constr{
|
|
||||||
tag: 0,
|
|
||||||
any_constructor: None,
|
|
||||||
fields: MaybeIndefArray::Indef(vec![
|
|
||||||
tx_out_ref,
|
|
||||||
tx_out
|
|
||||||
])
|
|
||||||
})
|
|
||||||
})
|
|
||||||
.collect();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,127 @@
|
||||||
|
use pallas_addresses::Address;
|
||||||
|
use pallas_codec::utils::{KeyValuePairs, MaybeIndefArray};
|
||||||
|
use pallas_primitives::babbage::{BigInt, Constr};
|
||||||
|
use uplc::PlutusData;
|
||||||
|
|
||||||
|
use crate::args::ResolvedInput;
|
||||||
|
|
||||||
|
pub fn get_tx_in_info(resolved_inputs: &[ResolvedInput]) -> anyhow::Result<Vec<PlutusData>> {
|
||||||
|
let mut tx_in_info = Vec::new();
|
||||||
|
|
||||||
|
for resolved_input in resolved_inputs {
|
||||||
|
let tx_out_ref = PlutusData::Constr(Constr {
|
||||||
|
tag: 0,
|
||||||
|
any_constructor: None,
|
||||||
|
fields: MaybeIndefArray::Indef(vec![
|
||||||
|
PlutusData::BoundedBytes(hex::decode(resolved_input.input.tx_hash.clone())?.into()),
|
||||||
|
PlutusData::BigInt(BigInt::Int(resolved_input.input.index.into())),
|
||||||
|
]),
|
||||||
|
});
|
||||||
|
|
||||||
|
let address = Address::from_bech32(&resolved_input.ouput.address)?;
|
||||||
|
|
||||||
|
let payment_tag = match address.typeid() % 2 {
|
||||||
|
0 => 0,
|
||||||
|
1 => 1,
|
||||||
|
_ => unreachable!(),
|
||||||
|
};
|
||||||
|
let stake_tag = match address.typeid() {
|
||||||
|
0 | 1 => Some(0),
|
||||||
|
2 | 3 => Some(1),
|
||||||
|
_ => None,
|
||||||
|
};
|
||||||
|
|
||||||
|
let (payment_part, stake_part) = match address {
|
||||||
|
Address::Shelley(s) => (s.payment().to_vec(), s.delegation().to_vec()),
|
||||||
|
_ => unreachable!(),
|
||||||
|
};
|
||||||
|
|
||||||
|
let lovelace = resolved_input.ouput.value.0;
|
||||||
|
|
||||||
|
let mut assets = resolved_input.ouput.value.1.clone();
|
||||||
|
|
||||||
|
assets.insert(
|
||||||
|
"".to_string(),
|
||||||
|
vec![("".to_string(), lovelace)].into_iter().collect(),
|
||||||
|
);
|
||||||
|
|
||||||
|
let tx_out = PlutusData::Constr(Constr {
|
||||||
|
tag: 0,
|
||||||
|
any_constructor: None,
|
||||||
|
fields: MaybeIndefArray::Indef(vec![
|
||||||
|
// txOutAddress
|
||||||
|
PlutusData::Constr(Constr {
|
||||||
|
tag: 0,
|
||||||
|
any_constructor: None,
|
||||||
|
fields: MaybeIndefArray::Indef(vec![
|
||||||
|
// addressCredential
|
||||||
|
PlutusData::Constr(Constr {
|
||||||
|
tag: payment_tag,
|
||||||
|
any_constructor: None,
|
||||||
|
fields: MaybeIndefArray::Indef(vec![PlutusData::BoundedBytes(
|
||||||
|
payment_part.into(),
|
||||||
|
)]),
|
||||||
|
}),
|
||||||
|
// addressStakingCredential
|
||||||
|
PlutusData::Constr(Constr {
|
||||||
|
tag: if stake_tag.is_some() { 0 } else { 1 },
|
||||||
|
any_constructor: None,
|
||||||
|
fields: MaybeIndefArray::Indef(match stake_tag {
|
||||||
|
Some(stake_tag) => vec![
|
||||||
|
// StakingCredential
|
||||||
|
PlutusData::Constr(Constr {
|
||||||
|
tag: 0,
|
||||||
|
any_constructor: None,
|
||||||
|
fields: MaybeIndefArray::Indef(vec![
|
||||||
|
// StakingHash
|
||||||
|
PlutusData::Constr(Constr {
|
||||||
|
tag: stake_tag,
|
||||||
|
any_constructor: None,
|
||||||
|
fields: MaybeIndefArray::Indef(vec![
|
||||||
|
PlutusData::BoundedBytes(stake_part.into()),
|
||||||
|
]),
|
||||||
|
}),
|
||||||
|
]),
|
||||||
|
}),
|
||||||
|
],
|
||||||
|
None => vec![],
|
||||||
|
}),
|
||||||
|
}),
|
||||||
|
]),
|
||||||
|
}),
|
||||||
|
// txOutValue
|
||||||
|
PlutusData::Map(KeyValuePairs::Def(
|
||||||
|
assets
|
||||||
|
.iter()
|
||||||
|
.map(|val| {
|
||||||
|
let currency_symbol =
|
||||||
|
PlutusData::BoundedBytes(hex::decode(val.0).unwrap().into());
|
||||||
|
let token_map = PlutusData::Map(KeyValuePairs::Def(
|
||||||
|
val.1
|
||||||
|
.iter()
|
||||||
|
.map(|token| {
|
||||||
|
(
|
||||||
|
PlutusData::BoundedBytes(
|
||||||
|
token.0.as_bytes().to_vec().into(),
|
||||||
|
),
|
||||||
|
PlutusData::BigInt(BigInt::Int((*token.1).into())),
|
||||||
|
)
|
||||||
|
})
|
||||||
|
.collect(),
|
||||||
|
));
|
||||||
|
(currency_symbol, token_map)
|
||||||
|
})
|
||||||
|
.collect(),
|
||||||
|
)),
|
||||||
|
]),
|
||||||
|
});
|
||||||
|
|
||||||
|
tx_in_info.push(PlutusData::Constr(Constr {
|
||||||
|
tag: 0,
|
||||||
|
any_constructor: None,
|
||||||
|
fields: MaybeIndefArray::Indef(vec![tx_out_ref, tx_out]),
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(tx_in_info)
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue