package protocol

import (
	"errors"
	"testing"
)

func TestDetectSSH(t *testing.T) {
	atomicFormats.Store([]format{{Both, "", detectSSH}})

	// 1. A standard OpenSSH banner
	openSSHBanner := []byte("SSH-2.0-OpenSSH_8.9p1 Ubuntu-3ubuntu0.4\r\n")

	// 2. An SSH banner with a pre-banner legal notice
	preBannerSSH := []byte(
		"*******************************************************************\r\n" +
			"* W A R N I N G                          *\r\n" +
			"* This system is for the use of authorized users only.          *\r\n" +
			"*******************************************************************\r\n" +
			"SSH-2.0-OpenSSH_7.6p1\r\n",
	)

	// 3. A different SSH implementation (Dropbear)
	dropbearBanner := []byte("SSH-2.0-dropbear_2020.81\r\n")

	// 4. An invalid banner (e.g., the MySQL banner from the previous example)
	mysqlBanner := []byte{
		0x0a, 0x38, 0x2e, 0x30, 0x2e, 0x33, 0x32, 0x00, 0x0d, 0x00, 0x00, 0x00,
	}

	// 5. A simple HTTP request
	httpBanner := []byte("GET / HTTP/1.1\r\nHost: example.com\r\n\r\n")

	t.Run("OpenSSH client", func(t *testing.T) {
		p, err := Detect(Server, openSSHBanner)
		if err != nil {
			t.Fatal(err)
			return
		}
		t.Logf("detected %s version %s", p.Name, p.Version)
		if p.Name != ProtocolSSH {
			t.Fatalf("expected ssh protocol, got %s", p.Name)
			return
		}
	})

	t.Run("OpenSSH server", func(t *testing.T) {
		p, err := Detect(Server, openSSHBanner)
		if err != nil {
			t.Fatal(err)
			return
		}
		t.Logf("detected %s version %s", p.Name, p.Version)
		if p.Name != ProtocolSSH {
			t.Fatalf("expected ssh protocol, got %s", p.Name)
			return
		}
	})

	t.Run("OpenSSH server with banner", func(t *testing.T) {
		p, err := Detect(Server, preBannerSSH)
		if err != nil {
			t.Fatal(err)
			return
		}
		t.Logf("detected %s version %s", p.Name, p.Version)
		if p.Name != ProtocolSSH {
			t.Fatalf("expected ssh protocol, got %s", p.Name)
			return
		}
	})

	t.Run("Dropbear server", func(t *testing.T) {
		p, err := Detect(Server, dropbearBanner)
		if err != nil {
			t.Fatal(err)
			return
		}
		t.Logf("detected %s version %s", p.Name, p.Version)
		if p.Name != ProtocolSSH {
			t.Fatalf("expected ssh protocol, got %s", p.Name)
			return
		}
	})

	t.Run("Invalid MySQL banner", func(t *testing.T) {
		_, err := Detect(Server, mysqlBanner)
		if !errors.Is(err, ErrUnknown) {
			t.Fatalf("expected unknown format, got error %T: %q", err, err)
		} else {
			t.Logf("error %q, as expected", err)
		}
	})

	t.Run("Invalid HTTP banner", func(t *testing.T) {
		_, err := Detect(Server, httpBanner)
		if !errors.Is(err, ErrUnknown) {
			t.Fatalf("expected unknown format, got error %T: %q", err, err)
		} else {
			t.Logf("error %q, as expected", err)
		}
	})
}