Asset plutus type

Models.go

@@ -19,7 +19,9 @@ type Unit struct {
 	Quantity int
 }
 
-/**
+/*
+*
+
 	ToValue converts a Unit to a Value object.
 
 	Returns:
@@ -35,7 +37,9 @@ func (u *Unit) ToValue() Value.Value {
 	return val
 }
 
-/**
+/*
+*
+
 	NewUnit creates a new Unit with the provided information.
 
 	Params:
@@ -69,7 +73,9 @@ type Payment struct {
 	IsInline  bool
 }
 
-/**
+/*
+*
+
 	PaymentFromTxOut creates a Payment object from a TransactionOutput.
 
 	Params:
@@ -110,10 +116,12 @@ func PaymentFromTxOut(txOut *TransactionOutput.TransactionOutput) *Payment {
 	return payment
 }
 
-/**
+/*
+*
+
 	NewPayment creates a new Payment object.
 
-	Params: 
+	Params:
 		receiver (string): The receiver's address.
 		lovelace (int): The amount in Lovelace.
 		units ([]Unit): The assets units to be included.
@@ -130,7 +138,9 @@ func NewPayment(receiver string, lovelace int, units []Unit) *Payment {
 	}
 }
 
-/**
+/*
+*
+
 	NewPaymentFromValue creates a new Payment object from an Address
 	and Value object.
 
@@ -158,7 +168,9 @@ func NewPaymentFromValue(receiver Address.Address, value Value.Value) *Payment {
 	return payment
 }
 
-/**
+/*
+*
+
 	ToValue converts a Payment to a Value object.
 
 	Returns:
@@ -173,7 +185,9 @@ func (p *Payment) ToValue() Value.Value {
 	return v
 }
 
-/**
+/*
+*
+
 	EnsureMinUTXO ensures that the payment amount meets the minimun UTXO requirement.
 
 	Params:
@@ -184,17 +198,23 @@ func (p *Payment) EnsureMinUTXO(cc Base.ChainContext) {
 		return
 	}
 	txOut := p.ToTxOut()
-	coins := Utils.MinLovelacePostAlonzo(*txOut, cc)
+	coins, err := Utils.MinLovelacePostAlonzo(*txOut, cc)
+	if err != nil {
+		return
+	}
+
 	if int64(p.Lovelace) < coins {
 		p.Lovelace = int(coins)
 	}
 }
 
-/**
-	ToTxOut converts a Payment object to a TransactionOutput object.
+/*
+*
 
-	Returns:
-   		*TransactionOutput.TransactionOutput: The created TransactionOutput object.
+		ToTxOut converts a Payment object to a TransactionOutput object.
+
+		Returns:
+	   		*TransactionOutput.TransactionOutput: The created TransactionOutput object.
 */
 func (p *Payment) ToTxOut() *TransactionOutput.TransactionOutput {
 	txOut := TransactionOutput.SimpleTransactionOutput(p.Receiver, p.ToValue())

apollotypes/plutus.go

@@ -36,7 +36,9 @@ type AikenPlutusJSON struct {
 	} `json:"definitions"`
 }
 
-/**
+/*
+*
+
 	GetScript retrives a Plutus V2 script by its name from an AikenPlutusJSON object.
 	It searches through the Validators and returns the script if found.
 

apollotypes/types.go

@@ -1,27 +1,33 @@
 package apollotypes
 
 import (
+	"bytes"
+
 	"github.com/Salvionied/apollo/serialization"
 	serAddress "github.com/Salvionied/apollo/serialization/Address"
 	"github.com/Salvionied/apollo/serialization/Key"
 	"github.com/Salvionied/apollo/serialization/Transaction"
 	"github.com/Salvionied/apollo/serialization/TransactionWitnessSet"
+	"github.com/Salvionied/apollo/serialization/UTxO"
 	"github.com/Salvionied/apollo/serialization/VerificationKeyWitness"
 	"github.com/Salvionied/apollo/txBuilding/Backend/Base"
 )
 
 type Wallet interface {
 	GetAddress() *serAddress.Address
-	SignTx(tx Transaction.Transaction) TransactionWitnessSet.TransactionWitnessSet
+	SignTx(tx Transaction.Transaction, usedUtxos []UTxO.UTxO) TransactionWitnessSet.TransactionWitnessSet
 	PkeyHash() serialization.PubKeyHash
+	SkeyHash() serialization.PubKeyHash
 	//SignMessage(address serAddress.Address, message []uint8) []uint8
 }
 
 type ExternalWallet struct {
 	Address serAddress.Address
 }
 
-/**
+/*
+*
+
 	GetAddress returns the address associated with an external wallet.
 
 	Returns:
@@ -31,7 +37,9 @@ func (ew *ExternalWallet) GetAddress() *serAddress.Address {
 	return &ew.Address
 }
 
-/**
+/*
+*
+
 	SignTx signs a transaction using an external wallet.
 
 	Params:
@@ -40,13 +48,15 @@ func (ew *ExternalWallet) GetAddress() *serAddress.Address {
 	Returns:
 		TransactionWitnessSet.TransactionWitnessSet: The withness set associated with the signed transaction.
 */
