def db_session() -> SessionLocal:
db = SessionLocal()
try:
yield db
finally:
db.close()
db.rollback()
async def handle(self, topic: str, payload: Dict):
db = SessionLocal()
try:
parts = topic.split("/")
if len(parts) < 2 or parts[0] != "moca":
return
# moca/...
if parts[1] == "via" and len(parts) >= 5:
# data from a service (e.g. moca-service-telegram)
# example: moca/via/telegram/4/contacts --> add or update contact(s)
service: str = parts[2]
connector_id: int = int(parts[3])
command: str = parts[4]
connector = crud.get_connector_by_connector_id(
db, service, connector_id)
if not connector:
return
if command == "contacts":
for contact_data in payload:
internal_contact_id = contact_data.get("contact_id")
contact = models.Contact(
contact_id=crud.get_id(connector.connector_id,
internal_contact_id),
internal_id=internal_contact_id,
service_id=service,
name=contact_data.get("name"),
username=contact_data.get("username"),
phone=contact_data.get("phone"),
avatar=None,
connector_id=connector.connector_id,
)
db.merge(contact)
db.commit()
elif command == "chats":
for chat_data in payload:
internal_chat_id = chat_data.get("chat_id")
chat = models.Chat(
chat_id=crud.get_id(connector.connector_id,
internal_chat_id),
connector_id=connector.connector_id,
internal_id=internal_chat_id,
name=chat_data.get("name"),
is_muted=False,
is_archived=False,
)
db.merge(chat)
db.commit()
chat_id = chat.chat_id
last_message = chat_data.get("last_message")
if last_message:
# Get contact of the sender, else ask for it
internal_contact_id = last_message.get(
"contact_id")
# check if contact exists, else request contact
maybe_contact = crud.get_contact_from_connector(
db,
connector.user_id,
connector.connector_id,
internal_contact_id,
)
if not maybe_contact:
contact = await self.get_contact(
connector.connector_type,
connector.connector_id,
internal_contact_id,
)
new_contact = models.Contact(
contact_id=crud.get_id(
connector.connector_id,
internal_contact_id),
internal_id=internal_contact_id,
service_id=connector.connector_type,
connector_id=connector.connector_id,
name=contact.get("name"),
username=contact.get("username"),
phone=contact.get("phone"),
avatar=contact.get("avatar"),
is_self=False,
)
db.merge(new_contact)
db.commit()
contact_id = new_contact.contact_id
else:
contact_id = maybe_contact.contact_id
# TODO: Reimplement last message
# new_last_message = models.Message(
# message_id=last_message.get("message_id"),
# contact_id=contact_id,
# internal_id=chat_id,
# message=json.dumps(last_message.get("message")),
# sent_datetime=datetime.fromisoformat(last_message.get("sent_datetime"))
# )
# db.merge(new_last_message)
# db.commit()
participants = chat_data.get("participants")
if participants:
for participant in participants:
c = crud.get_contact_from_connector(
db,
connector.user_id,
connector.connector_id,
participant,
)
if not c:
contact = await self.get_contact(
connector.connector_type,
connector.connector_id,
participant,
)
c = models.Contact(
contact_id=crud.get_id(
connector.connector_id,
participant),
internal_id=participant,
service_id=connector.connector_type,
connector_id=connector.connector_id,
name=contact.get("name"),
username=contact.get("username"),
phone=contact.get("phone"),
avatar=contact.get("avatar"),
is_self=False,
)
db.merge(c)
db.commit()
db.merge(
models.ContactsChatsRelationship(
contact_id=c.contact_id,
chat_id=chat_id))
try:
db.commit()
except sqlalchemy.exc.IntegrityError:
db.rollback()
elif command == "messages":
for message_data in payload:
# Get contact of the sender, else ask for it
internal_contact_id = message_data.get("contact_id")
# check if contact exists, else request contact
c = crud.get_contact_from_connector(
db,
connector.user_id,
connector.connector_id,
internal_contact_id,
)
if not c:
contact = await self.get_contact(
connector.connector_type,
connector.connector_id,
internal_contact_id,
)
c = models.Contact(
contact_id=crud.get_id(connector.connector_id,
internal_contact_id),
internal_id=internal_contact_id,
service_id=connector.connector_type,
connector_id=connector.connector_id,
name=contact.get("name"),
username=contact.get("username"),
phone=contact.get("phone"),
avatar=contact.get("avatar"),
is_self=False,
)
db.merge(c)
db.commit()
else:
pass
chat = (db.query(models.Chat).filter(
models.Chat.connector_id == connector_id,
models.Chat.internal_id == message_data.get(
"chat_id"),
).first())
if not chat:
chat = models.Chat(
chat_id=crud.get_id(
connector.connector_id,
message_data.get("chat_id")),
connector_id=connector_id,
internal_id=message_data.get("chat_id"),
name="Loading...",
is_muted=False,
is_archived=False,
)
db.merge(chat)
db.commit()
new_last_message = models.Message(
message_id=crud.get_message_id(
connector.connector_id,
message_data.get("message_id"), chat.chat_id),
internal_id=message_data.get("message_id"),
contact_id=crud.get_id(connector.connector_id,
internal_contact_id),
chat_id=chat.chat_id,
message=json.dumps(message_data.get("message")),
sent_datetime=datetime.fromisoformat(
message_data.get("sent_datetime").split("Z")
[0]),
)
db.merge(new_last_message)
db.commit()
finally:
db.close()
