package imapserver

import (
	"testing"

	"github.com/mjl-/mox/store"
)

func TestNumSetContains(t *testing.T) {
	num := func(v uint32) *setNumber {
		return &setNumber{v, false}
	}
	star := &setNumber{star: true}

	check := func(v bool) {
		t.Helper()
		if !v {
			t.Fatalf("bad")
		}
	}

	ss0 := numSet{true, nil} // "$"
	check(ss0.containsSeq(1, []store.UID{2}, []store.UID{2}))
	check(!ss0.containsSeq(1, []store.UID{2}, []store.UID{}))

	ss1 := numSet{false, []numRange{{*num(1), nil}}} // Single number 1.
	check(ss1.containsSeq(1, []store.UID{2}, nil))
	check(!ss1.containsSeq(2, []store.UID{1, 2}, nil))

	check(ss1.containsSeq(1, []store.UID{2}, nil))
	check(!ss1.containsSeq(2, []store.UID{1, 2}, nil))

	// 2:*
	ss2 := numSet{false, []numRange{{*num(2), star}}}
	check(ss2.containsSeq(1, []store.UID{2}, nil))
	check(!ss2.containsSeq(2, []store.UID{2}, nil))
	check(ss2.containsSeq(2, []store.UID{4, 5}, nil))
	check(ss2.containsSeq(3, []store.UID{4, 5, 6}, nil))
	check(!ss2.containsSeq(4, []store.UID{4, 5, 6}, nil))

	// *:2, same as 2:*
	ss3 := numSet{false, []numRange{{*star, num(2)}}}
	check(ss3.containsSeq(1, []store.UID{2}, nil))
	check(!ss3.containsSeq(2, []store.UID{2}, nil))
	check(ss3.containsSeq(2, []store.UID{4, 5}, nil))
	check(ss3.containsSeq(3, []store.UID{4, 5, 6}, nil))
	check(!ss3.containsSeq(4, []store.UID{4, 5, 6}, nil))
}

func TestNumSetInterpret(t *testing.T) {
	parseNumSet := func(s string) numSet {
		p := parser{upper: s}
		return p.xnumSet0(true, false)
	}

	checkEqual := func(lastUID store.UID, a, s string) {
		t.Helper()
		n := parseNumSet(a).xinterpretStar(func() store.UID { return lastUID })
		ns := n.String()
		if ns != s {
			t.Fatalf("%s != %s", ns, s)
		}
	}

	checkEqual(0, "1:*", "0:1")
	checkEqual(1, "1:*", "1")
	checkEqual(3, "1:*", "1:3")
	checkEqual(3, "4:*", "3:4")
	checkEqual(3, "*:4", "3:4")
	checkEqual(3, "*:4", "3:4")
	checkEqual(3, "*:1", "1:3")
	checkEqual(3, "1:*", "1:3")
	checkEqual(3, "1,2,3", "1,2,3")
	checkEqual(0, "1,2,3", "1,2,3")
	checkEqual(0, "1:3", "1:3")
	checkEqual(0, "3:1", "1:3")

	iter := parseNumSet("3:1").xinterpretStar(func() store.UID { return 2 }).newIter()
	v0, _ := iter.Next()
	v1, _ := iter.Next()
	v2, _ := iter.Next()
	_, ok := iter.Next()
	if v0 != 1 || v1 != 2 || v2 != 3 || ok {
		t.Fatalf("got %v %v %v %v, expected 1, 2, 3 false", v0, v1, v2, ok)
	}
}