consume-improved-response-transform-interface

Summary:

- Consume new reponse transform interface.
- Support for `json` response transforms.
- Support for `text`  response transforms.

Author: general-kroll-4-life

go.mod

@@ -19,7 +19,7 @@ require (
 	github.com/spf13/cobra v1.4.0
 	github.com/spf13/pflag v1.0.5
 	github.com/spf13/viper v1.10.1
-	github.com/stackql/any-sdk v0.1.3-alpha13
+	github.com/stackql/any-sdk v0.1.3-beta01
 	github.com/stackql/go-suffix-map v0.0.1-alpha01
 	github.com/stackql/psql-wire v0.1.1-beta23
 	github.com/stackql/stackql-parser v0.0.14-alpha05

go.sum

@@ -484,8 +484,8 @@ github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
 github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
 github.com/spf13/viper v1.10.1 h1:nuJZuYpG7gTj/XqiUwg8bA0cp1+M2mC3J4g5luUYBKk=
 github.com/spf13/viper v1.10.1/go.mod h1:IGlFPqhNAPKRxohIzWpI5QEy4kuI7tcl5WvR+8qy1rU=
-github.com/stackql/any-sdk v0.1.3-alpha13 h1:IJLnZojR7JU6f3cixL0U8KdJva+qociPb7Uu3Vctq9w=
-github.com/stackql/any-sdk v0.1.3-alpha13/go.mod h1:AKS/g28y7m4SWL/YW8veE9MCNy8XJgaicVibemVE9e8=
+github.com/stackql/any-sdk v0.1.3-beta01 h1:97q50CfwKrsU7niQvbASURxshlljoi16uyrnNLyphcI=
+github.com/stackql/any-sdk v0.1.3-beta01/go.mod h1:AKS/g28y7m4SWL/YW8veE9MCNy8XJgaicVibemVE9e8=
 github.com/stackql/go-suffix-map v0.0.1-alpha01 h1:TDUDS8bySu41Oo9p0eniUeCm43mnRM6zFEd6j6VUaz8=
 github.com/stackql/go-suffix-map v0.0.1-alpha01/go.mod h1:QAi+SKukOyf4dBtWy8UMy+hsXXV+yyEE4vmBkji2V7g=
 github.com/stackql/psql-wire v0.1.1-beta23 h1:1ayYMjZArfDcIMyEOKnm+Bp1zRCISw8pguvTFuUhhVQ=

internal/stackql/execution/mono_valent_execution.go

@@ -1,7 +1,6 @@
 package execution
 
 import (
-	"bytes"
 	"encoding/json"
 	"fmt"
 	"io"
@@ -1330,19 +1329,34 @@ func (mv *monoValentExecution) GetExecutor() (func(pc primitive.IPrimitiveCtx) i
 				expectedResponse, isExpectedResponse := m.GetResponse()
 				if isExpectedResponse {
 					responseTransform, responseTransformExists := expectedResponse.GetTransform()
-					if responseTransformExists && responseTransform.GetType() == "golang_template_v0.1.0" {
+					if responseTransformExists {
 						input := stdoutStr
-						tmpl := responseTransform.GetBody()
-						inStream := stream_transform.NewTextReader(bytes.NewBufferString(input))
-						outStream := bytes.NewBuffer(nil)
-						tfm, setupErr := stream_transform.NewTemplateStreamTransformer(tmpl, inStream, outStream)
-						if setupErr != nil {
-							return internaldto.NewErroneousExecutorOutput(fmt.Errorf("template stream transform error: %w", setupErr))
+						streamTransformerFactory := stream_transform.NewStreamTransformerFactory(
+							responseTransform.GetType(),
+							responseTransform.GetBody(),
+						)
+						if !streamTransformerFactory.IsTransformable() {
+							return internaldto.NewErroneousExecutorOutput(
+								fmt.Errorf("unsupported template type: %s", responseTransform.GetType()),
+							)
 						}
-						if tfErr := tfm.Transform(); tfErr != nil {
-							return internaldto.NewErroneousExecutorOutput(fmt.Errorf("failed to transform: %w", tfErr))
+						tfm, getTfmErr := streamTransformerFactory.GetTransformer(input)
+						if getTfmErr != nil {
+							return internaldto.NewErroneousExecutorOutput(
+								fmt.Errorf("failed to transform: %w", getTfmErr))
 						}
-						outputStr := outStream.String()
+						transformError := tfm.Transform()
+						if transformError != nil {
+							return internaldto.NewErroneousExecutorOutput(
+								fmt.Errorf("failed to transform: %w", transformError))
+						}
+						outStream := tfm.GetOutStream()
+						outputBytes, readErr := io.ReadAll(outStream)
+						if readErr != nil {
+							return internaldto.NewErroneousExecutorOutput(
+								fmt.Errorf("failed to read transformed stream: %w", readErr))
+						}
+						outputStr := string(outputBytes)
 						stdoutStr = outputStr
 					}
 				}