Joseph Doherty
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
from __future__ import annotations
|
|
import json
|
|
from sqlite3 import Connection
|
|
from chat.eventlog.projector import on
|
|
from chat.eventlog.log import Event
|
|
|
|
|
|
@on("group_node_initialized")
|
|
def _apply_group_node_initialized(conn: Connection, e: Event) -> None:
|
|
p = e.payload
|
|
conn.execute(
|
|
"INSERT OR REPLACE INTO group_node "
|
|
"(chat_id, members_json, summary, dynamic, threads_json) "
|
|
"VALUES (?, ?, ?, ?, ?)",
|
|
(
|
|
p["chat_id"],
|
|
json.dumps(p["members"]),
|
|
p.get("summary", ""),
|
|
p.get("dynamic", ""),
|
|
json.dumps(p.get("threads", [])),
|
|
),
|
|
)
|
|
|
|
|
|
@on("group_node_updated")
|
|
def _apply_group_node_updated(conn: Connection, e: Event) -> None:
|
|
p = e.payload
|
|
conn.execute(
|
|
"UPDATE group_node SET summary = ?, dynamic = ?, updated_at = datetime('now') "
|
|
"WHERE chat_id = ?",
|
|
(p.get("summary", ""), p.get("dynamic", ""), p["chat_id"]),
|
|
)
|
|
|
|
|
|
def get_group_node(conn: Connection, chat_id: str) -> dict | None:
|
|
row = conn.execute(
|
|
"SELECT chat_id, members_json, summary, dynamic, threads_json, updated_at "
|
|
"FROM group_node WHERE chat_id = ?",
|
|
(chat_id,),
|
|
).fetchone()
|
|
if not row:
|
|
return None
|
|
return {
|
|
"chat_id": row[0],
|
|
"members": json.loads(row[1]),
|
|
"summary": row[2],
|
|
"dynamic": row[3],
|
|
"threads": json.loads(row[4]),
|
|
"updated_at": row[5],
|
|
}
|