// Copyright 2015-2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may
// not use this file except in compliance with the License. A copy of the
// License is located at
//
//	http://aws.amazon.com/apache2.0/
//
// or in the "license" file accompanying this file. This file 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 (
	"os"
	"strings"

	"github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/flags"

	"github.com/aws/amazon-ecs-cli/ecs-cli/modules/cli/compose/factory"
	attributecheckercommand "github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/attributechecker"
	clusterCommand "github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/cluster"
	composeCommand "github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/compose"
	configureCommand "github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/configure"
	imageCommand "github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/image"
	licenseCommand "github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/license"
	localCommand "github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/local"
	logsCommand "github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/log"
	regcredsCommand "github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/regcreds"
	"github.com/aws/amazon-ecs-cli/ecs-cli/modules/utils/logger"
	"github.com/aws/amazon-ecs-cli/ecs-cli/modules/version"
	"github.com/sirupsen/logrus"
	"github.com/urfave/cli"
)

// cliArgsWithoutTestFlags returns command-line arguments without "go test" testflags.
func cliArgsWithoutTestFlags() []string {
	if !strings.HasSuffix(os.Args[0], "ecs-cli.test") {
		return os.Args
	}
	var args []string
	for _, arg := range os.Args {
		// Don't pass test flags used by Go to the ECS CLI. Otherwise, the ECS CLI will try to parse these
		// flags that are not defined and error.
		//
		// Example flag: -test.coverprofile=coverage.out
		// See "go help testflag" for full list.
		if strings.HasPrefix(arg, "-test.") {
			continue
		}
		args = append(args, arg)
	}
	return args
}

func main() {
	// Setup logrus for amazon-ecr-credential-helper
	logger.SetupLogger()

	app := cli.NewApp()
	app.Name = version.AppName
	app.Usage = "Command line interface for Amazon ECS"
	app.Version = version.String()
	app.Author = "Amazon Web Services"

	composeFactory := factory.NewProjectFactory()

	app.Commands = []cli.Command{
		configureCommand.ConfigureCommand(),
		clusterCommand.UpCommand(),
		clusterCommand.DownCommand(),
		clusterCommand.ScaleCommand(),
		clusterCommand.PsCommand(),
		imageCommand.PushCommand(),
		imageCommand.PullCommand(),
		imageCommand.ImagesCommand(),
		licenseCommand.LicenseCommand(),
		composeCommand.ComposeCommand(composeFactory),
		attributecheckercommand.AttributecheckerCommand(),
		logsCommand.LogCommand(),
		regcredsCommand.RegistryCredsCommand(),
		localCommand.LocalCommand(),
	}

	app.Flags = []cli.Flag{
		cli.StringFlag{
			Name:  flags.EndpointFlag,
			Usage: "Use a custom endpoint with the ECS CLI",
		},
	}

	err := app.Run(cliArgsWithoutTestFlags())
	if err != nil {
		logrus.Fatal(err)
	}
}