update build tx

minswap · Aug 3, 2024 · cb6915b · cb6915b
1 parent f7584f8
commit cb6915b
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/src/build-tx.ts b/src/build-tx.ts
@@ -95,6 +95,7 @@ export type BuildCreateTreasuryOptions = {
   validFrom: UnixTime;
   validTo: UnixTime;
   extraDatum?: Datum; // the datum of treasuryDatum.receiverDatum
+  extraUtxo?: UTxO;
 };
 
 export type BuildAddSellersOptions = {
@@ -154,6 +155,7 @@ export type BuildUpdateLBEOptions = {
   reserveBase?: bigint;
   revocable?: boolean;
   penaltyConfig?: { penaltyStartTime: bigint; percent: bigint };
+  extraInput?: UTxO;
 };
 
 export type BuildCancelLBEOptions = {
@@ -369,6 +371,7 @@ export class WarehouseBuilder {
       validTo,
       extraDatum,
       sellerAmount,
+      extraUtxo,
     } = options;
     const managerDatum: ManagerDatum = {
       factoryPolicyId: this.factoryHash,
@@ -439,8 +442,8 @@ export class WarehouseBuilder {
         if ("VerificationKeyCredential" in owner.paymentCredential) {
           this.tx.addSigner(plutusAddress2Address("Preprod", owner));
         } else {
-          // TODO: support cript hash
-          throw Error("Do not support Script owner")
+          invariant(extraUtxo);
+          this.tx.collectFrom([extraUtxo]);
         }
       },
     );
@@ -585,14 +588,16 @@ export class WarehouseBuilder {
       reserveBase,
       revocable,
       penaltyConfig,
+      extraInput,
     } = options;
     invariant(treasuryInput.datum);
     const inDatum = this.fromDatumTreasury(treasuryInput.datum);
+    const updatedOwner = owner ? address2PlutusAddress(owner) : inDatum.owner;
     const treasuryOutDatum: TreasuryDatum = {
       ...inDatum,
       startTime: startTime ?? inDatum.startTime,
       endTime: endTime ?? inDatum.endTime,
-      owner: owner ? address2PlutusAddress(owner) : inDatum.owner,
+      owner: updatedOwner,
       minimumOrderRaise: minimumOrderRaise ?? inDatum.minimumOrderRaise,
       minimumRaise: minimumRaise ?? inDatum.minimumRaise,
       maximumRaise: maximumRaise ?? inDatum.maximumRaise,
@@ -617,6 +622,14 @@ export class WarehouseBuilder {
       () => {
         this.tx.validFrom(validFrom).validTo(validTo);
       },
+      () => {
+        if ("VerificationKeyCredential" in updatedOwner.paymentCredential) {
+          this.tx.addSigner(plutusAddress2Address("Preprod", updatedOwner));
+        } else {
+          invariant(extraInput);
+          this.tx.collectFrom([extraInput]);
+        }
+      },
     );
     return this;
   }