#!/usr/bin/env python

# -*- coding: utf-8 -*-

# This file is part of Beremiz runtime.

#

# Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

#

# See COPYING.Runtime file for copyrights details.

#

# This library is free software; you can redistribute it and/or

# modify it under the terms of the GNU Lesser General Public

# License as published by the Free Software Foundation; either

# version 2.1 of the License, or (at your option) any later version.

# This library 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

# Lesser General Public License for more details.

# You should have received a copy of the GNU Lesser General Public

# License along with this library; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

import socket

import threading

import zeroconf

service_type = '_Beremiz._tcp.local.'

class ServicePublisher(object):

def __init__(self, protocol):

self.serviceproperties = {

'description': 'Beremiz remote PLC',

'protocol': protocol

}

self.name = None

self.ip_32b = None

self.port = None

self.server = None

self.service_name = None

self.retrytimer = None

def RegisterService(self, name, ip, port):

try:

self._RegisterService(name, ip, port)

except Exception as e:

print(f"Failed to register service ({str(e)}), retrying in 2 seconds")

self.retrytimer = threading.Timer(2, self.RegisterService, [name, ip, port])

self.retrytimer.start()

def _RegisterService(self, name, ip, port):

self.service_name = '%s.%s' % (name, service_type)

self.name = name

self.port = port

if ip == "0.0.0.0":

print("MDNS brodcasted on all interfaces")

interfaces = zeroconf.InterfaceChoice.All

ip = self.gethostaddr()

else:

interfaces = [ip]

self.server = zeroconf.Zeroconf(interfaces=interfaces)

print("MDNS brodcasted service address :" + ip)

self.ip_32b = socket.inet_aton(ip)

self.service_info = zeroconf.ServiceInfo(

service_type,

self.service_name,

self.port,

addresses=[self.ip_32b],

properties=self.serviceproperties)

self.server.register_service(self.service_info)

self.retrytimer = None

def UnRegisterService(self):

if self.retrytimer is not None:

self.retrytimer.cancel()

if self.server is not None:

self.server.unregister_service(self.service_info)

self.server.close()

self.server = None

def gethostaddr(self, dst='224.0.1.41'):

s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

try:

s.connect((dst, 7))

(host, _port) = s.getsockname()

s.close()

if host != '0.0.0.0':

return host

except Exception:

pass

return socket.gethostbyname(socket.gethostname())