Code uplift to replace old libraries and support asyncio
matrix-client was replaced with mautrix, issues #56 #59 websocket-client was replaced with websockets flask updated to support async early support for global timeline, issue #71
This commit is contained in:
parent
409b2a5a3c
commit
5638805481
4 changed files with 710 additions and 583 deletions
19
README.md
19
README.md
|
@ -7,19 +7,19 @@ This bridge will pass pnut.io channel messages through to Matrix, and Matrix mes
|
|||
|
||||
## Usage
|
||||
|
||||
The public bridge is once again online!
|
||||
|
||||
See [Using-the-public-bridge](https://gitlab.com/thrrgilag/pnut-matrix/-/wikis/Using-the-public-bridge) for details.
|
||||
The public bridge is not yet online pending uplift of this code base. Stay tuned!
|
||||
|
||||
|
||||
## Installation
|
||||
|
||||
**Warning! This code is extremely unstable and not yet ready for use on your matrix server.**
|
||||
|
||||
Currently pnut-matrix has been only tested with and confirmed to work with [synapse]. Please refer to the [synapse installation instructions] for details on how to setup your homeserver.
|
||||
|
||||
To install the latest version of pnut-matrix from source:
|
||||
|
||||
```sh
|
||||
git clone https://gitlab.com/thrrgilag/pnut-matrix.git
|
||||
git clone https://git.dreamfall.space/spacenerdmo/pnut-matrix.git
|
||||
cd pnut-matrix
|
||||
python3 -m venv env
|
||||
source env/bin/activate
|
||||
|
@ -43,12 +43,11 @@ curl --data '{"type": "m.login.application_service", "username": "your_sender_lo
|
|||
|
||||
## Contributing and support
|
||||
|
||||
You can open issues for bugs or feature requests and you can submit merge requests to this project on [GitLab]. You can also submit issues and patches directly to [morgan@mcmillian.dev].
|
||||
Please submit bugs, feature requests, and patches to [morgan@mcmillian.dev].
|
||||
|
||||
Join my public chat room for development discussion.
|
||||
Join my public chat room on pnut.io for development discussion.
|
||||
|
||||
- [pnut-matrix on pnut.io]
|
||||
- [#pnut_999:pnut-matrix.dreamfall.space]
|
||||
- [pnut-matrix]
|
||||
|
||||
|
||||
## License
|
||||
|
@ -58,9 +57,7 @@ GPLv3, see [LICENSE].
|
|||
[synapse]: https://github.com/matrix-org/synapse
|
||||
[synapse installation instructions]: https://matrix-org.github.io/synapse/latest/setup/installation.html
|
||||
[syanpse configuration]: https://matrix-org.github.io/synapse/latest/application_services.html
|
||||
[GitLab]: https://gitlab.com/thrrgilag/pnut-matrix/
|
||||
[morgan@mcmillian.dev]: mailto:morgan@mcmillian.dev
|
||||
[pnut-matrix on pnut.io]: https://patter.chat/999
|
||||
[#pnut_999:pnut-matrix.dreamfall.space]: https://matrix.to/#/#pnut_999:pnut-matrix.dreamfall.space
|
||||
[pnut-matrix]: https://patter.chat/999
|
||||
[LICENSE]: LICENSE
|
||||
[^1]: https://github.com/matrix-org/matrix-appservice-irc/issues/1270#issuecomment-849765090
|
||||
|
|
780
appservice.py
780
appservice.py
File diff suppressed because it is too large
Load diff
457
pnut-matrix.py
457
pnut-matrix.py
|
@ -1,5 +1,3 @@
|
|||
import websocket
|
||||
import threading
|
||||
import time
|
||||
import logging
|
||||
import logging.config
|
||||
|
@ -11,19 +9,22 @@ import magic
|
|||
import argparse
|
||||
import os
|
||||
import re
|
||||
import asyncio
|
||||
|
||||
from mautrix.client import ClientAPI
|
||||
from mautrix.types import TextMessageEventContent, Format, MessageType
|
||||
from mautrix.errors import MatrixConnectionError
|
||||
from mautrix.errors.request import MNotFound, MForbidden
|
||||
|
||||
from websockets.asyncio.client import connect
|
||||
from websockets.exceptions import ConnectionClosed
|
||||
|
||||
from matrix_client.api import MatrixHttpApi
|
||||
from matrix_client.api import MatrixError, MatrixRequestError
|
||||
from models import Avatars, Rooms, Events, DirectRooms, Users
|
||||
from database import db_session, init_db
|
||||
from sqlalchemy import and_
|
||||
from appservice import app
|
||||
|
||||
logger = logging.getLogger()
|
||||
|
||||
_shutdown = threading.Event()
|
||||
_reconnect = threading.Event()
|
||||
|
||||
class MLogFilter(logging.Filter):
|
||||
|
||||
ACCESS_TOKEN_RE = re.compile(r"(\?.*access(_|%5[Ff])token=)[^&]*(\s.*)$")
|
||||
|
@ -31,15 +32,17 @@ class MLogFilter(logging.Filter):
|
|||
|
||||
def filter(self, record):
|
||||
if record.name == "werkzeug" and len(record.args) > 0:
|
||||
redacted_uri = MLogFilter.ACCESS_TOKEN_RE.sub(r"\1<redacted>\3", record.args[0])
|
||||
redacted_uri = MLogFilter.ACCESS_TOKEN_RE.sub(r"\1<redacted>\3",
|
||||
record.args[0])
|
||||
record.args = (redacted_uri, ) + record.args[1:]
|
||||
elif record.name == "urllib3.connectionpool" and len(record.args) > 3:
|
||||
redacted_uri = MLogFilter.ACCESS_TOKEN_RE2.sub(r"\1<redacted>\3", record.args[4])
|
||||
redacted_uri = MLogFilter.ACCESS_TOKEN_RE2.sub(r"\1<redacted>\3",
|
||||
record.args[4])
|
||||
record.args = record.args[:4] + (redacted_uri,) + record.args[5:]
|
||||
|
||||
return True
|
||||
|
||||
def new_message(msg, meta):
|
||||
async def new_pnut_message(msg, meta):
|
||||
logger.debug("channel: " + msg.channel_id)
|
||||
logger.debug("username: " + msg.user.username)
|
||||
if 'name' in msg.user:
|
||||
|
@ -53,10 +56,18 @@ def new_message(msg, meta):
|
|||
if msg.source.id == config['PNUTCLIENT_ID']:
|
||||
return
|
||||
|
||||
matrix_id = matrix_id_from_pnut(msg.user.username)
|
||||
matrix_api = ClientAPI(config['MATRIX_AS_ID'],
|
||||
base_url=config['MATRIX_HOST'],
|
||||
token=config['MATRIX_AS_TOKEN'],
|
||||
as_user_id=matrix_id.lower())
|
||||
|
||||
if meta['channel_type'] == 'io.pnut.core.chat':
|
||||
room = Rooms.query.filter(Rooms.pnut_chan == msg.channel_id).one_or_none()
|
||||
room = Rooms.query.filter(Rooms.pnut_chan ==
|
||||
msg.channel_id).one_or_none()
|
||||
elif meta['channel_type'] == 'io.pnut.core.pm':
|
||||
room = DirectRooms.query.filter(DirectRooms.pnut_chan == msg.channel_id).one_or_none()
|
||||
room = DirectRooms.query.filter(DirectRooms.pnut_chan ==
|
||||
msg.channel_id).one_or_none()
|
||||
if room is None:
|
||||
# Do do an invite from the bridge user?
|
||||
logger.debug('new invite?')
|
||||
|
@ -64,12 +75,13 @@ def new_message(msg, meta):
|
|||
# subscribed_user_ids from meta
|
||||
logger.debug(meta['subscribed_user_ids'])
|
||||
pnut_user = matrix_id_from_pnut(msg.user.username)
|
||||
profile = get_matrix_profile(pnut_user)
|
||||
profile = await matrix_api.get_profile(matrix_id.lower())
|
||||
if not profile:
|
||||
new_matrix_user(msg.user.username)
|
||||
invitees=[]
|
||||
for pm_user in meta['subscribed_user_ids']:
|
||||
user = Users.query.filter(Users.pnut_user_id == pm_user).one_or_none()
|
||||
user = Users.query.filter(Users.pnut_user_id ==
|
||||
pm_user).one_or_none()
|
||||
if int(pm_user) == msg.user.id:
|
||||
continue
|
||||
if user is not None:
|
||||
|
@ -84,49 +96,39 @@ def new_message(msg, meta):
|
|||
logger.debug('-not_mapped-')
|
||||
return
|
||||
|
||||
matrix_id = matrix_id_from_pnut(msg.user.username)
|
||||
matrix_api = MatrixHttpApi(config['MATRIX_HOST'],
|
||||
token=config['MATRIX_AS_TOKEN'],
|
||||
identity=matrix_id)
|
||||
try:
|
||||
profile = await matrix_api.get_profile(matrix_id.lower())
|
||||
logger.debug(profile)
|
||||
|
||||
profile = get_matrix_profile(matrix_id)
|
||||
if not profile:
|
||||
except MNotFound:
|
||||
new_matrix_user(msg.user.username)
|
||||
logger.debug('-new_user-')
|
||||
profile = {'displayname': None}
|
||||
|
||||
if profile['displayname'] != matrix_display_from_pnut(msg.user):
|
||||
set_matrix_display(msg.user)
|
||||
await set_matrix_display(msg.user)
|
||||
logger.debug('-set_display-')
|
||||
|
||||
avatar = Avatars.query.filter(Avatars.pnut_user == msg.user.username).one_or_none()
|
||||
avatar = Avatars.query.filter(Avatars.pnut_user ==
|
||||
msg.user.username).one_or_none()
|
||||
if avatar is None or avatar.avatar != msg.user.content.avatar_image.url:
|
||||
set_matrix_avatar(msg.user)
|
||||
await set_matrix_avatar(msg.user)
|
||||
logger.debug('-set_avatar-')
|
||||
|
||||
# members = matrix_api.get_room_members(room.room_id)
|
||||
# logger.debug(members)
|
||||
# join_room(room.room_id, config['MATRIX_AS_ID'])
|
||||
# TODO: sort out room invite and join logic
|
||||
join_room(room.room_id, matrix_id)
|
||||
await join_room(room.room_id, matrix_id)
|
||||
|
||||
if 'content' in msg:
|
||||
text = msg.content.text + "\n"
|
||||
ts = int(time.time()) * 1000
|
||||
|
||||
lnktext = ""
|
||||
for link in msg.content.entities.links:
|
||||
if 'title' in link:
|
||||
lnktext += link.title + "\n"
|
||||
if 'url' in link:
|
||||
lnktext += link.url + "\n"
|
||||
|
||||
if len(lnktext) > 0:
|
||||
text += "\n" + lnktext
|
||||
|
||||
r = matrix_api.send_message(room.room_id, text, timestamp=ts)
|
||||
eventtext = TextMessageEventContent(msgtype=MessageType.TEXT,
|
||||
format=Format.HTML,
|
||||
body=msg.content.text,
|
||||
formatted_body=msg.content.html)
|
||||
rid = await matrix_api.send_message(room.room_id, eventtext)
|
||||
event = Events(
|
||||
event_id=r['event_id'],
|
||||
event_id=rid,
|
||||
room_id=room.room_id,
|
||||
pnut_msg_id=msg.id,
|
||||
pnut_user_id=msg.user.id,
|
||||
|
@ -137,14 +139,80 @@ def new_message(msg, meta):
|
|||
|
||||
if 'raw' in msg:
|
||||
logger.debug('-handle media uploads-')
|
||||
new_media(room.room_id, msg)
|
||||
await new_media(room.room_id, msg)
|
||||
|
||||
def new_media(room_id, msg):
|
||||
matrix_id = matrix_id_from_pnut(msg.user.username)
|
||||
matrix_api = MatrixHttpApi(config['MATRIX_HOST'],
|
||||
async def new_pnut_post(post, meta):
|
||||
|
||||
if not config['PNUT_GLOBAL']:
|
||||
return
|
||||
|
||||
if (config['PNUT_GLOBAL_HUMAN_ONLY'] and
|
||||
post.user.type in ['feed', 'bot']):
|
||||
logging.debug('-skipping non human post-')
|
||||
return
|
||||
|
||||
if 'content' in post:
|
||||
|
||||
text = ""
|
||||
if 'repost_of' in post:
|
||||
text += f"<{post.user.username}> reposted >> "
|
||||
post = post.repost_of
|
||||
|
||||
matrix_id = matrix_id_from_pnut(post.user.username)
|
||||
matrix_api = ClientAPI(config['MATRIX_AS_ID'],
|
||||
base_url=config['MATRIX_HOST'],
|
||||
token=config['MATRIX_AS_TOKEN'],
|
||||
identity=matrix_id)
|
||||
ts = int(time.time()) * 1000
|
||||
as_user_id=matrix_id.lower())
|
||||
try:
|
||||
profile = await matrix_api.get_profile(matrix_id.lower())
|
||||
|
||||
except MNotFound:
|
||||
new_matrix_user(post.user.username)
|
||||
|
||||
profile = {'displayname': None}
|
||||
|
||||
if profile['displayname'] != matrix_display_from_pnut(post.user):
|
||||
await set_matrix_display(post.user)
|
||||
logger.debug('-set_display-')
|
||||
|
||||
avatar = Avatars.query.filter(Avatars.pnut_user ==
|
||||
post.user.username).one_or_none()
|
||||
if (avatar is None or
|
||||
avatar.avatar != post.user.content.avatar_image.url):
|
||||
await set_matrix_avatar(post.user)
|
||||
logger.debug('-set_avatar-')
|
||||
|
||||
room_id = config['MATRIX_GLOBAL_ROOM']
|
||||
await join_room(room_id, matrix_id)
|
||||
postlink = f"https://posts.pnut.io/{post.id}"
|
||||
plaintext = f"{post.content.text}\n{postlink}"
|
||||
htmltext = (f"{post.content.html}"
|
||||
f" <a href='{postlink}'>[🔗]</a>")
|
||||
eventtext = TextMessageEventContent(msgtype=MessageType.TEXT,
|
||||
format=Format.HTML,
|
||||
body=plaintext,
|
||||
formatted_body=htmltext)
|
||||
rid = await matrix_api.send_message(room_id, eventtext)
|
||||
event = Events(
|
||||
event_id=rid,
|
||||
room_id=room_id,
|
||||
pnut_msg_id=post.id,
|
||||
pnut_user_id=post.user.id,
|
||||
pnut_chan_id=0,
|
||||
deleted=False)
|
||||
db_session.add(event)
|
||||
db_session.commit()
|
||||
|
||||
if 'raw' in post:
|
||||
logger.debug('-handle media uploads-')
|
||||
await new_media(room_id, post)
|
||||
|
||||
async def new_media(room_id, msg):
|
||||
matrix_id = matrix_id_from_pnut(msg.user.username)
|
||||
matrix_api = ClientAPI(config['MATRIX_AS_ID'],
|
||||
base_url=config['MATRIX_HOST'],
|
||||
token=config['MATRIX_AS_TOKEN'],
|
||||
as_user_id=matrix_id.lower())
|
||||
|
||||
if 'io.pnut.core.oembed' in msg.raw:
|
||||
|
||||
|
@ -157,16 +225,20 @@ def new_media(room_id, msg):
|
|||
info['h'] = oembed.height
|
||||
info['w'] = oembed.width
|
||||
elif oembed.type == 'audio':
|
||||
logger.debug("* recieved audio attachment")
|
||||
continue
|
||||
msgtype = 'm.audio'
|
||||
dl_url = oembed.url
|
||||
elif oembed.type == 'video':
|
||||
logger.debug("* recieved video attachment")
|
||||
continue
|
||||
msgtype = 'm.video'
|
||||
dl_url = oembed.url
|
||||
info['h'] = oembed.height
|
||||
info['w'] = oembed.width
|
||||
elif oembed.type == 'html5video':
|
||||
logger.debug("* recieved html5 video attachment")
|
||||
continue
|
||||
msgtype = 'm.video'
|
||||
dl_url = oembed.url
|
||||
info['h'] = oembed.height
|
||||
info['w'] = oembed.width
|
||||
elif oembed.type == 'rich':
|
||||
logger.debug("* recieved video attachment")
|
||||
logger.debug("* recieved rich attachment")
|
||||
continue
|
||||
else:
|
||||
logger.debug("* recieved unknown attachment")
|
||||
|
@ -177,128 +249,178 @@ def new_media(room_id, msg):
|
|||
with magic.Magic(flags=magic.MAGIC_MIME_TYPE) as m:
|
||||
info['mimetype'] = m.id_buffer(dl.content)
|
||||
info['size'] = len(dl.content)
|
||||
ul = matrix_api.media_upload(dl.content, info['mimetype'])
|
||||
ul = await matrix_api.upload_media(dl.content,
|
||||
mime_type=info['mimetype'])
|
||||
|
||||
if 'title' in oembed:
|
||||
title = oembed.title
|
||||
else:
|
||||
title = ""
|
||||
|
||||
r = matrix_api.send_content(room_id, ul['content_uri'], title, msgtype, extra_information=info, timestamp=ts)
|
||||
rid = await matrix_api.send_file(room_id, ul,
|
||||
file_name=title,
|
||||
file_type=msgtype,
|
||||
info=info)
|
||||
if 'channel_id' in msg:
|
||||
channel_id = msg.channel_id
|
||||
else:
|
||||
channel_id = 0
|
||||
event = Events(
|
||||
event_id=r['event_id'],
|
||||
event_id=rid,
|
||||
room_id=room_id,
|
||||
pnut_msg_id=msg.id,
|
||||
pnut_user_id=msg.user.id,
|
||||
pnut_chan_id=msg.channel_id,
|
||||
pnut_chan_id=channel_id,
|
||||
deleted=False)
|
||||
db_session.add(event)
|
||||
db_session.commit()
|
||||
|
||||
def delete_message(msg):
|
||||
async def delete_message(msg):
|
||||
matrix_id = matrix_id_from_pnut(msg.user.username)
|
||||
matrix_api = MatrixHttpApi(config['MATRIX_HOST'],
|
||||
matrix_api = ClientAPI(config['MATRIX_AS_ID'],
|
||||
base_url=config['MATRIX_HOST'],
|
||||
token=config['MATRIX_AS_TOKEN'],
|
||||
identity=matrix_id)
|
||||
as_user_id=matrix_id.lower())
|
||||
|
||||
events = Events.query.filter(and_(Events.pnut_msg_id == msg.id, Events.deleted == False)).all()
|
||||
events = Events.query.filter(and_(Events.pnut_msg_id ==
|
||||
msg.id, Events.deleted == False)).all()
|
||||
for event in events:
|
||||
matrix_api.redact_event(event.room_id, event.event_id)
|
||||
await matrix_api.redact(event.room_id, event.event_id)
|
||||
event.deleted = True
|
||||
db_session.commit()
|
||||
|
||||
def matrix_id_from_pnut(username):
|
||||
return "@" + config['MATRIX_PNUT_PREFIX'] + username + ":" + config['MATRIX_DOMAIN']
|
||||
matrix_id = (f"@{config['MATRIX_PNUT_PREFIX']}{username}"
|
||||
f":{config['MATRIX_DOMAIN']}")
|
||||
return matrix_id
|
||||
|
||||
def matrix_display_from_pnut(user):
|
||||
if user.type == 'bot':
|
||||
icon = ' 🤖'
|
||||
elif user.type == 'feed':
|
||||
icon = ' 📰'
|
||||
else:
|
||||
icon = ''
|
||||
|
||||
if 'name' in user:
|
||||
display = user.name + " (@" + user.username + ")"
|
||||
display = user.name + " (@" + user.username + ")" + icon
|
||||
else:
|
||||
display = "@" + user.username
|
||||
display = "@" + user.username + icon
|
||||
return display
|
||||
# return user.username + " (pnut)"
|
||||
|
||||
def get_matrix_profile(matrix_id):
|
||||
url = matrix_url + '/profile/' + matrix_id
|
||||
r = requests.get(url)
|
||||
|
||||
if r.status_code == 200:
|
||||
return r.json()
|
||||
else:
|
||||
return None
|
||||
|
||||
def set_matrix_display(user):
|
||||
async def set_matrix_display(user):
|
||||
matrix_id = matrix_id_from_pnut(user.username)
|
||||
matrix_api = MatrixHttpApi(config['MATRIX_HOST'],
|
||||
matrix_api = ClientAPI(config['MATRIX_AS_ID'],
|
||||
base_url=config['MATRIX_HOST'],
|
||||
token=config['MATRIX_AS_TOKEN'],
|
||||
identity=matrix_id)
|
||||
matrix_api.set_display_name(matrix_id, matrix_display_from_pnut(user))
|
||||
as_user_id=matrix_id.lower())
|
||||
await matrix_api.set_displayname(matrix_display_from_pnut(user))
|
||||
|
||||
def set_matrix_avatar(user):
|
||||
async def set_matrix_avatar(user):
|
||||
matrix_id = matrix_id_from_pnut(user.username)
|
||||
matrix_api = MatrixHttpApi(config['MATRIX_HOST'],
|
||||
matrix_api = ClientAPI(config['MATRIX_AS_ID'],
|
||||
base_url=config['MATRIX_HOST'],
|
||||
token=config['MATRIX_AS_TOKEN'],
|
||||
identity=matrix_id)
|
||||
as_user_id=matrix_id.lower())
|
||||
|
||||
dl = requests.get(user.content.avatar_image.url, stream=True)
|
||||
dl.raise_for_status()
|
||||
with magic.Magic(flags=magic.MAGIC_MIME_TYPE) as m:
|
||||
mtype = m.id_buffer(dl.content)
|
||||
ul = matrix_api.media_upload(dl.content, mtype)
|
||||
ul = await matrix_api.upload_media(dl.content, mtype)
|
||||
logger.debug(ul)
|
||||
|
||||
try:
|
||||
matrix_api.set_avatar_url(matrix_id, ul['content_uri'])
|
||||
avatar = Avatars.query.filter(Avatars.pnut_user == user.username).one_or_none()
|
||||
await matrix_api.set_avatar_url(ul)
|
||||
avatar = Avatars.query.filter(Avatars.pnut_user ==
|
||||
user.username).one_or_none()
|
||||
if avatar is None:
|
||||
avatar = Avatars(pnut_user=user.username, avatar=user.content.avatar_image.url)
|
||||
avatar = Avatars(pnut_user=user.username,
|
||||
avatar=user.content.avatar_image.url)
|
||||
db_session.add(avatar)
|
||||
else:
|
||||
avatar.avatar = user.content.avatar_image.url
|
||||
db_session.commit()
|
||||
|
||||
except MatrixRequestError:
|
||||
except Exception:
|
||||
logger.exception('failed to set user avatar')
|
||||
|
||||
def new_matrix_user(username):
|
||||
matrix_api = MatrixHttpApi(config['MATRIX_HOST'],
|
||||
token=config['MATRIX_AS_TOKEN'])
|
||||
endpoint = "/_matrix/client/v3/register"
|
||||
url = config['MATRIX_HOST'] + endpoint
|
||||
params = {'kind': 'user'}
|
||||
data = {
|
||||
'type': 'm.login.application_service',
|
||||
'username': config['MATRIX_PNUT_PREFIX'] + username
|
||||
}
|
||||
matrix_api.register(content=data)
|
||||
headers = {
|
||||
"Content-Type": "application/json",
|
||||
"Authorization": "Bearer " + config['MATRIX_AS_TOKEN']
|
||||
}
|
||||
logger.debug(data)
|
||||
r = requests.post(url, headers=headers, json=data, params=params)
|
||||
if r.status_code == 200:
|
||||
return
|
||||
|
||||
def join_room(room_id, matrix_id):
|
||||
matrix_api_as = MatrixHttpApi(config['MATRIX_HOST'],
|
||||
else:
|
||||
errmsg = f"- unable to register {username} -"
|
||||
logger.warning(errmsg)
|
||||
logger.debug(r.status_code)
|
||||
logger.debug(r.text)
|
||||
return
|
||||
|
||||
async def join_room(room_id, matrix_id):
|
||||
logging.debug('----- trying to join room -----')
|
||||
|
||||
matrix_api_as = ClientAPI(config['MATRIX_AS_ID'],
|
||||
base_url=config['MATRIX_HOST'],
|
||||
token=config['MATRIX_AS_TOKEN'])
|
||||
matrix_api = MatrixHttpApi(config['MATRIX_HOST'],
|
||||
|
||||
matrix_api = ClientAPI(config['MATRIX_AS_ID'],
|
||||
base_url=config['MATRIX_HOST'],
|
||||
token=config['MATRIX_AS_TOKEN'],
|
||||
identity=matrix_id)
|
||||
as_user_id=matrix_id.lower())
|
||||
|
||||
try:
|
||||
matrix_api.join_room(room_id)
|
||||
await matrix_api.join_room(room_id)
|
||||
# logging.debug('----- should be joined -----')
|
||||
|
||||
except MatrixRequestError as e:
|
||||
if e.code == 403:
|
||||
matrix_api_as.invite_user(room_id, matrix_id)
|
||||
matrix_api.join_room(room_id)
|
||||
else:
|
||||
except MForbidden:
|
||||
# logging.debug('------ got a forbidden ------')
|
||||
await matrix_api_as.invite_user(room_id, matrix_id.lower())
|
||||
await matrix_api.join_room(room_id)
|
||||
|
||||
except MatrixConnectionError:
|
||||
# logger.debug(e)
|
||||
# if 'code' in e and e.code == 403:
|
||||
# await matrix_api_as.invite_user(room_id, matrix_id)
|
||||
# await matrix_api.join_room(room_id)
|
||||
# else:
|
||||
logging.debug('------- moar join errors -------')
|
||||
logger.exception('failed to join room')
|
||||
logger.debug(f"{room_id}")
|
||||
|
||||
logger.debug('-room_join-')
|
||||
|
||||
def new_room(pnut_user, invitees, chan):
|
||||
dr = None
|
||||
matrix_api = MatrixHttpApi(config['MATRIX_HOST'],
|
||||
token=config['MATRIX_AS_TOKEN'],
|
||||
identity=pnut_user)
|
||||
url = matrix_url + '/createRoom'
|
||||
params = {"access_token": config['MATRIX_AS_TOKEN'], "user_id": pnut_user}
|
||||
content = {"visibility": "private", "is_direct": True, "invite": invitees}
|
||||
params = {
|
||||
"access_token": config['MATRIX_AS_TOKEN'],
|
||||
"user_id": pnut_user.lower()
|
||||
}
|
||||
content = {
|
||||
"visibility": "private",
|
||||
"is_direct": True,
|
||||
"invite": invitees
|
||||
}
|
||||
headers = {"Content-Type": "application/json"}
|
||||
r = requests.post(url, headers=headers, params=params,
|
||||
data=json.dumps(content))
|
||||
response = r.json()
|
||||
logger.debug(r.status_code)
|
||||
logger.debug(r.text)
|
||||
logger.debug(response)
|
||||
for bridge_user in invitees:
|
||||
dr = DirectRooms(room_id=response['room_id'],
|
||||
bridge_user=pnut_user, pnut_chan=chan)
|
||||
|
@ -308,83 +430,71 @@ def new_room(pnut_user, invitees, chan):
|
|||
|
||||
return dr
|
||||
|
||||
def on_message(ws, message):
|
||||
# logger.debug("on_message: " + message)
|
||||
async def on_message(message):
|
||||
logger.debug("on_message: " + message)
|
||||
msg = json.loads(message)
|
||||
logger.debug(msg['meta'])
|
||||
|
||||
if 'data' in msg:
|
||||
|
||||
if 'channel_type' in msg['meta']:
|
||||
|
||||
if msg['meta']['channel_type'] not in ['io.pnut.core.chat',
|
||||
'io.pnut.core.pm']:
|
||||
if 'meta' in msg:
|
||||
meta = msg['meta']
|
||||
else:
|
||||
return
|
||||
|
||||
for d_item in msg['data']:
|
||||
pmsg = pnutpy.models.Message.from_response_data(d_item)
|
||||
|
||||
if 'is_deleted' in msg['meta']:
|
||||
if msg['meta']['is_deleted']:
|
||||
logger.debug("message: delete")
|
||||
delete_message(pmsg)
|
||||
if 'data' in msg:
|
||||
data = msg['data']
|
||||
else:
|
||||
new_message(pmsg, msg['meta'])
|
||||
return
|
||||
|
||||
def on_error(ws, error):
|
||||
logger.error("on_error: !!! ERROR !!!")
|
||||
logger.error(error)
|
||||
if 'type' in meta:
|
||||
|
||||
def on_close(ws):
|
||||
logger.debug("on_close: ### CLOSED ###")
|
||||
if meta['type'] == "message":
|
||||
|
||||
def on_open(ws):
|
||||
channel_types = ['io.pnut.core.chat', 'io.pnut.core.pm']
|
||||
if meta['channel_type'] not in channel_types:
|
||||
return
|
||||
|
||||
def run(*args):
|
||||
while not _shutdown.isSet() and not _reconnect.isSet():
|
||||
time.sleep(3)
|
||||
for item in data:
|
||||
pnut_msg = pnutpy.models.Message.from_response_data(item)
|
||||
|
||||
if 'is_deleted' in meta and meta['is_deleted']:
|
||||
logger.debug("-message: delete-")
|
||||
delete_message(pnut_msg)
|
||||
|
||||
else:
|
||||
await new_pnut_message(pnut_msg, meta)
|
||||
|
||||
elif meta['type'] == "post":
|
||||
|
||||
for item in data:
|
||||
pnut_post = pnutpy.models.Post.from_response_data(item)
|
||||
await new_pnut_post(pnut_post, meta)
|
||||
|
||||
async def asmain():
|
||||
if config['MATRIX_ADMIN_ROOM']:
|
||||
logger.debug("- sould join admin room -")
|
||||
await join_room(config['MATRIX_ADMIN_ROOM'], config['MATRIX_AS_ID'])
|
||||
|
||||
ws_url = 'wss://stream.pnut.io/v1/app?access_token='
|
||||
ws_url += config['PNUT_APPTOKEN'] + '&key=' + config['PNUT_APPKEY']
|
||||
ws_url += '&include_raw=1'
|
||||
async for websocket in connect(uri=ws_url):
|
||||
try:
|
||||
ws.send(".")
|
||||
except websocket._exceptions.WebSocketConnectionClosedException:
|
||||
logger.debug('websocket closed exception caught...')
|
||||
_reconnect.set()
|
||||
async for message in websocket:
|
||||
await on_message(message)
|
||||
|
||||
time.sleep(1)
|
||||
logger.debug("*** terminate thread ***")
|
||||
await websocket.close()
|
||||
|
||||
t = threading.Thread(target=run)
|
||||
t.start()
|
||||
|
||||
def wsthreader(threadfunc):
|
||||
|
||||
def wrapper():
|
||||
|
||||
while not _shutdown.isSet():
|
||||
_reconnect.clear()
|
||||
logger.debug('threadfunc start...')
|
||||
running = threadfunc()
|
||||
logger.debug('threadfunc end...')
|
||||
if running:
|
||||
time.sleep(5)
|
||||
else:
|
||||
_shutdown.set()
|
||||
logger.debug('*** thread stopped ***')
|
||||
|
||||
return wrapper
|
||||
except ConnectionClosed:
|
||||
continue
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
|
||||
a_parser = argparse.ArgumentParser()
|
||||
a_parser.add_argument(
|
||||
'-d', action='store_true', dest='debug',
|
||||
help="debug logging"
|
||||
)
|
||||
# TODO: solve the database.py problem and enable this
|
||||
# a_parser.add_argument(
|
||||
# '-c', '--config', default="config.yaml",
|
||||
# help="configuration file"
|
||||
# )
|
||||
a_parser.add_argument('-c', '--config', dest='configyaml',
|
||||
default="config.yaml", help="configuration file")
|
||||
args = a_parser.parse_args()
|
||||
|
||||
configyaml = os.environ.get("CONFIG_FILE")
|
||||
|
@ -392,35 +502,16 @@ if __name__ == '__main__':
|
|||
with open(configyaml, "rb") as config_file:
|
||||
config = yaml.load(config_file, Loader=yaml.SafeLoader)
|
||||
|
||||
# websocket.enableTrace(True)
|
||||
logging.config.dictConfig(config['logging'])
|
||||
redact_filter = MLogFilter()
|
||||
logging.getLogger("werkzeug").addFilter(redact_filter)
|
||||
logging.getLogger("urllib3.connectionpool").addFilter(redact_filter)
|
||||
|
||||
ws_url = 'wss://stream.pnut.io/v1/app?access_token='
|
||||
ws_url += config['PNUT_APPTOKEN'] + '&key=' + config['PNUT_APPKEY']
|
||||
ws_url += '&include_raw=1'
|
||||
matrix_url = config['MATRIX_HOST'] + '/_matrix/client/r0'
|
||||
matrix_url = config['MATRIX_HOST'] + '/_matrix/client/v3'
|
||||
|
||||
# setup the database connection
|
||||
init_db()
|
||||
|
||||
# setup the websocket connection
|
||||
ws = websocket.WebSocketApp(ws_url, on_message=on_message,
|
||||
on_error=on_error, on_close=on_close)
|
||||
ws.on_open = on_open
|
||||
wst = threading.Thread(target=wsthreader(ws.run_forever))
|
||||
wst.start()
|
||||
|
||||
# setup the matrix app service
|
||||
if config['MATRIX_ADMIN_ROOM']:
|
||||
logger.debug("- sould join admin room -")
|
||||
join_room(config['MATRIX_ADMIN_ROOM'], config['MATRIX_AS_ID'])
|
||||
app.config.update(config)
|
||||
app.run(host=config['LISTEN_HOST'], port=config['LISTEN_PORT'])
|
||||
asyncio.run(asmain())
|
||||
|
||||
logger.info('!! shutdown initiated !!')
|
||||
_shutdown.set()
|
||||
ws.close()
|
||||
time.sleep(2)
|
||||
|
|
|
@ -1,8 +1,9 @@
|
|||
pyyaml
|
||||
requests
|
||||
matrix-client==0.3.2
|
||||
Flask
|
||||
pnutpy
|
||||
Flask[async]
|
||||
pnutpy>=0.5.0
|
||||
sqlalchemy
|
||||
websocket-client
|
||||
filemagic
|
||||
mautrix>=0.20.6,<0.21
|
||||
websockets
|
||||
asyncclick
|
||||
|
|
Loading…
Reference in a new issue