sergey/matrix-webhook
1
0
Fork 0
mirror of https://github.com/bsedin/matrix-webhook.git synced 2026-04-09 03:15:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add support for using predefined access token

Browse Source
This commit is contained in:
Alexander Olofsson
2023-03-08 14:41:52 +01:00
parent 1f072d31c8
commit a14f8fe625
4 changed files with 32 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
matrix_webhook/app.py
View File

@@ -17,8 +17,12 @@ async def main(event):
matrix client login & start web server
"""
LOGGER.info(f"Log in {conf.MATRIX_ID=} on {conf.MATRIX_URL=}")
await utils.CLIENT.login(conf.MATRIX_PW)
if conf.MATRIX_PW:
LOGGER.info(f"Log in {conf.MATRIX_ID=} on {conf.MATRIX_URL=}")
await utils.CLIENT.login(conf.MATRIX_PW)
else:
LOGGER.info(f"Restoring log in {conf.MATRIX_ID=} on {conf.MATRIX_URL=}")
utils.CLIENT.access_token = conf.MATRIX_TOKEN
server = web.Server(handler.matrix_webhook)
runner = web.ServerRunner(server)

19
matrix_webhook/conf.py
View File

@@ -36,11 +36,21 @@ parser.add_argument(
parser.add_argument(
"-p",
"--matrix-pw",
help="matrix password. Required. Environment variable: `MATRIX_PW`",
help="matrix password. Either this or token required. Environment variable: `MATRIX_PW`",
**(
{"default": os.environ["MATRIX_PW"]}
if "MATRIX_PW" in os.environ
else {"required": True}
else {}
),
)
parser.add_argument(
"-t",
"--matrix-token",
help="matrix access token. Either this or password required. Environment variable: `MATRIX_TOKEN`",
**(
{"default": os.environ["MATRIX_TOKEN"]}
if "MATRIX_TOKEN" in os.environ
else {}
),
)
parser.add_argument(
@@ -59,9 +69,14 @@ parser.add_argument(
args = parser.parse_args()
if not (args.matrix_pw or args.matrix_token):
print("Error: Either MATRIX_PW or MATRIX_TOKEN needs to be provided")
exit(1)
SERVER_ADDRESS = (args.host, args.port)
MATRIX_URL = args.matrix_url
MATRIX_ID = args.matrix_id
MATRIX_PW = args.matrix_pw
MATRIX_TOKEN = args.matrix_token
API_KEY = args.api_key
VERBOSE = args.verbose

6
matrix_webhook/utils.py
View File

@@ -44,7 +44,8 @@ async def join_room(room_id):
if isinstance(resp, JoinError):
if resp.status_code == "M_UNKNOWN_TOKEN":
LOGGER.warning("Reconnecting")
await CLIENT.login(conf.MATRIX_PW)
if conf.MATRIX_PW:
await CLIENT.login(conf.MATRIX_PW)
else:
return create_json_response(error_map(resp), resp.message)
else:
@@ -67,7 +68,8 @@ async def send_room_message(room_id, content):
if isinstance(resp, RoomSendError):
if resp.status_code == "M_UNKNOWN_TOKEN":
LOGGER.warning("Reconnecting")
await CLIENT.login(conf.MATRIX_PW)
if conf.MATRIX_PW:
await CLIENT.login(conf.MATRIX_PW)
else:
return create_json_response(error_map(resp), resp.message)
else: