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

    24

    25

from __future__ import absolute_import

    26

from __future__ import print_function

    27

import json

    28

    31

from autobahn.twisted import wamp

    32

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    34

from autobahn.wamp.serializer import MsgPackSerializer

    36

from twisted.internet.defer import inlineCallbacks

    37

from twisted.internet.protocol import ReconnectingClientFactory

    38

    39

    40

_WampSession = None

    41

_PySrv = None

    42

    43

ExposedCalls = [

    44

    "StartPLC",

    45

    "StopPLC",

    46

    "ForceReload",

    47

    "GetPLCstatus",

    48

    "NewPLC",

    49

    "MatchMD5",

    50

    "SetTraceVariablesList",

    51

    "GetTraceVariables",

    52

    "RemoteExec",

    53

    "GetLogMessage",

    54

    "ResetLogCount",

    55

]

    56

    57

SubscribedEvents = []

    58

    59

DoOnJoin = []

    60

    61

    62

def GetCallee(name):

    63

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

    64

    names = name.split('.')

    65

    obj = _PySrv.plcobj

    66

    while names:

    67

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

    68

    return obj

    69

    70

    71

class WampSession(wamp.ApplicationSession):

    87

    88

    @inlineCallbacks

    89

    def onJoin(self, details):

    90

        global _WampSession

    91

        _WampSession = self

    92

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

    93

        print('WAMP session joined by :', ID)

    94

        for name in ExposedCalls:

    95

            regoption = types.RegisterOptions(u'exact',u'last',None, None)

    96

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

    97

    98

        for name in SubscribedEvents:

    99

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

   100

   101

        for func in DoOnJoin:

   102

            yield func(self)

   103

   104

    def onLeave(self, details):

   105

        global _WampSession

   106

        _WampSession = None

   107

        print(_('WAMP session left'))

   108

   109

   110

class ReconnectingWampWebSocketClientFactory(WampWebSocketClientFactory, ReconnectingClientFactory):

   111

    def clientConnectionFailed(self, connector, reason):

   112

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

   113

        ReconnectingClientFactory.clientConnectionFailed(self, connector, reason)

   114

   115

    def clientConnectionLost(self, connector, reason):

   116

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

   117

        ReconnectingClientFactory.clientConnectionFailed(self, connector, reason)

   118

   119

   120

def LoadWampClientConf(wampconf):

   129

   139

   140

   142

   143

    WSClientConf = LoadWampClientConf(wampconf)

   144

   150

   151

    # create a WAMP application session factory

   152

    component_config = types.ComponentConfig(

   153

        realm=WSClientConf["realm"],

   155

    session_factory = wamp.ApplicationSessionFactory(

   156

        config=component_config)

   157

    session_factory.session = WampSession

   158

   159

    # create a WAMP-over-WebSocket transport client factory

   160

    transport_factory = ReconnectingWampWebSocketClientFactory(

   161

        session_factory,

   162

        url=WSClientConf["url"],

   164

   165

    # start the client from a Twisted endpoint

   166

    conn = connectWS(transport_factory)

   167

    print(_("WAMP client connecting to :"), _WSClientConf["url"])

   168

    return conn

   169

   170

   171

def GetSession():

   172

    return _WampSession

   173

   174

   175

def SetServer(pysrv):

   176

    global _PySrv

   177

    _PySrv = pysrv