package main

import (
	"encoding/binary"
	"encoding/json"
	"flag"
	"fmt"
	"net"
	"os"
	"io"
)

type CommandMap struct {
	Type		string `json:"Type"`
	Endpoint	string `json:"Endpoint"`
}

// Instance's new min_id value
type RunningInstance struct {
	Software		string `json:"software"`
	Status			int	`json:"status"`
	LastRun			string `json:"lastrun"`
	CaptureType		string `json:"capturetype"`
}

type ResponseBack struct {
	Type			string `json:"Type"`
	Message			string `json:"Message"`
	RunningInstances        map[string]RunningInstance `json:"RunningInstances"`
}

func main() {

	shutdownPtr := flag.Bool("shutdown", false, "Shutdown server")
	suspendPtr := flag.String("suspend", "", "Instance to Suspend")
	resumePtr := flag.String("resume", "", "Instance to Resume")
	addPtr := flag.String("add", "", "Instance to add")
	statusPtr := flag.Bool("status", false, "Check status")
	flag.Parse()

	/* Condition verification */
	totalflags := 0
	var commandMap CommandMap
	var responseback ResponseBack

	if *shutdownPtr {
		totalflags++
		commandMap.Type = "shutdown"
	}
	if *statusPtr {
		totalflags++
		commandMap.Type = "status"
	}
	if *addPtr != "" {
		totalflags++
		commandMap.Type = "add"
		commandMap.Endpoint = *addPtr
	}
	if *suspendPtr != "" {
		totalflags++
		commandMap.Type = "suspend"
	}
	if *resumePtr != "" {
		totalflags++
		commandMap.Type = "resume"
	}
	if totalflags > 1 {
		fmt.Println("Incompatible arguments, exiting.")
		os.Exit(1)
	} else if totalflags == 0 {
		fmt.Println("No options specified, exiting.")
		os.Exit(1)
	}

	commandByte, err := json.Marshal(commandMap)
	if err != nil {
		fmt.Println(err)
		return
	}
	c, err := net.Dial("tcp", "127.0.0.1:5555")
	if err != nil {
		fmt.Println(err)
		return
	}
	sizebytes := make([]byte, 4)
	b := len(commandByte)

	// Send the request
	binary.LittleEndian.PutUint32(sizebytes, uint32(b))
	_, err = c.Write(sizebytes)
	if err != nil {
		fmt.Println(err)
		return
	}
	_, err = c.Write(commandByte)
	if err != nil {
		fmt.Println(err)
		return
	}

	// Read the response
	n, err := io.ReadFull(c, sizebytes)
	if err != nil || n != 4 {
		fmt.Println("err", err, n)
	}
	jsonsize := int(binary.LittleEndian.Uint32(sizebytes))
	responsebytes := make([]byte, jsonsize)
	_, err = io.ReadFull(c, responsebytes)
	if err != nil {
		fmt.Println("Read Error", err)
	}
	err = json.Unmarshal(responsebytes, &responseback)
	if err != nil {
		fmt.Println("Unmarshal error", err)
	}

	switch commandMap.Type {
	case "add":
		fmt.Println(responseback.Message)
	case "status":
		fmt.Println("Status\t\tLastRun\t\t\tCaptureType\tEndpoint")
		for endpoint, runninginstance := range responseback.RunningInstances {
			if runninginstance.Status == 0 {
				fmt.Fprintf(os.Stdout, "New\t")
				fmt.Fprintf(os.Stdout, "\tNever\t\t\t")
			} else if runninginstance.Status == 200 {
				fmt.Fprintf(os.Stdout, "Running\t")
				fmt.Fprintf(os.Stdout, "\t%s\t", runninginstance.LastRun)
			} else if runninginstance.Status == 429 {
				fmt.Fprintf(os.Stdout, "TooManyRequests\t")
				fmt.Fprintf(os.Stdout, "%s\t", runninginstance.LastRun)
			} else if runninginstance.Status == 600 {
				fmt.Fprintf(os.Stdout, "ClientIssue")
				fmt.Fprintf(os.Stdout, "\t%s\t", runninginstance.LastRun)
			} else if runninginstance.Status == 602 {
				fmt.Fprintf(os.Stdout, "BadInstance")
				fmt.Fprintf(os.Stdout, "\t%s\t", runninginstance.LastRun)
			} else if runninginstance.Status == 605 {
				fmt.Fprintf(os.Stdout, "UnsupportedNode")
				fmt.Fprintf(os.Stdout, "\t%s\t", runninginstance.LastRun)
			} else {
				fmt.Fprintf(os.Stdout, "%d\t", runninginstance.Status)
				fmt.Fprintf(os.Stdout, "\t%s\t", runninginstance.LastRun)
			}
			if runninginstance.LastRun == "Queued" {
				fmt.Fprintf(os.Stdout, "\t\t")
			}
			fmt.Fprintf(os.Stdout, "\t%s\t", runninginstance.CaptureType)
			fmt.Fprintf(os.Stdout, "%s\n", endpoint)
		}
	}
}