Compare commits
2 Commits
c10c3ec280
...
8d63eb270b
Author | SHA1 | Date | |
---|---|---|---|
8d63eb270b | |||
1bdebec09d |
3
.gitignore
vendored
3
.gitignore
vendored
|
@ -1,2 +1 @@
|
||||||
/node_modules
|
/.venv
|
||||||
|
|
||||||
|
|
12
README.md
12
README.md
|
@ -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
1533
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
24
package.json
24
package.json
|
@ -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
8
pyproject.toml
Normal file
|
@ -0,0 +1,8 @@
|
||||||
|
[tool.pylint.'MESSAGES CONTROL']
|
||||||
|
disable = [
|
||||||
|
"line-too-long",
|
||||||
|
"unnecessary-semicolon",
|
||||||
|
"trailing-newlines",
|
||||||
|
"missing-docstring"
|
||||||
|
]
|
||||||
|
|
101
server.py
101
server.py
|
@ -1,6 +1,8 @@
|
||||||
import asyncio
|
import asyncio
|
||||||
|
import string
|
||||||
|
import secrets
|
||||||
import json
|
import json
|
||||||
from websockets import serve
|
import websockets
|
||||||
|
|
||||||
clients = {
|
clients = {
|
||||||
"FD3ZJFHQ7r": {
|
"FD3ZJFHQ7r": {
|
||||||
|
@ -12,36 +14,85 @@ clients = {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
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())
|
|
@ -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",
|
|
||||||
]
|
|
||||||
}
|
|
Loading…
Reference in New Issue
Block a user