Compare commits

..

2 Commits

Author SHA1 Message Date
8d63eb270b
:3 2024-09-09 04:04:18 +03:00
1bdebec09d
:3 2024-08-28 06:42:21 +03:00
8 changed files with 99 additions and 1619 deletions

3
.gitignore vendored
View File

@ -1,2 +1 @@
/node_modules /.venv

View File

@ -2,12 +2,12 @@
```text ```text
GET /api/ws GET /api/ws
GET /api/{uuid}/keypad/code GET /api/cc/{uuid}/keypad/code
GET /api/{uuid}/keypad/get_user_info/{uid} GET /api/cc/{uuid}/keypad/get_user_info/{uid}
POST /api/{uuid}/keypad/log/door_open/{uid} POST /api/cc/{uuid}/keypad/log/door_open/{uid}
POST /api/{uuid}/keypad/set_code POST /api/cc/{uuid}/keypad/set_code
POST /api/{uuid}/core/reboot POST /api/cc/{uuid}/core/reboot
POST /api/{uuid}/core/update POST /api/cc/{uuid}/core/update
``` ```
## websocket ## websocket

1533
package-lock.json generated

File diff suppressed because it is too large Load Diff

View File

@ -1,24 +0,0 @@
{
"name": "rtmc-be",
"version": "1.0.0",
"main": "index.js",
"scripts": {
"test": "echo \"Error: no test specified\" && exit 1"
},
"author": "",
"license": "ISC",
"description": "",
"devDependencies": {
"@types/express": "^4.17.21",
"@types/knex": "^0.15.2",
"@types/node": "^22.5.0",
"@types/ws": "^8.5.12",
"typescript": "^5.5.4"
},
"dependencies": {
"express": "^4.19.2",
"knex": "^3.1.0",
"pug": "^3.0.3",
"ws": "^8.18.0"
}
}

8
pyproject.toml Normal file
View File

@ -0,0 +1,8 @@
[tool.pylint.'MESSAGES CONTROL']
disable = [
"line-too-long",
"unnecessary-semicolon",
"trailing-newlines",
"missing-docstring"
]

113
server.py
View File

@ -1,47 +1,98 @@
import asyncio import asyncio
import string
import secrets
import json import json
from websockets import serve import websockets
clients = { clients = {
"FD3ZJFHQ7r" : { "FD3ZJFHQ7r": {
"doorOpen" : False, "doorOpen": False,
"lockdown" : False, "lockdown": False,
"lastOpened" : 1, #unix timestamp "lastOpened": 1, #unix timestamp
"secLevel" : 1, "secLevel": 1,
"openedBy" : "user" "openedBy": "user"
} }
} }
async def echo(websocket): def rand_id():
print("Connection established") return ''.join(secrets.choice(string.ascii_letters + string.digits) for _ in range(16))
authenticated = False
await websocket.send(json.dumps({
"type" : "SAuthReq"
}))
async for msg in websocket: class WsClient:
msg_json = json.loads(msg) def __init__(self, ws):
if authenticated == False: self.ws = ws
if msg_json["type"] == "CAuth": self.module_info = {}
if msg_json["token"] in clients: self.pcid = None
await websocket.send(json.dumps({
"type" : "SAuthStatus", def send(self, message):
"failed" : False, self.ws.send(json.dumps(message))
"reason" : "Authentication Successful"
}))
authenticated = True
else: class WebSocketHandler:
await websocket.send(json.dumps({ def __init__(self):
"type" : "SAuthStatus", self.clients = {}
"failed" : True,
"reason" : "Invalid token" def on_message(self, client, message):
})) message_type = message.get('type')
# Route messages based on type using match-case
match message_type:
case "CAuth":
client.pcid = message["pcid"]
case "CLog":
print(f"[CC] [{message["level"]}] {message["msg"]}")
case "CModule":
match message["module"]:
case "keypad":
self.handle_keypad_msg(client, message["message"]);
def handle_keypad_msg(self, client: WsClient, message):
match message["type"]:
case "CDoorOpened":
print(f"[CC][KEYPAD] Door was opened on {message["time"]} {"manually" if message["manual"] else ""}")
case "CGetUserStat":
client.send({
"type": "SUserStat",
# user info
})
case "CGetDoorStat":
stats = {}
for pc in self.clients:
i = pc.module_info["keypad"]
stats[pc.uid] = i if i else {}
client.send({
"type": "CDoorStat",
"stats": stats
})
async def new_client(self, ws):
uid = rand_id()
self.clients[uid] = WsClient(ws)
self.clients[uid].send({
"type" : "SAuthReq"
})
async for msg_t in ws:
msg = json.loads(msg_t)
self.on_message(self.clients[uid], msg)
def stop(self):
for c in self.clients:
c.ws.close()
async def main(): async def main():
async with serve(echo, "localhost", 8765): wsh = WebSocketHandler()
async with websockets.serve(wsh.new_client, "localhost", 8765):
await asyncio.get_running_loop().create_future() await asyncio.get_running_loop().create_future()
asyncio.run(main()) asyncio.run(main())

View File

View File

@ -1,21 +0,0 @@
{
"$schema": "http://json.schemastore.org/tsconfig",
"compilerOptions": {
"typeRoots": ["./node_modules/@types", "./src/types"],
"strict": true,
"target": "ESNext",
"module": "ESNext",
"moduleResolution": "NodeNext",
"allowSyntheticDefaultImports": true,
"esModuleInterop": true,
"types": [""],
"noEmit": true,
"isolatedModules": true,
"noImplicitAny": true,
"alwaysStrict": true,
"outDir": "build",
},
"include": [
"src/**/*.ts",
]
}