2
0
mirror of https://github.com/ComradCollective/Comrad synced 2024-11-16 00:12:48 +00:00
Comrad/komrade/backend/switchboard.py
2020-09-06 23:40:53 +01:00

85 lines
2.5 KiB
Python

# internal imports
import os,sys; sys.path.append(os.path.abspath(os.path.join(os.path.abspath(os.path.join(os.path.dirname(__file__),'..')),'..')))
from komrade import *
from komrade.backend import *
# external imports
from flask import Flask, request, jsonify
from flask_classful import FlaskView
OPERATOR = None
TELEPHONE = None
from flask_classful import FlaskView, route
class TheSwitchboard(FlaskView, Logger):
default_methods = ['GET']
excluded_methods = ['phone','op']
@property
def phone(self):
global TELEPHONE
from komrade.backend.the_telephone import TheTelephone
if not TELEPHONE: TELEPHONE=TheTelephone()
return TELEPHONE
@property
def op(self):
global OPERATOR
from komrade.backend.the_operator import TheOperator
if not OPERATOR: OPERATOR=TheOperator()
return OPERATOR
def get(self,msg):
self.log('Incoming call!:',msg)
if not msg:
self.log('empty request!')
return OPERATOR_INTERCEPT_MESSAGE
# unenescape
msg = msg.replace('_','/')
if not isBase64(msg):
self.log('not valid input!')
return OPERATOR_INTERCEPT_MESSAGE
encr_b64_str = msg
# first try to get from string to bytes
self.log('incoming <--',encr_b64_str)
try:
encr_b64_b = encr_b64_str.encode('utf-8')
self.log('encr_b64_b',encr_b64_b)
except UnicodeEncodeError:
self.log('not valid unicode?')
return OPERATOR_INTERCEPT_MESSAGE
# then try to get from b64 bytes to raw bytes
try:
data = b64decode(encr_b64_b)
self.log('data',data)
self.log(f'successfully understood input')
except binascii.Error as e:
self.log('not valid b64?')
return OPERATOR_INTERCEPT_MESSAGE
# # step 3: give to The Operator
try:
# return 'Success! your message was: '+str(data)
res = self.op.receive(data)
return res
except AssertionError as e:
self.log('got exception!!',e)
return OPERATOR_INTERCEPT_MESSAGE
# return response to caller
return OPERATOR_INTERCEPT_MESSAGE
def run_forever(port='8080'):
global OPERATOR,TELEPHONE
TELEPHONE = TheTelephone(allow_builtin=False)
OPERATOR = TheOperator(allow_builtin=False)
app = Flask(__name__)
TheSwitchboard.register(app, route_base='/op/', route_prefix=None)
app.run(debug=True, port=port, host='0.0.0.0')