automod: basic set storage API

automod/engine.go

···
 	"fmt"
 	"log/slog"
 	"strings"
-	"sync"
 
 	appbsky "github.com/bluesky-social/indigo/api/bsky"
 	"github.com/bluesky-social/indigo/atproto/identity"
 	"github.com/bluesky-social/indigo/atproto/syntax"
-	"github.com/bluesky-social/indigo/xrpc"
 )
 
 // runtime for executing rules, managing state, and recording moderation actions.
 //
 // TODO: careful when initializing: several fields should not be null or zero, even though they are pointer type.
 type Engine struct {
-	Logger  *slog.Logger
+	Logger    *slog.Logger
 	Directory identity.Directory
 	// current rule sets. will eventually be possible to swap these out at runtime
-	RulesMap  sync.Map
+	// TODO: RulesMap  sync.Map
 	// used to persist moderation actions in mod service (optional)
-	AdminClient *xrpc.Client
-	CountStore  CountStore
+	// TODO: AdminClient *xrpc.Client
+	Counters CountStore
+	Sets     SetStore
 }
 
 func (e *Engine) ProcessIdentityEvent(ctx context.Context, t string, did syntax.DID) error {
···
 
 func (e *Engine) NewPostEvent(ident *identity.Identity, path string, post *appbsky.FeedPost) PostEvent {
 	return PostEvent{
-		RecordEvent {
+		RecordEvent{
 			Event{
 				Engine:  e,
 				Account: AccountMeta{Identity: ident},
···
 }
 
 func (e *Engine) GetCount(key, period string) (int, error) {
-	return e.CountStore.GetCount(context.TODO(), key, period)
+	return e.Counters.GetCount(context.TODO(), key, period)
 }
 
-func (e *Engine) InSet(setName, val string) (bool, error) {
-	// XXX: implement
-	return false, nil
+// checks if `val` is an element of set `name`
+func (e *Engine) InSet(name, val string) (bool, error) {
+	return e.Sets.InSet(context.TODO(), name, val)
 }

automod/setstore.go

···
+package automod
+
+import (
+	"context"
+	"fmt"
+)
+
+type SetStore interface {
+	InSet(ctx context.Context, name, val string) (bool, error)
+}
+
+// TODO: this implementation isn't race-safe (yet)!
+type MemSetStore struct {
+	Sets map[string]map[string]bool
+}
+
+func NewMemSetStore() MemSetStore {
+	return MemSetStore{
+		Sets: make(map[string]map[string]bool),
+	}
+}
+
+func (s MemSetStore) InSet(ctx context.Context, name, val string) (bool, error) {
+	set, ok := s.Sets[name]
+	if !ok {
+		return false, fmt.Errorf("not a known set: %s", name)
+	}
+	_, ok = set[val]
+	return ok, nil
+}

cmd/hepa/otel.go

···
 package main
 
 import (
-	"os"
-	"log/slog"
-	"log"
 	"context"
+	"log"
+	"log/slog"
+	"os"
 	"time"
 
 	"go.opentelemetry.io/otel"

cmd/hepa/server.go

···
 	}
 
 	engine := automod.Engine{
-		Logger: logger,
+		Logger:    logger,
 		Directory: dir,
-		CountStore: automod.NewMemCountStore(),
+		Counters:  automod.NewMemCountStore(),
+		Sets:      automod.NewMemSetStore(),
 		// TODO: RulesMap (loaded/config from somewhere)
 		// TODO: AdminClient (XRPC with mod access)
 	}