-func (ew *ExternalWallet) SignTx(tx Transaction.Transaction) TransactionWitnessSet.TransactionWitnessSet {
+func (ew *ExternalWallet) SignTx(tx Transaction.Transaction, usedUtxos []UTxO.UTxO) TransactionWitnessSet.TransactionWitnessSet {
 	return tx.TransactionWitnessSet
 }
 
-/**
+/*
+*
+
 	PkeyHash returns the public key hash assoicated with an external wallet.
-	It computes and returns the public key hash based on the PaymentPart 
+	It computes and returns the public key hash based on the PaymentPart
 	of the wallet's address.
 
 	Returns:
@@ -57,6 +67,10 @@ func (ew *ExternalWallet) PkeyHash() serialization.PubKeyHash {
 	return res
 }
 
+func (ew *ExternalWallet) SkeyHash() serialization.PubKeyHash {
+	return serialization.PubKeyHash{}
+}
+
 type GenericWallet struct {
 	SigningKey           Key.SigningKey
 	VerificationKey      Key.VerificationKey
@@ -65,20 +79,29 @@ type GenericWallet struct {
 	StakeVerificationKey Key.StakeVerificationKey
 }
 
-/**
-	PkeyHash calculates and returns the public key hash associated with a generic wallet.
-	It computes the public key hash by calling the Hash() method on the wallet's VerificationKey.
-	Then it returns as a serialization.PubKeyHas type.
-
-	Returns:
-   		serialization.PubKeyHash: The public key hash of the generic wallet.
+/*
+*
+
+		PkeyHash calculates and returns the public key hash associated with a generic wallet.
+		It computes the public key hash by calling the Hash() method on the wallet's VerificationKey.
+		Then it returns as a serialization.PubKeyHas type.
+
+		Returns:
+	   		serialization.PubKeyHash: The public key hash of the generic wallet.
 */
 func (gw *GenericWallet) PkeyHash() serialization.PubKeyHash {
 	res, _ := gw.VerificationKey.Hash()
 	return res
 }
 
-/**
+func (gw *GenericWallet) SkeyHash() serialization.PubKeyHash {
+	res, _ := gw.StakeVerificationKey.Hash()
+	return res
+}
+
+/*
+*
+
 	GetAddress returns the address associated with a generic wallet.
 
 	Returns:
@@ -88,27 +111,103 @@ func (gw *GenericWallet) GetAddress() *serAddress.Address {
 	return &gw.Address
 }
 
-/**
-	SignTx signs a transaction using a generic wallet and returns the updated TransactionWitnessSet.
-	It takes a transaction of type Transaction.Transaction and signs it using the wallet's SigningKey.
-	Then it appends the corresponding VerificationKeyWitness to the TransactionWitnessSet and returns
-	the updated witness set.
+/*
+*
 
-	Parameters:
-	   	wallet (*GenericWallet): A pointer to a generic wallet.
-		tx (Transaction.Transaction): The transaction to be signed.
+		SignTx signs a transaction using a generic wallet and returns the updated TransactionWitnessSet.
+		It takes a transaction of type Transaction.Transaction and signs it using the wallet's SigningKey.
+		Then it appends the corresponding VerificationKeyWitness to the TransactionWitnessSet and returns
+		the updated witness set.
 
-	Returns:
-   		TransactionWitnessSet.TransactionWitnessSet: The updated TransactionWitnessSet after signing the transaction.
+		Parameters:
+		   	wallet (*GenericWallet): A pointer to a generic wallet.
+			tx (Transaction.Transaction): The transaction to be signed.
+
+		Returns:
+	   		TransactionWitnessSet.TransactionWitnessSet: The updated TransactionWitnessSet after signing the transaction.
 */
-func (wallet *GenericWallet) SignTx(tx Transaction.Transaction) TransactionWitnessSet.TransactionWitnessSet {
+func (wallet *GenericWallet) SignTx(
+	tx Transaction.Transaction,
+	usedUtxos []UTxO.UTxO,
+) TransactionWitnessSet.TransactionWitnessSet {
 	witness_set := tx.TransactionWitnessSet
 	txHash, _ := tx.TransactionBody.Hash()
-	signature, _ := wallet.SigningKey.Sign(txHash)
-	witness_set.VkeyWitnesses = append(witness_set.VkeyWitnesses, VerificationKeyWitness.VerificationKeyWitness{Vkey: wallet.VerificationKey, Signature: signature})
+	if isKeyHashUsedFromUtxos(usedUtxos, wallet.PkeyHash()) ||
+		isKeyHashUsedFromTx(tx, wallet.PkeyHash()) {
+		signature, _ := wallet.SigningKey.Sign(txHash)
+
+		witness_set.VkeyWitnesses = append(
+			witness_set.VkeyWitnesses,
+			VerificationKeyWitness.VerificationKeyWitness{
+				Vkey:      wallet.VerificationKey,
+				Signature: signature,
+			},
+		)
+	}
+
+	if isKeyHashUsedFromUtxos(usedUtxos, wallet.SkeyHash()) ||
+		isKeyHashUsedFromTx(tx, wallet.SkeyHash()) {
+		signature, _ := wallet.StakeSigningKey.Sign(txHash)
+
+		witness_set.VkeyWitnesses = append(
+			witness_set.VkeyWitnesses,
+			VerificationKeyWitness.VerificationKeyWitness{
+				Vkey:      Key.VerificationKey(wallet.StakeVerificationKey),
+				Signature: signature,
+			},
+		)
+	}
+
 	return witness_set
 }
 
+func isKeyHashUsedFromUtxos(
+	usedUtxos []UTxO.UTxO,
+	keyHash serialization.PubKeyHash,
+) bool {
+	for _, utxo := range usedUtxos {
+		utxoKeyHash := serialization.PubKeyHash(utxo.Output.GetAddress().PaymentPart)
+		if utxoKeyHash == keyHash {
+			return true
+		}
+	}
+	return false
+}
+
+func isKeyHashUsedFromTx(
+	tx Transaction.Transaction,
+	keyHash serialization.PubKeyHash,
+) bool {
+	keyHashBytes := keyHash[:]
+	if tx.TransactionBody.Certificates != nil {
+		for _, certificate := range *tx.TransactionBody.Certificates {
+			if certificate.StakeCredential.KeyHash() == keyHash {
+				return true
+			}
+		}
+	}
+	if tx.TransactionBody.Withdrawals != nil {
+		for withdrawal := range *tx.TransactionBody.Withdrawals {
+			withdrawalBytes := withdrawal[1:]
+			if bytes.Equal(withdrawalBytes, keyHashBytes) {
+				return true
+			}
+		}
+	}
+	for _, requiredSigner := range tx.TransactionBody.RequiredSigners {
+		if requiredSigner == keyHash {
+			return true
+		}
+	}
+	for _, nativeScript := range tx.TransactionWitnessSet.NativeScripts {
+		if bytes.Equal(nativeScript.KeyHash, keyHashBytes) {
+			return true
+		}
+	}
+
+	return false
+}
+
 type Backend Base.ChainContext
 
 type Address serAddress.Address

backends.go

@@ -47,28 +47,28 @@ func NewBlockfrostBackend(
 			constants.BLOCKFROST_BASE_URL_MAINNET,
 			int(constants.MAINNET),
 			projectId,
-		), nil
+		)
 	case constants.TESTNET:
 
 		return BlockFrostChainContext.NewBlockfrostChainContext(
 			constants.BLOCKFROST_BASE_URL_TESTNET,
 			int(constants.TESTNET),
 			projectId,
-		), nil
+		)
 	case constants.PREVIEW:
 
 		return BlockFrostChainContext.NewBlockfrostChainContext(
 			constants.BLOCKFROST_BASE_URL_PREVIEW,
 			int(constants.TESTNET),
 			projectId,
-		), nil
+		)
 	case constants.PREPROD:
 
 		return BlockFrostChainContext.NewBlockfrostChainContext(
 			constants.BLOCKFROST_BASE_URL_PREPROD,
 			int(constants.TESTNET),
 			projectId,
-		), nil
+		)
 	default:
 		return BlockFrostChainContext.BlockFrostChainContext{}, fmt.Errorf("Invalid network")
 	}