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

import sys

    25

#from twisted.python import log

    26

from autobahn.twisted import wamp

    27

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    28

from twisted.internet.defer import inlineCallbacks

    29

from autobahn.wamp import types

    30

from autobahn.wamp.serializer import MsgPackSerializer

    31

from twisted.internet.protocol import ReconnectingClientFactory

    32

import json

    33

    34

_WampSession = None

    35

_PySrv = None

    36

    37

ExposedCalls = ["StartPLC",

    38

                "StopPLC",

    39

                "ForceReload",

    40

                "GetPLCstatus",

    41

                "NewPLC",

    42

                "MatchMD5",

    43

                "SetTraceVariablesList",

    44

                "GetTraceVariables",

    45

                "RemoteExec",

    46

                "GetLogMessage",

    47

                "ResetLogCount",

    48

                ]

    49

    50

SubscribedEvents = []

    51

    52

DoOnJoin = []

    53

    54

def GetCallee(name):

    55

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

    56

    global _PySrv

    57

    names = name.split('.')

    58

    obj = _PySrv.plcobj

    59

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

    60

    return obj

    61

    62

class WampSession(wamp.ApplicationSession):

    63

    64

    @inlineCallbacks

    65

    def onJoin(self, details):

    66

        global _WampSession

    67

        _WampSession = self

    68

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

    69

        print 'WAMP session joined by :', ID

    70

        for name in ExposedCalls:

    71

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

    72

    73

        for name in SubscribedEvents:

    74

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

    75

    76

        for func in DoOnJoin:

    77

            yield func(self)

    78

    79

    def onLeave(self, details):

    80

        global _WampSession

    81

        _WampSession = None

    82

        print 'WAMP session left'

    83

    84

class ReconnectingWampWebSocketClientFactory(WampWebSocketClientFactory, ReconnectingClientFactory):

    85

    def clientConnectionFailed(self, connector, reason):

    86

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

    87

        self.retry(connector)

    88

    def clientConnectionLost(self, connector, reason):

    89

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

    90

        self.retry(connector)

    91

    92

def LoadWampClientConf(wampconf):

    93

    94

    WSClientConf = json.load(open(wampconf))

    95

    return WSClientConf

    96

    97

def RegisterWampClient(wampconf):

    98

    99

    WSClientConf = LoadWampClientConf(wampconf)

   100

   101

    ## start logging to console

   102

    # log.startLogging(sys.stdout)

   103

   104

    # create a WAMP application session factory

   105

    component_config = types.ComponentConfig(

   106

        realm = WSClientConf["realm"],

   107

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

   108

    session_factory = wamp.ApplicationSessionFactory(

   109

        config = component_config)

   110

    session_factory.session = WampSession

   111

   112

    # create a WAMP-over-WebSocket transport client factory

   113

    transport_factory = ReconnectingWampWebSocketClientFactory(

   114

        session_factory,

   115

        url = WSClientConf["url"],

   116

        serializers = [MsgPackSerializer()],

   117

        debug = False,

   118

        debug_wamp = False)

   119

   120

    # start the client from a Twisted endpoint

   121

    conn = connectWS(transport_factory)

   122

    print "WAMP client connecting to :",WSClientConf["url"]

   123

    return conn

   124

   125

def GetSession():

   126

    global _WampSession

   127

    return _WampSession

   128

   129

def SetServer(pysrv):

   130

    global _PySrv

   131

    _PySrv = pysrv

   132