package server

import (
	"context"

	"coopcloud.tech/abra/cli/formatter"
	"github.com/hetznercloud/hcloud-go/hcloud"
	"github.com/sirupsen/logrus"
	"github.com/urfave/cli/v2"
)

var hetznerCloudType string
var hetznerCloudImage string
var hetznerCloudSSHKeys cli.StringSlice
var hetznerCloudLocation string
var hetznerCloudAPIToken string
var serverNewHetznerCloudCommand = &cli.Command{
	Name:      "hetzner",
	Usage:     "Create a new Hetzner virtual server",
	ArgsUsage: "<name>",
	Description: `
Create a new Hetzner virtual server.

This command uses the uses the Hetzner Cloud API bindings to send a server
creation request. You must already have a Hetzner Cloud account and an account
API token before using this command.

Your token can be loaded from the environment using the HCLOUD_TOKEN
environment variable or otherwise passing the "--env/-e" flag.
`,
	Flags: []cli.Flag{
		&cli.StringFlag{
			Name:        "type",
			Aliases:     []string{"t"},
			Usage:       "Server type",
			Destination: &hetznerCloudType,
			Value:       "cx11",
		},
		&cli.StringFlag{
			Name:        "image",
			Aliases:     []string{"i"},
			Usage:       "Image type",
			Value:       "debian-10",
			Destination: &hetznerCloudImage,
		},
		&cli.StringSliceFlag{
			Name:        "ssh-keys",
			Aliases:     []string{"s"},
			Usage:       "SSH keys",
			Destination: &hetznerCloudSSHKeys,
		},
		&cli.StringFlag{
			Name:        "location",
			Aliases:     []string{"l"},
			Usage:       "Server location",
			Value:       "hel1",
			Destination: &hetznerCloudLocation,
		},
		&cli.StringFlag{
			Name:        "token",
			Aliases:     []string{"T"},
			Usage:       "Hetzner Cloud API token",
			EnvVars:     []string{"HCLOUD_TOKEN"},
			Destination: &hetznerCloudAPIToken,
		},
	},
	Action: func(c *cli.Context) error {
		name := c.Args().First()
		if name == "" {
			return cli.ShowSubcommandHelp(c)
		}

		if hetznerCloudAPIToken == "" {
			logrus.Fatal("API token is missing, cannot continue")
		}

		ctx := context.Background()
		client := hcloud.NewClient(hcloud.WithToken(hetznerCloudAPIToken))

		var sshKeys []*hcloud.SSHKey
		for _, sshKey := range c.StringSlice("ssh-keys") {
			sshKey, _, err := client.SSHKey.GetByName(ctx, sshKey)
			if err != nil {
				logrus.Fatal(err)
			}
			sshKeys = append(sshKeys, sshKey)
		}

		serverOpts := hcloud.ServerCreateOpts{
			Name:       name,
			ServerType: &hcloud.ServerType{Name: hetznerCloudType},
			Image:      &hcloud.Image{Name: hetznerCloudImage},
			SSHKeys:    sshKeys,
			Location:   &hcloud.Location{Name: hetznerCloudLocation},
		}
		res, _, err := client.Server.Create(ctx, serverOpts)
		if err != nil {
			logrus.Fatal(err)
		}

		tableColumns := []string{"Name", "IPv4", "Root Password"}
		table := formatter.CreateTable(tableColumns)
		if len(sshKeys) > 0 {
			table.Append([]string{name, res.Server.PublicNet.IPv4.IP.String(), "N/A (using SSH keys)"})
		} else {
			table.Append([]string{name, res.Server.PublicNet.IPv4.IP.String(), res.RootPassword})
		}
		table.Render()

		return nil
	},
}

var serverNewCommand = &cli.Command{
	Name:        "new",
	Usage:       "Create a new server using a 3rd party provider",
	Description: "Use a provider plugin to create a new server which can then be used to house a new Co-op Cloud installation.",
	ArgsUsage:   "<provider>",
	Subcommands: []*cli.Command{
		serverNewHetznerCloudCommand,
	},
}