refactor(core): encapsulates context extension

fleet-sdk · Dec 12, 2022 · 8d0ebcd · 8d0ebcd
1 parent d24494f
commit 8d0ebcd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/packages/core/src/models/ergoUnsignedInput.spec.ts b/packages/core/src/models/ergoUnsignedInput.spec.ts
@@ -19,7 +19,7 @@ describe("Construction", () => {
   });
 });
 
-describe("extension", () => {
+describe("Tweaking", () => {
   it("Should set extension", () => {
     const input = new ErgoUnsignedInput(regularBoxesMock[0]);
     input.setContextVars({ 0: "0402", 1: "0580c0fc82aa02" });

diff --git a/packages/core/src/models/ergoUnsignedInput.ts b/packages/core/src/models/ergoUnsignedInput.ts
@@ -14,22 +14,26 @@ type InputType<T> = T extends "default" ? UnsignedInput : EIP12UnsignedInput;
 type DataInputType<T> = T extends "default" ? DataInput : EIP12UnsignedDataInput;
 
 export class ErgoUnsignedInput extends ErgoBox {
-  extension?: ContextExtension;
+  private _extension?: ContextExtension;
+
+  public get extension(): ContextExtension | undefined {
+    return this._extension;
+  }
 
   constructor(box: Box<Amount>) {
     super(box);
   }
 
   public setContextVars(extension: ContextExtension): ErgoUnsignedInput {
-    this.extension = extension;
+    this._extension = extension;
 
     return this;
   }
 
   public toUnsignedInputObject<T extends BuildOutputType>(type: T): InputType<T> {
     return {
       ...this.toObject(type),
-      extension: this.extension || {}
+      extension: this._extension || {}
     } as InputType<T>;
   }