PC-Admin/redlight
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

variable renames, allow list of tokens in config

Browse Source
This commit is contained in:
PC-Admin 2023-08-18 01:24:49 +08:00
parent 095667ea99
commit 63f3c3df7b
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
redlight_server_module.py
View File

@ -30,9 +30,9 @@ logger.propagate = False
class SourceDataManager: class SourceDataManager:
def __init__(self, module, config): def __init__(self, module, config):
self._module = module self._module = module
self._source_repo_url = config.get("source_repo_url", "") self._source_repo_url = config.get("redlight_source_repo_url", "")
self._git_token = config.get("git_token", "") self._git_token = config.get("redlight_git_token", "")
self._source_list_file_path = config.get("source_list_file_path", "dist/summaries.json") self._source_list_file_path = config.get("redlight_source_list_file_path", "dist/summaries.json")
self._filtered_tags = config.get("filtered_tags", []) self._filtered_tags = config.get("filtered_tags", [])
self._source_dict = {} self._source_dict = {}
self._source_dict_last_update = None self._source_dict_last_update = None
@ -103,7 +103,8 @@ class RedlightServerResource:
self._module = module self._module = module
self._data_manager = SourceDataManager(module, config) self._data_manager = SourceDataManager(module, config)
self._source_dict = self._data_manager.get_data() self._source_dict = self._data_manager.get_data()
self._api_tokens = ["stong-access-token"] self._client_api_tokens = config.get("redlight_client_tokens", [])
self._filtered_tags = config.get("filtered_tags", [])
# Logging for debug purposes # Logging for debug purposes
logger.debug(f"Filtered room_id_hashes: {list(self._source_dict.keys())}") logger.debug(f"Filtered room_id_hashes: {list(self._source_dict.keys())}")
@ -152,7 +153,7 @@ class RedlightServerResource:
api_token = data["api_token"] api_token = data["api_token"]
# Check if the provided API token is valid. # Check if the provided API token is valid.
if api_token not in self._api_tokens: if api_token not in self._client_api_tokens:
logger.warning(f"Invalid API token provided by {request.getClientIP()}.") logger.warning(f"Invalid API token provided by {request.getClientIP()}.")
request.setResponseCode(401) request.setResponseCode(401)
defer.returnValue(json.dumps({"error": "Unauthorized"}).encode("utf-8")) defer.returnValue(json.dumps({"error": "Unauthorized"}).encode("utf-8"))