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
|
||||
GET /api/ws
|
||||
GET /api/{uuid}/keypad/code
|
||||
GET /api/{uuid}/keypad/get_user_info/{uid}
|
||||
POST /api/{uuid}/keypad/log/door_open/{uid}
|
||||
POST /api/{uuid}/keypad/set_code
|
||||
POST /api/{uuid}/core/reboot
|
||||
POST /api/{uuid}/core/update
|
||||
GET /api/cc/{uuid}/keypad/code
|
||||
GET /api/cc/{uuid}/keypad/get_user_info/{uid}
|
||||
POST /api/cc/{uuid}/keypad/log/door_open/{uid}
|
||||
POST /api/cc/{uuid}/keypad/set_code
|
||||
POST /api/cc/{uuid}/core/reboot
|
||||
POST /api/cc/{uuid}/core/update
|
||||
```
|
||||
|
||||
## 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 string
|
||||
import secrets
|
||||
import json
|
||||
from websockets import serve
|
||||
import websockets
|
||||
|
||||
clients = {
|
||||
"FD3ZJFHQ7r": {
|
||||
|
@ -12,36 +14,85 @@ clients = {
|
|||
}
|
||||
}
|
||||
|
||||
async def echo(websocket):
|
||||
print("Connection established")
|
||||
authenticated = False
|
||||
await websocket.send(json.dumps({
|
||||
"type" : "SAuthReq"
|
||||
}))
|
||||
def rand_id():
|
||||
return ''.join(secrets.choice(string.ascii_letters + string.digits) for _ in range(16))
|
||||
|
||||
async for msg in websocket:
|
||||
msg_json = json.loads(msg)
|
||||
if authenticated == False:
|
||||
if msg_json["type"] == "CAuth":
|
||||
if msg_json["token"] in clients:
|
||||
await websocket.send(json.dumps({
|
||||
"type" : "SAuthStatus",
|
||||
"failed" : False,
|
||||
"reason" : "Authentication Successful"
|
||||
}))
|
||||
authenticated = True
|
||||
else:
|
||||
await websocket.send(json.dumps({
|
||||
"type" : "SAuthStatus",
|
||||
"failed" : True,
|
||||
"reason" : "Invalid token"
|
||||
}))
|
||||
class WsClient:
|
||||
def __init__(self, ws):
|
||||
self.ws = ws
|
||||
self.module_info = {}
|
||||
self.pcid = None
|
||||
|
||||
def send(self, message):
|
||||
self.ws.send(json.dumps(message))
|
||||
|
||||
|
||||
|
||||
class WebSocketHandler:
|
||||
def __init__(self):
|
||||
self.clients = {}
|
||||
|
||||
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 with serve(echo, "localhost", 8765):
|
||||
wsh = WebSocketHandler()
|
||||
|
||||
async with websockets.serve(wsh.new_client, "localhost", 8765):
|
||||
await asyncio.get_running_loop().create_future()
|
||||
|
||||
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