shareable/unshareable coordinator

Author: cheggaaa

accountlimit/accountlimit.go

@@ -1,3 +1,4 @@
+//go:generate mockgen -destination mock_accountlimit/mock_accountlimit.go github.com/anyproto/any-sync-coordinator/accountlimit AccountLimit
 package accountlimit
 
 import (

accountlimit/mock_accountlimit/mock_accountlimit.go

@@ -12,6 +12,7 @@ import (
 	"github.com/anyproto/any-sync/app/logger"
 	"github.com/anyproto/any-sync/commonspace"
 	"github.com/anyproto/any-sync/commonspace/object/accountdata"
+	"github.com/anyproto/any-sync/commonspace/object/acl/list"
 	"github.com/anyproto/any-sync/commonspace/spacesyncproto"
 	"github.com/anyproto/any-sync/consensus/consensusproto"
 	"github.com/anyproto/any-sync/coordinator/coordinatorproto"
@@ -298,3 +299,73 @@ func (c *coordinator) AclAddRecord(ctx context.Context, spaceId string, payload
 	}
 	return rawRecordWithId, nil
 }
+
+func (c *coordinator) MakeSpaceShareable(ctx context.Context, spaceId string) (err error) {
+	pubKey, err := peer.CtxPubKey(ctx)
+	if err != nil {
+		return coordinatorproto.ErrForbidden
+	}
+	statusEntry, err := c.spaceStatus.Status(ctx, spaceId)
+	if err != nil {
+		return
+	}
+	if statusEntry.Identity != pubKey.Account() {
+		return coordinatorproto.ErrForbidden
+	}
+	if statusEntry.IsShareable {
+		return nil
+	}
+
+	limits, err := c.accountLimit.GetLimitsBySpace(ctx, spaceId)
+	if err != nil {
+		return err
+	}
+	return c.spaceStatus.MakeShareable(ctx, spaceId, limits.SharedSpacesLimit)
+}
+
+func (c *coordinator) MakeSpaceUnshareable(ctx context.Context, spaceId, aclHead string) (err error) {
+	pubKey, err := peer.CtxPubKey(ctx)
+	if err != nil {
+		return coordinatorproto.ErrForbidden
+	}
+	statusEntry, err := c.spaceStatus.Status(ctx, spaceId)
+	if err != nil {
+		return
+	}
+	if statusEntry.Identity != pubKey.Account() {
+		return coordinatorproto.ErrForbidden
+	}
+	if !statusEntry.IsShareable {
+		return nil
+	}
+
+	hasRecord, err := c.acl.HasRecord(ctx, spaceId, aclHead)
+	if err != nil {
+		return err
+	}
+	if !hasRecord {
+		return coordinatorproto.ErrAclHeadIsMissing
+	}
+
+	var (
+		activeMembers int
+		hasInvites    bool
+	)
+	err = c.acl.ReadState(ctx, spaceId, func(s *list.AclState) error {
+		for _, acc := range s.CurrentAccounts() {
+			if acc.Permissions.NoPermissions() {
+				continue
+			}
+			activeMembers++
+		}
+		if len(s.Invites()) > 0 {
+			hasInvites = true
+		}
+		return nil
+	})
+	if hasInvites || activeMembers > 1 {
+		return coordinatorproto.ErrAclNonEmpty
+	}
+
+	return c.spaceStatus.MakeUnshareable(ctx, spaceId)
+}