#!/bin/bash

#set -x

rm -f ./CLI_OK ./PLC_OK ./PLC_CONNECTED

export BEREMIZ_APPDATA=`pwd`/AppData

mkdir -p $BEREMIZ_APPDATA

KEYSTORE=$BEREMIZ_APPDATA/keystore

# Set BEREMIZ_LOCAL_HOST to localhost if not already set

: ${BEREMIZ_LOCAL_HOST:=localhost}

# Start runtime one first time to generate PLC PSK

$BEREMIZPYTHONPATH $BEREMIZPATH/Beremiz_service.py -s psk.txt -n test_wamp_ID -x 0 &

PLC_PID=$!

res=110 # default to ETIMEDOUT

c=5

while ((c--)); do

if [[ -a psk.txt ]]; then

echo got PSK.

res=0 # OK success

break

else

echo waiting PSK.... $c

sleep 1

fi

done

kill $PLC_PID

if [ "$res" != "0" ] ; then

echo timeout generating PSK.

exit $res

fi

IFS=':' read -r PLC_wamp_ID PLC_wamp_secret < psk.txt

URI="WAMPS://${BEREMIZ_LOCAL_HOST}:8888/ws#Automation#${PLC_wamp_ID}"

# Prepare test project

cp -a $BEREMIZPATH/tests/projects/wamp .

sed -i "s,TEST_URI,${URI},g" wamp/beremiz.xml

# Start CLI one first time to generate IDE PSK

IDE_PSK=$KEYSTORE/own/default.psk

rm -f $IDE_PSK

$BEREMIZPYTHONPATH $BEREMIZPATH/Beremiz_cli.py -k \

--project-home wamp connect &

CLI_PID=$!

res=110 # default to ETIMEDOUT

c=5

while ((c--)); do

if [[ -a $IDE_PSK ]]; then

echo got IDE PSK.

res=0 # OK success

break

else

echo waiting IDE PSK.... $c

sleep 1

fi

done

kill $CLI_PID

if [ "$res" != "0" ] ; then

echo timeout generating IDE PSK.

exit $res

fi

IFS=':' read -r IDE_wamp_ID IDE_wamp_secret < $IDE_PSK

# Prepare crossbar server configuration

mkdir -p certs/server .crossbar

openssl req -nodes -new -x509 -keyout certs/server/server.key \

-subj "/C=FR/L=Paris/O=Beremiz/OU=server/CN=${BEREMIZ_LOCAL_HOST}" \

-addext "subjectAltName=DNS:${BEREMIZ_LOCAL_HOST}" \

-out certs/server/server.crt

cp certs/server/server.key ./.crossbar/server.key

cp certs/server/server.crt ./.crossbar/server.crt

cat > .crossbar/config.json <<JsonEnd

{

"version": 2,

"workers": [

{

"type": "router",

"id": "automation_router",

"realms": [

{

"name": "Automation",

"roles": [

{

"name": "authenticated",

"permissions": [

{

"uri": "",

"match": "prefix",

"allow": {

"call": true,

"register": true,

"publish": true,

"subscribe": true

},

"disclose": {

"caller": false,

"publisher": false

},

"cache": true

}

]

}

]

}

],

"transports": [

{

"type": "web",

"endpoint": {

"type": "tcp",

"port": 8888,

"tls": {

"certificate": "server.crt",

"key": "server.key"

}

},

"paths": {

"ws": {

"type": "websocket",

"auth": {

"wampcra": {

"type": "static",

"users": {

"${IDE_wamp_ID}": {

"secret": "${IDE_wamp_secret}",

"role": "authenticated"

},

"${PLC_wamp_ID}": {

"secret": "${PLC_wamp_secret}",

"role": "authenticated"

}

}

}

}

}

}

}

]

}

]

}

JsonEnd

crossbar start &> crossbar_log.txt &

SERVER_PID=$!

res=110 # default to ETIMEDOUT

c=15

while ((c--)); do

if [[ -a .crossbar/node.pid ]]; then

echo found crossbar pid

res=0 # OK success

break

else

echo wait for crossbar to start.... $c

sleep 1

fi

done

if [ "$res" != "0" ] ; then

kill $SERVER_PID

echo timeout starting crossbar.

exit $res

fi

# give more time to crossbar

sleep 3

# Prepare runtime Wamp config

cat > wampconf.json <<JsonEnd

{

"ID": "${PLC_wamp_ID}",

"active": true,

"protocolOptions": {

"autoPingInterval": 60,

"autoPingTimeout": 20

},

"realm": "Automation",

"authentication": "PSK",

"url": "wss://${BEREMIZ_LOCAL_HOST}:8888/ws"

}

JsonEnd

# Re-use self-signed server cert for client

cp .crossbar/server.crt wampTrustStore.crt

# Start Beremiz runtime again, with wamp enabled

$BEREMIZPYTHONPATH $BEREMIZPATH/Beremiz_service.py -c wampconf.json -s psk.txt -n test_wamp_ID -x 0 &> >(

echo "Start PLC loop"

while read line; do

# Wait for server to print modified value

echo "PLC>> $line"

if [[ "$line" =~ "WAMP session joined" ]]; then

echo "PLC is connected"

touch ./PLC_CONNECTED

fi

if [[ "$line" == "PLCobject : PLC started" ]]; then

echo "PLC was programmed"

touch ./PLC_OK

fi

done

echo "End PLC loop"

) &

PLC_PID=$!

echo wait for runtime to come up

res=110 # default to ETIMEDOUT

c=30

while ((c--)); do

if [[ -a ./PLC_CONNECTED ]]; then

res=0 # OK success

break

else

sleep 1

fi

done

if [ "$res" != "0" ] ; then

kill $SERVER_PID

kill $PLC_PID

echo timeout connecting PLC to crossbar.

exit $res

fi

# Re-use self-signed server cert for client in test project

IDE_CERT=$KEYSTORE/cert

mkdir -p $IDE_CERT

cp .crossbar/server.crt $IDE_CERT/${BEREMIZ_LOCAL_HOST}.crt

# Use CLI to build transfer and start PLC

$BEREMIZPYTHONPATH $BEREMIZPATH/Beremiz_cli.py -k \

--project-home wamp build transfer run &> >(

echo "Start CLI loop"

while read line; do

# Wait for PLC runtime to output expected value on stdout

echo "CLI>> $line"

if [[ "$line" == "PLC installed successfully." ]]; then

echo "CLI did transfer PLC program"

touch ./CLI_OK

fi

done

echo "End CLI loop"

) &

CLI_PID=$!

echo all subprocess started, start polling results

res=110 # default to ETIMEDOUT

c=30

while ((c--)); do

if [[ -a ./CLI_OK && -a ./PLC_OK ]]; then

echo got results.

res=0 # OK success

break

else

echo waiting.... $c

sleep 1

fi

done

# Kill PLC and subprocess

echo will kill PLC:$PLC_PID, SERVER:$SERVER_PID and CLI:$CLI_PID

kill $PLC_PID

kill $CLI_PID

kill $SERVER_PID

exit $res