1

#!/usr/bin/env python

     2

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

     3

#

     4

#Copyright (C) 2015: Edouard TISSERANT

     5

#

     6

#See COPYING file for copyrights details.

     7

#

     8

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

     9

#modify it under the terms of the GNU General Public

    10

#License as published by the Free Software Foundation; either

    11

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

    12

#

    13

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

    14

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

    15

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

    16

#General Public License for more details.

    17

#

    18

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

    19

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

    20

#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    21

    22

import sys, traceback, atexit

    23

#from twisted.python import log

    24

from twisted.internet import reactor, threads

    25

from autobahn.twisted import wamp

    26

from autobahn.twisted.websocket import WampWebSocketClientFactory, connectWS

    27

from autobahn.wamp import types

    28

from autobahn.wamp.exception import TransportLost

    29

from autobahn.wamp.serializer import MsgPackSerializer

    30

from threading import Thread, Event

    31

    32

_WampSession = None

    33

_WampConnection = None

    34

_WampSessionEvent = Event()

    35

    36

class WampSession(wamp.ApplicationSession):

    37

    def onJoin(self, details):

    38

        global _WampSession, _WampSessionEvent

    39

        _WampSession = self

    40

        _WampSessionEvent.set()

    41

        print 'WAMP session joined for :', self.config.extra["ID"]

    42

    43

    def onLeave(self, details):

    44

        global _WampSession, _WampSessionEvent

    45

        _WampSessionEvent.clear()

    46

        _WampSession = None

    47

        print 'WAMP session left'

    48

    49

PLCObjDefaults = { "StartPLC": False,

    50

                   "GetTraceVariables" : ("Broken",None),

    51

                   "GetPLCstatus" : ("Broken",None),

    52

                   "RemoteExec" : (-1, "RemoteExec script failed!")}

    53

    54

def WAMP_connector_factory(uri, confnodesroot):

    55

    """

    56

    WAMP://127.0.0.1:12345/path#realm#ID

    57

    WAMPS://127.0.0.1:12345/path#realm#ID

    58

    """

    59

    servicetype, location = uri.split("://")

    60

    urlpath, realm, ID = location.split('#')

    61

    urlprefix = {"WAMP":"ws",

    62

                 "WAMPS":"wss"}[servicetype]

    63

    url = urlprefix+"://"+urlpath

    64

    65

    def RegisterWampClient():

    66

    67

        ## start logging to console

    68

        # log.startLogging(sys.stdout)

    69

    70

        # create a WAMP application session factory

    71

        component_config = types.ComponentConfig(

    72

            realm = realm,

    73

            extra = {"ID":ID})

    74

        session_factory = wamp.ApplicationSessionFactory(

    75

            config = component_config)

    76

        session_factory.session = WampSession

    77

    78

        # create a WAMP-over-WebSocket transport client factory

    79

        transport_factory = WampWebSocketClientFactory(

    80

            session_factory,

    81

            url = url,

    82

            serializers = [MsgPackSerializer()],

    83

            debug = False,

    84

            debug_wamp = False)

    85

    86

        # start the client from a Twisted endpoint

    87

        conn = connectWS(transport_factory)

    88

        confnodesroot.logger.write(_("WAMP connecting to URL : %s\n")%url)

    89

        return conn

    90

    91

    AddToDoBeforeQuit = confnodesroot.AppFrame.AddToDoBeforeQuit

    92

    def ThreadProc():

    93

        global _WampConnection

    94

        _WampConnection = RegisterWampClient()

    95

        AddToDoBeforeQuit(reactor.stop)

    96

        reactor.run(installSignalHandlers=False)

    97

    98

    def WampSessionProcMapper(funcname):

    99

        wampfuncname = '.'.join((ID,funcname))

   100

        def catcher_func(*args,**kwargs):

   101

            global _WampSession

   102

            if _WampSession is not None :

   103

                try:

   104

                    return threads.blockingCallFromThread(

   105

                        reactor, _WampSession.call, wampfuncname,

   106

                        *args,**kwargs)

   107

                except TransportLost, e:

   108

                    confnodesroot.logger.write_error("Connection lost!\n")

   109

                    confnodesroot._SetConnector(None)

   110

                except Exception,e:

   111

                    errmess = traceback.format_exc()

   112

                    confnodesroot.logger.write_error(errmess+"\n")

   113

                    print errmess

   114

                    #confnodesroot._SetConnector(None)

   115

            return PLCObjDefaults.get(funcname)

   116

        return catcher_func

   117

   118

    class WampPLCObjectProxy(object):

   119

        def __init__(self):

   120

            global _WampSessionEvent, _WampConnection

   121

            if not reactor.running:

   122

                Thread(target=ThreadProc).start()

   123

            else:

   124

                _WampConnection = threads.blockingCallFromThread(

   125

                    reactor, RegisterWampClient)

   126

            if not _WampSessionEvent.wait(5):

   127

                _WampConnection = stopConnecting()

   128

                raise Exception, _("WAMP connection timeout")

   129

   130

        def __del__(self):

   131

            global _WampConnection

   132

            _WampConnection.disconnect()

   133

            #

   134

            # reactor.stop()

   135

   136

        def __getattr__(self, attrName):

   137

            member = self.__dict__.get(attrName, None)

   138

            if member is None:

   139

                member = WampSessionProcMapper(attrName)

   140

                self.__dict__[attrName] = member

   141

            return member

   142

   143

    # Try to get the proxy object

   144

    try :

   145

        return WampPLCObjectProxy()

   146

    except Exception, msg:

   147

        confnodesroot.logger.write_error(_("WAMP connection to '%s' failed.\n")%location)

   148

        confnodesroot.logger.write_error(traceback.format_exc())

   149

        return None

   150

   151

   152

   153