mirror of
https://github.com/ComradCollective/Comrad
synced 2024-11-01 21:40:32 +00:00
86 lines
2.6 KiB
Python
86 lines
2.6 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:
|
|
self.log('Success! your message was: '+str(data))
|
|
res = self.op.receive(data)
|
|
self.log('Your return result should be:',res)
|
|
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') |