package wayland

import (
	"testing"
	"time"

	"github.com/bnema/neferwl/internal/ports"
	"github.com/bnema/purego-libwayland/protocol/kdedecoration"
	"github.com/bnema/purego-libwayland/protocol/wayland"
	"github.com/bnema/wlturbo"
)

type kdeModeProxy struct {
	wlturbo.BaseProxy
	modes chan uint32
}

func (p *kdeModeProxy) Dispatch(e *wlturbo.Event) { p.modes <- e.Uint32() }

// KDE decoration defaults to server-side and acknowledges the mode a client
// asks for: answering server to Firefox's client request made it ask again
// in a tight loop.
func TestKDEDecorationAcknowledgesMode(t *testing.T) {
	s, _, _, dir := dmabufServer(t, ports.DMABufSupport{})
	c := protocolClient(t, s, dir)
	comp := bindProtocol(t, c, "wl_compositor")
	mgr := bindProtocol(t, c, "org_kde_kwin_server_decoration_manager")
	registerProtocol(t, c, mgr)
	surf := c.AllocateID()
	requestProtocol(t, c, comp, wayland.CompositorRequestCreateSurface, surf)
	deco := c.AllocateID()
	proxy := &kdeModeProxy{modes: make(chan uint32, 8)}
	proxy.SetID(deco)
	c.Context().Register(proxy)
	requestProtocol(t, c, mgr, kdedecoration.OrgKdeKwinServerDecorationManagerRequestCreate, deco, surf)
	requestProtocol(t, c, deco, kdedecoration.OrgKdeKwinServerDecorationRequestRequestMode, uint32(kdedecoration.OrgKdeKwinServerDecorationModeClient))
	// An undefined mode gets no answer.
	requestProtocol(t, c, deco, kdedecoration.OrgKdeKwinServerDecorationRequestRequestMode, uint32(3))
	if err := c.Roundtrip(); err != nil {
		t.Fatal(err)
	}
	for _, want := range []kdedecoration.OrgKdeKwinServerDecorationMode{kdedecoration.OrgKdeKwinServerDecorationModeServer, kdedecoration.OrgKdeKwinServerDecorationModeClient} {
		select {
		case got := <-proxy.modes:
			if got != uint32(want) {
				t.Fatalf("mode %d want %d", got, want)
			}
		case <-time.After(2 * time.Second):
			t.Fatalf("no mode %d", want)
		}
	}
	select {
	case got := <-proxy.modes:
		t.Fatalf("extra mode %d", got)
	default:
	}
}