Compare commits
65 commits
9c5eab51af
...
f70b2147cd
Author | SHA1 | Date | |
---|---|---|---|
|
f70b2147cd | ||
|
298d7fda72 | ||
|
8918b5205d | ||
|
67b4d949fd | ||
|
89ae8a7290 | ||
|
91376c0eba | ||
|
b2ca6dd59a | ||
|
dac05a3ed8 | ||
|
fe8e7d246f | ||
|
a47530180d | ||
|
e0cf42f8f6 | ||
fa21ce201d | |||
62cc80033d | |||
3626949020 | |||
bad06c4f23 | |||
2d7107e745 | |||
e41c965538 | |||
70c8fd7a0c | |||
8569f28f3c | |||
956d5927cd | |||
67afa22e2e | |||
cbdbefb5fc | |||
0a2f7fe00d | |||
3de6a9ac3d | |||
9f9c034461 | |||
4c2b3f589f | |||
0c2e3754da | |||
c2fd435622 | |||
3a09b481e5 | |||
b7d82d9117 | |||
66a91b5deb | |||
dce2c9072b | |||
71b9cd9d16 | |||
a0ed6ded41 | |||
da2ca4cfec | |||
3fdf64c6ed | |||
7b9a5cf428 | |||
6ea5529ef5 | |||
eaa6308906 | |||
7d0ddc492c | |||
de7670204a | |||
846130771e | |||
4f853df32c | |||
8fb91f7754 | |||
37ac7dbb0c | |||
26f23a1354 | |||
36c65b3ae6 | |||
f3a32f1176 | |||
6912758a44 | |||
25a72b3002 | |||
d210f44efc | |||
bdd2f20b84 | |||
0690ac5212 | |||
e8774cb8bd | |||
d9027356ab | |||
4a7c9239fc | |||
8a331e0c7b | |||
47272aee4f | |||
ff20e26821 | |||
e53bd87d81 | |||
9be92afce3 | |||
d2a7e523e8 | |||
45dd6f10e3 | |||
24309ce900 | |||
6b6777cb89 |
35 changed files with 1563 additions and 502 deletions
8
.gitignore
vendored
8
.gitignore
vendored
|
@ -147,6 +147,7 @@ venv.bak/
|
||||||
/site
|
/site
|
||||||
|
|
||||||
# mypy
|
# mypy
|
||||||
|
.mypy.ini
|
||||||
.mypy_cache/
|
.mypy_cache/
|
||||||
.dmypy.json
|
.dmypy.json
|
||||||
dmypy.json
|
dmypy.json
|
||||||
|
@ -181,5 +182,8 @@ cython_debug/
|
||||||
.cursorindexingignore
|
.cursorindexingignore
|
||||||
|
|
||||||
# Custom stuff
|
# Custom stuff
|
||||||
gacha_game.db
|
gacha_game*.db
|
||||||
config.ini
|
gacha_game*.db.*
|
||||||
|
config*.ini
|
||||||
|
|
||||||
|
.idea
|
||||||
|
|
1
.tool-versions
Normal file
1
.tool-versions
Normal file
|
@ -0,0 +1 @@
|
||||||
|
nodejs 23.4.0
|
|
@ -1,55 +1,58 @@
|
||||||
import requests
|
import requests
|
||||||
from misskey.exceptions import MisskeyAPIException
|
from fediverse_factory import get_fediverse_service
|
||||||
from client import client_connection
|
|
||||||
from db_utils import get_db_connection
|
from db_utils import get_db_connection
|
||||||
|
from config import RARITY_TO_WEIGHT
|
||||||
|
|
||||||
def add_character(name: str, rarity: int, weight: float, image_url: str) -> tuple[int, str]:
|
def add_character(name: str, rarity: int, weight: float, image_url: str) -> tuple[int, str]:
|
||||||
"""
|
"""
|
||||||
Adds a character to the database, uploading the image from a public URL to the bot's Misskey Drive.
|
Adds a character to the database, uploading the image from a public URL to the Fediverse instance.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
name (str): Character name.
|
name (str): Character name.
|
||||||
rarity (int): Character rarity (e.g., 1-5).
|
rarity (int): Character rarity (e.g., 1-5).
|
||||||
weight (float): Pull weight (e.g., 0.02).
|
weight (float): Pull weight (e.g., 0.02).
|
||||||
image_url (str): Public URL of the image from the post (e.g., from note['files'][i]['url']).
|
image_url (str): Public URL of the image from the post.
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
tuple[int, str]: Character ID and bot's Drive file_id.
|
tuple[int, str]: Character ID and file_id.
|
||||||
|
|
||||||
Raises:
|
Raises:
|
||||||
ValueError: If inputs are invalid.
|
ValueError: If inputs are invalid.
|
||||||
RuntimeError: If image download/upload or database operation fails.
|
RuntimeError: If image download/upload or database operation fails.
|
||||||
"""
|
"""
|
||||||
try:
|
|
||||||
# Validate inputs
|
|
||||||
if not name or not name.strip():
|
|
||||||
raise ValueError("Character name cannot be empty.")
|
|
||||||
if not isinstance(rarity, int) or rarity < 1:
|
|
||||||
raise ValueError("Rarity must be a positive integer.")
|
|
||||||
if not isinstance(weight, (int, float)) or weight <= 0:
|
|
||||||
raise ValueError("Weight must be a positive number.")
|
|
||||||
if not image_url:
|
|
||||||
raise ValueError("Image URL must be provided.")
|
|
||||||
|
|
||||||
|
stripped_name = name.strip()
|
||||||
|
|
||||||
|
# Validate inputs
|
||||||
|
if not stripped_name:
|
||||||
|
raise ValueError('Character name cannot be empty.')
|
||||||
|
if rarity < 1:
|
||||||
|
raise ValueError('Rarity must be a positive integer.')
|
||||||
|
if rarity not in RARITY_TO_WEIGHT.keys():
|
||||||
|
raise ValueError(f'Invalid rarity: {rarity}')
|
||||||
|
if not image_url:
|
||||||
|
raise ValueError('Image URL must be provided.')
|
||||||
|
|
||||||
|
try:
|
||||||
# Download image
|
# Download image
|
||||||
response = requests.get(image_url, stream=True)
|
response = requests.get(image_url, stream=True, timeout=30)
|
||||||
if response.status_code != 200:
|
if response.status_code != 200:
|
||||||
raise RuntimeError(f"Failed to download image from {image_url}")
|
raise RuntimeError(f"Failed to download image from {image_url}")
|
||||||
|
|
||||||
# Upload to bot's Drive
|
# Upload to Fediverse instance
|
||||||
mk = client_connection()
|
fediverse_service = get_fediverse_service()
|
||||||
try:
|
try:
|
||||||
media = mk.drive_files_create(response.raw)
|
uploaded_file = fediverse_service.upload_file(response.raw)
|
||||||
file_id = media["id"]
|
file_id = uploaded_file.id
|
||||||
except MisskeyAPIException as e:
|
except RuntimeError as e:
|
||||||
raise RuntimeError(f"Failed to upload image to bot's Drive: {e}") from e
|
raise RuntimeError(f"Failed to upload image: {e}") from e
|
||||||
|
|
||||||
# Insert into database
|
# Insert into database
|
||||||
conn = get_db_connection()
|
conn = get_db_connection()
|
||||||
cur = conn.cursor()
|
cur = conn.cursor()
|
||||||
cur.execute(
|
cur.execute(
|
||||||
'INSERT INTO characters (name, rarity, weight, file_id) VALUES (?, ?, ?, ?)',
|
'INSERT INTO characters (name, rarity, weight, file_id) VALUES (?, ?, ?, ?)',
|
||||||
(name.strip(), rarity, float(weight), file_id)
|
(stripped_name, rarity, float(weight), file_id)
|
||||||
)
|
)
|
||||||
conn.commit()
|
conn.commit()
|
||||||
character_id = cur.lastrowid
|
character_id = cur.lastrowid
|
||||||
|
|
|
@ -1,58 +1,49 @@
|
||||||
import time
|
import time
|
||||||
import traceback
|
import traceback
|
||||||
import misskey
|
|
||||||
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_config, set_config
|
||||||
from client import client_connection
|
from fediverse_factory import get_fediverse_service
|
||||||
|
import config
|
||||||
# Initialize the Misskey client
|
|
||||||
client = client_connection()
|
|
||||||
|
|
||||||
# Define your whitelist
|
|
||||||
# TODO: move to config
|
|
||||||
whitelisted_instances: list[str] = []
|
|
||||||
|
|
||||||
def stream_notifications():
|
def stream_notifications():
|
||||||
print("Starting filtered notification stream...")
|
# Initialize the Fediverse service
|
||||||
|
fediverse_service = get_fediverse_service()
|
||||||
|
|
||||||
|
# Get the last seen notification ID from the database
|
||||||
last_seen_id = get_config("last_seen_notif_id")
|
last_seen_id = get_config("last_seen_notif_id")
|
||||||
|
whitelisted_instances = getattr(config, 'TRUSTED_INSTANCES', [])
|
||||||
|
|
||||||
|
print('Listening for notifications...')
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
# May be able to mark notifications as read using misskey.py and
|
# Get notifications from the fediverse service
|
||||||
# filter them out here. This function also takes a since_id we
|
notifications = fediverse_service.get_notifications(since_id=last_seen_id)
|
||||||
# could use as well
|
|
||||||
notifications = client.i_notifications()
|
|
||||||
|
|
||||||
if notifications:
|
if notifications:
|
||||||
# Oldest to newest
|
|
||||||
notifications.reverse()
|
|
||||||
|
|
||||||
new_last_seen_id = last_seen_id
|
new_last_seen_id = last_seen_id
|
||||||
|
|
||||||
for notification in notifications:
|
for notification in notifications:
|
||||||
notif_id = notification.get("id")
|
notif_id = notification.id
|
||||||
|
|
||||||
# Skip old or same ID notifications
|
# Skip old or same ID notifications
|
||||||
if last_seen_id is not None and notif_id <= last_seen_id:
|
if last_seen_id is not None and notif_id <= last_seen_id:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
user = notification.get("user", {})
|
username = notification.user.username
|
||||||
username = user.get("username", "unknown")
|
host = notification.user.host
|
||||||
host = user.get("host") # None if local user
|
|
||||||
|
|
||||||
instance = host if host else "local"
|
instance = host if host else "local"
|
||||||
|
|
||||||
if instance in whitelisted_instances or instance == "local":
|
if instance in whitelisted_instances or instance == "local":
|
||||||
note = notification.get("note", {}).get("text", "")
|
note = notification.post.text if notification.post else ""
|
||||||
notif_type = notification.get("type", "unknown")
|
notif_type = notification.type.value
|
||||||
|
|
||||||
print(f"📨 [{notif_type}] from @{username}@{instance}")
|
print(f"📨 [{notif_type}] from @{username}@{instance}")
|
||||||
print(f"💬 {note}")
|
print(f"💬 {note}")
|
||||||
print("-" * 30)
|
print("-" * 30)
|
||||||
|
|
||||||
# 🧠 Send to the parser
|
# 🧠 Send to the parser
|
||||||
parse_notification(notification,client)
|
parse_notification(notification, fediverse_service)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
print(f"⚠️ Blocked notification from untrusted instance: {host}")
|
print(f"⚠️ Blocked notification from untrusted instance: {host}")
|
||||||
|
@ -72,8 +63,5 @@ def stream_notifications():
|
||||||
print(f"An exception has occured: {e}\n{traceback.format_exc()}")
|
print(f"An exception has occured: {e}\n{traceback.format_exc()}")
|
||||||
time.sleep(5)
|
time.sleep(5)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
stream_notifications()
|
stream_notifications()
|
||||||
|
|
|
@ -1,5 +0,0 @@
|
||||||
import misskey
|
|
||||||
import config
|
|
||||||
|
|
||||||
def client_connection():
|
|
||||||
return misskey.Misskey(address=config.INSTANCE, i=config.KEY)
|
|
|
@ -1,22 +1,75 @@
|
||||||
'''Essentials for the bot to function'''
|
'''Essentials for the bot to function'''
|
||||||
import configparser
|
import configparser
|
||||||
|
from os import environ, path
|
||||||
|
|
||||||
|
|
||||||
|
class ConfigError(Exception):
|
||||||
|
'''Could not find config file'''
|
||||||
|
|
||||||
|
|
||||||
|
def get_config_file() -> str:
|
||||||
|
'''Gets the path to the config file in the current environment'''
|
||||||
|
env: str | None = environ.get('KEMOVERSE_ENV')
|
||||||
|
if not env:
|
||||||
|
raise ConfigError('Error: KEMOVERSE_ENV is unset')
|
||||||
|
if not (env in ['prod', 'dev']):
|
||||||
|
raise ConfigError(f'Error: Invalid environment: {env}')
|
||||||
|
|
||||||
|
config_path: str = f'config_{env}.ini'
|
||||||
|
|
||||||
|
if not path.isfile(config_path):
|
||||||
|
raise ConfigError(f'Could not find {config_path}')
|
||||||
|
return config_path
|
||||||
|
|
||||||
|
def get_rarity_to_weight(config_section):
|
||||||
|
"""Parses Rarity_X keys from config and returns a {rarity: weight} dict."""
|
||||||
|
rarity_weights = {}
|
||||||
|
for key, value in config_section.items():
|
||||||
|
if key.startswith("rarity_"):
|
||||||
|
rarity = int(key.removeprefix("rarity_"))
|
||||||
|
rarity_weights[rarity] = float(value)
|
||||||
|
return rarity_weights
|
||||||
|
|
||||||
|
|
||||||
config = configparser.ConfigParser()
|
config = configparser.ConfigParser()
|
||||||
config.read('config.ini')
|
config.read(get_config_file())
|
||||||
|
|
||||||
# Username for the bot
|
# Username for the bot
|
||||||
USER = config['application']['BotUser']
|
USER = config['credentials']['User'].lower()
|
||||||
|
|
||||||
# API key for the bot
|
# API key for the bot
|
||||||
KEY = config['application']['ApiKey']
|
KEY = config['credentials']['Token']
|
||||||
# Bot's Misskey instance URL
|
# Bot's Misskey/Pleroma instance URL
|
||||||
INSTANCE = config['application']['InstanceUrl']
|
INSTANCE = config['credentials']['Instance'].lower()
|
||||||
|
|
||||||
# SQLite Database location
|
# Instance type validation
|
||||||
DB_PATH = config['application']['DatabaseLocation']
|
if 'InstanceType' not in config['application']:
|
||||||
|
raise ValueError("InstanceType must be specified in config.ini")
|
||||||
|
|
||||||
|
instance_type = config['application']['InstanceType'].lower()
|
||||||
|
if instance_type not in ('misskey', 'pleroma'):
|
||||||
|
raise ValueError("InstanceType must be either 'misskey' or 'pleroma'")
|
||||||
|
|
||||||
|
INSTANCE_TYPE = instance_type
|
||||||
|
|
||||||
|
# Web server port
|
||||||
|
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
|
||||||
# Fedi handles in the traditional 'user@domain.tld' style, allows these users
|
# Fedi handles in the traditional 'user@domain.tld' style, allows these users
|
||||||
# to use extra admin exclusive commands with the bot'''
|
# to use extra admin exclusive commands with the bot
|
||||||
ADMINS = config['application']['DefaultAdmins']
|
ADMINS = config['application']['DefaultAdmins']
|
||||||
|
# SQLite Database location
|
||||||
|
DB_PATH = config['application']['DatabaseLocation']
|
||||||
|
|
||||||
|
NOTIFICATION_POLL_INTERVAL = int(config['notification']['PollInterval'])
|
||||||
|
NOTIFICATION_BATCH_SIZE = int(config['notification']['BatchSize'])
|
||||||
|
|
||||||
|
GACHA_ROLL_INTERVAL = int(config['gacha']['RollInterval'])
|
||||||
|
|
||||||
|
RARITY_TO_WEIGHT = get_rarity_to_weight(config['gacha'])
|
21
bot/custom_types.py
Normal file
21
bot/custom_types.py
Normal file
|
@ -0,0 +1,21 @@
|
||||||
|
from typing import TypedDict, List, Dict, Any
|
||||||
|
|
||||||
|
BotResponse = TypedDict('BotResponse', {
|
||||||
|
'message': str,
|
||||||
|
'attachment_urls': List[str] | None
|
||||||
|
})
|
||||||
|
|
||||||
|
Character = TypedDict('Character', {
|
||||||
|
'id': int,
|
||||||
|
'name': str,
|
||||||
|
'rarity': int,
|
||||||
|
'weight': float,
|
||||||
|
'image_url': str
|
||||||
|
})
|
||||||
|
|
||||||
|
ParsedNotification = TypedDict('ParsedNotification', {
|
||||||
|
'author': str,
|
||||||
|
'command': str | None,
|
||||||
|
'arguments': List[str],
|
||||||
|
'note_obj': Dict[str, Any]
|
||||||
|
})
|
143
bot/db_utils.py
143
bot/db_utils.py
|
@ -1,58 +1,121 @@
|
||||||
|
from random import choices
|
||||||
import sqlite3
|
import sqlite3
|
||||||
import random
|
|
||||||
import config
|
import config
|
||||||
|
from custom_types import Character
|
||||||
|
|
||||||
DB_PATH = config.DB_PATH
|
DB_PATH = config.DB_PATH
|
||||||
|
CONNECTION: sqlite3.Connection
|
||||||
|
CURSOR: sqlite3.Cursor
|
||||||
|
|
||||||
def get_db_connection():
|
|
||||||
|
def connect() -> None:
|
||||||
'''Creates a connection to the database'''
|
'''Creates a connection to the database'''
|
||||||
conn = sqlite3.connect(DB_PATH)
|
print('Connecting to the database...')
|
||||||
conn.row_factory = sqlite3.Row
|
global CONNECTION
|
||||||
return conn
|
global CURSOR
|
||||||
|
CONNECTION = sqlite3.connect(DB_PATH, autocommit=True)
|
||||||
|
CONNECTION.row_factory = sqlite3.Row
|
||||||
|
CURSOR = CONNECTION.cursor()
|
||||||
|
|
||||||
def get_or_create_user(username):
|
|
||||||
'''Retrieves an ID for a given user, if the user does not exist, it will be
|
def get_random_character() -> Character | None:
|
||||||
created.'''
|
''' Gets a random character from the database'''
|
||||||
conn = get_db_connection()
|
CURSOR.execute('SELECT * FROM characters')
|
||||||
conn.row_factory = sqlite3.Row
|
characters = CURSOR.fetchall()
|
||||||
cur = conn.cursor()
|
|
||||||
cur.execute('SELECT id FROM users WHERE username = ?', (username,))
|
if not characters:
|
||||||
user = cur.fetchone()
|
return None
|
||||||
|
|
||||||
|
weights = [config.RARITY_TO_WEIGHT[c['rarity']] for c in characters]
|
||||||
|
chosen = choices(characters, weights=weights, k=1)[0]
|
||||||
|
|
||||||
|
return {
|
||||||
|
'id': chosen['id'],
|
||||||
|
'name': chosen['name'],
|
||||||
|
'rarity': chosen['rarity'],
|
||||||
|
'weight': config.RARITY_TO_WEIGHT[chosen['rarity']],
|
||||||
|
'image_url': chosen['file_id']
|
||||||
|
}
|
||||||
|
|
||||||
|
def get_player(username: str) -> int:
|
||||||
|
'''Retrieve a player ID by username, or return None if not found.'''
|
||||||
|
CURSOR.execute('SELECT id FROM users WHERE username = ?', (username,))
|
||||||
|
user = CURSOR.fetchone()
|
||||||
if user:
|
if user:
|
||||||
conn.close()
|
return int(user[0])
|
||||||
return user[0]
|
|
||||||
|
|
||||||
# New user starts with has_rolled = False
|
def insert_player(username: str) -> int:
|
||||||
cur.execute(
|
'''Insert a new player with default has_rolled = False and return their user ID.'''
|
||||||
|
CURSOR.execute(
|
||||||
'INSERT INTO users (username, has_rolled) VALUES (?, ?)',
|
'INSERT INTO users (username, has_rolled) VALUES (?, ?)',
|
||||||
(username, False)
|
(username, False)
|
||||||
)
|
)
|
||||||
conn.commit()
|
return CURSOR.lastrowid
|
||||||
user_id = cur.lastrowid
|
|
||||||
conn.close()
|
|
||||||
return user_id
|
|
||||||
|
|
||||||
def add_pull(user_id, character_id):
|
def delete_player(username: str) -> bool:
|
||||||
|
'''Permanently deletes a user and all their pulls.'''
|
||||||
|
CURSOR.execute(
|
||||||
|
'SELECT id FROM users WHERE username = ?',
|
||||||
|
(username,)
|
||||||
|
)
|
||||||
|
user = CURSOR.fetchone()
|
||||||
|
|
||||||
|
user_id = user[0]
|
||||||
|
|
||||||
|
# Delete pulls
|
||||||
|
CURSOR.execute(
|
||||||
|
'DELETE FROM pulls WHERE user_id = ?',
|
||||||
|
(user_id,)
|
||||||
|
)
|
||||||
|
|
||||||
|
# Delete user
|
||||||
|
CURSOR.execute(
|
||||||
|
'DELETE FROM users WHERE id = ?',
|
||||||
|
(user_id,)
|
||||||
|
)
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def insert_character(
|
||||||
|
name: str, rarity: int, weight: float, file_id: str) -> int:
|
||||||
|
'''Inserts a character'''
|
||||||
|
CURSOR.execute(
|
||||||
|
'INSERT INTO characters (name, rarity, weight, file_id) VALUES \
|
||||||
|
(?, ?, ?, ?)',
|
||||||
|
(name, rarity, weight, file_id)
|
||||||
|
)
|
||||||
|
character_id = CURSOR.lastrowid
|
||||||
|
return character_id if character_id else 0
|
||||||
|
|
||||||
|
|
||||||
|
def insert_pull(user_id: int, character_id: int) -> None:
|
||||||
'''Creates a pull in the database'''
|
'''Creates a pull in the database'''
|
||||||
conn = get_db_connection()
|
CURSOR.execute(
|
||||||
cur = conn.cursor()
|
'INSERT INTO pulls (user_id, character_id) VALUES (?, ?)',
|
||||||
cur.execute('INSERT INTO pulls (user_id, character_id) VALUES (?, ?)', (user_id, character_id))
|
(user_id, character_id)
|
||||||
conn.commit()
|
)
|
||||||
conn.close()
|
|
||||||
|
|
||||||
def get_config(key):
|
|
||||||
|
def get_last_rolled_at(user_id: int) -> int:
|
||||||
|
'''Gets the timestamp when the user last rolled'''
|
||||||
|
CURSOR.execute(
|
||||||
|
"SELECT timestamp FROM pulls WHERE user_id = ? ORDER BY timestamp \
|
||||||
|
DESC",
|
||||||
|
(user_id,))
|
||||||
|
row = CURSOR.fetchone()
|
||||||
|
return row[0] if row else 0
|
||||||
|
|
||||||
|
|
||||||
|
def get_config(key: str) -> str:
|
||||||
'''Reads the value for a specified config key from the db'''
|
'''Reads the value for a specified config key from the db'''
|
||||||
conn = get_db_connection()
|
CURSOR.execute("SELECT value FROM config WHERE key = ?", (key,))
|
||||||
cur = conn.cursor()
|
row = CURSOR.fetchone()
|
||||||
cur.execute("SELECT value FROM config WHERE key = ?", (key,))
|
return row[0] if row else ''
|
||||||
row = cur.fetchone()
|
|
||||||
conn.close()
|
|
||||||
return row[0] if row else None
|
|
||||||
|
|
||||||
def set_config(key, value):
|
|
||||||
|
def set_config(key: str, value: str) -> None:
|
||||||
'''Writes the value for a specified config key to the db'''
|
'''Writes the value for a specified config key to the db'''
|
||||||
conn = get_db_connection()
|
CURSOR.execute("INSERT OR REPLACE INTO config (key, value) VALUES (?, ?)",
|
||||||
cur = conn.cursor()
|
(key, value))
|
||||||
cur.execute("INSERT OR REPLACE INTO config (key, value) VALUES (?, ?)", (key, value))
|
|
||||||
conn.commit()
|
|
||||||
conn.close()
|
|
||||||
|
|
31
bot/fediverse_factory.py
Normal file
31
bot/fediverse_factory.py
Normal file
|
@ -0,0 +1,31 @@
|
||||||
|
from fediverse_service import FediverseService
|
||||||
|
from misskey_service import MisskeyService
|
||||||
|
from pleroma_service import PleromaService
|
||||||
|
import config
|
||||||
|
|
||||||
|
|
||||||
|
class FediverseServiceFactory:
|
||||||
|
"""Factory for creating FediverseService implementations based on configuration"""
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def create_service() -> FediverseService:
|
||||||
|
"""
|
||||||
|
Create a FediverseService implementation based on the configured instance type.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
FediverseService implementation (MisskeyService or PleromaService)
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
ValueError: If the instance type is not supported
|
||||||
|
"""
|
||||||
|
if config.INSTANCE_TYPE == "misskey":
|
||||||
|
return MisskeyService()
|
||||||
|
elif config.INSTANCE_TYPE == "pleroma":
|
||||||
|
return PleromaService()
|
||||||
|
else:
|
||||||
|
raise ValueError(f"Unsupported instance type: {config.INSTANCE_TYPE}")
|
||||||
|
|
||||||
|
|
||||||
|
def get_fediverse_service() -> FediverseService:
|
||||||
|
"""Convenience function to get a FediverseService instance"""
|
||||||
|
return FediverseServiceFactory.create_service()
|
74
bot/fediverse_service.py
Normal file
74
bot/fediverse_service.py
Normal file
|
@ -0,0 +1,74 @@
|
||||||
|
from abc import ABC, abstractmethod
|
||||||
|
from typing import List, Optional, Union, BinaryIO
|
||||||
|
from fediverse_types import FediverseNotification, FediversePost, FediverseFile, Visibility
|
||||||
|
|
||||||
|
|
||||||
|
class FediverseService(ABC):
|
||||||
|
"""Abstract interface for Fediverse platform services (Misskey, Pleroma, etc.)"""
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
def get_notifications(self, since_id: Optional[str] = None) -> List[FediverseNotification]:
|
||||||
|
"""
|
||||||
|
Retrieve notifications from the Fediverse instance.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
since_id: Optional ID to get notifications newer than this ID
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List of FediverseNotification objects
|
||||||
|
"""
|
||||||
|
pass
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
def create_post(
|
||||||
|
self,
|
||||||
|
text: str,
|
||||||
|
reply_to_id: Optional[str] = None,
|
||||||
|
visibility: Visibility = Visibility.HOME,
|
||||||
|
file_ids: Optional[List[str]] = None,
|
||||||
|
visible_user_ids: Optional[List[str]] = None
|
||||||
|
) -> str:
|
||||||
|
"""
|
||||||
|
Create a new post on the Fediverse instance.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
text: The text content of the post
|
||||||
|
reply_to_id: Optional ID of post to reply to
|
||||||
|
visibility: Visibility level for the post
|
||||||
|
file_ids: Optional list of file IDs to attach
|
||||||
|
visible_user_ids: Optional list of user IDs who can see the post (for specified visibility)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
ID of the created post
|
||||||
|
"""
|
||||||
|
pass
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
def get_post_by_id(self, post_id: str) -> Optional[FediversePost]:
|
||||||
|
"""
|
||||||
|
Retrieve a specific post by its ID.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
post_id: The ID of the post to retrieve
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
FediversePost object if found, None otherwise
|
||||||
|
"""
|
||||||
|
pass
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
def upload_file(self, file_data: Union[BinaryIO, bytes], filename: Optional[str] = None) -> FediverseFile:
|
||||||
|
"""
|
||||||
|
Upload a file to the Fediverse instance.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
file_data: File data as binary stream or bytes
|
||||||
|
filename: Optional filename for the uploaded file
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
FediverseFile object with ID, URL, and other metadata
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
RuntimeError: If file upload fails
|
||||||
|
"""
|
||||||
|
pass
|
73
bot/fediverse_types.py
Normal file
73
bot/fediverse_types.py
Normal file
|
@ -0,0 +1,73 @@
|
||||||
|
from dataclasses import dataclass
|
||||||
|
from typing import Optional, List, Dict, Any
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
|
||||||
|
class NotificationType(Enum):
|
||||||
|
MENTION = "mention"
|
||||||
|
REPLY = "reply"
|
||||||
|
FOLLOW = "follow"
|
||||||
|
FAVOURITE = "favourite"
|
||||||
|
REBLOG = "reblog"
|
||||||
|
POLL = "poll"
|
||||||
|
OTHER = "other"
|
||||||
|
|
||||||
|
|
||||||
|
class Visibility(Enum):
|
||||||
|
PUBLIC = "public"
|
||||||
|
UNLISTED = "unlisted"
|
||||||
|
HOME = "home"
|
||||||
|
FOLLOWERS = "followers"
|
||||||
|
SPECIFIED = "specified"
|
||||||
|
DIRECT = "direct"
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class FediverseUser:
|
||||||
|
"""Common user representation across Fediverse platforms"""
|
||||||
|
id: str
|
||||||
|
username: str
|
||||||
|
host: Optional[str] = None # None for local users
|
||||||
|
display_name: Optional[str] = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def full_handle(self) -> str:
|
||||||
|
"""Returns the full fediverse handle (@user@domain or @user for local)"""
|
||||||
|
if self.host:
|
||||||
|
return f"@{self.username}@{self.host}"
|
||||||
|
return f"@{self.username}"
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class FediverseFile:
|
||||||
|
"""Common file/attachment representation"""
|
||||||
|
id: str
|
||||||
|
url: str
|
||||||
|
type: Optional[str] = None
|
||||||
|
name: Optional[str] = None
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class FediversePost:
|
||||||
|
"""Common post representation across Fediverse platforms"""
|
||||||
|
id: str
|
||||||
|
text: Optional[str]
|
||||||
|
user: FediverseUser
|
||||||
|
visibility: Visibility
|
||||||
|
created_at: Optional[str] = None
|
||||||
|
files: List[FediverseFile] = None
|
||||||
|
reply_to_id: Optional[str] = None
|
||||||
|
|
||||||
|
def __post_init__(self):
|
||||||
|
if self.files is None:
|
||||||
|
self.files = []
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class FediverseNotification:
|
||||||
|
"""Common notification representation across Fediverse platforms"""
|
||||||
|
id: str
|
||||||
|
type: NotificationType
|
||||||
|
user: FediverseUser
|
||||||
|
post: Optional[FediversePost] = None
|
||||||
|
created_at: Optional[str] = None
|
|
@ -1,69 +0,0 @@
|
||||||
import random
|
|
||||||
from db_utils import get_or_create_user, add_pull, get_db_connection
|
|
||||||
from add_character import add_character
|
|
||||||
|
|
||||||
def get_character():
|
|
||||||
''' Gets a random character from the database'''
|
|
||||||
conn = get_db_connection()
|
|
||||||
cur = conn.cursor()
|
|
||||||
cur.execute('SELECT * FROM characters')
|
|
||||||
characters = cur.fetchall()
|
|
||||||
conn.close()
|
|
||||||
|
|
||||||
if not characters:
|
|
||||||
return None, None, None, None
|
|
||||||
|
|
||||||
weights = [c['weight'] for c in characters]
|
|
||||||
chosen = random.choices(characters, weights=weights, k=1)[0]
|
|
||||||
|
|
||||||
return chosen['id'], chosen['name'], chosen['file_id'], chosen['rarity']
|
|
||||||
|
|
||||||
def is_float(val):
|
|
||||||
try:
|
|
||||||
float(val)
|
|
||||||
return True
|
|
||||||
except ValueError:
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
# TODO: See issue #3, separate command parsing from game logic.
|
|
||||||
def gacha_response(command,full_user, arguments,note_obj):
|
|
||||||
'''Parses a given command with arguments, processes the game state and
|
|
||||||
returns a response'''
|
|
||||||
|
|
||||||
if command == "roll":
|
|
||||||
user_id = get_or_create_user(full_user)
|
|
||||||
character_id, character_name, file_id, rarity = get_character()
|
|
||||||
|
|
||||||
if not character_id:
|
|
||||||
#TODO: Can't have tuples of a single element
|
|
||||||
# Return these as a dict or object instead.
|
|
||||||
return(f"@{full_user} Uwaaa... something went wrong! No characters found. 😿")
|
|
||||||
|
|
||||||
add_pull(user_id,character_id)
|
|
||||||
stars = '⭐️' * rarity
|
|
||||||
return([f"@{full_user} 🎲 Congrats! You rolled {stars} **{character_name}**\nShe's all yours now~ 💖✨",[file_id]])
|
|
||||||
|
|
||||||
if command == "create":
|
|
||||||
# Example call from bot logic
|
|
||||||
image_url = note_obj.get("files", [{}])[0].get("url") if note_obj.get("files") else None
|
|
||||||
if not image_url:
|
|
||||||
return "You need an image to create a character, dumbass."
|
|
||||||
|
|
||||||
if len(arguments) != 3:
|
|
||||||
return "Please specify the following attributes in order: name, rarity, drop weighting"
|
|
||||||
|
|
||||||
if not (arguments[1].isnumeric() and 1 <= int(arguments[1]) <= 5):
|
|
||||||
return f"Invalid rarity: '{arguments[1]}' must be a number between 1 and 5"
|
|
||||||
|
|
||||||
if not (is_float(arguments[2]) and 0.0 < float(arguments[2]) <= 1.0):
|
|
||||||
return f"Invalid drop weight: '{arguments[2]}' must be a decimal value between 0.0 and 1.0"
|
|
||||||
|
|
||||||
character_id, file_id = add_character(
|
|
||||||
name=arguments[0],
|
|
||||||
rarity=int(arguments[1]),
|
|
||||||
weight=float(arguments[2]),
|
|
||||||
image_url=image_url
|
|
||||||
)
|
|
||||||
return([f"Added {arguments[0]}, ID {character_id}.",[file_id]])
|
|
||||||
return None
|
|
156
bot/misskey_service.py
Normal file
156
bot/misskey_service.py
Normal file
|
@ -0,0 +1,156 @@
|
||||||
|
import misskey
|
||||||
|
from typing import List, Optional, Dict, Any, Union, BinaryIO
|
||||||
|
from fediverse_service import FediverseService
|
||||||
|
from fediverse_types import (
|
||||||
|
FediverseNotification, FediversePost, FediverseUser, FediverseFile,
|
||||||
|
NotificationType, Visibility
|
||||||
|
)
|
||||||
|
import config
|
||||||
|
|
||||||
|
|
||||||
|
class MisskeyService(FediverseService):
|
||||||
|
"""Misskey implementation of FediverseService"""
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self.client = misskey.Misskey(address=config.INSTANCE, i=config.KEY)
|
||||||
|
|
||||||
|
def _convert_misskey_user(self, user_data: Dict[str, Any]) -> FediverseUser:
|
||||||
|
"""Convert Misskey user data to FediverseUser"""
|
||||||
|
return FediverseUser(
|
||||||
|
id=user_data.get("id", ""),
|
||||||
|
username=user_data.get("username", "unknown"),
|
||||||
|
host=user_data.get("host"),
|
||||||
|
display_name=user_data.get("name")
|
||||||
|
)
|
||||||
|
|
||||||
|
def _convert_misskey_file(self, file_data: Dict[str, Any]) -> FediverseFile:
|
||||||
|
"""Convert Misskey file data to FediverseFile"""
|
||||||
|
return FediverseFile(
|
||||||
|
id=file_data.get("id", ""),
|
||||||
|
url=file_data.get("url", ""),
|
||||||
|
type=file_data.get("type"),
|
||||||
|
name=file_data.get("name")
|
||||||
|
)
|
||||||
|
|
||||||
|
def _convert_misskey_visibility(self, visibility: str) -> Visibility:
|
||||||
|
"""Convert Misskey visibility to our enum"""
|
||||||
|
visibility_map = {
|
||||||
|
"public": Visibility.PUBLIC,
|
||||||
|
"unlisted": Visibility.UNLISTED,
|
||||||
|
"home": Visibility.HOME,
|
||||||
|
"followers": Visibility.FOLLOWERS,
|
||||||
|
"specified": Visibility.SPECIFIED
|
||||||
|
}
|
||||||
|
return visibility_map.get(visibility, Visibility.HOME)
|
||||||
|
|
||||||
|
def _convert_to_misskey_visibility(self, visibility: Visibility) -> str:
|
||||||
|
"""Convert our visibility enum to Misskey visibility"""
|
||||||
|
visibility_map = {
|
||||||
|
Visibility.PUBLIC: "public",
|
||||||
|
Visibility.UNLISTED: "unlisted",
|
||||||
|
Visibility.HOME: "home",
|
||||||
|
Visibility.FOLLOWERS: "followers",
|
||||||
|
Visibility.SPECIFIED: "specified",
|
||||||
|
Visibility.DIRECT: "specified" # Map direct to specified for Misskey
|
||||||
|
}
|
||||||
|
return visibility_map.get(visibility, "home")
|
||||||
|
|
||||||
|
def _convert_misskey_notification_type(self, notif_type: str) -> NotificationType:
|
||||||
|
"""Convert Misskey notification type to our enum"""
|
||||||
|
type_map = {
|
||||||
|
"mention": NotificationType.MENTION,
|
||||||
|
"reply": NotificationType.REPLY,
|
||||||
|
"follow": NotificationType.FOLLOW,
|
||||||
|
"favourite": NotificationType.FAVOURITE,
|
||||||
|
"reblog": NotificationType.REBLOG,
|
||||||
|
"poll": NotificationType.POLL
|
||||||
|
}
|
||||||
|
return type_map.get(notif_type, NotificationType.OTHER)
|
||||||
|
|
||||||
|
def _convert_misskey_post(self, note_data: Dict[str, Any]) -> FediversePost:
|
||||||
|
"""Convert Misskey note data to FediversePost"""
|
||||||
|
files = []
|
||||||
|
if note_data.get("files"):
|
||||||
|
files = [self._convert_misskey_file(f) for f in note_data["files"]]
|
||||||
|
|
||||||
|
return FediversePost(
|
||||||
|
id=note_data.get("id", ""),
|
||||||
|
text=note_data.get("text"),
|
||||||
|
user=self._convert_misskey_user(note_data.get("user", {})),
|
||||||
|
visibility=self._convert_misskey_visibility(note_data.get("visibility", "home")),
|
||||||
|
created_at=note_data.get("createdAt"),
|
||||||
|
files=files,
|
||||||
|
reply_to_id=note_data.get("replyId")
|
||||||
|
)
|
||||||
|
|
||||||
|
def _convert_misskey_notification(self, notification_data: Dict[str, Any]) -> FediverseNotification:
|
||||||
|
"""Convert Misskey notification data to FediverseNotification"""
|
||||||
|
post = None
|
||||||
|
if notification_data.get("note"):
|
||||||
|
post = self._convert_misskey_post(notification_data["note"])
|
||||||
|
|
||||||
|
return FediverseNotification(
|
||||||
|
id=notification_data.get("id", ""),
|
||||||
|
type=self._convert_misskey_notification_type(notification_data.get("type", "")),
|
||||||
|
user=self._convert_misskey_user(notification_data.get("user", {})),
|
||||||
|
post=post,
|
||||||
|
created_at=notification_data.get("createdAt")
|
||||||
|
)
|
||||||
|
|
||||||
|
def get_notifications(self, since_id: Optional[str] = None) -> List[FediverseNotification]:
|
||||||
|
"""Get notifications from Misskey instance"""
|
||||||
|
params = {
|
||||||
|
'include_types': ['mention', 'reply'],
|
||||||
|
'limit': 50
|
||||||
|
}
|
||||||
|
if since_id:
|
||||||
|
params["since_id"] = since_id
|
||||||
|
|
||||||
|
notifications = self.client.i_notifications(**params)
|
||||||
|
return [self._convert_misskey_notification(notif) for notif in notifications]
|
||||||
|
|
||||||
|
def create_post(
|
||||||
|
self,
|
||||||
|
text: str,
|
||||||
|
reply_to_id: Optional[str] = None,
|
||||||
|
visibility: Visibility = Visibility.HOME,
|
||||||
|
file_ids: Optional[List[str]] = None,
|
||||||
|
visible_user_ids: Optional[List[str]] = None
|
||||||
|
) -> str:
|
||||||
|
"""Create a post on Misskey instance"""
|
||||||
|
params = {
|
||||||
|
"text": text,
|
||||||
|
"visibility": self._convert_to_misskey_visibility(visibility)
|
||||||
|
}
|
||||||
|
|
||||||
|
if reply_to_id:
|
||||||
|
params["reply_id"] = reply_to_id
|
||||||
|
|
||||||
|
if file_ids:
|
||||||
|
params["file_ids"] = file_ids
|
||||||
|
|
||||||
|
if visible_user_ids and visibility == Visibility.SPECIFIED:
|
||||||
|
params["visible_user_ids"] = visible_user_ids
|
||||||
|
|
||||||
|
response = self.client.notes_create(**params)
|
||||||
|
return response.get("createdNote", {}).get("id", "")
|
||||||
|
|
||||||
|
def get_post_by_id(self, post_id: str) -> Optional[FediversePost]:
|
||||||
|
"""Get a specific post by ID from Misskey instance"""
|
||||||
|
try:
|
||||||
|
note = self.client.notes_show(noteId=post_id)
|
||||||
|
return self._convert_misskey_post(note)
|
||||||
|
except Exception:
|
||||||
|
return None
|
||||||
|
|
||||||
|
def upload_file(self, file_data: Union[BinaryIO, bytes], filename: Optional[str] = None) -> FediverseFile:
|
||||||
|
"""Upload a file to Misskey Drive"""
|
||||||
|
try:
|
||||||
|
from misskey.exceptions import MisskeyAPIException
|
||||||
|
|
||||||
|
media = self.client.drive_files_create(file_data)
|
||||||
|
return self._convert_misskey_file(media)
|
||||||
|
except MisskeyAPIException as e:
|
||||||
|
raise RuntimeError(f"Failed to upload file to Misskey Drive: {e}") from e
|
||||||
|
except Exception as e:
|
||||||
|
raise RuntimeError(f"Unexpected error during file upload: {e}") from e
|
|
@ -1,34 +1,38 @@
|
||||||
import random, re
|
import re
|
||||||
import config
|
import config
|
||||||
from gacha_response import gacha_response
|
from response import generate_response
|
||||||
|
from fediverse_factory import get_fediverse_service
|
||||||
|
from fediverse_types import FediverseNotification, NotificationType, Visibility
|
||||||
|
from custom_types import ParsedNotification
|
||||||
|
|
||||||
def parse_notification(notification,client):
|
def parse_notification(notification: FediverseNotification, fediverse_service=None):
|
||||||
'''Oarses any notifications received by the bot and sends any commands to
|
'''Parses any notifications received by the bot and sends any commands to
|
||||||
gacha_response()'''
|
generate_response()'''
|
||||||
|
|
||||||
|
if fediverse_service is None:
|
||||||
|
fediverse_service = get_fediverse_service()
|
||||||
|
|
||||||
# We get the type of notification to filter the ones that we actually want
|
# We get the type of notification to filter the ones that we actually want
|
||||||
# to parse
|
# to parse
|
||||||
|
if notification.type not in (NotificationType.MENTION, NotificationType.REPLY):
|
||||||
notif_type = notification.get("type")
|
|
||||||
if not notif_type in ('mention', 'reply'):
|
|
||||||
return # Ignore anything that isn't a mention
|
return # Ignore anything that isn't a mention
|
||||||
|
|
||||||
|
# Return early if no post attached
|
||||||
|
if not notification.post:
|
||||||
|
return
|
||||||
|
|
||||||
# We want the visibility to be related to the type that was received (so if
|
# We want the visibility to be related to the type that was received (so if
|
||||||
# people don't want to dump a bunch of notes on home they don't have to)
|
# people don't want to dump a bunch of notes on home they don't have to)
|
||||||
visibility = notification["note"]["visibility"]
|
if notification.post.visibility != Visibility.SPECIFIED:
|
||||||
if visibility != "specified":
|
visibility = Visibility.HOME
|
||||||
visibility = "home"
|
else:
|
||||||
|
visibility = Visibility.SPECIFIED
|
||||||
|
|
||||||
# Get the full Activitypub ID of the user
|
# Get the full Activitypub ID of the user
|
||||||
user = notification.get("user", {})
|
full_user = notification.user.full_handle
|
||||||
username = user.get("username", "unknown")
|
|
||||||
host = user.get("host")
|
|
||||||
# Local users may not have a hostname attached
|
|
||||||
full_user = f"@{username}" if not host else f"@{username}@{host}"
|
|
||||||
|
|
||||||
note_obj = notification.get("note", {})
|
note_text = notification.post.text
|
||||||
note_text = note_obj.get("text")
|
note_id = notification.post.id
|
||||||
note_id = note_obj.get("id")
|
|
||||||
|
|
||||||
note = note_text.strip().lower() if note_text else ""
|
note = note_text.strip().lower() if note_text else ""
|
||||||
|
|
||||||
|
@ -40,32 +44,41 @@ def parse_notification(notification,client):
|
||||||
|
|
||||||
# Make sure the notification text explicitly mentions the bot
|
# Make sure the notification text explicitly mentions the bot
|
||||||
if not any(variant in note for variant in username_variants):
|
if not any(variant in note for variant in username_variants):
|
||||||
return
|
return None
|
||||||
|
|
||||||
# Find command and arguments after the mention
|
# Find command and arguments after the mention
|
||||||
# Removes all mentions (regex = mentions that start with @ and may contain @domain)
|
# Removes all mentions
|
||||||
|
# regex = mentions that start with @ and may contain @domain
|
||||||
cleaned_text = re.sub(r"@\w+(?:@\S+)?", "", note).strip()
|
cleaned_text = re.sub(r"@\w+(?:@\S+)?", "", note).strip()
|
||||||
parts = cleaned_text.split()
|
parts = cleaned_text.split()
|
||||||
|
|
||||||
command = parts[0].lower() if parts else None
|
command = parts[0].lower() if parts else None
|
||||||
arguments = parts[1:] if len(parts) > 1 else []
|
arguments = parts[1:] if len(parts) > 1 else []
|
||||||
|
|
||||||
# TODO: move response generation to a different function
|
# Create ParsedNotification object for the new response system
|
||||||
response = gacha_response(command.lower(),full_user, arguments, note_obj)
|
parsed_notification: ParsedNotification = {
|
||||||
|
'author': full_user,
|
||||||
|
'command': command,
|
||||||
|
'arguments': arguments,
|
||||||
|
'note_obj': {
|
||||||
|
'id': note_id,
|
||||||
|
'text': note_text,
|
||||||
|
'files': [{'url': f.url} for f in notification.post.files] if notification.post.files else []
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
# Generate response using the new system
|
||||||
|
response = generate_response(parsed_notification)
|
||||||
if not response:
|
if not response:
|
||||||
return
|
return
|
||||||
|
|
||||||
if isinstance(response, str):
|
# Handle attachment URLs (convert to file IDs if needed)
|
||||||
client.notes_create(
|
file_ids = response['attachment_urls'] if response['attachment_urls'] else None
|
||||||
text=response,
|
|
||||||
reply_id=note_id,
|
fediverse_service.create_post(
|
||||||
visibility=visibility
|
text=response['message'],
|
||||||
)
|
reply_to_id=note_id,
|
||||||
else:
|
|
||||||
client.notes_create(
|
|
||||||
text=response[0],
|
|
||||||
reply_id=note_id,
|
|
||||||
visibility=visibility,
|
visibility=visibility,
|
||||||
file_ids=response[1]
|
file_ids=file_ids
|
||||||
#visible_user_ids=[] #todo: write actual visible users ids so pleromers can use the bot privately
|
#visible_user_ids=[] #todo: write actual visible users ids so pleromers can use the bot privately
|
||||||
)
|
)
|
||||||
|
|
188
bot/pleroma_service.py
Normal file
188
bot/pleroma_service.py
Normal file
|
@ -0,0 +1,188 @@
|
||||||
|
from mastodon import Mastodon
|
||||||
|
from typing import List, Optional, Dict, Any, Union, BinaryIO
|
||||||
|
import io
|
||||||
|
import filetype
|
||||||
|
from fediverse_service import FediverseService
|
||||||
|
from fediverse_types import (
|
||||||
|
FediverseNotification, FediversePost, FediverseUser, FediverseFile,
|
||||||
|
NotificationType, Visibility
|
||||||
|
)
|
||||||
|
import config
|
||||||
|
|
||||||
|
|
||||||
|
class PleromaService(FediverseService):
|
||||||
|
"""Pleroma implementation of FediverseService using Mastodon.py"""
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self.client = Mastodon(
|
||||||
|
access_token=config.KEY,
|
||||||
|
api_base_url=config.INSTANCE
|
||||||
|
)
|
||||||
|
|
||||||
|
def _convert_mastodon_user(self, user_data: Dict[str, Any]) -> FediverseUser:
|
||||||
|
"""Convert Mastodon/Pleroma user data to FediverseUser"""
|
||||||
|
acct = user_data.get("acct", "")
|
||||||
|
if "@" in acct:
|
||||||
|
username, host = acct.split("@", 1)
|
||||||
|
else:
|
||||||
|
username = acct
|
||||||
|
host = None
|
||||||
|
|
||||||
|
return FediverseUser(
|
||||||
|
id=str(user_data.get("id", "")),
|
||||||
|
username=username,
|
||||||
|
host=host,
|
||||||
|
display_name=user_data.get("display_name")
|
||||||
|
)
|
||||||
|
|
||||||
|
def _convert_mastodon_file(self, file_data: Dict[str, Any]) -> FediverseFile:
|
||||||
|
"""Convert Mastodon/Pleroma media attachment to FediverseFile"""
|
||||||
|
return FediverseFile(
|
||||||
|
id=str(file_data.get("id", "")),
|
||||||
|
url=file_data.get("url", ""),
|
||||||
|
type=file_data.get("type"),
|
||||||
|
name=file_data.get("description")
|
||||||
|
)
|
||||||
|
|
||||||
|
def _convert_mastodon_visibility(self, visibility: str) -> Visibility:
|
||||||
|
"""Convert Mastodon/Pleroma visibility to our enum"""
|
||||||
|
visibility_map = {
|
||||||
|
"public": Visibility.PUBLIC,
|
||||||
|
"unlisted": Visibility.UNLISTED,
|
||||||
|
"private": Visibility.FOLLOWERS,
|
||||||
|
"direct": Visibility.DIRECT
|
||||||
|
}
|
||||||
|
return visibility_map.get(visibility, Visibility.PUBLIC)
|
||||||
|
|
||||||
|
def _convert_to_mastodon_visibility(self, visibility: Visibility) -> str:
|
||||||
|
"""Convert our visibility enum to Mastodon/Pleroma visibility"""
|
||||||
|
visibility_map = {
|
||||||
|
Visibility.PUBLIC: "public",
|
||||||
|
Visibility.UNLISTED: "unlisted",
|
||||||
|
Visibility.HOME: "unlisted", # Map home to unlisted for Pleroma
|
||||||
|
Visibility.FOLLOWERS: "private",
|
||||||
|
Visibility.SPECIFIED: "direct", # Map specified to direct for Pleroma
|
||||||
|
Visibility.DIRECT: "direct"
|
||||||
|
}
|
||||||
|
return visibility_map.get(visibility, "public")
|
||||||
|
|
||||||
|
def _convert_mastodon_notification_type(self, notif_type: str) -> NotificationType:
|
||||||
|
"""Convert Mastodon/Pleroma notification type to our enum"""
|
||||||
|
type_map = {
|
||||||
|
"mention": NotificationType.MENTION,
|
||||||
|
"follow": NotificationType.FOLLOW,
|
||||||
|
"favourite": NotificationType.FAVOURITE,
|
||||||
|
"reblog": NotificationType.REBLOG,
|
||||||
|
"poll": NotificationType.POLL
|
||||||
|
}
|
||||||
|
return type_map.get(notif_type, NotificationType.OTHER)
|
||||||
|
|
||||||
|
def _convert_mastodon_status(self, status_data: Dict[str, Any]) -> FediversePost:
|
||||||
|
"""Convert Mastodon/Pleroma status data to FediversePost"""
|
||||||
|
files = []
|
||||||
|
if status_data.get("media_attachments"):
|
||||||
|
files = [self._convert_mastodon_file(f) for f in status_data["media_attachments"]]
|
||||||
|
|
||||||
|
# Extract plain text from HTML content
|
||||||
|
content = status_data.get("content", "")
|
||||||
|
# Basic HTML stripping - in production you might want to use a proper HTML parser
|
||||||
|
import re
|
||||||
|
plain_text = re.sub(r'<[^>]+>', '', content) if content else None
|
||||||
|
|
||||||
|
return FediversePost(
|
||||||
|
id=str(status_data.get("id", "")),
|
||||||
|
text=plain_text,
|
||||||
|
user=self._convert_mastodon_user(status_data.get("account", {})),
|
||||||
|
visibility=self._convert_mastodon_visibility(status_data.get("visibility", "public")),
|
||||||
|
created_at=status_data.get("created_at"),
|
||||||
|
files=files,
|
||||||
|
reply_to_id=str(status_data["in_reply_to_id"]) if status_data.get("in_reply_to_id") else None
|
||||||
|
)
|
||||||
|
|
||||||
|
def _convert_mastodon_notification(self, notification_data: Dict[str, Any]) -> FediverseNotification:
|
||||||
|
"""Convert Mastodon/Pleroma notification data to FediverseNotification"""
|
||||||
|
post = None
|
||||||
|
if notification_data.get("status"):
|
||||||
|
post = self._convert_mastodon_status(notification_data["status"])
|
||||||
|
|
||||||
|
return FediverseNotification(
|
||||||
|
id=str(notification_data.get("id", "")),
|
||||||
|
type=self._convert_mastodon_notification_type(notification_data.get("type", "")),
|
||||||
|
user=self._convert_mastodon_user(notification_data.get("account", {})),
|
||||||
|
post=post,
|
||||||
|
created_at=notification_data.get("created_at")
|
||||||
|
)
|
||||||
|
|
||||||
|
def get_notifications(self, since_id: Optional[str] = None) -> List[FediverseNotification]:
|
||||||
|
"""Get notifications from Pleroma instance"""
|
||||||
|
params = {}
|
||||||
|
if since_id:
|
||||||
|
params["since_id"] = since_id
|
||||||
|
|
||||||
|
notifications = self.client.notifications(**params)
|
||||||
|
return [self._convert_mastodon_notification(notif) for notif in notifications]
|
||||||
|
|
||||||
|
def create_post(
|
||||||
|
self,
|
||||||
|
text: str,
|
||||||
|
reply_to_id: Optional[str] = None,
|
||||||
|
visibility: Visibility = Visibility.HOME,
|
||||||
|
file_ids: Optional[List[str]] = None,
|
||||||
|
visible_user_ids: Optional[List[str]] = None
|
||||||
|
) -> str:
|
||||||
|
"""Create a post on Pleroma instance"""
|
||||||
|
params = {
|
||||||
|
"status": text,
|
||||||
|
"visibility": self._convert_to_mastodon_visibility(visibility)
|
||||||
|
}
|
||||||
|
|
||||||
|
if reply_to_id:
|
||||||
|
params["in_reply_to_id"] = reply_to_id
|
||||||
|
|
||||||
|
if file_ids:
|
||||||
|
params["media_ids"] = file_ids
|
||||||
|
|
||||||
|
# Note: Pleroma/Mastodon doesn't have direct equivalent to visible_user_ids
|
||||||
|
# For direct messages, you typically mention users in the status text
|
||||||
|
|
||||||
|
response = self.client.status_post(**params)
|
||||||
|
return str(response.get("id", ""))
|
||||||
|
|
||||||
|
def get_post_by_id(self, post_id: str) -> Optional[FediversePost]:
|
||||||
|
"""Get a specific post by ID from Pleroma instance"""
|
||||||
|
try:
|
||||||
|
status = self.client.status(post_id)
|
||||||
|
return self._convert_mastodon_status(status)
|
||||||
|
except Exception:
|
||||||
|
return None
|
||||||
|
|
||||||
|
def upload_file(self, file_data: Union[BinaryIO, bytes], filename: Optional[str] = None) -> FediverseFile:
|
||||||
|
"""Upload a file to Pleroma instance"""
|
||||||
|
try:
|
||||||
|
# Convert file_data to bytes for MIME detection
|
||||||
|
if hasattr(file_data, 'read'):
|
||||||
|
# Check if we can seek back
|
||||||
|
try:
|
||||||
|
current_pos = file_data.tell()
|
||||||
|
file_bytes = file_data.read()
|
||||||
|
file_data.seek(current_pos)
|
||||||
|
file_data = io.BytesIO(file_bytes)
|
||||||
|
except (io.UnsupportedOperation, OSError):
|
||||||
|
# Non-seekable stream, already read all data
|
||||||
|
file_data = io.BytesIO(file_bytes)
|
||||||
|
else:
|
||||||
|
file_bytes = file_data
|
||||||
|
file_data = io.BytesIO(file_bytes)
|
||||||
|
|
||||||
|
# Use filetype library for robust MIME detection
|
||||||
|
kind = filetype.guess(file_bytes)
|
||||||
|
if kind is not None:
|
||||||
|
mime_type = kind.mime
|
||||||
|
else:
|
||||||
|
# Fallback to image/jpeg if detection fails
|
||||||
|
mime_type = 'image/jpeg'
|
||||||
|
|
||||||
|
media = self.client.media_post(file_data, mime_type=mime_type, description=filename)
|
||||||
|
return self._convert_mastodon_file(media)
|
||||||
|
except Exception as e:
|
||||||
|
raise RuntimeError(f"Failed to upload file to Pleroma: {e}") from e
|
215
bot/response.py
Normal file
215
bot/response.py
Normal file
|
@ -0,0 +1,215 @@
|
||||||
|
from datetime import datetime, timedelta, timezone
|
||||||
|
from typing import TypedDict, Any, List, Dict
|
||||||
|
from db_utils import get_player, insert_player, delete_player, insert_pull, get_last_rolled_at, \
|
||||||
|
get_random_character
|
||||||
|
from add_character import add_character
|
||||||
|
from config import GACHA_ROLL_INTERVAL
|
||||||
|
from custom_types import BotResponse, ParsedNotification
|
||||||
|
|
||||||
|
|
||||||
|
def do_roll(author: str) -> BotResponse:
|
||||||
|
'''Determines whether the user can roll, then pulls a random character'''
|
||||||
|
user_id = get_player(author)
|
||||||
|
if not user_id:
|
||||||
|
return {
|
||||||
|
'message':f'{author} 🛑 You haven’t signed up yet! Use the `signup` command to start playing.',
|
||||||
|
'attachment_urls': None
|
||||||
|
}
|
||||||
|
# Get date of user's last roll
|
||||||
|
date = get_last_rolled_at(user_id)
|
||||||
|
|
||||||
|
# No date means it's users first roll
|
||||||
|
if date:
|
||||||
|
# SQLite timestamps returned by the DB are always in UTC
|
||||||
|
# Below timestamps are to be converted to UTC
|
||||||
|
prev = datetime.strptime(str(date) + '+0000', '%Y-%m-%d %H:%M:%S%z')
|
||||||
|
now = datetime.now(timezone.utc)
|
||||||
|
|
||||||
|
time_since_last_roll = now - prev
|
||||||
|
roll_interval = timedelta(seconds=GACHA_ROLL_INTERVAL)
|
||||||
|
duration = roll_interval - time_since_last_roll
|
||||||
|
|
||||||
|
# User needs to wait before they can roll again
|
||||||
|
if time_since_last_roll < roll_interval:
|
||||||
|
remaining_duration = None
|
||||||
|
if duration.seconds > 3600:
|
||||||
|
remaining_duration = f'{-(duration.seconds // -3600)} hours'
|
||||||
|
elif duration.seconds > 60:
|
||||||
|
remaining_duration = f'{-(duration.seconds // -60)} minutes'
|
||||||
|
else:
|
||||||
|
remaining_duration = f'{duration.seconds} seconds'
|
||||||
|
|
||||||
|
return {
|
||||||
|
'message': f'{author} ⏱️ Please wait another \
|
||||||
|
{remaining_duration} before rolling again.',
|
||||||
|
'attachment_urls': None
|
||||||
|
}
|
||||||
|
|
||||||
|
character = get_random_character()
|
||||||
|
|
||||||
|
if not character:
|
||||||
|
return {
|
||||||
|
'message': f'{author} Uwaaa... something went wrong! No \
|
||||||
|
characters found. 😿',
|
||||||
|
'attachment_urls': None
|
||||||
|
}
|
||||||
|
|
||||||
|
insert_pull(user_id, character['id'])
|
||||||
|
stars = '⭐️' * character['rarity']
|
||||||
|
return {
|
||||||
|
'message': f'{author} 🎲 Congrats! You rolled {stars} \
|
||||||
|
**{character['name']}**\nShe\'s all yours now~ 💖✨',
|
||||||
|
'attachment_urls': [character['image_url']]
|
||||||
|
}
|
||||||
|
|
||||||
|
def do_signup(author: str) -> BotResponse:
|
||||||
|
'''Registers a new user if they haven’t signed up yet.'''
|
||||||
|
user_id = get_player(author)
|
||||||
|
|
||||||
|
if user_id:
|
||||||
|
return {
|
||||||
|
'message':f'{author} 👀 You’re already signed up! Let the rolling begin~ 🎲',
|
||||||
|
'attachment_urls': None
|
||||||
|
}
|
||||||
|
|
||||||
|
new_user_id = insert_player(author)
|
||||||
|
return {
|
||||||
|
'message': f'{author} ✅ Signed up successfully! Your gacha destiny begins now... ✨ Use the roll command to start!',
|
||||||
|
'attachment_urls': None
|
||||||
|
}
|
||||||
|
|
||||||
|
def is_float(val: Any) -> bool:
|
||||||
|
'''Returns true if `val` can be converted to a float'''
|
||||||
|
try:
|
||||||
|
float(val)
|
||||||
|
return True
|
||||||
|
except ValueError:
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
def do_create(
|
||||||
|
author: str,
|
||||||
|
arguments: List[str],
|
||||||
|
note_obj: Dict[str, Any]) -> BotResponse:
|
||||||
|
'''Creates a character'''
|
||||||
|
# Example call from bot logic
|
||||||
|
image_url = note_obj.get('files', [{}])[0].get('url') \
|
||||||
|
if note_obj.get('files') else None
|
||||||
|
|
||||||
|
if not image_url:
|
||||||
|
return {
|
||||||
|
'message': f'{author} You need an image to create a character, \
|
||||||
|
dumbass.',
|
||||||
|
'attachment_urls': None
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(arguments) != 3:
|
||||||
|
return {
|
||||||
|
'message': f'{author} Please specify the following attributes \
|
||||||
|
in order: name, rarity, weight',
|
||||||
|
'attachment_urls': None
|
||||||
|
}
|
||||||
|
|
||||||
|
if not (arguments[1].isnumeric() and 1 <= int(arguments[1]) <= 5):
|
||||||
|
return {
|
||||||
|
'message': f'{author} Invalid rarity: \'{arguments[1]}\' must \
|
||||||
|
be a number between 1 and 5',
|
||||||
|
'attachment_urls': None
|
||||||
|
}
|
||||||
|
if not (is_float(arguments[2]) and 0.0 < float(arguments[2]) <= 1.0):
|
||||||
|
return {
|
||||||
|
'message': f'{author} Invalid drop weight: \'{arguments[2]}\' \
|
||||||
|
must be a decimal value between 0.0 and 1.0',
|
||||||
|
'attachment_urls': None
|
||||||
|
}
|
||||||
|
|
||||||
|
character_id, file_id = add_character(
|
||||||
|
name=arguments[0],
|
||||||
|
rarity=int(arguments[1]),
|
||||||
|
weight=float(arguments[2]),
|
||||||
|
image_url=image_url
|
||||||
|
)
|
||||||
|
return {
|
||||||
|
'message': f'{author} Added {arguments[0]}, ID {character_id}.',
|
||||||
|
'attachment_urls': [file_id]
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def do_help(author: str) -> BotResponse:
|
||||||
|
'''Provides a list of commands that the bot can do.'''
|
||||||
|
return {
|
||||||
|
'message':f'{author} Here\'s what I can do:\n \
|
||||||
|
- `roll` Pulls a random character.\
|
||||||
|
- `create <name> <rarity>` Creates a character using a given image.\
|
||||||
|
- `signup` Registers your account.\
|
||||||
|
- `delete_account` Deletes your account.\
|
||||||
|
- `help` Shows this message',
|
||||||
|
'attachment_urls': None
|
||||||
|
}
|
||||||
|
|
||||||
|
def delete_account(author: str) -> BotResponse:
|
||||||
|
return {
|
||||||
|
'message':f'{author} ⚠️ This will permanently delete your account and all your cards.\n'
|
||||||
|
'If you’re sure, reply with `confirm_delete` to proceed.\n\n'
|
||||||
|
'**There is no undo.** Your gacha luck will be lost to the void... 💀✨',
|
||||||
|
'attachment_urls': None
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
def confirm_delete(author: str) -> BotResponse:
|
||||||
|
|
||||||
|
delete_player(author)
|
||||||
|
|
||||||
|
return {
|
||||||
|
'message':f'{author} 🧼 Your account and all your cards have been deleted. RIP your gacha history 🕊️✨',
|
||||||
|
'attachment_urls': None
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def generate_response(notification: ParsedNotification) -> BotResponse | None:
|
||||||
|
'''Given a command with arguments, processes the game state and
|
||||||
|
returns a response'''
|
||||||
|
|
||||||
|
# Temporary response variable
|
||||||
|
res: BotResponse | None = None
|
||||||
|
# TODO: Check if the user has an account
|
||||||
|
author = notification['author']
|
||||||
|
user_id = get_player(author)
|
||||||
|
command = notification['command']
|
||||||
|
# Check if the user is an administrator
|
||||||
|
# user_is_administrator = user_is_administrator()
|
||||||
|
|
||||||
|
# Unrestricted commands
|
||||||
|
match command:
|
||||||
|
case 'signup':
|
||||||
|
res = do_signup(author)
|
||||||
|
case 'help':
|
||||||
|
res = do_help(author)
|
||||||
|
case 'roll':
|
||||||
|
res = do_roll(author)
|
||||||
|
case _:
|
||||||
|
pass
|
||||||
|
|
||||||
|
if not user_id:
|
||||||
|
return res
|
||||||
|
|
||||||
|
# User commands
|
||||||
|
match command:
|
||||||
|
case 'create':
|
||||||
|
res = do_create(
|
||||||
|
author,
|
||||||
|
notification['arguments'],
|
||||||
|
notification['note_obj']
|
||||||
|
)
|
||||||
|
case 'signup':
|
||||||
|
res = do_signup(author)
|
||||||
|
case 'delete_account':
|
||||||
|
res = delete_account(author)
|
||||||
|
case 'confirm_delete':
|
||||||
|
res = confirm_delete(author)
|
||||||
|
case _:
|
||||||
|
pass
|
||||||
|
# if not user_is_administrator:
|
||||||
|
return res
|
||||||
|
|
||||||
|
# Administrator commands go here
|
61
db.py
61
db.py
|
@ -1,61 +0,0 @@
|
||||||
import sqlite3
|
|
||||||
|
|
||||||
# Connect to SQLite database (or create it if it doesn't exist)
|
|
||||||
conn = sqlite3.connect('gacha_game.db')
|
|
||||||
cursor = conn.cursor()
|
|
||||||
|
|
||||||
# Create tables
|
|
||||||
cursor.execute('''
|
|
||||||
CREATE TABLE IF NOT EXISTS users (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
username TEXT UNIQUE NOT NULL,
|
|
||||||
has_rolled BOOLEAN NOT NULL DEFAULT 0
|
|
||||||
)
|
|
||||||
''')
|
|
||||||
|
|
||||||
cursor.execute('''
|
|
||||||
CREATE TABLE IF NOT EXISTS characters (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
name TEXT NOT NULL,
|
|
||||||
rarity INTEGER NOT NULL,
|
|
||||||
weight REAL NOT NULL,
|
|
||||||
file_id TEXT NOT NULL
|
|
||||||
)
|
|
||||||
''')
|
|
||||||
|
|
||||||
cursor.execute('''
|
|
||||||
CREATE TABLE IF NOT EXISTS pulls (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
user_id INTEGER,
|
|
||||||
character_id INTEGER,
|
|
||||||
timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
|
||||||
FOREIGN KEY (user_id) REFERENCES users(id),
|
|
||||||
FOREIGN KEY (character_id) REFERENCES characters(id)
|
|
||||||
)
|
|
||||||
''')
|
|
||||||
|
|
||||||
cursor.execute("""
|
|
||||||
CREATE TABLE IF NOT EXISTS config (
|
|
||||||
key TEXT PRIMARY KEY,
|
|
||||||
value TEXT
|
|
||||||
)
|
|
||||||
""")
|
|
||||||
|
|
||||||
""" # Insert example characters into the database if they don't already exist
|
|
||||||
characters = [
|
|
||||||
('Murakami-san', 1, 0.35),
|
|
||||||
('Mastodon-kun', 2, 0.25),
|
|
||||||
('Pleroma-tan', 3, 0.2),
|
|
||||||
('Misskey-tan', 4, 0.15),
|
|
||||||
('Syuilo-mama', 5, 0.05)
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
cursor.executemany('''
|
|
||||||
INSERT OR IGNORE INTO characters (name, rarity, weight) VALUES (?, ?, ?)
|
|
||||||
''', characters)
|
|
||||||
"""
|
|
||||||
|
|
||||||
# Commit changes and close
|
|
||||||
conn.commit()
|
|
||||||
conn.close()
|
|
24
docs/index.md
Normal file
24
docs/index.md
Normal file
|
@ -0,0 +1,24 @@
|
||||||
|
# 🎲 Kemoverse Documentation
|
||||||
|
|
||||||
|
Welcome to the developer documentation for **Kemoverse**, a gacha trading card game in the Fediverse!
|
||||||
|
|
||||||
|
Features collectible cards, rarity-based pulls, and integration with Misskey.
|
||||||
|
|
||||||
|
Name comes from Kemonomimi and Fediverse.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📁 Table of Contents
|
||||||
|
|
||||||
|
- [Installation](./install.md)
|
||||||
|
- [Game Design](./design.md)
|
||||||
|
- [Bot Architecture](./bot.md)
|
||||||
|
- [Database Structure](./database.md)
|
||||||
|
- [Card System](./cards.md)
|
||||||
|
- [Web UI](./web.md)
|
||||||
|
- [Theming and Assets](./theme.md)
|
||||||
|
- [Contributing](./contributing.md)
|
||||||
|
- [FAQ](./faq.md)
|
||||||
|
|
||||||
|
---
|
||||||
|
|
83
docs/install.md
Normal file
83
docs/install.md
Normal file
|
@ -0,0 +1,83 @@
|
||||||
|
|
||||||
|
## 🧪 Installation
|
||||||
|
|
||||||
|
### Download and install dependencies
|
||||||
|
|
||||||
|
Clone the repo
|
||||||
|
|
||||||
|
```sh
|
||||||
|
git clone https://git.waifuism.life/waifu/kemoverse.git
|
||||||
|
cd kemoverse
|
||||||
|
```
|
||||||
|
|
||||||
|
Setup a virtual environment (Optional, recommended)
|
||||||
|
|
||||||
|
```sh
|
||||||
|
python3 -m venv venv
|
||||||
|
source venv/bin/activate
|
||||||
|
```
|
||||||
|
|
||||||
|
Install project dependencies via pip
|
||||||
|
|
||||||
|
```sh
|
||||||
|
python3 -m pip install -r requirements.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
### Setup config file
|
||||||
|
|
||||||
|
A sample config file is included with the project as a template: `example_config.ini`
|
||||||
|
|
||||||
|
Create a copy of this file and replace its' values with your own. Consult the
|
||||||
|
template for more information about individual config values and their meaning.
|
||||||
|
|
||||||
|
Config files are environment-specific. Use `config_dev.ini` for development and
|
||||||
|
`config_prod.ini` for production. Switch between environments using the
|
||||||
|
`KEMOVERSE_ENV` environment variable.
|
||||||
|
|
||||||
|
```sh
|
||||||
|
cp example_config.ini config_dev.ini
|
||||||
|
# Edit config_dev.ini
|
||||||
|
```
|
||||||
|
|
||||||
|
### Setup database
|
||||||
|
|
||||||
|
To set up the database, run:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
KEMOVERSE_ENV=dev python3 setup_db.py
|
||||||
|
```
|
||||||
|
|
||||||
|
### Run the bot
|
||||||
|
|
||||||
|
```sh
|
||||||
|
KEMOVERSE_ENV=dev ./startup.sh
|
||||||
|
```
|
||||||
|
|
||||||
|
If all goes well, you should now be able to interact with the bot.
|
||||||
|
|
||||||
|
### Running in production
|
||||||
|
|
||||||
|
To run the the in a production environment, use `KEMOVERSE_ENV=prod`. You will
|
||||||
|
also need to create a `config_prod.ini` file and run the database setup step
|
||||||
|
again if pointing prod to a different database. (you are pointing dev and prod
|
||||||
|
to different databases, right? 🤨)
|
||||||
|
|
||||||
|
### Updating
|
||||||
|
|
||||||
|
To update the bot, first pull new changes from upstream:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
git pull
|
||||||
|
```
|
||||||
|
|
||||||
|
Then run any database migrations. We recommend testing in dev beforehand to
|
||||||
|
make sure nothing breaks in the update process.
|
||||||
|
|
||||||
|
**Always backup your prod database before running any migrations!**
|
||||||
|
|
||||||
|
```sh
|
||||||
|
# Backup database file
|
||||||
|
cp gacha_game_dev.db gacha_game_dev.db.bak
|
||||||
|
# Run migrations
|
||||||
|
KEMOVERSE_ENV=dev python3 setup_db.py
|
||||||
|
```
|
33
docs/theme.md
Normal file
33
docs/theme.md
Normal file
|
@ -0,0 +1,33 @@
|
||||||
|
Welcome to the **Visual Identity** guide for the Kemoverse. This page contains the standard colors, logos, and graphic elements used across the game (cards, UI, web presence, bots, etc). Please follow these guidelines to ensure consistency.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🟢 Primary Color Palette
|
||||||
|
|
||||||
|
| Color Name | Hex Code | Usage |
|
||||||
|
|----------------|------------|--------------------------------------|
|
||||||
|
| Green | `#5aa02c` | Main buttons, links, headers |
|
||||||
|
| Midnight Black | `#1A1A1A` | Backgrounds, dark mode |
|
||||||
|
| Misty White | `#FAFAFA` | Default backgrounds, light text bg |
|
||||||
|
| Soft Gray | `#CCCCCC` | Borders, placeholders, separators |
|
||||||
|
| Highlight Green | `#8dd35f` | Alerts, emphasis, icons |
|
||||||
|
| Rarity Gold | `#FFD700` | Special rare cards, SSR outlines |
|
||||||
|
| Rarity Silver | `#C0C0C0` | Rare card text, stat glow effects |
|
||||||
|
|
||||||
|
> ✅ Use `Green` and `Misty White` for the standard UI. Avoid mixing in extra palettes unless explicitly needed.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🖼 Logos
|
||||||
|
|
||||||
|
### Main Logo
|
||||||
|
|
||||||
|
<p align="center">
|
||||||
|
<img src="../web/static/logo.png" alt="Fediverse Gacha Bot Logo" width="300" height="auto">
|
||||||
|
</p>
|
||||||
|
|
||||||
|
- File: `web/static/logo.png`
|
||||||
|
- Usage: Website header, favicon, bot avatar, watermark
|
||||||
|
|
||||||
|
|
||||||
|
---
|
|
@ -1,17 +1,44 @@
|
||||||
; Rename me to config.ini and put your values in here
|
; Rename me to config.ini and put your values in here
|
||||||
[application]
|
[application]
|
||||||
; Full fedi handle of the bot user
|
|
||||||
BotUser = @bot@example.tld
|
|
||||||
|
|
||||||
; API key for the bot
|
|
||||||
; Generate one by going to Settings > API > Generate access token
|
|
||||||
ApiKey = abcdefghijklmnopqrstuvwxyz012345
|
|
||||||
|
|
||||||
; Fully qualified URL of the instance hosting the bot
|
|
||||||
InstanceUrl = http://example.tld
|
|
||||||
|
|
||||||
; Comma separated list of fedi handles for any administrator users
|
; Comma separated list of fedi handles for any administrator users
|
||||||
|
; More can be added through the application
|
||||||
DefaultAdmins = ['admin@example.tld']
|
DefaultAdmins = ['admin@example.tld']
|
||||||
|
|
||||||
; SQLite Database location
|
; SQLite Database location
|
||||||
DatabaseLocation = ./gacha_game.db
|
DatabaseLocation = ./gacha_game.db
|
||||||
|
|
||||||
|
[gacha]
|
||||||
|
; Number of seconds players have to wait between rolls
|
||||||
|
RollInterval = 72000
|
||||||
|
; Rarity drop weights (1 to 5 stars)
|
||||||
|
; Format: rarity=weight per line
|
||||||
|
; In order: common, uncommon, rare, epic and legendary (Example values below)
|
||||||
|
Rarity_1 = 0.7
|
||||||
|
Rarity_2 = 0.2
|
||||||
|
Rarity_3 = 0.08
|
||||||
|
Rarity_4 = 0.015
|
||||||
|
Rarity_5 = 0.005
|
||||||
|
|
||||||
|
[notification]
|
||||||
|
; Number of seconds to sleep while awaiting new notifications
|
||||||
|
PollInterval = 5
|
||||||
|
; Number of notifications to process at once (max 100)
|
||||||
|
BatchSize = 10
|
||||||
|
|
||||||
|
[credentials]
|
||||||
|
; Fully qualified URL of the instance hosting the bot
|
||||||
|
Instance = http://example.tld
|
||||||
|
; Full fedi handle of the bot user
|
||||||
|
User = @bot@example.tld
|
||||||
|
; API key for the bot
|
||||||
|
; Generate one by going to Settings > API > Generate access token
|
||||||
|
Token = abcdefghijklmnopqrstuvwxyz012345
|
||||||
|
|
||||||
|
; Instance type - either "misskey" or "pleroma"
|
||||||
|
InstanceType = misskey
|
||||||
|
|
||||||
|
; Web server port (default: 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 =
|
||||||
|
|
28
migrations/0000_setup.sql
Normal file
28
migrations/0000_setup.sql
Normal file
|
@ -0,0 +1,28 @@
|
||||||
|
CREATE TABLE IF NOT EXISTS users (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
username TEXT UNIQUE NOT NULL,
|
||||||
|
has_rolled BOOLEAN NOT NULL DEFAULT 0
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS characters (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
name TEXT NOT NULL,
|
||||||
|
rarity INTEGER NOT NULL,
|
||||||
|
weight REAL NOT NULL,
|
||||||
|
file_id TEXT NOT NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS pulls (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
user_id INTEGER,
|
||||||
|
character_id INTEGER,
|
||||||
|
timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
FOREIGN KEY (user_id) REFERENCES users(id),
|
||||||
|
FOREIGN KEY (character_id) REFERENCES characters(id)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS config (
|
||||||
|
key TEXT PRIMARY KEY,
|
||||||
|
value TEXT
|
||||||
|
);
|
||||||
|
INSERT OR IGNORE INTO config VALUES ("schema_version", 0);
|
1
migrations/0001_fix_notif_id.sql
Normal file
1
migrations/0001_fix_notif_id.sql
Normal file
|
@ -0,0 +1 @@
|
||||||
|
INSERT OR IGNORE INTO config VALUES ("last_seen_notif_id", 0);
|
1
migrations/0002_weigh_infer.sql
Normal file
1
migrations/0002_weigh_infer.sql
Normal file
|
@ -0,0 +1 @@
|
||||||
|
ALTER TABLE characters DROP COLUMN weight;
|
59
readme.md
59
readme.md
|
@ -1,47 +1,59 @@
|
||||||
# Kemoverse
|
# Kemoverse
|
||||||
|
|
||||||
A gacha-style bot for the Fediverse built with Python. Users can roll for characters, trade, duel, and perhaps engage with popularity-based mechanics. Currently designed for use with Misskey. Name comes from Kemonomimi and Fediverse.
|
A gacha-style bot for the Fediverse built with Python. Users can roll for characters, trade, duel, and perhaps engage with popularity-based mechanics. Supports both Misskey and Pleroma instances. Name comes from Kemonomimi and Fediverse.
|
||||||
|
<p align="center">
|
||||||
|
<img src="./web/static/logo.png" alt="Fediverse Gacha Bot Logo" width="300" height="auto">
|
||||||
|
</p>
|
||||||
|
|
||||||

|
## 📝 Docs
|
||||||
|
|
||||||
|
👉 [**Start reading the docs**](./docs/index.md)
|
||||||
|
|
||||||
|
🤌 [**Install instructions for those in a rush**](docs/install.md)
|
||||||
|
|
||||||
## 🔧 Features
|
## 🔧 Features
|
||||||
|
|
||||||
### ✅ Implemented
|
### ✅ Implemented
|
||||||
- 🎲 Character roll system
|
- 🎲 Character roll system
|
||||||
- 🎴 Cards stats system
|
- 🧠 Core database structure for cards
|
||||||
- 🧠 Core database structure for characters and stats
|
- 📦 Basic support for storing pulls per player
|
||||||
- 📦 Basic support for storing pulls per user
|
- ⏱️ Time-based limitations on rolls
|
||||||
|
- ⚠️ Explicit account creation/deletion
|
||||||
|
|
||||||
### 🧩 In Progress
|
### 🧩 In Progress
|
||||||
- 📝 Whitelist system to limit access
|
- 📝 Whitelist system to limit access
|
||||||
- ⏱️ Time-based limitations on rolls
|
|
||||||
- ⚔️ Dueling system
|
|
||||||
|
|
||||||
## 🧠 Planned Features (Long Term)
|
|
||||||
|
## 🧠 Roadmap
|
||||||
|
|
||||||
|
[See our v2.0 board for more details](https://git.waifuism.life/waifu/kemoverse/projects/3)
|
||||||
|
|
||||||
### 🛒 Gameplay & Collection
|
### 🛒 Gameplay & Collection
|
||||||
- 🔁 **Trading system** between users
|
- 🔁 **Trading system** between players
|
||||||
- ⭐ **Favorite characters** (pin them or set profiles)
|
- ⭐ **Favorite characters** (pin them or set profiles)
|
||||||
- 📢 **Public post announcements** for rare card pulls
|
- 📢 **Public post announcements** for rare card pulls
|
||||||
|
- 📊 **Stats** for cards
|
||||||
|
- 🎮 **Games** to play
|
||||||
|
- ⚔️ Dueling
|
||||||
- 🧮 **Leaderboards**
|
- 🧮 **Leaderboards**
|
||||||
- Most traded Characters
|
- Most traded cards
|
||||||
- Most owned Characters
|
- Most owned cards
|
||||||
- Most voted Characters
|
- Most voted cards
|
||||||
- Most popular Characters (via usage-based popularity metrics)
|
- Most popular cards (via usage-based popularity metrics)
|
||||||
- Users with the rarest Characters
|
- Users with the rarest cards
|
||||||
|
|
||||||
### 🎨 Card Aesthetics
|
### 🎨 Card Aesthetics
|
||||||
- 🖼️ Simple card template for character rendering
|
- 🖼️ Simple card template for character rendering
|
||||||
- 🌐 Web app to generate cards from images
|
- 🌐 Web app to generate cards from images
|
||||||
|
|
||||||
### 🌍 Fediverse Support
|
### 🌍 Fediverse Support
|
||||||
✅ Anyone from the fediverse can play, but the server only works using a Misskey instance. Want to rewrite the program in Elixir for Pleroma? Let us know!
|
✅ Anyone from the fediverse can play! The bot supports both Misskey and Pleroma instances through configurable backends.
|
||||||
|
|
||||||
## 🗃️ Tech Stack
|
## 🗃️ Tech Stack
|
||||||
|
|
||||||
- Python (3.11+)
|
- Python (3.12+)
|
||||||
- SQLite
|
- SQLite
|
||||||
- Fediverse API integration (via Misskey endpoints)
|
- Fediverse API integration (Misskey and Pleroma support)
|
||||||
- Flask
|
- Flask
|
||||||
- Modular DB design for extensibility
|
- Modular DB design for extensibility
|
||||||
|
|
||||||
|
@ -49,21 +61,16 @@ A gacha-style bot for the Fediverse built with Python. Users can roll for charac
|
||||||
|
|
||||||
The bot is meant to feel *light, fun, and competitive*. Mixing social, gacha and duel tactics.
|
The bot is meant to feel *light, fun, and competitive*. Mixing social, gacha and duel tactics.
|
||||||
|
|
||||||
## 🧪 Getting Started (coming soon)
|
|
||||||
|
|
||||||
Instructions on installing dependencies, initializing the database, and running the bot locally will go here.
|
|
||||||
|
|
||||||
|
|
||||||
```mermaid
|
```mermaid
|
||||||
flowchart TD
|
flowchart TD
|
||||||
|
|
||||||
subgraph Player Interaction
|
subgraph Player Interaction
|
||||||
A1[Misskey bot]
|
A1[Fediverse bot]
|
||||||
A2[Web]
|
A2[Web]
|
||||||
end
|
end
|
||||||
|
|
||||||
subgraph Misskey
|
subgraph Fediverse
|
||||||
B1[Misskey instance]
|
B1[Fediverse instance]
|
||||||
end
|
end
|
||||||
|
|
||||||
subgraph Bot
|
subgraph Bot
|
||||||
|
@ -71,7 +78,7 @@ flowchart TD
|
||||||
C2[Notification parser]
|
C2[Notification parser]
|
||||||
C3[Gacha roll logic]
|
C3[Gacha roll logic]
|
||||||
C4[Database interface]
|
C4[Database interface]
|
||||||
C5[Misskey API poster]
|
C5[Fediverse API poster]
|
||||||
end
|
end
|
||||||
|
|
||||||
subgraph Website
|
subgraph Website
|
||||||
|
|
|
@ -6,3 +6,5 @@ Jinja2==3.1.6
|
||||||
MarkupSafe==3.0.2
|
MarkupSafe==3.0.2
|
||||||
Werkzeug==3.1.3
|
Werkzeug==3.1.3
|
||||||
Misskey.py==4.1.0
|
Misskey.py==4.1.0
|
||||||
|
Mastodon.py==1.8.1
|
||||||
|
filetype==1.2.0
|
||||||
|
|
129
setup_db.py
Normal file
129
setup_db.py
Normal file
|
@ -0,0 +1,129 @@
|
||||||
|
import sqlite3
|
||||||
|
import traceback
|
||||||
|
import os
|
||||||
|
import argparse
|
||||||
|
from configparser import ConfigParser
|
||||||
|
from typing import List, Tuple
|
||||||
|
|
||||||
|
class DBNotFoundError(Exception):
|
||||||
|
'''Could not find the database location'''
|
||||||
|
|
||||||
|
class InvalidMigrationError(Exception):
|
||||||
|
'''Migration file has an invalid name'''
|
||||||
|
|
||||||
|
class KemoverseEnvUnset(Exception):
|
||||||
|
'''KEMOVERSE_ENV is not set or has an invalid value'''
|
||||||
|
|
||||||
|
class ConfigError(Exception):
|
||||||
|
'''Could not find the config file for the current environment'''
|
||||||
|
|
||||||
|
def get_migrations() -> List[Tuple[int, str]] | InvalidMigrationError:
|
||||||
|
'''Returns a list of migration files in numeric order.'''
|
||||||
|
# Store transaction id and filename separately
|
||||||
|
sql_files: List[Tuple[int, str]] = []
|
||||||
|
migrations_dir = 'migrations'
|
||||||
|
|
||||||
|
for filename in os.listdir(migrations_dir):
|
||||||
|
joined_path = os.path.join(migrations_dir, filename)
|
||||||
|
|
||||||
|
# Ignore anything that isn't a .sql file
|
||||||
|
if not (os.path.isfile(joined_path) and filename.endswith('.sql')):
|
||||||
|
print(f'{filename} is not a .sql file, ignoring...')
|
||||||
|
continue
|
||||||
|
|
||||||
|
parts = filename.split('_', 1)
|
||||||
|
|
||||||
|
# Invalid filename format
|
||||||
|
if len(parts) < 2 or not parts[0].isdigit():
|
||||||
|
raise InvalidMigrationError(f'Invalid migration file: {filename}')
|
||||||
|
|
||||||
|
sql_files.append((int(parts[0]), joined_path))
|
||||||
|
|
||||||
|
# Get sorted list of files by migration number
|
||||||
|
sql_files.sort(key=lambda x: x[0])
|
||||||
|
return sql_files
|
||||||
|
|
||||||
|
def perform_migration(cursor: sqlite3.Cursor, migration: tuple[int, str]) -> None:
|
||||||
|
'''Performs a migration on the DB'''
|
||||||
|
print(f'Performing migration {migration[1]}...')
|
||||||
|
|
||||||
|
# Open and execute the sql script
|
||||||
|
with open(migration[1], encoding='utf-8') as file:
|
||||||
|
script = file.read()
|
||||||
|
cursor.executescript(script)
|
||||||
|
# Update the schema version
|
||||||
|
cursor.execute('UPDATE config SET value = ? WHERE key = "schema_version"', (migration[0],))
|
||||||
|
|
||||||
|
def get_db_path() -> str | DBNotFoundError:
|
||||||
|
'''Gets the DB path from config.ini'''
|
||||||
|
env = os.environ.get('KEMOVERSE_ENV')
|
||||||
|
if not (env and env in ['prod', 'dev']):
|
||||||
|
raise KemoverseEnvUnset
|
||||||
|
|
||||||
|
print(f'Running in "{env}" mode')
|
||||||
|
|
||||||
|
config_path = f'config_{env}.ini'
|
||||||
|
|
||||||
|
if not os.path.isfile(config_path):
|
||||||
|
raise ConfigError(f'Could not find {config_path}')
|
||||||
|
|
||||||
|
config = ConfigParser()
|
||||||
|
config.read(config_path)
|
||||||
|
db_path = config['application']['DatabaseLocation']
|
||||||
|
if not db_path:
|
||||||
|
raise DBNotFoundError()
|
||||||
|
return db_path
|
||||||
|
|
||||||
|
def get_current_migration(cursor: sqlite3.Cursor) -> int:
|
||||||
|
'''Gets the current schema version of the database'''
|
||||||
|
try:
|
||||||
|
cursor.execute('SELECT value FROM config WHERE key = ?', ('schema_version',))
|
||||||
|
version = cursor.fetchone()
|
||||||
|
return -1 if not version else int(version[0])
|
||||||
|
except sqlite3.Error:
|
||||||
|
print('Error getting schema version')
|
||||||
|
# Database has not been initialized yet
|
||||||
|
return -1
|
||||||
|
|
||||||
|
def main():
|
||||||
|
'''Does the thing'''
|
||||||
|
# Connect to the DB
|
||||||
|
db_path = ''
|
||||||
|
try:
|
||||||
|
db_path = get_db_path()
|
||||||
|
except ConfigError as ex:
|
||||||
|
print(ex)
|
||||||
|
return
|
||||||
|
except KemoverseEnvUnset:
|
||||||
|
print('Error: KEMOVERSE_ENV is either not set or has an invalid value.')
|
||||||
|
print('Please set KEMOVERSE_ENV to either "dev" or "prod" before running.')
|
||||||
|
print(traceback.format_exc())
|
||||||
|
return
|
||||||
|
|
||||||
|
conn = sqlite3.connect(db_path, autocommit=False)
|
||||||
|
conn.row_factory = sqlite3.Row
|
||||||
|
cursor = conn.cursor()
|
||||||
|
|
||||||
|
# Obtain list of migrations to run
|
||||||
|
migrations = get_migrations()
|
||||||
|
# Determine schema version
|
||||||
|
current_migration = get_current_migration(cursor)
|
||||||
|
print(f'Current schema version: {current_migration}')
|
||||||
|
|
||||||
|
# Run any migrations newer than current schema
|
||||||
|
for migration in migrations:
|
||||||
|
if migration[0] <= current_migration:
|
||||||
|
print(f'Migration already up: {migration[1]}')
|
||||||
|
continue
|
||||||
|
try:
|
||||||
|
perform_migration(cursor, migration)
|
||||||
|
conn.commit()
|
||||||
|
except Exception as ex:
|
||||||
|
print(f'An error occurred while applying {migration[1]}: {ex}, aborting...')
|
||||||
|
print(traceback.format_exc())
|
||||||
|
conn.rollback()
|
||||||
|
break
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
24
web/app.py
24
web/app.py
|
@ -1,14 +1,30 @@
|
||||||
from flask import Flask, render_template
|
|
||||||
import sqlite3
|
import sqlite3
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
|
||||||
|
# Add bot directory to path to import config
|
||||||
|
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'bot'))
|
||||||
|
import config
|
||||||
|
|
||||||
|
from flask import Flask, render_template, abort
|
||||||
|
from werkzeug.exceptions import HTTPException
|
||||||
|
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
DB_PATH = "./gacha_game.db" # Adjust path if needed
|
DB_PATH = config.DB_PATH
|
||||||
|
|
||||||
def get_db_connection():
|
def get_db_connection():
|
||||||
conn = sqlite3.connect(DB_PATH)
|
conn = sqlite3.connect(DB_PATH)
|
||||||
conn.row_factory = sqlite3.Row
|
conn.row_factory = sqlite3.Row
|
||||||
return conn
|
return conn
|
||||||
|
|
||||||
|
@app.errorhandler(HTTPException)
|
||||||
|
def handle_exception(error):
|
||||||
|
return render_template("_error.html", error=error), error.code
|
||||||
|
|
||||||
|
@app.route("/i404")
|
||||||
|
def i404():
|
||||||
|
return abort(404)
|
||||||
|
|
||||||
@app.route('/')
|
@app.route('/')
|
||||||
def index():
|
def index():
|
||||||
conn = get_db_connection()
|
conn = get_db_connection()
|
||||||
|
@ -33,6 +49,8 @@ def user_profile(user_id):
|
||||||
|
|
||||||
cursor.execute('SELECT * FROM users WHERE id = ?', (user_id,))
|
cursor.execute('SELECT * FROM users WHERE id = ?', (user_id,))
|
||||||
user = cursor.fetchone()
|
user = cursor.fetchone()
|
||||||
|
if user is None:
|
||||||
|
abort(404)
|
||||||
|
|
||||||
cursor.execute('''
|
cursor.execute('''
|
||||||
SELECT pulls.timestamp, characters.name as character_name, characters.rarity
|
SELECT pulls.timestamp, characters.name as character_name, characters.rarity
|
||||||
|
@ -56,4 +74,4 @@ def submit_character():
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
app.run(host='0.0.0.0', port=5000, debug=True)
|
app.run(host='0.0.0.0', port=config.WEB_PORT, debug=True)
|
||||||
|
|
92
web/static/style.css
Normal file
92
web/static/style.css
Normal file
|
@ -0,0 +1,92 @@
|
||||||
|
body {
|
||||||
|
font-family: "Segoe UI", Tahoma, Geneva, Verdana, sans-serif;
|
||||||
|
background-color: #f4f6fa;
|
||||||
|
color: #333;
|
||||||
|
margin: 0;
|
||||||
|
padding: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
header {
|
||||||
|
background-color: #7289da;
|
||||||
|
color: white;
|
||||||
|
padding: 20px;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
header h1 {
|
||||||
|
margin: 0;
|
||||||
|
font-size: 2.5em;
|
||||||
|
}
|
||||||
|
|
||||||
|
header p {
|
||||||
|
margin-top: 5px;
|
||||||
|
font-size: 1.1em;
|
||||||
|
}
|
||||||
|
|
||||||
|
.container {
|
||||||
|
max-width: 800px;
|
||||||
|
margin: 30px auto;
|
||||||
|
padding: 20px;
|
||||||
|
background-color: #ffffff;
|
||||||
|
border-radius: 10px;
|
||||||
|
box-shadow: 0 3px 10px rgba(0, 0, 0, 0.07);
|
||||||
|
}
|
||||||
|
|
||||||
|
h2 {
|
||||||
|
border-bottom: 1px solid #ccc;
|
||||||
|
padding-bottom: 8px;
|
||||||
|
margin-top: 30px;
|
||||||
|
}
|
||||||
|
|
||||||
|
ul {
|
||||||
|
list-style-type: none;
|
||||||
|
padding-left: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
li {
|
||||||
|
margin: 10px 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
a {
|
||||||
|
text-decoration: none;
|
||||||
|
color: #2c3e50;
|
||||||
|
font-weight: bold;
|
||||||
|
background-color: #e3eaf3;
|
||||||
|
padding: 8px 12px;
|
||||||
|
border-radius: 6px;
|
||||||
|
display: inline-block;
|
||||||
|
transition: background-color 0.2s;
|
||||||
|
}
|
||||||
|
|
||||||
|
a:hover {
|
||||||
|
background-color: #cdd8e6;
|
||||||
|
}
|
||||||
|
|
||||||
|
.leaderboard-entry {
|
||||||
|
margin-bottom: 8px;
|
||||||
|
padding: 6px 10px;
|
||||||
|
background: #f9fafc;
|
||||||
|
border-left: 4px solid #7289da;
|
||||||
|
border-radius: 5px;
|
||||||
|
}
|
||||||
|
|
||||||
|
footer {
|
||||||
|
text-align: center;
|
||||||
|
margin-top: 40px;
|
||||||
|
font-size: 0.9em;
|
||||||
|
color: #888;
|
||||||
|
}
|
||||||
|
|
||||||
|
.note {
|
||||||
|
background: #fcfcf0;
|
||||||
|
border: 1px dashed #bbb;
|
||||||
|
padding: 10px;
|
||||||
|
border-radius: 8px;
|
||||||
|
margin-top: 30px;
|
||||||
|
font-size: 0.95em;
|
||||||
|
color: #666;
|
||||||
|
}
|
||||||
|
|
||||||
|
.footer-link {
|
||||||
|
margin: 0 10px;
|
||||||
|
}
|
30
web/templates/_base.html
Normal file
30
web/templates/_base.html
Normal file
|
@ -0,0 +1,30 @@
|
||||||
|
<!DOCTYPE html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<link rel="stylesheet" type="text/css" href="{{ url_for('static', filename='style.css') }}">
|
||||||
|
<link rel="shortcut icon" href="{{ url_for('static', filename='logo.png') }}">
|
||||||
|
<title>
|
||||||
|
{% if title %}
|
||||||
|
{{ title }}
|
||||||
|
{% else %}
|
||||||
|
{% block title %}{% endblock %}
|
||||||
|
{% endif %}
|
||||||
|
| Kemoverse
|
||||||
|
</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<header>
|
||||||
|
{% block header %}{% endblock %}
|
||||||
|
</header>
|
||||||
|
|
||||||
|
<div class="container">
|
||||||
|
{% block content %}{% endblock %}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<footer>
|
||||||
|
<a class="footer-link" href="{{ url_for('about') }}">About</a>
|
||||||
|
<a class="footer-link" href="{{ url_for('submit_character') }}">Submit a Character</a>
|
||||||
|
{% block footer_extra %}{% endblock %}
|
||||||
|
</footer>
|
||||||
|
</body>
|
||||||
|
</html>
|
8
web/templates/_error.html
Normal file
8
web/templates/_error.html
Normal file
|
@ -0,0 +1,8 @@
|
||||||
|
{% extends "_base.html" %}
|
||||||
|
{% block title %}
|
||||||
|
{{ error.code }}
|
||||||
|
{% endblock %}
|
||||||
|
{% block content %}
|
||||||
|
<h2>{{ error.code }} - {{ error.name }} </h2>
|
||||||
|
<p>{{ error.description }}</p>
|
||||||
|
{% endblock %}
|
|
@ -1,13 +1,9 @@
|
||||||
<!DOCTYPE html>
|
{% extends "_base.html" %}
|
||||||
<html>
|
|
||||||
<head>
|
{% block content %}
|
||||||
<title>About - Misskey Gacha Center</title>
|
|
||||||
</head>
|
|
||||||
<body>
|
|
||||||
<h1>About This Gacha</h1>
|
<h1>About This Gacha</h1>
|
||||||
<p>This is a playful Misskey-themed gacha tracker made with Flask and SQLite.</p>
|
<p>This is a playful Misskey-themed gacha tracker made with Flask and SQLite.</p>
|
||||||
<p>All rolls are stored, stats are tracked, and characters are added manually for now.</p>
|
<p>All rolls are stored, stats are tracked, and characters are added manually for now.</p>
|
||||||
<p>Built with love, chaos, and way too much caffeine ☕.</p>
|
<p>Built with love, chaos, and way too much caffeine ☕.</p>
|
||||||
<a href="{{ url_for('index') }}">← Back to Home</a>
|
<a href="{{ url_for('index') }}">← Back to Home</a>
|
||||||
</body>
|
{% endblock %}
|
||||||
</html>
|
|
|
@ -1,110 +1,11 @@
|
||||||
<!DOCTYPE html>
|
{% extends "_base.html" %}
|
||||||
<html>
|
|
||||||
<head>
|
|
||||||
<title>Misskey Gacha Center</title>
|
|
||||||
<style>
|
|
||||||
body {
|
|
||||||
font-family: "Segoe UI", Tahoma, Geneva, Verdana, sans-serif;
|
|
||||||
background-color: #f4f6fa;
|
|
||||||
color: #333;
|
|
||||||
margin: 0;
|
|
||||||
padding: 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
header {
|
{% block header %}
|
||||||
background-color: #7289da;
|
|
||||||
color: white;
|
|
||||||
padding: 20px;
|
|
||||||
text-align: center;
|
|
||||||
}
|
|
||||||
|
|
||||||
header h1 {
|
|
||||||
margin: 0;
|
|
||||||
font-size: 2.5em;
|
|
||||||
}
|
|
||||||
|
|
||||||
header p {
|
|
||||||
margin-top: 5px;
|
|
||||||
font-size: 1.1em;
|
|
||||||
}
|
|
||||||
|
|
||||||
.container {
|
|
||||||
max-width: 800px;
|
|
||||||
margin: 30px auto;
|
|
||||||
padding: 20px;
|
|
||||||
background-color: #ffffff;
|
|
||||||
border-radius: 10px;
|
|
||||||
box-shadow: 0 3px 10px rgba(0, 0, 0, 0.07);
|
|
||||||
}
|
|
||||||
|
|
||||||
h2 {
|
|
||||||
border-bottom: 1px solid #ccc;
|
|
||||||
padding-bottom: 8px;
|
|
||||||
margin-top: 30px;
|
|
||||||
}
|
|
||||||
|
|
||||||
ul {
|
|
||||||
list-style-type: none;
|
|
||||||
padding-left: 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
li {
|
|
||||||
margin: 10px 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
a {
|
|
||||||
text-decoration: none;
|
|
||||||
color: #2c3e50;
|
|
||||||
font-weight: bold;
|
|
||||||
background-color: #e3eaf3;
|
|
||||||
padding: 8px 12px;
|
|
||||||
border-radius: 6px;
|
|
||||||
display: inline-block;
|
|
||||||
transition: background-color 0.2s;
|
|
||||||
}
|
|
||||||
|
|
||||||
a:hover {
|
|
||||||
background-color: #cdd8e6;
|
|
||||||
}
|
|
||||||
|
|
||||||
.leaderboard-entry {
|
|
||||||
margin-bottom: 8px;
|
|
||||||
padding: 6px 10px;
|
|
||||||
background: #f9fafc;
|
|
||||||
border-left: 4px solid #7289da;
|
|
||||||
border-radius: 5px;
|
|
||||||
}
|
|
||||||
|
|
||||||
footer {
|
|
||||||
text-align: center;
|
|
||||||
margin-top: 40px;
|
|
||||||
font-size: 0.9em;
|
|
||||||
color: #888;
|
|
||||||
}
|
|
||||||
|
|
||||||
.note {
|
|
||||||
background: #fcfcf0;
|
|
||||||
border: 1px dashed #bbb;
|
|
||||||
padding: 10px;
|
|
||||||
border-radius: 8px;
|
|
||||||
margin-top: 30px;
|
|
||||||
font-size: 0.95em;
|
|
||||||
color: #666;
|
|
||||||
}
|
|
||||||
|
|
||||||
.footer-link {
|
|
||||||
margin: 0 10px;
|
|
||||||
}
|
|
||||||
</style>
|
|
||||||
</head>
|
|
||||||
<body>
|
|
||||||
|
|
||||||
<header>
|
|
||||||
<h1>Misskey Gacha Center</h1>
|
<h1>Misskey Gacha Center</h1>
|
||||||
<p>Track your luck. Compare your pulls. Compete with friends.</p>
|
<p>Track your luck. Compare your pulls. Compete with friends.</p>
|
||||||
</header>
|
{% endblock %}
|
||||||
|
|
||||||
<div class="container">
|
{% block content %}
|
||||||
|
|
||||||
<h2>🎖️ Leaderboard: Most Rolls</h2>
|
<h2>🎖️ Leaderboard: Most Rolls</h2>
|
||||||
{% for user in top_users %}
|
{% for user in top_users %}
|
||||||
|
@ -125,13 +26,4 @@
|
||||||
<div class="note">
|
<div class="note">
|
||||||
🚀 This is a fun little gacha tracker! More features coming soon. Want to help shape it?
|
🚀 This is a fun little gacha tracker! More features coming soon. Want to help shape it?
|
||||||
</div>
|
</div>
|
||||||
|
{% endblock %}
|
||||||
</div>
|
|
||||||
|
|
||||||
<footer>
|
|
||||||
<a class="footer-link" href="{{ url_for('about') }}">About</a>
|
|
||||||
<a class="footer-link" href="{{ url_for('submit_character') }}">Submit a Character</a>
|
|
||||||
</footer>
|
|
||||||
|
|
||||||
</body>
|
|
||||||
</html>
|
|
|
@ -1,12 +1,8 @@
|
||||||
<!DOCTYPE html>
|
{% extends "_base.html" %}
|
||||||
<html>
|
|
||||||
<head>
|
{% block content %}
|
||||||
<title>Submit a Character - Misskey Gacha Center</title>
|
|
||||||
</head>
|
|
||||||
<body>
|
|
||||||
<h1>Submit a Character</h1>
|
<h1>Submit a Character</h1>
|
||||||
<p>Want to add a new character to the gacha pool?</p>
|
<p>Want to add a new character to the gacha pool?</p>
|
||||||
<p>This feature will be available soon. Stay tuned!</p>
|
<p>This feature will be available soon. Stay tuned!</p>
|
||||||
<a href="{{ url_for('index') }}">← Back to Home</a>
|
<a href="{{ url_for('index') }}">← Back to Home</a>
|
||||||
</body>
|
{% endblock %}
|
||||||
</html>
|
|
|
@ -1,57 +1,5 @@
|
||||||
<!DOCTYPE html>
|
{% extends "_base.html" %}
|
||||||
<html>
|
{% block content %}
|
||||||
<head>
|
|
||||||
<title>{{ user['username'] }}'s Rolls</title>
|
|
||||||
<style>
|
|
||||||
body {
|
|
||||||
font-family: Arial, sans-serif;
|
|
||||||
background-color: #f4f4f8;
|
|
||||||
margin: 0;
|
|
||||||
padding: 20px;
|
|
||||||
}
|
|
||||||
.profile, .pulls {
|
|
||||||
background-color: white;
|
|
||||||
padding: 15px;
|
|
||||||
border-radius: 10px;
|
|
||||||
box-shadow: 0 0 10px rgba(0,0,0,0.1);
|
|
||||||
margin-bottom: 20px;
|
|
||||||
}
|
|
||||||
h1, h2 {
|
|
||||||
margin-top: 0;
|
|
||||||
}
|
|
||||||
ul {
|
|
||||||
list-style-type: none;
|
|
||||||
padding: 0;
|
|
||||||
}
|
|
||||||
li {
|
|
||||||
padding: 10px 0;
|
|
||||||
border-bottom: 1px solid #eee;
|
|
||||||
}
|
|
||||||
.rarity {
|
|
||||||
color: gold;
|
|
||||||
font-weight: bold;
|
|
||||||
margin-left: 8px;
|
|
||||||
}
|
|
||||||
.timestamp {
|
|
||||||
color: #888;
|
|
||||||
font-size: 0.9em;
|
|
||||||
}
|
|
||||||
a {
|
|
||||||
display: inline-block;
|
|
||||||
margin-top: 20px;
|
|
||||||
color: #333;
|
|
||||||
text-decoration: none;
|
|
||||||
background-color: #ddd;
|
|
||||||
padding: 8px 12px;
|
|
||||||
border-radius: 5px;
|
|
||||||
}
|
|
||||||
a:hover {
|
|
||||||
background-color: #bbb;
|
|
||||||
}
|
|
||||||
</style>
|
|
||||||
</head>
|
|
||||||
<body>
|
|
||||||
|
|
||||||
<div class="profile">
|
<div class="profile">
|
||||||
<h1>{{ user['username'] }}'s Gacha Rolls</h1>
|
<h1>{{ user['username'] }}'s Gacha Rolls</h1>
|
||||||
<p>User ID: {{ user['id'] }}</p>
|
<p>User ID: {{ user['id'] }}</p>
|
||||||
|
@ -72,6 +20,4 @@
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<a href="{{ url_for('index') }}">← Back to Users</a>
|
<a href="{{ url_for('index') }}">← Back to Users</a>
|
||||||
|
{% endblock %}
|
||||||
</body>
|
|
||||||
</html>
|
|
Loading…
Add table
Reference in a new issue