feat(ledger): Add RawScriptBytes to Script interface (#1210)

It's rather awkward to get the script bytes right now, as you have to try casting it to different script types; If you just need the bytes (ex: to store those in a database, etc.) then the RawScriptBytes method gives you access directly to those bytes (without the version wrapper). Might also consider adding a RawBytes to give *exactly* what's on chain, and/or a Version, to indicate which script version it is, but I held off on that for now as this is all that I need to unblock myself.

Signed-off-by: Pi Lanningham <[email protected]>

Author: Quantumplation

ledger/common/script.go

@@ -37,6 +37,7 @@ type ScriptHash = Blake2b224
 type Script interface {
 	isScript()
 	Hash() ScriptHash
+	RawScriptBytes() []byte
 }
 
 type ScriptRef struct {
@@ -114,6 +115,10 @@ func (s PlutusV1Script) Hash() ScriptHash {
 	)
 }
 
+func (s PlutusV1Script) RawScriptBytes() []byte {
+	return []byte(s)
+}
+
 type PlutusV2Script []byte
 
 func (PlutusV2Script) isScript() {}
@@ -127,6 +132,10 @@ func (s PlutusV2Script) Hash() ScriptHash {
 	)
 }
 
+func (s PlutusV2Script) RawScriptBytes() []byte {
+	return []byte(s)
+}
+
 type PlutusV3Script []byte
 
 func (PlutusV3Script) isScript() {}
@@ -140,6 +149,10 @@ func (s PlutusV3Script) Hash() ScriptHash {
 	)
 }
 
+func (s PlutusV3Script) RawScriptBytes() []byte {
+	return []byte(s)
+}
+
 func (s PlutusV3Script) Evaluate(
 	scriptContext data.PlutusData,
 	budget ExUnits,
@@ -238,6 +251,10 @@ func (s NativeScript) Hash() ScriptHash {
 	)
 }
 
+func (s NativeScript) RawScriptBytes() []byte {
+	return s.Cbor()
+}
+
 type NativeScriptPubkey struct {
 	cbor.StructAsArray
 	Type uint

ledger/common/script_test.go

@@ -15,6 +15,7 @@
 package common_test
 
 import (
+	"bytes"
 	"encoding/hex"
 	"reflect"
 	"testing"
@@ -40,6 +41,13 @@ func TestScriptRefDecodeEncode(t *testing.T) {
 			&expectedScript,
 		)
 	}
+	if !bytes.Equal(testScriptRef.Script.RawScriptBytes(), scriptCbor) {
+		t.Fatalf(
+			"did not get expected raw script bytes\n     got: %x\n  wanted: %x",
+			testScriptRef.Script.RawScriptBytes(),
+			scriptCbor,
+		)
+	}
 	scriptRefCbor, err := cbor.Encode(testScriptRef)
 	if err != nil {
 		t.Fatalf("unexpected error: %s", err)