Generate member lists

zooid/groups.go

@@ -4,6 +4,8 @@ import (
 	"fiatjaf.com/nostr"
 )
 
+// Utils
+
 func GetGroupIDFromEvent(event nostr.Event) string {
 	tag := event.Tags.Find("h")
 
@@ -14,11 +16,16 @@ func GetGroupIDFromEvent(event nostr.Event) string {
 	return ""
 }
 
+// Struct definition
+
 type GroupStore struct {
-	Config *Config
-	Events *EventStore
+	Config     *Config
+	Events     *EventStore
+	Management *ManagementStore
 }
 
+// Metadata
+
 func (g *GroupStore) GetMetadata(h string) nostr.Event {
 	filter := nostr.Filter{
 		Kinds: []nostr.Kind{nostr.KindSimpleGroupMetadata},
@@ -34,6 +41,81 @@ func (g *GroupStore) GetMetadata(h string) nostr.Event {
 	return nostr.Event{}
 }
 
+func (g *GroupStore) SetMetadataFromEvent(event nostr.Event) error {
+	tags := nostr.Tags{}
+
+	for _, tag := range event.Tags {
+		if len(tag) >= 2 && tag[0] == "h" {
+			tags = append(tags, nostr.Tag{"d", tag[1]})
+		} else {
+			tags = append(tags, tag)
+		}
+	}
+
+	metadataEvent := nostr.Event{
+		Kind:      nostr.KindSimpleGroupMetadata,
+		CreatedAt: event.CreatedAt,
+		Tags:      tags,
+	}
+
+	return g.Events.SignAndStoreEvent(&metadataEvent, true)
+}
+
+// Deletion
+
+func (g *GroupStore) DeleteGroup(h string) {
+	filters := []nostr.Filter{
+		{
+			Tags: nostr.TagMap{
+				"d": []string{h},
+			},
+		},
+		{
+			Tags: nostr.TagMap{
+				"h": []string{h},
+			},
+		},
+	}
+
+	for _, filter := range filters {
+		for event := range g.Events.QueryEvents(filter, 0) {
+			g.Events.DeleteEvent(event.ID)
+		}
+	}
+}
+
+// Admins
+
+func (g *GroupStore) IsAdmin(h string, pubkey nostr.PubKey) bool {
+	return g.Management.IsAdmin(pubkey)
+}
+
+func (g *GroupStore) GetAdmins(h string) []nostr.PubKey {
+	return g.Management.GetAdmins()
+}
+
+func (g *GroupStore) GenerateAdminsEvent(h string) nostr.Event {
+	tags := nostr.Tags{
+		nostr.Tag{"-"},
+	}
+
+	for _, pubkey := range g.GetAdmins(h) {
+		tags = append(tags, nostr.Tag{"p", pubkey.Hex()})
+	}
+
+	event := nostr.Event{
+		Kind:      nostr.KindSimpleGroupAdmins,
+		CreatedAt: nostr.Now(),
+		Tags:      tags,
+	}
+
+	g.Config.Sign(&event)
+
+	return event
+}
+
+// Membership
+
 func (g *GroupStore) AddMember(h string, pubkey nostr.PubKey) error {
 	event := nostr.Event{
 		Kind:      nostr.KindSimpleGroupPutUser,
@@ -60,47 +142,6 @@ func (g *GroupStore) RemoveMember(h string, pubkey nostr.PubKey) error {
 	return g.Events.SignAndStoreEvent(&event, true)
 }
 
-func (g *GroupStore) SetMetadataFromEvent(event nostr.Event) error {
-	tags := nostr.Tags{}
-
-	for _, tag := range event.Tags {
-		if len(tag) >= 2 && tag[0] == "h" {
-			tags = append(tags, nostr.Tag{"d", tag[1]})
-		} else {
-			tags = append(tags, tag)
-		}
-	}
-
-	metadataEvent := nostr.Event{
-		Kind:      nostr.KindSimpleGroupMetadata,
-		CreatedAt: event.CreatedAt,
-		Tags:      tags,
-	}
-
-	return g.Events.SignAndStoreEvent(&metadataEvent, true)
-}
-
-func (g *GroupStore) DeleteGroup(h string) {
-	filters := []nostr.Filter{
-		{
-			Tags: nostr.TagMap{
-				"d": []string{h},
-			},
-		},
-		{
-			Tags: nostr.TagMap{
-				"h": []string{h},
-			},
-		},
-	}
-
-	for _, filter := range filters {
-		for event := range g.Events.QueryEvents(filter, 0) {
-			g.Events.DeleteEvent(event.ID)
-		}
-	}
-}
-
 func (g *GroupStore) IsMember(h string, pubkey nostr.PubKey) bool {
 	filter := nostr.Filter{
 		Kinds: []nostr.Kind{nostr.KindSimpleGroupPutUser, nostr.KindSimpleGroupRemoveUser},
@@ -123,10 +164,65 @@ func (g *GroupStore) IsMember(h string, pubkey nostr.PubKey) bool {
 	return false
 }
 
+func (g *GroupStore) GetMembers(h string) []nostr.PubKey {
+	filter := nostr.Filter{
+		Kinds: []nostr.Kind{nostr.KindSimpleGroupPutUser, nostr.KindSimpleGroupRemoveUser},
+		Tags: nostr.TagMap{
+			"h": []string{h},
+		},
+	}
+
+	members := make([]nostr.PubKey, 0)
+
+	for event := range g.Events.QueryEvents(filter, 0) {
+		for hex := range event.Tags.FindAll("p") {
+			if pubkey, err := nostr.PubKeyFromHex(hex[1]); err != nil {
+				if event.Kind == nostr.KindSimpleGroupPutUser {
+					members = append(members, pubkey)
+				} else {
+					members = Remove(members, pubkey)
+				}
+			}
+		}
+	}
+
+	return members
+}
+
+func (g *GroupStore) GenerateMembersEvent(h string) nostr.Event {
+	tags := nostr.Tags{
+		nostr.Tag{"-"},
+	}
+
+	for _, pubkey := range g.GetMembers(h) {
+		tags = append(tags, nostr.Tag{"p", pubkey.Hex()})
+	}
+
+	event := nostr.Event{
+		Kind:      nostr.KindSimpleGroupMembers,
+		CreatedAt: nostr.Now(),
+		Tags:      tags,
+	}
+
+	g.Config.Sign(&event)
+
+	return event
+}
+
+// Other stuff
+
 func (g *GroupStore) HasAccess(h string, pubkey nostr.PubKey) bool {
 	if !HasTag(g.GetMetadata(h).Tags, "closed") {
 		return true
 	}
 
-	return g.IsMember(h, pubkey)
+	if g.IsAdmin(h, pubkey) {
+		return true
+	}
+
+	if g.IsMember(h, pubkey) {
+		return true
+	}
+
+	return false
 }