logos-messaging-nim/waku/waku_rln_relay/group_manager/static/group_manager.nim

import
    ../group_manager_base,
    ../../rln,
    std/sequtils

export
  group_manager_base

type
    StaticGroupManager* = ref object of GroupManager
      groupKeys*: seq[IdentityCredential]
      groupSize*: uint

template initializedGuard*(g: StaticGroupManager): untyped =
  if not g.initialized:
      raise newException(ValueError, "StaticGroupManager is not initialized")

method init*(g: StaticGroupManager): Future[void] {.async,gcsafe.} =
  let
    groupSize = g.groupSize
    groupKeys = g.groupKeys
    membershipIndex = if g.membershipIndex.isSome(): g.membershipIndex.get() 
                      else: raise newException(ValueError, "Membership index is not set") 

  if membershipIndex < MembershipIndex(0) or membershipIndex >= MembershipIndex(groupSize):
    raise newException(ValueError, "Invalid membership index. Must be within 0 and " & $(groupSize - 1) & "but was " & $membershipIndex)
  g.idCredentials = some(groupKeys[membershipIndex])

  # Seed the received commitments into the merkle tree
  let idCommitments = groupKeys.mapIt(it.idCommitment)
  let membersInserted = g.rlnInstance.insertMembers(g.latestIndex, idCommitments)
  if not membersInserted:
    raise newException(ValueError, "Failed to insert members into the merkle tree")

  discard g.slideRootQueue()

  g.latestIndex += MembershipIndex(idCommitments.len() - 1)

  g.initialized = true

  return

method startGroupSync*(g: StaticGroupManager): Future[void] =
  initializedGuard(g)
  var retFuture = newFuture[void]("StaticGroupManager.startGroupSync")
  # No-op
  retFuture.complete()
  return retFuture

method register*(g: StaticGroupManager, idCommitment: IDCommitment): Future[void] {.async.} =
  initializedGuard(g)

  await g.registerBatch(@[idCommitment])


method registerBatch*(g: StaticGroupManager, idCommitments: seq[IDCommitment]): Future[void] {.async.} =
  initializedGuard(g)

  let membersInserted = g.rlnInstance.insertMembers(g.latestIndex + 1, idCommitments)
  if not membersInserted:
    raise newException(ValueError, "Failed to insert members into the merkle tree")

  if g.registerCb.isSome():
    var memberSeq = newSeq[Membership]()
    for i in 0..<idCommitments.len():
      memberSeq.add(Membership(idCommitment: idCommitments[i], index: g.latestIndex + MembershipIndex(i) + 1))
    await g.registerCb.get()(memberSeq)

  discard g.slideRootQueue()

  g.latestIndex += MembershipIndex(idCommitments.len())

  return

method withdraw*(g: StaticGroupManager, idSecretHash: IdentitySecretHash): Future[void] {.async.} =
  initializedGuard(g)

  let groupKeys = g.groupKeys

  for i in 0..<groupKeys.len():
    if groupKeys[i].idSecretHash == idSecretHash:
        let idCommitment = groupKeys[i].idCommitment
        let index = MembershipIndex(i)
        let memberRemoved = g.rlnInstance.removeMember(index)
        if not memberRemoved:
          raise newException(ValueError, "Failed to remove member from the merkle tree")

        if g.withdrawCb.isSome():
          await g.withdrawCb.get()(@[Membership(idCommitment: idCommitment, index: index)])

        return


method withdrawBatch*(g: StaticGroupManager, idSecretHashes: seq[IdentitySecretHash]): Future[void] {.async.} =
  initializedGuard(g)

  # call withdraw on each idSecretHash
  for idSecretHash in idSecretHashes:
      await g.withdraw(idSecretHash)

method onRegister*(g: StaticGroupManager, cb: OnRegisterCallback) {.gcsafe.} =
  g.registerCb = some(cb)

method onWithdraw*(g: StaticGroupManager, cb: OnWithdrawCallback) {.gcsafe.} =
  g.withdrawCb = some(cb)

method stop*(g: StaticGroupManager): Future[void] =
  initializedGuard(g)
  # No-op
  var retFut = newFuture[void]("StaticGroupManager.stop")
  retFut.complete()
  return retFut

method isReady*(g: StaticGroupManager): Future[bool] {.gcsafe.} =
  initializedGuard(g)
  var retFut = newFuture[bool]("StaticGroupManager.isReady")
  retFut.complete(true)
  return retFut