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 session list lock

    19

svghmi_sessions = []

    20

    21

class HMISession(object):

    22

    def __init__(self, protocol_instance):

    23

        global svghmi_sessions

    24

        svghmi_sessions.append(self)

    25

    26

        # TODO multiclient :

    27

        # get a unique bit index amont other svghmi_sessions,

    28

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

    29

    30

    def __del__(self):

    31

        global svghmi_sessions

    32

        svghmi_sessions.remove(self)

    33

    35

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

    36

        #    update HMITree

    37

        #        - values

    38

        #        - refresh rates / subsriptions

    39

    40

        # TODO multiclient : pass client index as well

    41

        pass

    42

    43

    44

class HMIProtocol(WebSocketServerProtocol):

    45

    46

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

    47

        self._hmi_session = None

    48

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

    49

    50

    def onOpen(self):

    51

        self._hmi_session = HMISession(self)

    52

        print "open"

    53

    54

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

    55

        del self._hmi_session

    56

        self._hmi_session = None

    57

        print "close"

    58

    59

    def onMessage(self, msg, isBinary):

    60

        self._hmi_session.onMessage(msg)

    61

        print msg

    62

        #self.sendMessage(msg, binary)

    63

    64

svghmi_root = None

    65

svghmi_listener = None

    66

    67

# Called by PLCObject at start

    68

def _runtime_svghmi0_start():

    69

    global svghmi_listener, svghmi_root

    70

    71

    svghmi_root = Resource()

    72

    73

    wsfactory = WebSocketServerFactory()

    74

    wsfactory.protocol = HMIProtocol

    75

    76

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

    77

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

    78

    79

    sitefactory = Site(svghmi_root)

    80

    81

    svghmi_listener = reactor.listenTCP(8008, sitefactory)

    82

    83

    # TODO

    84

    # start a thread that call the C part of SVGHMI

    85

    86

    87

# Called by PLCObject at stop

    88

def _runtime_svghmi0_stop():

    89

    global svghmi_listener