ForcePilot/backend/package/yuxi/channel/extensions/tlon/channel_mgmt.py

import time
import logging

from yuxi.channel.extensions.tlon.utils import normalize_ship

logger = logging.getLogger(__name__)


async def create_channel(api, host_ship: str, group_flag: str,
                         channel_type: str = "chat",
                         name: str = "", description: str = "",
                         readers: str = "all", writers: str = "all") -> dict:
    payload = {
        "channel": {
            "add": {
                "flag": group_flag,
                "channel": {
                    "added": int(time.time() * 1000),
                    "type": channel_type,
                    "readers": readers,
                    "writers": writers,
                }
            }
        }
    }
    if name:
        payload["channel"]["add"]["channel"]["name"] = name
    if description:
        payload["channel"]["add"]["channel"]["description"] = description

    await api.poke(app="groups", mark="group-action-4", json_data=payload)
    nest = f"{channel_type}/{host_ship}/{name or 'new'}"
    return {"nest": nest}


async def delete_channel_group(api, host_ship: str, group_flag: str, channel_nest: str) -> None:
    payload = {
        "channel": {
            "del": {
                "flag": group_flag,
                "nest": channel_nest,
            }
        }
    }
    await api.poke(app="groups", mark="group-action-4", json_data=payload)


async def rename_channel(api, host_ship: str, group_flag: str,
                         channel_nest: str, new_name: str) -> None:
    payload = {
        "channel": {
            "edit": {
                "flag": group_flag,
                "nest": channel_nest,
                "channel": {"name": new_name},
            }
        }
    }
    await api.poke(app="groups", mark="group-action-4", json_data=payload)


async def set_channel_permissions(api, host_ship: str, group_flag: str,
                                  channel_nest: str, readers: str = "",
                                  writers: str = "") -> None:
    edit_payload = {}
    if readers:
        edit_payload["readers"] = readers
    if writers:
        edit_payload["writers"] = writers
    if not edit_payload:
        return

    payload = {
        "channel": {
            "edit": {
                "flag": group_flag,
                "nest": channel_nest,
                "channel": edit_payload,
            }
        }
    }
    await api.poke(app="groups", mark="group-action-4", json_data=payload)


async def create_group(api, ship: str, name: str, title: str = "",
                       description: str = "", is_public: bool = False) -> dict:
    flag = f"~{ship}/{name}"
    payload = {
        "group": {
            "create": {
                "group": {
                    "name": name,
                    "title": title or name,
                    "description": description,
                    "privacy": "public" if is_public else "private",
                }
            }
        }
    }
    await api.poke(app="groups", mark="group-action-4", json_data=payload)
    return {"flag": flag}


async def invite_to_group(api, group_flag: str, ships: list[str]) -> None:
    payload = {
        "group": {
            "invite": {
                "flag": group_flag,
                "ships": [normalize_ship(s) for s in ships],
            }
        }
    }
    await api.poke(app="groups", mark="group-action-4", json_data=payload)


async def kick_from_group(api, group_flag: str, ships: list[str]) -> None:
    for ship in ships:
        payload = {
            "group": {
                "kick": {
                    "flag": group_flag,
                    "ship": normalize_ship(ship),
                }
            }
        }
        await api.poke(app="groups", mark="group-action-4", json_data=payload)


async def ban_from_group(api, group_flag: str, ships: list[str]) -> None:
    for ship in ships:
        payload = {
            "group": {
                "ban": {
                    "flag": group_flag,
                    "ship": normalize_ship(ship),
                }
            }
        }
        await api.poke(app="groups", mark="group-action-4", json_data=payload)


async def set_group_roles(api, group_flag: str, ship: str, roles: list[str]) -> None:
    payload = {
        "group": {
            "role": {
                "flag": group_flag,
                "ship": normalize_ship(ship),
                "roles": roles,
            }
        }
    }
    await api.poke(app="groups", mark="group-action-4", json_data=payload)


async def set_group_privacy(api, group_flag: str, privacy: str) -> None:
    payload = {
        "group": {
            "edit": {
                "flag": group_flag,
                "group": {"privacy": privacy},
            }
        }
    }
    await api.poke(app="groups", mark="group-action-4", json_data=payload)


async def delete_group(api, group_flag: str) -> None:
    payload = {
        "group": {
            "delete": {"flag": group_flag}
        }
    }
    await api.poke(app="groups", mark="group-action-4", json_data=payload)


async def revoke_group_invite(api, group_flag: str, ships: list[str]) -> None:
    for ship in ships:
        payload = {
            "group": {
                "del-invite": {
                    "flag": group_flag,
                    "ship": normalize_ship(ship),
                }
            }
        }
        await api.poke(app="groups", mark="group-action-4", json_data=payload)