231 lines
8.6 KiB
Python
231 lines
8.6 KiB
Python
"""
|
|
|
|
AUTHOR:
|
|
|
|
Khushal P Soonderji
|
|
|
|
DATE:
|
|
|
|
Create: Tuesday, 30th Sept., 2025
|
|
|
|
OBJECTIVE:
|
|
|
|
Simple helper functions to work with the Socket.IO server process.
|
|
|
|
REFERENCES:
|
|
|
|
N/A
|
|
|
|
DOWNLOADS:
|
|
|
|
N/A
|
|
|
|
"""
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** IMPORT ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
# To make sibling directories accessible for imports:
|
|
import sys
|
|
sys.path.append(".")
|
|
sys.path.append("..")
|
|
|
|
# My utils:
|
|
from utils_v2.string import regex
|
|
from utils_v2.queue.kafka.controllers.async_kafka import ConsumerKafka, get_ssl_context
|
|
from utils_v2.cache.async_redis_cache_v3 import AsyncRedisCache
|
|
from utils_v2.serialization.json_serializer import JSONSerializer
|
|
|
|
# To maintain the app's state:
|
|
from wsio_v2.app_state import AppState
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** MACROS / ONE-TIME INIT ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
# --- Nothing Yet
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** VARIABLES ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
# --- Nothing Yet
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** FUNCTIONS ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
def origin_is_allowed(origin: str, app_state: AppState) -> bool:
|
|
|
|
"""
|
|
To check if a given origin is in the allowed list.
|
|
:param origin: The origin of your request.
|
|
:return: True if allowed, else False.
|
|
"""
|
|
|
|
# For debugging:
|
|
app_state.printer("Validating origin.")
|
|
|
|
# Start by assuming failure:
|
|
is_allowed = False
|
|
|
|
# Check through all the allowed origins:
|
|
for allowed in app_state.ALLOWED_ORIGINS:
|
|
try:
|
|
if regex.match(origin, allowed):
|
|
is_allowed = True
|
|
break
|
|
except Exception as exception:
|
|
app_state.printer(exception)
|
|
|
|
# Done here:
|
|
app_state.printer(is_allowed)
|
|
return is_allowed
|
|
|
|
|
|
# ---------------------------------------------------------------------------------------------------------------------
|
|
|
|
|
|
async def init(
|
|
script_id: str,
|
|
debug: bool,
|
|
app_state: AppState,
|
|
):
|
|
|
|
"""
|
|
To initialize all credentials, instances, and connectivity for this whole script.
|
|
:param script_id: The id to use to load cred and data from the internal service.
|
|
:param debug: Whether, or not, you would like to print the debug messages.
|
|
:return: True if initialized successfully, else False.
|
|
"""
|
|
|
|
# Basic stuff:
|
|
if debug: app_state.printer.enable()
|
|
app_state.printer("Initializing.")
|
|
|
|
# ┏┓ • •
|
|
# ┃┃┏┓┓┏┓┓┏┓┏
|
|
# ┗┛┛ ┗┗┫┗┛┗┛
|
|
# ┛
|
|
|
|
response = await app_state.http_client.post(
|
|
url = r"https://api.thecaoffice.com/ca/get/title",
|
|
headers = {"Origin": "https://thecaoffice.com/"},
|
|
data = {
|
|
"domainName": "127.0.0.1:1234",
|
|
"screenWidth": 1920,
|
|
"screenHeight": 1080
|
|
}
|
|
)
|
|
if response.status_code not in [200]:
|
|
print("FATAL: ALLOWED ORIGINS NOT FETCHED!")
|
|
return False
|
|
ALLOWED_ORIGINS = [origin["domain"] for origin in response.json().get("data", {}).get("rs2", [])]
|
|
app_state.printer(ALLOWED_ORIGINS)
|
|
if len(ALLOWED_ORIGINS) < 1:
|
|
print("FATAL: ALLOWED ORIGINS IS EMPTY!")
|
|
return False
|
|
|
|
# ┏┓ ┓ ┓ ┳┓
|
|
# ┃ ┏┓┏┓┏┫ ┏┓┏┓┏┫ ┃┃┏┓╋┏┓
|
|
# ┗┛┛ ┗ ┗┻ ┗┻┛┗┗┻ ┻┛┗┻┗┗┻
|
|
|
|
# Get the script credentials:
|
|
response = await app_state.http_client.get(
|
|
url = r"https://nexcom.ditscentre.in/internal/cred/get",
|
|
headers = {"X-Script-Id": script_id}
|
|
)
|
|
if response.status_code not in [200]:
|
|
print("FATAL: SCRIPT CREDENTIALS LOADING FAILED!")
|
|
return False
|
|
script_cred = response.json().get("data")
|
|
|
|
# Get the script data:
|
|
response = await app_state.http_client.get(
|
|
url = r"https://nexcom.ditscentre.in/internal/data/get",
|
|
headers = {"X-Script-Id": script_id}
|
|
)
|
|
if response.status_code not in [200]:
|
|
print("FATAL: SCRIPT DATA LOADING FAILED!")
|
|
return False
|
|
app_state.SCRIPT_DATA = response.json().get("data")
|
|
|
|
# Done with this step:
|
|
app_state.printer("Cred and Data loaded.")
|
|
|
|
# ┓┏┓ ┏┓ ┏┓┓•
|
|
# ┃┫ ┏┓╋┃┏┏┓ ┃ ┃┓┏┓┏┓╋┏
|
|
# ┛┗┛┗┻┛┛┗┗┻ ┗┛┗┗┗ ┛┗┗┛
|
|
|
|
# Create the consumer that will listen to changes in watchlist:
|
|
consumer_creds = script_cred["kafka"]["consumer"]
|
|
app_state.kafka_consumer = ConsumerKafka(
|
|
topic = consumer_creds["topic"],
|
|
bootstrap_servers = consumer_creds["config"]["bootstrapServers"],
|
|
security_protocol = consumer_creds["config"].get("securityProtocol", "PLAINTEXT"),
|
|
ssl_context = get_ssl_context(
|
|
ca_file = consumer_creds["config"].get("caFile"),
|
|
cert_file = consumer_creds["config"].get("certFile"),
|
|
key_file = consumer_creds["config"].get("keyFile"),
|
|
),
|
|
serializer = JSONSerializer(),
|
|
debug = debug
|
|
)
|
|
if not await app_state.kafka_consumer.connect():
|
|
print("FATAL: KAFKA CONSUMER NOT CREATED!")
|
|
return False
|
|
app_state.printer("Kafka consumer ready.")
|
|
|
|
# ┳┓ ┓• ┏┓ ┓
|
|
# ┣┫┏┓┏┫┓┏ ━━ ┃ ┏┓┏┣┓┏┓
|
|
# ┛┗┗ ┗┻┗┛ ┗┛┗┻┗┛┗┗
|
|
|
|
app_state.redis_cache = AsyncRedisCache(
|
|
connection_string = script_cred["redisCache"]["general"]["sentinelJson"],
|
|
# connection_string = script_cred["redisCache"]["general"]["connectionString"],
|
|
# serializer = JSONSerializer(),
|
|
debug = debug,
|
|
debug_prefix = "General Cache | "
|
|
)
|
|
if not await app_state.redis_cache.connect():
|
|
print("FATAL: REDIS CACHE NOT CREATED!")
|
|
return False
|
|
app_state.printer("Redis cache ready.")
|
|
|
|
# ┳┓
|
|
# ┃┃┏┓┏┓┏┓
|
|
# ┻┛┗┛┛┗┗
|
|
|
|
# If everything went well, we return with success:
|
|
app_state.printer("Initialization done.")
|
|
return True
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** MAIN PROGRAM ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
pass
|