#!/usr/bin/env python
#
# Copyright (C) 2017 pilino1234
#  This Source Code Form is subject to the terms of the Mozilla Public
#  License, v. 2.0. If a copy of the MPL was not distributed with this
#  file, You can obtain one at http://mozilla.org/MPL/2.0/.
#

"""Example Python 2 client for pilight daemon socket API

Runs a SSDP discovery for pilight daemons on the network, and connects to it.
Identifies as a "receiver" type client, and prints received messages until
terminated.

"""

import socket
import struct
import re


def discover(service, retries=1):
    """SSDP discovery function

    :param service: The name of the SSDP service to run discovery for
    :type service: str
    :param retries: The number of discovery retries. Defaults to 1.
    :type retries: int

    """

    group = ("239.255.255.250", 1900)
    message = "\r\n".join([
        'M-SEARCH * HTTP/1.1',
        'HOST: {0}:{1}'.format(*group),
        'MAN: "ssdp:discover"',
        'ST: {st}', 'MX: 3', '', ''])

    ssdp_responses = {}

    for i in range(retries):
        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM,
                             socket.IPPROTO_UDP)
        sock.setsockopt(socket.SOL_SOCKET, socket.SO_RCVTIMEO,
                        struct.pack('LL', 0, 10000))
        sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, 2)
        sock.sendto(message.format(st=service), group)

        while True:
            try:
                ssdp_responses[i] = sock.recv(1024)
                break
            except socket.timeout:
                break
            except IOError:
                print "no pilight ssdp connections found"
                break

    sock.close()
    return ssdp_responses


if __name__ == "__main__":

    RESPONSES = discover("urn:schemas-upnp-org:service:pilight:1")

    if len(RESPONSES) > 0:
        for resp in RESPONSES:
            locationsrc = re.search('Location:([0-9.]+):(.*)',
                                    str(RESPONSES[0]), re.IGNORECASE)

            if locationsrc:
                location = locationsrc.group(1)
                port = locationsrc.group(2)

            s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            socket.setdefaulttimeout(0)
            try:
                s.connect((location, int(port)))
            except ConnectionRefusedError:
                # Connection was refused, try next SSDP response
                print "connection refused"
                continue

            s.send('{"action": "identify", "options": {"core": 1}}')
            text = ""
            while True:
                line = s.recv(1024)
                text += line
                if "\n\n" in line[-2:]:
                    text = text[:-2]
                    break

            if text == '{"status":"success"}':
                text = ""
                while True:
                    line = s.recv(1024)
                    text += line
                    if "\n\n" in line[-2:]:
                        text = text[:-2]
                        for f in iter(text.splitlines()):
                            print f
                        text = ""
            elif text == '{"status":"failure"}':
                # connection failed, try next SSDP response
                print "connection failed"
                continue
            s.close()