// Copyright 2017 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 (
	"flag"
	"fmt"
	"os"
	"runtime"

	"github.com/aws/amazon-ecs-cni-plugins/pkg/capabilities"
	"github.com/aws/amazon-ecs-cni-plugins/pkg/logger"
	"github.com/aws/amazon-ecs-cni-plugins/pkg/version"
	"github.com/aws/amazon-ecs-cni-plugins/plugins/eni/commands"
	"github.com/aws/amazon-ecs-cni-plugins/plugins/eni/version/cnispec"
	log "github.com/cihub/seelog"
	"github.com/containernetworking/cni/pkg/skel"
)

const (
	defaultLogFilePath = "/log/ecs-cni-eni-plugin.log"
)

func init() {
	// This is to ensure that all the namespace operations are performed for
	// a single thread
	runtime.LockOSThread()
}

func main() {
	defer log.Flush()
	logger.SetupLogger(logger.GetLogFileLocation(defaultLogFilePath))

	var printVersion bool
	var printCapabilities bool
	flag.BoolVar(&printVersion, "version", false, "prints version and exits")
	flag.BoolVar(&printCapabilities, capabilities.Command, false, "print a list of supported features")
	flag.Parse()

	if printVersion {
		if err := printVersionInfo(); err != nil {
			os.Stderr.WriteString(
				fmt.Sprintf("Error getting version string: %s", err.Error()))
			os.Exit(1)
		}
		return
	}

	if printCapabilities {
		// capabilities: awsvpc-network-mode
		capability := capabilities.New(capabilities.TaskENICapability)
		err := capability.Print()
		if err != nil {
			os.Stderr.WriteString(err.Error())
			os.Exit(1)
		}
		return
	}
	skel.PluginMain(commands.Add, commands.Del, cnispec.GetSpecVersionSupported())
}

func printVersionInfo() error {
	versionInfo, err := version.String()
	if err != nil {
		return err
	}
	fmt.Println(versionInfo)
	return nil
}