from flask import Response, request
from io import BytesIO
import xml.etree.ElementTree

from fHDHR.tools import sub_el


class RMG_Devices_Probe():
    endpoints = ["/rmg/devices/probe"]
    endpoint_name = "rmg_devices_probe"
    endpoint_methods = ["GET", "POST"]
    endpoint_default_parameters = {
                                    "uri": "<base_url>"
                                    }

    def __init__(self, fhdhr):
        self.fhdhr = fhdhr

    def __call__(self, *args):
        return self.get(*args)

    def get(self, *args):
        """Probes a specific URI for a network device, and returns a device, if it exists at the given URI."""

        base_url = request.url_root[:-1]

        uri = request.args.get('uri', default=None, type=str)

        out = xml.etree.ElementTree.Element('MediaContainer')
        out.set('size', str(len(list(self.fhdhr.origins.origins_dict.keys()))))

        for origin in list(self.fhdhr.origins.origins_dict.keys()):

            if uri == "%s/rmg/%s%s" % (base_url, self.fhdhr.config.dict["main"]["uuid"], origin):

                if self.fhdhr.origins.origins_dict[origin].setup_success:
                    alive_status = "alive"
                else:
                    alive_status = "dead"

                sub_el(out, 'Device',
                       key="%s%s" % (self.fhdhr.config.dict["main"]["uuid"], origin),
                       make=self.fhdhr.config.dict["rmg"]["reporting_manufacturer"],
                       model=self.fhdhr.config.dict["rmg"]["reporting_model"],
                       modelNumber=self.fhdhr.config.internal["versions"]["fHDHR"],
                       protocol="livetv",
                       status=alive_status,
                       title="%s %s" % (self.fhdhr.config.dict["fhdhr"]["friendlyname"], origin),
                       tuners=str(self.fhdhr.origins.origins_dict[origin].tuners),
                       uri="%s/rmg/%s%s" % (base_url, self.fhdhr.config.dict["main"]["uuid"], origin),
                       uuid="device://tv.plex.grabbers.fHDHR/%s%s" % (self.fhdhr.config.dict["main"]["uuid"], origin),
                       thumb="favicon.ico",
                       interface='network'
                       )

        fakefile = BytesIO()
        fakefile.write(b'<?xml version="1.0" encoding="UTF-8"?>\n')
        fakefile.write(xml.etree.ElementTree.tostring(out, encoding='UTF-8'))
        device_xml = fakefile.getvalue()

        return Response(status=200,
                        response=device_xml,
                        mimetype='application/xml')