WKS-2076 - Add base64 encoding support for source code

Author: ehl-jf

commands/deploy_cmd.go

@@ -1,6 +1,7 @@
 package commands
 
 import (
+	"encoding/base64"
 	"encoding/json"
 	"fmt"
 	"net/http"
@@ -39,6 +40,7 @@ func GetDeployCommand() components.Command {
 			model.GetChangesVersionFlag(),
 			model.GetChangesDescriptionFlag(),
 			model.GetChangesCommitShaFlag(),
+			model.GetBase64Flag(),
 		},
 		Action: func(c *components.Context) error {
 			server, err := model.GetServerDetails(c)
@@ -151,6 +153,10 @@ func prepareDeployRequest(ctx *components.Context, manifest *model.Manifest, act
 	}
 	sourceCode = common.CleanImports(sourceCode)
 
+	if ctx.GetBoolFlagValue(model.FlagBase64) {
+		sourceCode = "base64:" + base64.StdEncoding.EncodeToString([]byte(sourceCode))
+	}
+
 	var secrets []*model.Secret
 
 	if !ctx.GetBoolFlagValue(model.FlagNoSecrets) {

commands/deploy_cmd_test.go

@@ -4,6 +4,7 @@
 package commands
 
 import (
+	"encoding/base64"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -155,6 +156,31 @@ func TestDeployCommand(t *testing.T) {
 				),
 			commandArgs: []string{"--" + model.FlagChangesVersion, "version number", "--" + model.FlagChangesDescription, "version description", "--" + model.FlagChangesCommitSha, "version commitsha"},
 		},
+		{
+			name:         "create with base64 encoding",
+			workerAction: "BEFORE_UPLOAD",
+			workerName:   "wk-0",
+			commandArgs:  []string{"--" + model.FlagBase64},
+			serverBehavior: common.NewServerStub(t).
+				WithGetOneEndpoint().
+				WithCreateEndpoint(
+					expectDeployRequestBase64(actionsMeta, "wk-0", "BEFORE_UPLOAD", ""),
+				),
+		},
+		{
+			name:         "update with base64 encoding",
+			workerAction: "GENERIC_EVENT",
+			workerName:   "wk-1",
+			commandArgs:  []string{"--" + model.FlagBase64},
+			serverBehavior: common.NewServerStub(t).
+				WithGetOneEndpoint().
+				WithUpdateEndpoint(
+					expectDeployRequestBase64(actionsMeta, "wk-1", "GENERIC_EVENT", ""),
+				).
+				WithWorkers(&model.WorkerDetails{
+					Key: "wk-1",
+				}),
+		},
 		{
 			name: "fails when version invalid",
 			serverBehavior: common.NewServerStub(t).
@@ -239,6 +265,17 @@ func expectDeployRequest(actionsMeta common.ActionsMetadata, workerName, actionN
 	}
 }
 
+func expectDeployRequestBase64(actionsMeta common.ActionsMetadata, workerName, actionName, projectKey string, secrets ...*model.Secret) common.BodyValidator {
+	return func(t require.TestingT, body []byte) {
+		want := getExpectedDeployRequestForAction(t, actionsMeta, workerName, actionName, projectKey, secrets...)
+		want.SourceCode = "base64:" + base64.StdEncoding.EncodeToString([]byte(want.SourceCode))
+		got := &deployRequest{}
+		err := json.Unmarshal(body, got)
+		require.NoError(t, err)
+		assertDeployRequestEquals(t, want, got)
+	}
+}
+
 func getExpectedDeployRequestForAction(
 	t require.TestingT,
 	actionsMeta common.ActionsMetadata,

model/flags.go

@@ -25,6 +25,7 @@ const (
 	FlagChangesVersion     = "changes-version"
 	FlagChangesDescription = "changes-description"
 	FlagChangesCommitSha   = "changes-commitsha"
+	FlagBase64             = "base64"
 	defaultTimeoutMillis   = 5000
 )
 
@@ -119,6 +120,10 @@ func GetChangesCommitShaFlag() components.StringFlag {
 	return components.NewStringFlag(FlagChangesCommitSha, "Commit identifier or your change in your VCS.", components.WithStrDefaultValue(""))
 }
 
+func GetBase64Flag() components.BoolFlag {
+	return components.NewBoolFlag(FlagBase64, "Encode the worker source code in base64 before sending.", components.WithBoolDefaultValue(false))
+}
+
 func GetServerDetails(c *components.Context) (*config.ServerDetails, error) {
 	serverURLFromEnv, envHasServerURL := os.LookupEnv(EnvKeyServerURL)
 	accessTokenFromEnv, envHasAccessToken := os.LookupEnv(EnvKeyAccessToken)