1

#!/usr/bin/env python

     2

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

     3

     4

# This file is part of Beremiz runtime.

     5

#

     6

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

     7

#

     8

# See COPYING.Runtime file for copyrights details.

     9

#

    10

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

    11

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

    12

# License as published by the Free Software Foundation; either

    13

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

    14

    15

# This library is distributed in the hope that it will be useful,

    16

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    17

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    18

# Lesser General Public License for more details.

    19

    20

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

    21

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

    22

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

    23

    26

import sys

    27

from autobahn.twisted import wamp

    28

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    29

from twisted.internet.defer import inlineCallbacks

    30

from autobahn.wamp import types

    31

from autobahn.wamp.serializer import MsgPackSerializer

    32

from twisted.internet.protocol import ReconnectingClientFactory

    33

import json

    34

    35

_WampSession = None

    36

_PySrv = None

    37

    38

ExposedCalls = ["StartPLC",

    39

                "StopPLC",

    40

                "ForceReload",

    41

                "GetPLCstatus",

    42

                "NewPLC",

    43

                "MatchMD5",

    44

                "SetTraceVariablesList",

    45

                "GetTraceVariables",

    46

                "RemoteExec",

    47

                "GetLogMessage",

    48

                "ResetLogCount",

    49

                ]

    50

    51

SubscribedEvents = []

    52

    53

DoOnJoin = []

    54

    55

    56

def GetCallee(name):

    57

    """ Get Callee or Subscriber corresponding to '.' spearated object path """

    58

    global _PySrv

    59

    names = name.split('.')

    60

    obj = _PySrv.plcobj

    61

    while names:

    62

        obj = getattr(obj, names.pop(0))

    63

    return obj

    64

    65

    66

class WampSession(wamp.ApplicationSession):

    67

    68

    @inlineCallbacks

    69

    def onJoin(self, details):

    70

        global _WampSession

    71

        _WampSession = self

    72

        ID = self.config.extra["ID"]

    74

        for name in ExposedCalls:

    75

            reg = yield self.register(GetCallee(name), '.'.join((ID, name)))

    76

    77

        for name in SubscribedEvents:

    78

            reg = yield self.subscribe(GetCallee(name), name)

    79

    80

        for func in DoOnJoin:

    81

            yield func(self)

    82

    83

    def onLeave(self, details):

    84

        global _WampSession

    85

        _WampSession = None

    87

    88

    89

class ReconnectingWampWebSocketClientFactory(WampWebSocketClientFactory, ReconnectingClientFactory):

    90

    def clientConnectionFailed(self, connector, reason):

    91

        print("WAMP Client connection failed .. retrying ..")

    92

        self.retry(connector)

    93

    94

    def clientConnectionLost(self, connector, reason):

    95

        print("WAMP Client connection lost .. retrying ..")

    96

        self.retry(connector)

    97

    98

    99

def LoadWampClientConf(wampconf):

   100

   101

    WSClientConf = json.load(open(wampconf))

   102

    return WSClientConf

   103

   104

   105

def RegisterWampClient(wampconf):

   106

   107

    WSClientConf = LoadWampClientConf(wampconf)

   108

   109

    # start logging to console

   110

    # log.startLogging(sys.stdout)

   111

   112

    # create a WAMP application session factory

   113

    component_config = types.ComponentConfig(

   114

        realm=WSClientConf["realm"],

   115

        extra={"ID": WSClientConf["ID"]})

   116

    session_factory = wamp.ApplicationSessionFactory(

   117

        config=component_config)

   118

    session_factory.session = WampSession

   119

   120

    # create a WAMP-over-WebSocket transport client factory

   121

    transport_factory = ReconnectingWampWebSocketClientFactory(

   122

        session_factory,

   123

        url=WSClientConf["url"],

   124

        serializers=[MsgPackSerializer()],

   125

        debug=False,

   126

        debug_wamp=False)

   127

   128

    # start the client from a Twisted endpoint

   129

    conn = connectWS(transport_factory)

   131

    return conn

   132

   133

   134

def GetSession():

   135

    global _WampSession

   136

    return _WampSession

   137

   138

   139

def SetServer(pysrv):

   140

    global _PySrv

   141

    _PySrv = pysrv