101 lines
3.8 KiB
Python
101 lines
3.8 KiB
Python
# maubot - A plugin-based Matrix bot system.
|
|
# Copyright (C) 2022 Tulir Asokan
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
import random
|
|
import re
|
|
import string
|
|
|
|
import bcrypt
|
|
|
|
from mautrix.util.config import BaseFileConfig, ConfigUpdateHelper
|
|
|
|
bcrypt_regex = re.compile(r"^\$2[ayb]\$.{56}$")
|
|
|
|
|
|
class Config(BaseFileConfig):
|
|
@staticmethod
|
|
def _new_token() -> str:
|
|
return "".join(random.choices(string.ascii_lowercase + string.digits, k=64))
|
|
|
|
def do_update(self, helper: ConfigUpdateHelper) -> None:
|
|
base = helper.base
|
|
copy = helper.copy
|
|
|
|
if "database" in self and self["database"].startswith("sqlite:///"):
|
|
helper.base["database"] = self["database"].replace("sqlite:///", "sqlite:")
|
|
else:
|
|
copy("database")
|
|
copy("database_opts")
|
|
if isinstance(self["crypto_database"], dict):
|
|
if self["crypto_database.type"] == "postgres":
|
|
base["crypto_database"] = self["crypto_database.postgres_uri"]
|
|
else:
|
|
copy("crypto_database")
|
|
copy("plugin_directories.upload")
|
|
copy("plugin_directories.load")
|
|
copy("plugin_directories.trash")
|
|
if "plugin_directories.db" in self:
|
|
base["plugin_databases.sqlite"] = self["plugin_directories.db"]
|
|
else:
|
|
copy("plugin_databases.sqlite")
|
|
copy("plugin_databases.postgres")
|
|
copy("plugin_databases.postgres_opts")
|
|
copy("server.hostname")
|
|
copy("server.port")
|
|
copy("server.public_url")
|
|
copy("server.listen")
|
|
copy("server.ui_base_path")
|
|
copy("server.plugin_base_path")
|
|
copy("server.override_resource_path")
|
|
shared_secret = self["server.unshared_secret"]
|
|
if shared_secret is None or shared_secret == "generate":
|
|
base["server.unshared_secret"] = self._new_token()
|
|
else:
|
|
base["server.unshared_secret"] = shared_secret
|
|
if "registration_secrets" in self:
|
|
base["homeservers"] = self["registration_secrets"]
|
|
else:
|
|
copy("homeservers")
|
|
copy("admins")
|
|
for username, password in base["admins"].items():
|
|
if password and not bcrypt_regex.match(password):
|
|
if password == "password":
|
|
password = self._new_token()
|
|
base["admins"][username] = bcrypt.hashpw(
|
|
password.encode("utf-8"), bcrypt.gensalt()
|
|
).decode("utf-8")
|
|
copy("api_features.login")
|
|
copy("api_features.plugin")
|
|
copy("api_features.plugin_upload")
|
|
copy("api_features.instance")
|
|
copy("api_features.instance_database")
|
|
copy("api_features.client")
|
|
copy("api_features.client_proxy")
|
|
copy("api_features.client_auth")
|
|
copy("api_features.dev_open")
|
|
copy("api_features.log")
|
|
copy("logging")
|
|
|
|
def is_admin(self, user: str) -> bool:
|
|
return user == "root" or user in self["admins"]
|
|
|
|
def check_password(self, user: str, passwd: str) -> bool:
|
|
if user == "root":
|
|
return False
|
|
passwd_hash = self["admins"].get(user, None)
|
|
if not passwd_hash:
|
|
return False
|
|
return bcrypt.checkpw(passwd.encode("utf-8"), passwd_hash.encode("utf-8"))
|