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

    29

from autobahn.twisted import wamp

    30

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    31

from autobahn.wamp import types

    32

from autobahn.wamp.serializer import MsgPackSerializer

    33

from twisted.internet.defer import inlineCallbacks

    34

from twisted.internet.protocol import ReconnectingClientFactory

    35

    36

    37

_WampSession = None

    38

_PySrv = None

    39

    40

ExposedCalls = [

    41

    "StartPLC",

    42

    "StopPLC",

    43

    "ForceReload",

    44

    "GetPLCstatus",

    45

    "NewPLC",

    46

    "MatchMD5",

    47

    "SetTraceVariablesList",

    48

    "GetTraceVariables",

    49

    "RemoteExec",

    50

    "GetLogMessage",

    51

    "ResetLogCount",

    52

]

    53

    54

SubscribedEvents = []

    55

    56

DoOnJoin = []

    57

    58

    59

def GetCallee(name):

    60

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

    61

    names = name.split('.')

    62

    obj = _PySrv.plcobj

    63

    while names:

    64

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

    65

    return obj

    66

    67

    68

class WampSession(wamp.ApplicationSession):

    69

    70

    @inlineCallbacks

    71

    def onJoin(self, details):

    72

        global _WampSession

    73

        _WampSession = self

    74

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

    75

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

    76

        for name in ExposedCalls:

    79

    80

        for name in SubscribedEvents:

    81

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

    82

    83

        for func in DoOnJoin:

    84

            yield func(self)

    85

    86

    def onLeave(self, details):

    87

        global _WampSession

    88

        _WampSession = None

    89

        print(_('WAMP session left'))

    90

    91

    92

class ReconnectingWampWebSocketClientFactory(WampWebSocketClientFactory, ReconnectingClientFactory):

    93

    def clientConnectionFailed(self, connector, reason):

    94

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

    95

        ReconnectingClientFactory.clientConnectionFailed(self, connector, reason)

    96

    97

    def clientConnectionLost(self, connector, reason):

    98

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

    99

        ReconnectingClientFactory.clientConnectionFailed(self, connector, reason)

   100

   101

   102

def LoadWampClientConf(wampconf):

   103

   104

    WSClientConf = json.load(open(wampconf))

   105

    return WSClientConf

   106

   107

   108

def RegisterWampClient(wampconf):

   109

   110

    WSClientConf = LoadWampClientConf(wampconf)

   111

   112

    # start logging to console

   113

    # log.startLogging(sys.stdout)

   114

   115

    # create a WAMP application session factory

   116

    component_config = types.ComponentConfig(

   117

        realm=WSClientConf["realm"],

   118

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

   119

    session_factory = wamp.ApplicationSessionFactory(

   120

        config=component_config)

   121

    session_factory.session = WampSession

   122

   123

    # create a WAMP-over-WebSocket transport client factory

   124

    transport_factory = ReconnectingWampWebSocketClientFactory(

   125

        session_factory,

   126

        url=WSClientConf["url"],

   127

        serializers=[MsgPackSerializer()],

   128

        debug=False,

   129

        debug_wamp=False)

   130

   131

    # start the client from a Twisted endpoint

   132

    conn = connectWS(transport_factory)

   133

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

   134

    return conn

   135

   136

   137

def GetSession():

   138

    return _WampSession

   139

   140

   141

def SetServer(pysrv):

   142

    global _PySrv

   143

    _PySrv = pysrv