Skip to content

Commit

Permalink
cmd/commands: don't error out on replacement failure
Browse files Browse the repository at this point in the history
  • Loading branch information
guggero committed Oct 15, 2024
1 parent 5d60da5 commit 8c79940
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 20 deletions.
16 changes: 7 additions & 9 deletions cmd/commands/commands.go
Original file line number Diff line number Diff line change
Expand Up @@ -54,10 +54,10 @@ var (

// replaceCustomData replaces the custom channel data hex string with the
// decoded custom channel data in the JSON response.
func replaceCustomData(jsonBytes []byte) ([]byte, error) {
func replaceCustomData(jsonBytes []byte) []byte {
// If there's nothing to replace, return the original JSON.
if !customDataPattern.Match(jsonBytes) {
return jsonBytes, nil
return jsonBytes
}

replacedBytes := customDataPattern.ReplaceAllFunc(
Expand All @@ -78,10 +78,12 @@ func replaceCustomData(jsonBytes []byte) ([]byte, error) {
var buf bytes.Buffer
err := json.Indent(&buf, replacedBytes, "", " ")
if err != nil {
return nil, err
// If we can't indent the JSON, it likely means the replacement
// data wasn't correct, so we return the original JSON.
return jsonBytes
}

return buf.Bytes(), nil
return buf.Bytes()
}

func getContext() context.Context {
Expand Down Expand Up @@ -118,11 +120,7 @@ func printRespJSON(resp proto.Message) {
return
}

jsonBytesReplaced, err := replaceCustomData(jsonBytes)
if err != nil {
fmt.Println("unable to replace custom data: ", err)
jsonBytesReplaced = jsonBytes
}
jsonBytesReplaced := replaceCustomData(jsonBytes)

fmt.Printf("%s\n", jsonBytesReplaced)
}
Expand Down
14 changes: 3 additions & 11 deletions cmd/commands/commands_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,6 @@ func TestReplaceCustomData(t *testing.T) {
data string
replaceData string
expected string
expectedErr string
}{
{
name: "no replacement necessary",
Expand Down Expand Up @@ -175,7 +174,8 @@ func TestReplaceCustomData(t *testing.T) {
name: "invalid json",
data: "this ain't json, " +
"\"custom_channel_data\":\"a\"",
expectedErr: "invalid character 'h' in literal true",
expected: "this ain't json, " +
"\"custom_channel_data\":\"a\"",
},
{
name: "valid json, invalid hex, just formatted",
Expand All @@ -186,15 +186,7 @@ func TestReplaceCustomData(t *testing.T) {

for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
result, err := replaceCustomData([]byte(tc.data))

if tc.expectedErr != "" {
require.ErrorContains(t, err, tc.expectedErr)
return
}

require.NoError(t, err)

result := replaceCustomData([]byte(tc.data))
require.Equal(t, tc.expected, string(result))
})
}
Expand Down

0 comments on commit 8c79940

Please sign in to comment.