add(set-env-var): cmd for set env var

Author: JayByeun

.DS_Store

@@ -7,6 +7,7 @@
 *.dll
 *.so
 *.dylib
+*.xml
 
 # Test binary, built with `go test -c`
 *.test

.gitignore

@@ -0,0 +1,78 @@
+// Copyright (c) Codesphere Inc.
+// SPDX-License-Identifier: Apache-2.0
+
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/codesphere-cloud/cs-go/api"
+	"github.com/codesphere-cloud/cs-go/pkg/out"
+	"github.com/jedib0t/go-pretty/v6/table"
+
+	"github.com/spf13/cobra"
+)
+
+type SetEnvVarCmd struct {
+    opts SetEnvVarOptions
+	cmd  *cobra.Command
+}
+
+type SetEnvVarOptions struct {
+	GlobalOptions
+	WorkspaceId *int
+	EnvVar map[string]string
+}
+
+func addSetEnvVarCmd(p *cobra.Command, opts GlobalOptions) {
+	l := ListTeamsCmd{
+		cmd: &cobra.Command{
+			Use:   "env vars",
+			Short: "set env vars",
+			Long:  `set an environment variable for your workspace`,
+			Example: `
+Set environment variable:
+
+$ cs set env var --workspace-id <workspace-id> --name <env-var-name> --value <env-var-value>
+			`,
+		},
+		opts: opts,
+	}
+	l.cmd.RunE = l.RunE
+	p.AddCommand(l.cmd)
+}
+
+func (l *SetEnvVarCmd) RunE(_ *cobra.Command, args []string) (err error) {
+	client, err := NewClient(l.opts)
+	if err != nil {
+		return fmt.Errorf("Failed to create Codesphere client: %w", err)
+	}
+    workspaceId, err := l.getWorkspaceId(client.workspaceId)
+    if (err !== nil) {
+        return nil, fmt.Errorf("Failed to get workspaceId (%s): %w", client, err)
+    }
+    err := client.SetEnvVarOnWorkspace(workspaceId, *l.Opts.EnvVar)
+    if (err != nil) {
+        return nil, fmt.Errorf("Failed to set environment variable: %s", *l.Opts.EnvVar)
+    }
+
+    return nil
+}
+
+func (l *SetEnvVarCmd) getWorkspaceId(client Client) (workspaceId int, err error) {
+    if l.Opts.WorkspaceId != nil && *l.Opts.WorkspaceId >= 0 {
+        workspaceId := *l.Opts.WorkspaceId
+        return
+    }
+	var allWorkspaces []api.Workspace{}
+	allWorkspaces, err = client.ListWorkspaces()
+	if err != nil {
+		return
+	}
+	for _, w := range allWorkspaces {
+		if int(w.id) == *l.Opts.WorkspaceId
+		    workspaceId := int(w.id)
+		    return
+	}
+	return
+}