waifu/kemoverse
waifu/kemoverse
3
2
Fork 3
Code Issues 12 Pull requests 3 Projects 1 Releases Packages Wiki Activity

named some damn properties wrong

Browse source
This commit is contained in:
Moon 2025-06-12 13:25:04 +09:00
parent 34df5fc6ee
commit 99442b4e01
5 changed files with 22 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
.gitignore vendored
View file

@ -181,5 +181,5 @@ cython_debug/
.cursorindexingignore .cursorindexingignore
# Custom stuff # Custom stuff
gacha_game.db *.db
config.ini *.ini

6
bot/bot_app.py
View file

@ -3,13 +3,13 @@ import traceback
from parsing import parse_notification from parsing import parse_notification
from db_utils import get_or_create_user, add_pull, get_config, set_config from db_utils import get_or_create_user, add_pull, get_config, set_config
from fediverse_factory import get_fediverse_service from fediverse_factory import get_fediverse_service
import config
# Initialize the Fediverse service # Initialize the Fediverse service
fediverse_service = get_fediverse_service() fediverse_service = get_fediverse_service()
# Define your whitelist # Get trusted instances from config
# TODO: move to config whitelisted_instances = config.TRUSTED_INSTANCES
whitelisted_instances: list[str] = []
def stream_notifications(): def stream_notifications():
print("Starting filtered notification stream...") print("Starting filtered notification stream...")

4
bot/config.py
View file

@ -27,6 +27,10 @@ INSTANCE_TYPE = instance_type
# Web server port # Web server port
WEB_PORT = config['application'].getint('WebPort', 5000) WEB_PORT = config['application'].getint('WebPort', 5000)
# Trusted instances
trusted_instances_str = config['application'].get('TrustedInstances', '')
TRUSTED_INSTANCES = [instance.strip() for instance in trusted_instances_str.split(',') if instance.strip()]
# Extra stuff for control of the bot # Extra stuff for control of the bot
# TODO: move this to db # TODO: move this to db

13
bot/misskey_service.py
View file

@ -99,9 +99,12 @@ class MisskeyService(FediverseService):
def get_notifications(self, since_id: Optional[str] = None) -> List[FediverseNotification]: def get_notifications(self, since_id: Optional[str] = None) -> List[FediverseNotification]:
"""Get notifications from Misskey instance""" """Get notifications from Misskey instance"""
params = {} params = {
'include_types': ['mention', 'reply'],
'limit': 50
}
if since_id: if since_id:
params["sinceId"] = since_id params["since_id"] = since_id
notifications = self.client.i_notifications(**params) notifications = self.client.i_notifications(**params)
return [self._convert_misskey_notification(notif) for notif in notifications] return [self._convert_misskey_notification(notif) for notif in notifications]
@ -121,13 +124,13 @@ class MisskeyService(FediverseService):
} }
if reply_to_id: if reply_to_id:
params["replyId"] = reply_to_id params["reply_id"] = reply_to_id
if file_ids: if file_ids:
params["fileIds"] = file_ids params["file_ids"] = file_ids
if visible_user_ids and visibility == Visibility.SPECIFIED: if visible_user_ids and visibility == Visibility.SPECIFIED:
params["visibleUserIds"] = visible_user_ids params["visible_user_ids"] = visible_user_ids
response = self.client.notes_create(**params) response = self.client.notes_create(**params)
return response.get("createdNote", {}).get("id", "") return response.get("createdNote", {}).get("id", "")

6
example_config.ini
View file

@ -20,4 +20,8 @@ DefaultAdmins = ['admin@example.tld']
DatabaseLocation = ./gacha_game.db DatabaseLocation = ./gacha_game.db
; Web server port (default: 5000) ; Web server port (default: 5000)
WebPort = 5000 WebPort = 5000
; Comma-separated list of trusted fediverse instances (leave empty to allow only local users)
; Example: TrustedInstances = mastodon.social,misskey.io,pleroma.example.com
TrustedInstances =