internal/cmd/integrations/publish.go

// Copyright 2021 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package integrations

import (
	"errors"
	"fmt"
	"internal/apiclient"
	"internal/client/integrations"
	"internal/clilog"
	"internal/cmd/utils"
	"os"
	"strconv"

	"github.com/spf13/cobra"
	"github.com/spf13/pflag"
)

// PublishVerCmd to publish an integration flow version
var PublishVerCmd = &cobra.Command{
	Use:   "publish",
	Short: "Publish an integration flow version",
	Long:  "Publish an integration flow version",
	Args: func(cmd *cobra.Command, args []string) (err error) {
		cmdProject := cmd.Flag("proj")
		cmdRegion := cmd.Flag("reg")
		version := utils.GetStringParam(cmd.Flag("ver"))
		userLabel := utils.GetStringParam(cmd.Flag("user-label"))
		snapshot := utils.GetStringParam(cmd.Flag("snapshot"))
		latest, _ := strconv.ParseBool(utils.GetStringParam(cmd.Flag("latest")))

		if err = apiclient.SetRegion(utils.GetStringParam(cmdRegion)); err != nil {
			return err
		}
		if err = validate(version, userLabel, snapshot, latest); err != nil {
			return err
		}
		cmd.Flags().VisitAll(func(f *pflag.Flag) {
			clilog.Debug.Printf("%s: %s\n", f.Name, f.Value)
		})
		return apiclient.SetProjectID(utils.GetStringParam(cmdProject))
	},
	RunE: func(cmd *cobra.Command, args []string) (err error) {
		cmd.SilenceUsage = true

		version := utils.GetStringParam(cmd.Flag("ver"))
		userLabel := utils.GetStringParam(cmd.Flag("user-label"))
		snapshot := utils.GetStringParam(cmd.Flag("snapshot"))
		name := utils.GetStringParam(cmd.Flag("name"))
		configVarsJson := utils.GetStringParam(cmd.Flag("config-vars-json"))
		configVarsFile := utils.GetStringParam(cmd.Flag("config-vars"))

		var contents []byte
		var info string

		if configVarsFile != "" {
			if _, err := os.Stat(configVarsFile); os.IsNotExist(err) {
				return err
			}

			contents, err = os.ReadFile(configVarsFile)
			if err != nil {
				return err
			}
		}

		if configVarsJson != "" {
			contents = []byte(configVarsJson)
		}

		latest := ignoreLatest(version, userLabel, snapshot)

		if latest {
			apiclient.DisableCmdPrintHttpResponse()
			// list integration versions, order by state=SNAPSHOT, page size = 1 and return basic info
			respBody, err := integrations.ListVersions(name, 1, "", "state=SNAPSHOT",
				"snapshot_number", false, false, true)
			if err != nil {
				return fmt.Errorf("unable to list versions: %v", err)
			}
			if string(respBody) == "{}" {
				if respBody, err = integrations.ListVersions(name, 1, "", "state=DRAFT",
					"snapshot_number", false, false, true); err != nil {
					return fmt.Errorf("unable to list versions: %v", err)
				}
			}
			version, err = integrations.GetIntegrationVersion(respBody)
			if err != nil {
				return err
			}
			apiclient.EnableCmdPrintHttpResponse()
			_, err = integrations.Publish(name, version, contents)
			info = "version " + version
		} else if version != "" {
			_, err = integrations.Publish(name, version, contents)
			info = "version " + version
		} else if userLabel != "" {
			_, err = integrations.PublishUserLabel(name, userLabel, contents)
			info = "user label " + userLabel
		} else if snapshot != "" {
			_, err = integrations.PublishSnapshot(name, snapshot, contents)
			info = "snapshot number " + snapshot
		}
		if err == nil {
			clilog.Info.Printf("Integration %s %s published successfully\n", name, info)
		}
		return err
	},
	Example: `Publishes an integration vesion with the highest snapshot in SNAPSHOT state: ` + GetExample(14) + `
Publishes an integration version that matches user supplied snapshot number: ` + GetExample(15),
}

func init() {
	var name, version, userLabel, snapshot, configVars, configVarsJson string
	var latest bool

	PublishVerCmd.Flags().StringVarP(&name, "name", "n",
		"", "Integration flow name")
	PublishVerCmd.Flags().StringVarP(&version, "ver", "v",
		"", "Integration flow version")
	PublishVerCmd.Flags().StringVarP(&userLabel, "user-label", "u",
		"", "Integration flow user label")
	PublishVerCmd.Flags().StringVarP(&snapshot, "snapshot", "s",
		"", "Integration flow snapshot number")
	PublishVerCmd.Flags().StringVarP(&configVars, "config-vars", "",
		"", "Path to file containing config variables")
	PublishVerCmd.Flags().StringVarP(&configVarsJson, "config-vars-json", "",
		"", "JSON string containing the config variables.")
	PublishVerCmd.Flags().BoolVarP(&latest, "latest", "",
		true, "Publishes the integeration version with the highest snapshot number in SNAPSHOT state; default is true")

	_ = PublishVerCmd.MarkFlagRequired("name")
}

func validate(version string, userLabel string, snapshot string, latest bool) (err error) {
	switch {
	case !latest && (version == "" && userLabel == "" && snapshot == ""):
		return errors.New("must pass oneOf version, snapshot or user-label")
	case !latest && (version != "" && (userLabel != "" || snapshot != "")):
		return errors.New("must pass oneOf version, snapshot or user-label")
	case !latest && (userLabel != "" && (version != "" || snapshot != "")):
		return errors.New("must pass oneOf version, snapshot or user-label")
	case !latest && (snapshot != "" && (userLabel != "" || version != "")):
		return errors.New("must pass oneOf version, snapshot or user-label")
	}
	return nil
}

func ignoreLatest(version string, userLabel string, snapshot string) (latest bool) {
	if version != "" || userLabel != "" || snapshot != "" {
		return false
	}
	return true
}