package integration

import (
	"code.fbi.h-da.de/cocsn/gosdn/cli"
	"code.fbi.h-da.de/cocsn/gosdn/forks/goarista/gnmi"
	"code.fbi.h-da.de/cocsn/gosdn/nucleus"
	"code.fbi.h-da.de/cocsn/gosdn/nucleus/util/proto"
	guuid "github.com/google/uuid"
	gpb "github.com/openconfig/gnmi/proto/gnmi"
	log "github.com/sirupsen/logrus"
	"github.com/spf13/viper"
	pb "google.golang.org/protobuf/proto"
	"os"
	"testing"
)

const unreachable = "203.0.113.10:6030"

var testAddress = "141.100.70.171:6030"
var testAPIEndpoint = "http://141.100.70.171:8080"
var testUsername = "admin"
var testPassword = "arista"
var defaultPath = []string{"/system/config/hostname"}
var opt *nucleus.GnmiTransportOptions

func testSetupIntegration() {
	if os.Getenv("GOSDN_LOG") == "nolog" {
		log.SetLevel(log.PanicLevel)
	}

	a := os.Getenv("GOSDN_TEST_ENDPOINT")
	if a != "" {
		testAddress = a
	}
	api := os.Getenv("GOSDN_TEST_API_ENDPOINT")
	if api != "" {
		testAPIEndpoint = api
	}
	u := os.Getenv("GOSDN_TEST_USER")
	if u != "" {
		testUsername = u
	}
	p := os.Getenv("GOSDN_TEST_PASSWORD")
	if p != "" {
		testPassword = p
	}

	gnmiMessages = map[string]pb.Message{
		"../proto/cap-resp-arista-ceos":                  &gpb.CapabilityResponse{},
		"../proto/req-full-node":                         &gpb.GetRequest{},
		"../proto/req-full-node-arista-ceos":             &gpb.GetRequest{},
		"../proto/req-interfaces-arista-ceos":            &gpb.GetRequest{},
		"../proto/req-interfaces-interface-arista-ceos":  &gpb.GetRequest{},
		"../proto/req-interfaces-wildcard":               &gpb.GetRequest{},
		"../proto/resp-full-node":                        &gpb.GetResponse{},
		"../proto/resp-full-node-arista-ceos":            &gpb.GetResponse{},
		"../proto/resp-interfaces-arista-ceos":           &gpb.GetResponse{},
		"../proto/resp-interfaces-interface-arista-ceos": &gpb.GetResponse{},
		"../proto/resp-interfaces-wildcard":              &gpb.GetResponse{},
		"../proto/resp-set-system-config-hostname":       &gpb.SetResponse{},
	}
	for k, v := range gnmiMessages {
		if err := proto.Read(k, v); err != nil {
			log.Fatalf("error parsing %v: %v", k, err)
		}
	}

	opt = &nucleus.GnmiTransportOptions{
		Config: gnmi.Config{
			Addr:     testAddress,
			Username: "admin",
			Password: "arista",
			Encoding: gpb.Encoding_JSON_IETF,
		},
		RespChan: make(chan *gpb.SubscribeResponse),
	}
}

func TestCliIntegration(t *testing.T) {
	if testing.Short() {
		t.Skip("skipping integration test")
	}
	tests := []struct {
		name    string
		wantErr bool
	}{
		{
			name:    "default",
			wantErr: false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			defer viper.Reset()
			if err := cli.HTTPGet(testAPIEndpoint, "init"); (err != nil) != tt.wantErr {
				switch err.(type) {
				case viper.ConfigFileNotFoundError:
				default:
					t.Errorf("gosdn cli init error = %v, wantErr %v", err, tt.wantErr)
					return
				}
			}
			cliPnd := viper.GetString("CLI_PND")
			cliSbi := viper.GetString("CLI_SBI")

			if err := cli.HTTPGet(
				testAPIEndpoint,
				"addDevice",
				"address="+testAddress,
				"password="+testPassword,
				"username="+testUsername,
				"sbi="+cliSbi,
				"pnd="+cliPnd,
			); (err != nil) != tt.wantErr {
				t.Errorf("gosdn cli add-device error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			did := viper.GetString("LAST_DEVICE_UUID")

			if err := cli.HTTPGet(
				testAPIEndpoint,
				"request",
				"uuid="+did,
				"sbi="+cliSbi,
				"pnd="+cliPnd,
				"path=/system/config/hostname",
			); (err != nil) != tt.wantErr {
				t.Errorf("gosdn cli request error = %v, wantErr %v", err, tt.wantErr)
				return
			}

			if err := cli.HTTPGet(
				testAPIEndpoint,
				"getDevice",
				"address="+testAddress,
				"uuid="+did,
				"sbi="+cliSbi,
				"pnd="+cliPnd,
			); (err != nil) != tt.wantErr {
				t.Errorf("gosdn cli get-device error = %v, wantErr %v", err, tt.wantErr)
				return
			}

			hostname := guuid.New().String()
			if err := cli.HTTPGet(
				testAPIEndpoint,
				"set",
				"address="+testAddress,
				"uuid="+did,
				"sbi="+cliSbi,
				"pnd="+cliPnd,
				"path=/system/config/hostname",
				"value="+hostname,
			); (err != nil) != tt.wantErr {
				t.Errorf("gosdn cli set error = %v, wantErr %v", err, tt.wantErr)
				return
			}

			resp, err := cli.Get(testAddress, testUsername, testPassword, "/system/config/hostname")
			if (err != nil) != tt.wantErr {
				t.Errorf("cli.Get() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			got := resp.Notification[0].Update[0].Val.GetStringVal()
			if got != hostname {
				t.Errorf("integration test failed = got: %v, want: %v", got, hostname)
			}
		})
	}
}