Listgroup command

dcp/commands/__init__.py

@@ -68,5 +68,8 @@ def init():
     from dcp.commands.messagedelete import MessageDelete
     type_registry[MessageDelete.name] = MessageDelete()
 
+    from dcp.commands.listgroups import ListGroups
+    type_registry[ListGroups.name] = ListGroups()
+
     from dcp.commands.quit import Quit
     type_registry[Quit.name] = Quit()

dcp/commands/listgroups.py

+from sqlalchemy.future import select
+from sqlalchemy.exc import NoResultFound
+from dcp.config import conf
+from dcp.db import (StorageUser,StorageGroup,StorageGroupAcl,
+                    StorageGroupMember,StorageUserLine,StorageGroupLine,
+                    async_session)
+from dcp.frame import Frame
+from dcp.client import username_to_client
+from dcp.send import sendto_user, sendto_group, filt_exclude_client
+from dcp.commands import Command, CommandError
+
+class ListGroups(Command):
+    name = 'listgroups'
+    async def pre_reg(self, _client, _frame):
+        raise CommandError("You have not identified")
+
+    async def reg(self, client, frame):
+        target = "none"
+        print(frame.dst)
+        if len(frame.dst)>1:
+            target=frame.dst
+            if target != 'user' and target != 'none':
+                raise CommandError(f"{target} is Invalid usage. Accepts user or none, for listing groups you are a member of, or all publicly listed groups on the server")
+        async with async_session() as session:
+            async with session.begin():
+                if target == 'user':
+                    stmt = (select(StorageGroup)
+                            .join(StorageGroupMember,StorageGroupMember.group_id==StorageGroup.id)
+                            .where(StorageUser.username==client.username))
+                else:
+                    stmt = (select(StorageGroup))
+                result = await session.execute(stmt)
+
+            reply = Frame.create(f'@{client.username}', frame.dst,
+                                'listgroups',frame.related,
+                                {'groups':[i.StorageGroup.name for i in result]})
+
+            await sendto_user(client.username,reply)

protodoc/dcpv4-rough-draft.md

@@ -125,6 +125,15 @@ Format:
 
 See the Registration flow for more information.
 
+### ```listgroups```
+Client Format:
+```{... 'type': 'listgroups', 'dst':('user'|'group')}```
+Server Format:
+```{... 'type': 'listgroups', 'dst':('user'|'group'),'data':{'groups':[<strings>]}}```
+
+List the groups on the server when the destination is specified as `group` or not at all,
+or the groups the user is a part of when `user` is specified.
+
 ### ```login```
 Format:
 Client: ```{... 'type': 'login', 'data': {'auth': 'password', 'password': <string>}}```