cmd_tidy.go

// Copyright 2019 Aporeto Inc.
// 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 main

import (
	"fmt"
	"os"
	"os/exec"

	"github.com/spf13/cobra"
	"github.com/spf13/viper"
	"go.aporeto.io/remod/internal/remod"
)

var cmdTidy = &cobra.Command{
	Use:     "tidy",
	Aliases: []string{"g"},
	Short:   "Run a wrapped go mod tidy command",
	Long: `This wraps a go mod tody command while remod is on.

This can be used to tidy the go.mod file. Every argument is passed
to the underlying go mod tidy command.
`,
	DisableFlagParsing: true,
	Example:            `remod tidy`,
	PreRunE: func(cmd *cobra.Command, args []string) error {
		return viper.BindPFlags(cmd.Flags())
	},
	Args: cobra.MaximumNArgs(1),
	RunE: func(cmd *cobra.Command, args []string) error {

		if len(args) == 1 && (args[0] == "-h" || args[0] == "--help") {
			return cmd.Usage()
		}

		if remod.Enabled() {
			if err := remod.Off(); err != nil {
				return fmt.Errorf("unable to set remod to off: %s", err)
			}

			defer func() {
				if err := remod.On(); err != nil {
					panic(err)
				}
			}()
		}

		c := exec.Command("go", append([]string{"mod", "tidy"}, args...)...)
		c.Stdin = os.Stdin
		c.Stderr = os.Stderr
		c.Stdout = os.Stdout

		if err := c.Start(); err != nil {
			return err
		}

		return c.Wait()
	},
}