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 autobahn.twisted import wamp

    26

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    27

from twisted.internet.defer import inlineCallbacks

    28

from autobahn.wamp import types

    29

from autobahn.wamp.serializer import MsgPackSerializer

    30

from twisted.internet.protocol import ReconnectingClientFactory

    31

import json

    32

    33

_WampSession = None

    34

_PySrv = None

    35

    36

ExposedCalls = ["StartPLC",

    37

                "StopPLC",

    38

                "ForceReload",

    39

                "GetPLCstatus",

    40

                "NewPLC",

    41

                "MatchMD5",

    42

                "SetTraceVariablesList",

    43

                "GetTraceVariables",

    44

                "RemoteExec",

    45

                "GetLogMessage",

    46

                "ResetLogCount",

    47

                ]

    48

    49

SubscribedEvents = []

    50

    51

DoOnJoin = []

    52

    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:

    60

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

    61

    return obj

    62

    63

    64

class WampSession(wamp.ApplicationSession):

    65

    66

    @inlineCallbacks

    67

    def onJoin(self, details):

    68

        global _WampSession

    69

        _WampSession = self

    70

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

    71

        print 'WAMP session joined by :', ID

    72

        for name in ExposedCalls:

    73

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

    74

    75

        for name in SubscribedEvents:

    76

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

    77

    78

        for func in DoOnJoin:

    79

            yield func(self)

    80

    81

    def onLeave(self, details):

    82

        global _WampSession

    83

        _WampSession = None

    84

        print 'WAMP session left'

    85

    86

    87

class ReconnectingWampWebSocketClientFactory(WampWebSocketClientFactory, ReconnectingClientFactory):

    88

    def clientConnectionFailed(self, connector, reason):

    89

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

    90

        self.retry(connector)

    91

    92

    def clientConnectionLost(self, connector, reason):

    93

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

    94

        self.retry(connector)

    95

    96

    97

def LoadWampClientConf(wampconf):

    98

    99

    WSClientConf = json.load(open(wampconf))

   100

    return WSClientConf

   101

   102

   103

def RegisterWampClient(wampconf):

   104

   105

    WSClientConf = LoadWampClientConf(wampconf)

   106

   107

    # start logging to console

   108

    # log.startLogging(sys.stdout)

   109

   110

    # create a WAMP application session factory

   111

    component_config = types.ComponentConfig(

   112

        realm=WSClientConf["realm"],

   113

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

   114

    session_factory = wamp.ApplicationSessionFactory(

   115

        config=component_config)

   116

    session_factory.session = WampSession

   117

   118

    # create a WAMP-over-WebSocket transport client factory

   119

    transport_factory = ReconnectingWampWebSocketClientFactory(

   120

        session_factory,

   121

        url=WSClientConf["url"],

   122

        serializers=[MsgPackSerializer()],

   123

        debug=False,

   124

        debug_wamp=False)

   125

   126

    # start the client from a Twisted endpoint

   127

    conn = connectWS(transport_factory)

   128

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

   129

    return conn

   130

   131

   132

def GetSession():

   133

    global _WampSession

   134

    return _WampSession

   135

   136

   137

def SetServer(pysrv):

   138

    global _PySrv

   139

    _PySrv = pysrv