// Copyright (c) Authors of Clover
//
// All rights reserved. This program and the accompanying materials
// are made available under the terms of the Apache License, Version 2.0
// which accompanies this distribution, and is available at
// http://www.apache.org/licenses/LICENSE-2.0

package cmd

import (
    "fmt"
    "os"
    "strings"

    homedir "github.com/mitchellh/go-homedir"
    "github.com/spf13/cobra"
    "github.com/spf13/viper"
    "cloverkube"
)

var cfgFile string
var controllerIP string
var cloverFile string

// rootCmd represents the base command when called without any subcommands
var rootCmd = &cobra.Command{
    Use:   "cloverctl",
    Short: "Command-Line Interface (CLI) for Clover",
    Long: ``,
    // Uncomment the following line if your bare application
    // has an action associated with it:
    //Run: func(cmd *cobra.Command, args []string) {
    //},
}

// Execute adds all child commands to the root command and sets flags
// appropriately. This is called by main.main(). It only needs to happen
// once to the rootCmd.
func Execute() {
    if err := rootCmd.Execute(); err != nil {
        fmt.Println(err)
        os.Exit(1)
    }
}

func init() {
    cobra.OnInitialize(initConfig)

    // Here you will define your flags and configuration settings.
    // Cobra supports persistent flags, which, if defined here,
    // will be global for your application.
    rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "",
                              "config file (default is $HOME/.cloverctl.yaml)")

    // Cobra also supports local flags, which will only run
    // when this action is called directly.
    rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}

// initConfig reads in config file and ENV variables if set.
func initConfig() {
    if cfgFile != "" {
    // Use config file from the flag.
        viper.SetConfigFile(cfgFile)
    } else {
    // Find home directory.
        home, err := homedir.Dir()
        if err != nil {
            fmt.Println(err)
            os.Exit(1)
        }

        // Search config in home directory with name ".cloverctl"
        viper.AddConfigPath(home)
        viper.SetConfigName(".cloverctl")
    }

    viper.AutomaticEnv() // read in environment variables that match

    cip_file := ""
    // If a config file is found in home, read it in.
    if err := viper.ReadInConfig(); err == nil {
        fmt.Println("Using config file:", viper.ConfigFileUsed())
        cip_file = viper.GetString("ControllerIP")
    } else {
        // Check for file in path from cloverctl
        ep, err := os.Executable()
        if err == nil {
            exe_path := strings.Replace(ep, "cloverctl", "", -1)
            viper.AddConfigPath(exe_path)
            if err := viper.ReadInConfig(); err == nil {
                fmt.Println("Using config file:", viper.ConfigFileUsed())
                cip_file = viper.GetString("ControllerIP")
            }
        }
    }

    cPort, cip_kube := cloverkube.GetServicesPortIP("clover-controller")
    // If IP in file
    if cip_file != "" {
        // Nodeport
        controllerIP = "http://" + cip_file + ":" + fmt.Sprint(cPort)
    }
    // Override IP, if LB IP found
    if cip_kube != "" {
        fmt.Printf("IP %v", cip_kube)
        controllerIP = "http://" + cip_kube
    }
}

func checkControllerIP() {
    // controllerIP exists
    service := "clover-controller"
    if controllerIP == "" {
        fmt.Printf("%s address unspecified or cannot be found\n", service)
        os.Exit(1)
    } else {
        fmt.Printf("%s address: %s\n", service, controllerIP)
    }
}