// 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 imageCommand defines the commands for image workflows
package imageCommand

import (
	"github.com/aws/amazon-ecs-cli/ecs-cli/modules"
	"github.com/aws/amazon-ecs-cli/ecs-cli/modules/cli/image"
	"github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/flags"
	"github.com/aws/amazon-ecs-cli/ecs-cli/modules/commands/usage"
	"github.com/urfave/cli"
)

// PushCommand push ECR image
func PushCommand() cli.Command {
	return cli.Command{
		Name:         "push",
		Usage:        usage.Push,
		ArgsUsage:    image.PushImageFormat,
		Before:       app.BeforeApp,
		Action:       image.ImagePush,
		Flags:        flags.AppendFlags(imagePushFlags(), flags.OptionalRegionAndProfileFlags(), flags.DebugFlag(), fipsEndpointFlag()),
		OnUsageError: flags.UsageErrorFactory("push"),
	}
}

// PullCommand pull ECR image
func PullCommand() cli.Command {
	return cli.Command{
		Name:         "pull",
		Usage:        usage.Pull,
		ArgsUsage:    image.PullImageFormat,
		Before:       app.BeforeApp,
		Action:       image.ImagePull,
		Flags:        flags.AppendFlags(imagePullFlags(), flags.OptionalRegionAndProfileFlags(), flags.DebugFlag(), fipsEndpointFlag()),
		OnUsageError: flags.UsageErrorFactory("pull"),
	}
}

// ImagesCommand list images in ECR
func ImagesCommand() cli.Command {
	return cli.Command{
		Name:         "images",
		Usage:        usage.Images,
		ArgsUsage:    image.ListImageFormat,
		Before:       app.BeforeApp,
		Action:       image.ImageList,
		Flags:        flags.AppendFlags(imageListFlags(), flags.OptionalRegionAndProfileFlags(), flags.DebugFlag()),
		OnUsageError: flags.UsageErrorFactory("images"),
	}
}

func imagePushFlags() []cli.Flag {
	return []cli.Flag{
		cli.StringFlag{
			Name:  flags.RegistryIdFlag,
			Usage: "[Optional] Specifies the Amazon ECR registry ID to push the image to. By default, images are pushed to the current AWS account.",
		},
		cli.StringFlag{
			Name:  flags.ResourceTagsFlag,
			Usage: "[Optional] Specify AWS Resource tags which will be to your ECR repository. Specify in the format 'key1=value1,key2=value2,key3=value3.",
		},
	}
}

func imagePullFlags() []cli.Flag {
	return []cli.Flag{
		cli.StringFlag{
			Name:  flags.RegistryIdFlag,
			Usage: "[Optional] Specifies the the Amazon ECR registry ID to pull the image from. By default, images are pulled from the current AWS account.",
		},
	}
}

func imageListFlags() []cli.Flag {
	return []cli.Flag{
		cli.BoolFlag{
			Name:  flags.TaggedFlag,
			Usage: "[Optional] Filters the result to show only tagged images",
		},
		cli.BoolFlag{
			Name:  flags.UntaggedFlag,
			Usage: "[Optional] Filters the result to show only untagged images",
		},
	}
}

func fipsEndpointFlag() []cli.Flag {
	return []cli.Flag{
		cli.BoolFlag{
			Name:  flags.UseFIPSFlag + ",fips",
			Usage: "[Optional] Routes calls to ECR through FIPS endpoints.",
		},
	}
}