1

#!/usr/bin/env python

     2

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

     3

     4

# This file is part of Beremiz

     5

# Copyright (C) 2019: Edouard TISSERANT

     6

# See COPYING file for copyrights details.

     7

     8

from __future__ import absolute_import

     9

    10

from twisted.web.server import Site

    11

from twisted.web.resource import Resource

    12

from twisted.internet import reactor

    13

from twisted.web.static import File

    14

    15

from autobahn.twisted.websocket import WebSocketServerFactory, WebSocketServerProtocol

    16

from autobahn.twisted.resource import  WebSocketResource

    17

    18

# TODO multiclient :

    19

# session list lock

    20

# svghmi_sessions = []

    21

    22

svghmi_session = None

    23

    48

class HMISession(object):

    49

    def __init__(self, protocol_instance):

    50

        global svghmi_session

    51

    52

        # TODO: kill existing session for robustness

    53

        assert(svghmi_session is None)

    54

    55

        svghmi_session = self

    56

        self.protocol_instance = protocol_instance

    57

    58

        # TODO multiclient :

    59

        # svghmi_sessions.append(self)

    60

        # get a unique bit index amont other svghmi_sessions,

    61

        # so that we can match flags passed by C->python callback

    62

    63

    def __del__(self):

    64

        global svghmi_session

    65

        assert(svghmi_session)

    66

        svghmi_session = None

    67

    68

        # TODO multiclient :

    69

        # svghmi_sessions.remove(self)

    70

    71

    def onMessage(self, msg):

    72

        # TODO :  pass it to the C side recieve_message()

    73

        #    update HMITree

    74

        #        - values

    75

        #        - refresh rates / subsriptions

    76

    77

        # TODO multiclient : pass client index as well

    78

        pass

    79

    82

    83

class HMIProtocol(WebSocketServerProtocol):

    84

    85

    def __init__(self, *args, **kwargs):

    86

        self._hmi_session = None

    87

        WebSocketServerProtocol.__init__(self, *args, **kwargs)

    88

    89

    def onOpen(self):

    90

        self._hmi_session = HMISession(self)

    91

        print "open"

    92

    93

    def onClose(self, wasClean, code, reason):

    94

        del self._hmi_session

    95

        self._hmi_session = None

    96

        print "close"

    97

    98

    def onMessage(self, msg, isBinary):

    99

        self._hmi_session.onMessage(msg)

   100

        print msg

   101

        #self.sendMessage(msg, binary)

   102

   103

svghmi_root = None

   104

svghmi_listener = None

   105

   106

# Called by PLCObject at start

   107

def _runtime_svghmi0_start():

   108

    global svghmi_listener, svghmi_root

   109

   110

    svghmi_root = Resource()

   111

   112

    wsfactory = WebSocketServerFactory()

   113

    wsfactory.protocol = HMIProtocol

   114

   115

    # svghmi_root.putChild("",File(".svg"))

   116

    svghmi_root.putChild("ws",WebSocketResource(wsfactory))

   117

   118

    sitefactory = Site(svghmi_root)

   119

   120

    svghmi_listener = reactor.listenTCP(8008, sitefactory)

   121

   122

    # TODO

   123

    # start a thread that call the C part of SVGHMI

   124

   125

   126

# Called by PLCObject at stop

   127

def _runtime_svghmi0_stop():

   128

    global svghmi_listener

   129

    svghmi_listener.stopListening()