#!/usr/bin/env python3

# Libervia: an XMPP client
# Copyright (C) 2009-2023 Jérôme Poisson (goffi@goffi.org)

# 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 <http://www.gnu.org/licenses/>.


from twisted.words.xish import domish

from libervia.backend.plugins.plugin_xep_0339 import NS_JINGLE_RTP_SSMA, XEP_0339
from libervia.backend.tools.xml_tools import parse


class TestXEP0339:
    def test_parse_sdp(self, host):
        """'ssrc' and 'ssrc-group' attributes in SDP are correctly parsed"""
        xep_0339 = XEP_0339(host)

        call_data = {}
        metadata = {}
        media_type = "video"
        application_data = {}
        transport_data = {}

        # SDP: a=ssrc:123 label:stream_label
        attribute = "ssrc"
        parts = ["123", "label:stream_label"]

        xep_0339._parse_sdp_a_trigger(
            attribute,
            parts,
            call_data,
            metadata,
            media_type,
            application_data,
            transport_data,
        )

        assert application_data == {"ssrc": {123: {"label": "stream_label"}}}

        application_data = {}

        # SDP: a=ssrc-group:FID 123 456
        attribute = "ssrc-group"
        parts = ["FID", "123", "456"]

        xep_0339._parse_sdp_a_trigger(
            attribute,
            parts,
            call_data,
            metadata,
            media_type,
            application_data,
            transport_data,
        )

        assert application_data == {"ssrc-group": {"FID": [123, 456]}}

    def test_generate_sdp(self, host):
        """'ssrc' and 'ssrc-group' attributes in SDP are correctly generated"""
        xep_0339 = XEP_0339(host)

        media_data = {
            "ssrc": {123: {"label": "stream_label"}},
            "ssrc-group": {"FID": [123, 456]},
        }
        sdp_lines = []

        xep_0339._generate_sdp_content_trigger(
            {}, True, 1, {}, sdp_lines, {}, "", media_data, "video"
        )

        assert sdp_lines == ["a=ssrc:123 label:stream_label", "a=ssrc-group:FID 123 456"]

    def test_parse_description(self, host):
        """'<source>' and '<ssrc-group>' elements are correctly parsed"""
        xep_0339 = XEP_0339(host)

        media_data = {}
        raw_xml = """
        <description xmlns='urn:xmpp:jingle:apps:rtp:1' media='audio'>
          <source xmlns='urn:xmpp:jingle:apps:rtp:ssma:0' ssrc='1'>
            <parameter name='cname' value='some_name'/>
            <parameter name='msid' value='media'/>
          </source>
          <ssrc-group xmlns='urn:xmpp:jingle:apps:rtp:ssma:0' semantics='FEC'>
            <source ssrc='2'/>
            <source ssrc='3'/>
          </ssrc-group>
        </description>
        """
        desc_elt = parse(raw_xml)

        xep_0339._parse_description_trigger(desc_elt, media_data)

        assert media_data == {
            "msid": "media",
            "ssrc": {
                1: {
                    "cname": "some_name",
                    "msid": "media",
                }
            },
            "ssrc-group": {"FEC": [2, 3]},
        }

    def test_generate_description(self, host):
        """'<source>' and '<ssrc-group>' elements are correctly generated"""
        xep_0339 = XEP_0339(host)

        media_data = {
            "ssrc": {
                1: {
                    "cname": "some_name",
                    "msid": "media",
                }
            },
            "ssrc-group": {"FEC": [2, 3]},
        }
        session = {}
        desc_elt = domish.Element((None, "description"))

        xep_0339._build_description_trigger(desc_elt, media_data, session)

        source_elts = list(desc_elt.elements(NS_JINGLE_RTP_SSMA, "source"))
        assert len(source_elts) == 1
        source_elt = source_elts[0]
        assert source_elt["ssrc"] == "1"
        param_elts = list(source_elt.elements(NS_JINGLE_RTP_SSMA, "parameter"))
        assert len(param_elts) == 2
        assert param_elts[0]["name"] == "cname"
        assert param_elts[0]["value"] == "some_name"
        assert param_elts[1]["name"] == "msid"
        assert param_elts[1]["value"] == "media"

        ssrc_group_elts = list(desc_elt.elements(NS_JINGLE_RTP_SSMA, "ssrc-group"))
        assert len(ssrc_group_elts) == 1
        ssrc_group_elt = ssrc_group_elts[0]
        assert ssrc_group_elt["semantics"] == "FEC"
        ssrc_ids = [
            int(source_elt["ssrc"])
            for source_elt in ssrc_group_elt.elements(NS_JINGLE_RTP_SSMA, "source")
        ]
        assert ssrc_ids == [2, 3]