feat(edge): PostClient with HMAC+nonce signed event POST

This commit is contained in:
유창욱 2026-07-11 21:27:38 +09:00
parent 5da838ef0b
commit 67464971bf
2 changed files with 86 additions and 0 deletions

44
edge/alert.py Normal file
View file

@ -0,0 +1,44 @@
import base64
import hashlib
import hmac
import json
from edge.events import Event
def sign(secret: str, nonce: str, ts: str, body: bytes) -> str:
msg = nonce.encode() + ts.encode() + body
return hmac.new(secret.encode(), msg, hashlib.sha256).hexdigest()
class PostClient:
def __init__(self, url: str, device_id: str, secret: str,
nonce_fn, now_fn, http_post=None, timeout: float = 5.0):
self.url = url
self.device_id = device_id
self.secret = secret
self.nonce_fn = nonce_fn
self.now_fn = now_fn
self.timeout = timeout
if http_post is None:
import requests
http_post = requests.post
self._post = http_post
def send(self, event: Event, snapshots: list[bytes]) -> bool:
payload = event.to_payload(self.device_id)
payload["snapshots"] = [base64.b64encode(s).decode() for s in snapshots]
body = json.dumps(payload, sort_keys=True)
nonce = self.nonce_fn()
ts = str(self.now_fn())
headers = {
"Content-Type": "application/json",
"X-Nonce": nonce,
"X-Timestamp": ts,
"X-Signature": sign(self.secret, nonce, ts, body.encode()),
}
try:
resp = self._post(self.url, data=body, headers=headers, timeout=self.timeout)
except Exception:
return False
return 200 <= resp.status_code < 300

42
tests/test_alert.py Normal file
View file

@ -0,0 +1,42 @@
import json
from edge.events import Event, EventType, Severity
from edge.alert import sign, PostClient
def _event():
return Event(EventType.FALL, Severity.HIGH, "cam-01", 7, None, 1000.0, 3.0,
{"torso_angle": 78.0}, None)
class _FakeResp:
def __init__(self, code): self.status_code = code
def test_sign_is_deterministic():
s1 = sign("secret", "n1", "1000", b"body")
s2 = sign("secret", "n1", "1000", b"body")
assert s1 == s2 and len(s1) == 64
def test_send_posts_signed_payload():
captured = {}
def fake_post(url, data=None, headers=None, timeout=None):
captured["url"] = url
captured["headers"] = headers
captured["body"] = data
return _FakeResp(200)
client = PostClient("https://x/api/events", "pi-01", "secret",
nonce_fn=lambda: "n1", now_fn=lambda: 1000.0,
http_post=fake_post)
ok = client.send(_event(), [b"\xff\xd8jpeg"])
assert ok is True
assert captured["headers"]["X-Nonce"] == "n1"
assert captured["headers"]["X-Timestamp"] == "1000.0"
payload = json.loads(captured["body"])
assert payload["device_id"] == "pi-01"
assert payload["event_type"] == "fall"
assert len(payload["snapshots"]) == 1
expected = sign("secret", "n1", "1000.0", captured["body"].encode())
assert captured["headers"]["X-Signature"] == expected
def test_send_returns_false_on_error_code():
client = PostClient("https://x", "pi-01", "s", lambda: "n", lambda: 1.0,
http_post=lambda *a, **k: _FakeResp(500))
assert client.send(_event(), []) is False