// RAINBOND, Application Management Platform
// Copyright (C) 2014-2020 Goodrain Co., Ltd.

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version. For any non-GPL usage of Rainbond,
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
// must be obtained first.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.

package cmd

import (
	"encoding/json"
	"fmt"
	"io"
	"net"
	"net/http"
	"os"
	"strings"

	"github.com/goodrain/rainbond/gateway/controller/openresty/model"
	"github.com/gosuri/uitable"
	"github.com/urfave/cli"
)

//NewCmdGateway gateway cmd
func NewCmdGateway() cli.Command {
	c := cli.Command{
		Name:  "gateway",
		Usage: "Gateway management related commands",
		Subcommands: []cli.Command{
			cli.Command{
				Name:  "endpoints",
				Usage: "list gateway http endpoints",
				Subcommands: []cli.Command{
					cli.Command{
						Name:  "http",
						Usage: "list gateway http endpoints",
						Flags: []cli.Flag{
							cli.IntFlag{
								Name:  "port",
								Usage: "gateway http endpoint query port",
								Value: 18080,
							},
						},
						Action: func(c *cli.Context) error {
							return listHTTPEndpoint(c)
						},
					},
					cli.Command{
						Name:  "stream",
						Usage: "list gateway stream endpoints",
						Flags: []cli.Flag{
							cli.IntFlag{
								Name:  "port",
								Usage: "gateway stream endpoint query port",
								Value: 18081,
							},
						},
						Action: func(c *cli.Context) error {
							return listStreamEndpoint(c)
						},
					},
				},
			},
		},
	}
	return c
}

func listStreamEndpoint(c *cli.Context) error {
	return tcpGetAndPrint(fmt.Sprintf("127.0.0.1:%d", c.Int("port")))
}

func listHTTPEndpoint(c *cli.Context) error {
	return httpGetAndPrint(fmt.Sprintf("http://127.0.0.1:%d/config/backends", c.Int("port")))
}
func tcpGetAndPrint(address string) error {
	conn, err := net.Dial("tcp", address)
	if err != nil {
		return err
	}
	defer conn.Close()
	_, err = conn.Write([]byte("GET\r\n"))
	if err != nil {
		return err
	}
	print(conn)
	return nil
}
func httpGetAndPrint(url string) error {
	res, err := http.Get(url)
	if err != nil {
		fmt.Println(err.Error())
		os.Exit(1)
	}
	if res.Body != nil {
		defer res.Body.Close()
		print(res.Body)
	}
	return nil
}

func print(reader io.Reader) {
	decoder := json.NewDecoder(reader)
	var backends []*model.Backend
	if err := decoder.Decode(&backends); err != nil {
		fmt.Println(err.Error())
		os.Exit(1)
	}
	table := uitable.New()
	table.Wrap = true // wrap columns
	for _, b := range backends {
		table.AddRow(b.Name, strings.Join(func() []string {
			var re []string
			for _, e := range b.Endpoints {
				re = append(re, fmt.Sprintf("%s:%s %d", e.Address, e.Port, e.Weight))
			}
			return re
		}(), ";"))
	}
	fmt.Println(table)
}