forked from Drift/pvv-nixos-config
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
|
from typing import Awaitable, Callable, Optional, Tuple
|
||
|
|
|
||
|
|
from smtplib import SMTP_SSL as SMTP
|
||
|
|
|
||
|
|
import synapse
|
||
|
|
from synapse import module_api
|
||
|
|
|
||
|
|
|
||
|
|
class SMTPAuthProvider:
|
||
|
|
def __init__(self, config: dict, api: module_api):
|
||
|
|
self.api = api
|
||
|
|
|
||
|
|
self.config = config
|
||
|
|
|
||
|
|
api.register_password_auth_provider_callbacks(
|
||
|
|
auth_checkers={
|
||
|
|
("m.login.password", ("password",)): self.check_pass,
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
async def check_pass(
|
||
|
|
self,
|
||
|
|
username: str,
|
||
|
|
login_type: str,
|
||
|
|
login_dict: "synapse.module_api.JsonDict",
|
||
|
|
):
|
||
|
|
if login_type != "m.login.password":
|
||
|
|
return None
|
||
|
|
|
||
|
|
result = False
|
||
|
|
with SMTP(self.config["smtp_host"]) as smtp:
|
||
|
|
password = login_dict.get("password")
|
||
|
|
try:
|
||
|
|
smtp.login(username, password)
|
||
|
|
result = True
|
||
|
|
except:
|
||
|
|
return None
|
||
|
|
|
||
|
|
if result == True:
|
||
|
|
userid = self.api.get_qualified_user_id(username)
|
||
|
|
if not self.api.check_user_exists(userid):
|
||
|
|
self.api.register_user(username)
|
||
|
|
return (userid, None)
|
||
|
|
else:
|
||
|
|
return None
|