Fix quoting on yaml marshaled uint128 values (#149)

Author: cmmarslender

pkg/tls/tls.go

@@ -94,12 +94,11 @@ func GenerateAllCerts(outDir string, privateCACert *x509.Certificate, privateCAK
 		return errors.New("you must provide the CA cert and key if providing a CA, or set both to nil and a new CA will be generated")
 	} else {
 		// Must have non-nil values for both, so ensure the cert and key match
-		if ! CertMatchesPrivateKey(privateCACert, privateCAKey) {
+		if !CertMatchesPrivateKey(privateCACert, privateCAKey) {
 			return errors.New("provided private CA Cert and Key do not match")
 		}
 	}
 
-
 	for _, node := range publicNodeNames {
 		_, _, err = GenerateCASignedCert(chiaCACert, chiaCAKey, path.Join(outDir, node, fmt.Sprintf("public_%s", node)))
 		if err != nil {

pkg/types/uint128.go

@@ -31,6 +31,8 @@ import (
 	"math"
 	"math/big"
 	"math/bits"
+
+	"gopkg.in/yaml.v3"
 )
 
 // Uint128Zero is a zero-valued uint128.
@@ -494,7 +496,12 @@ func (u Uint128) MarshalJSON() ([]byte, error) {
 
 // MarshalYAML marshals the uint128 value to yaml
 func (u Uint128) MarshalYAML() (interface{}, error) {
-	return u.String(), nil
+	// Return a YAML node with ScalarNode type to ensure the value is treated as a plain number and not quoted
+	return &yaml.Node{
+		Kind:  yaml.ScalarNode,
+		Value: u.Big().String(), // Use the string value without quotes
+		Style: yaml.TaggedStyle, // Prevent quoting
+	}, nil
 }
 
 // FitsInUint64 returns true if the value of the Uint128 will fit in Uint64

pkg/types/uint128_test.go

@@ -33,6 +33,7 @@ import (
 	"math/big"
 	"testing"
 
+	"github.com/stretchr/testify/assert"
 	"gopkg.in/yaml.v3"
 
 	"github.com/chia-network/go-chia-libs/pkg/types"
@@ -543,3 +544,31 @@ func TestUint128_UnmarshalYAML_Null(t *testing.T) {
 		t.Error("`null` did not unmarshal as zero")
 	}
 }
+
+func TestNewUint128_MarshalJSON(t *testing.T) {
+	num := types.Uint128From64(12345)
+	out, err := json.Marshal(num)
+	assert.NoError(t, err)
+	assert.Equal(t, "12345", string(out))
+
+	bigInt := new(big.Int)
+	bigInt.SetString("18446744073709551616", 10)
+	num = types.Uint128FromBig(bigInt)
+	out, err = json.Marshal(num)
+	assert.NoError(t, err)
+	assert.Equal(t, "18446744073709551616", string(out))
+}
+
+func TestNewUint128_MarshalYAML(t *testing.T) {
+	num := types.Uint128From64(12345)
+	out, err := yaml.Marshal(num)
+	assert.NoError(t, err)
+	assert.Equal(t, "12345\n", string(out))
+
+	bigInt := new(big.Int)
+	bigInt.SetString("18446744073709551616", 10)
+	num = types.Uint128FromBig(bigInt)
+	out, err = yaml.Marshal(num)
+	assert.NoError(t, err)
+	assert.Equal(t, "18446744073709551616\n", string(out))
+}