# matrix-appservice-kakaotalk - A Matrix-KakaoTalk puppeting bridge.
# Copyright (C) 2022 Tulir Asokan, Andrew Ferrazzutti
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.
"""Custom wrapper classes around types defined by the KakaoTalk API."""

from typing import Optional, NewType, Union

from attr import dataclass

from mautrix.types import SerializableAttrs, JSON, deserializer

from ..types.channel.channel_info import NormalChannelInfo
from ..types.openlink.open_channel_info import OpenChannelInfo
from ..types.user.channel_user_info import NormalChannelUserInfo, OpenChannelUserInfo


ChannelInfoUnion = NewType("ChannelInfoUnion", Union[NormalChannelInfo, OpenChannelInfo])

@deserializer(ChannelInfoUnion)
def deserialize_channel_info_union(data: JSON) -> ChannelInfoUnion:
    if "openLink" in data:
        return OpenChannelInfo.deserialize(data)
    else:
        return NormalChannelInfo.deserialize(data)

setattr(ChannelInfoUnion, "deserialize", deserialize_channel_info_union)


UserInfoUnion = NewType("UserInfoUnion", Union[NormalChannelUserInfo, OpenChannelUserInfo])

@deserializer(UserInfoUnion)
def deserialize_user_info_union(data: JSON) -> UserInfoUnion:
    if "perm" in data:
        return OpenChannelUserInfo.deserialize(data)
    else:
        return NormalChannelUserInfo.deserialize(data)

setattr(UserInfoUnion, "deserialize", deserialize_user_info_union)


@dataclass
class PortalChannelInfo(SerializableAttrs):
    name: str
    participants: list[UserInfoUnion]
    # TODO Image
    channel_info: Optional[ChannelInfoUnion] = None # Should be set manually by caller