Better trie implementations

dataset/domain.go

@@ -1,11 +1,29 @@
 package dataset
 
 import (
+	"fmt"
 	"strings"
 
+	"git.maze.io/maze/styx/dataset/dnstrie"
 	"github.com/miekg/dns"
 )
 
+type DomainTrie struct {
+	*dnstrie.ValueTrie[bool]
+}
+
+func NewDomainTrie(permit bool, domains ...string) (*DomainTrie, error) {
+	trie := &DomainTrie{
+		ValueTrie: dnstrie.NewValue[bool](),
+	}
+	for _, domain := range domains {
+		if err := trie.Insert(domain, permit); err != nil {
+			return nil, fmt.Errorf("dataset: error inserting %s: %w", domain, err)
+		}
+	}
+	return trie, nil
+}
+
 type DomainTree struct {
 	root *domainTreeNode
 }