pokken: add database tables

titles/pokken/database.py

@@ -1,7 +1,13 @@
 from core.data import Data
 from core.config import CoreConfig
 
+from .schema import *
 
 class PokkenData(Data):
     def __init__(self, cfg: CoreConfig) -> None:
         super().__init__(cfg)
+
+        self.profile = PokkenProfileData(cfg, self.session)
+        self.match = PokkenMatchData(cfg, self.session)
+        self.item = PokkenItemData(cfg, self.session)
+        self.static = PokkenStaticData(cfg, self.session)

titles/pokken/schema/__init__.py

@@ -0,0 +1,4 @@
+from .profile import PokkenProfileData
+from .match import PokkenMatchData
+from .item import PokkenItemData
+from .static import PokkenStaticData

titles/pokken/schema/item.py

@@ -0,0 +1,27 @@
+from typing import Optional, Dict, List
+from sqlalchemy import Table, Column, UniqueConstraint, PrimaryKeyConstraint, and_, case
+from sqlalchemy.types import Integer, String, TIMESTAMP, Boolean, JSON
+from sqlalchemy.schema import ForeignKey
+from sqlalchemy.sql import func, select, update, delete
+from sqlalchemy.engine import Row
+from sqlalchemy.dialects.mysql import insert
+
+from core.data.schema import BaseData, metadata
+
+item = Table(
+    'pokken_item',
+    metadata,
+    Column('id', Integer, primary_key=True, nullable=False),
+    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False, unique=True),
+    Column('category', Integer),
+    Column('content', Integer),
+    Column('type', Integer),
+    UniqueConstraint('user', 'category', 'content', 'type', name='pokken_item_uk'),
+    mysql_charset="utf8mb4",
+)
+
+class PokkenItemData(BaseData):
+    """
+    Items obtained as rewards
+    """
+    pass

titles/pokken/schema/match.py

@@ -0,0 +1,45 @@
+from typing import Optional, Dict, List
+from sqlalchemy import Table, Column, UniqueConstraint, PrimaryKeyConstraint, and_, case
+from sqlalchemy.types import Integer, String, TIMESTAMP, Boolean, JSON
+from sqlalchemy.schema import ForeignKey
+from sqlalchemy.sql import func, select, update, delete
+from sqlalchemy.engine import Row
+from sqlalchemy.dialects.mysql import insert
+
+from core.data.schema import BaseData, metadata
+
+# Pokken sends depressingly little match data...
+match_data = Table(
+    'pokken_match_data',
+    metadata,
+    Column('id', Integer, primary_key=True, nullable=False),
+    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column('play_mode', Integer),
+    Column('result', Integer),
+    Column('ex_ko_num', Integer),
+    Column('wko_num', Integer),
+    Column('timeup_win_num', Integer),
+    Column('cool_ko_num', Integer),
+    Column('perfect_ko_num', Integer),
+    Column('use_navi', Integer),
+    Column('use_navi_cloth', Integer),
+    Column('use_aid_skill', Integer),
+    Column('play_date', TIMESTAMP),
+    mysql_charset="utf8mb4",
+)
+
+class PokkenMatchData(BaseData):
+    """
+    Match logs
+    """
+    def save_match(self, user_id: int, match_data: Dict) -> Optional[int]:
+        pass
+
+    def get_match(self, match_id: int) -> Optional[Row]:
+        pass
+
+    def get_matches_by_user(self, user_id: int) -> Optional[List[Row]]:
+        pass
+
+    def get_matches(self, limit: int = 20) -> Optional[List[Row]]:
+        pass

titles/pokken/schema/profile.py

@@ -0,0 +1,192 @@
+from typing import Optional, Dict, List
+from sqlalchemy import Table, Column, UniqueConstraint, PrimaryKeyConstraint, and_, case
+from sqlalchemy.types import Integer, String, TIMESTAMP, Boolean, JSON
+from sqlalchemy.schema import ForeignKey
+from sqlalchemy.sql import func, select, update, delete
+from sqlalchemy.engine import Row
+from sqlalchemy.dialects.mysql import insert
+
+from core.data.schema import BaseData, metadata
+
+# Some more of the repeated fields could probably be their own tables, for now I just did the ones that made sense to me
+# Having the profile table be this massive kinda blows for updates but w/e, **kwargs to the rescue
+profile = Table(
+    'pokken_profile',
+    metadata,
+    Column('id', Integer, primary_key=True, nullable=False),
+    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False, unique=True),
+    Column('trainer_name', String(16)), # optional
+    Column('home_region_code', Integer),
+    Column('home_loc_name', String(255)),
+    Column('pref_code', Integer),
+    Column('navi_newbie_flag', Boolean),
+    Column('navi_enable_flag', Boolean),
+    Column('pad_vibrate_flag', Boolean),
+    Column('trainer_rank_point', Integer),
+    Column('wallet', Integer),
+    Column('fight_money', Integer),
+    Column('score_point', Integer),
+    Column('grade_max_num', Integer),
+    Column('extra_counter', Integer), # Optional
+    Column('total_play_days', Integer),
+    Column('play_date_time', Integer),
+    Column('lucky_box_fail_num', Integer),
+    Column('event_reward_get_flag', Integer),
+    Column('rank_pvp_all', Integer),
+    Column('rank_pvp_loc', Integer),
+    Column('rank_cpu_all', Integer),
+    Column('rank_cpu_loc', Integer),
+    Column('rank_event', Integer),
+    Column('awake_num', Integer),
+    Column('use_support_num', Integer),
+    Column('rankmatch_flag', Integer),
+    Column('rankmatch_max', Integer), # Optional
+    Column('rankmatch_success', Integer), # Optional
+    Column('beat_num', Integer), # Optional
+    Column('title_text_id', Integer),
+    Column('title_plate_id', Integer),
+    Column('title_decoration_id', Integer),
+    Column('support_pokemon_list', JSON), # Repeated, Integer
+    Column('support_set_1', JSON), # Repeated, Integer
+    Column('support_set_2', JSON), # Repeated, Integer
+    Column('support_set_3', JSON), # Repeated, Integer
+    Column('navi_trainer', Integer),
+    Column('navi_version_id', Integer),
+    Column('aid_skill_list', JSON), # Repeated, Integer
+    Column('aid_skill', Integer),
+    Column('comment_text_id', Integer),
+    Column('comment_word_id', Integer),
+    Column('latest_use_pokemon', Integer),
+    Column('ex_ko_num', Integer),
+    Column('wko_num', Integer),
+    Column('timeup_win_num', Integer),
+    Column('cool_ko_num', Integer),
+    Column('cool_ko_num', Integer),
+    Column('perfect_ko_num', Integer),
+    Column('record_flag', Integer),
+    Column('continue_num', Integer),
+    Column('avatar_body', Integer), # Optional
+    Column('avatar_gender', Integer), # Optional
+    Column('avatar_background', Integer), # Optional
+    Column('avatar_head', Integer), # Optional
+    Column('avatar_battleglass', Integer), # Optional
+    Column('avatar_face0', Integer), # Optional
+    Column('avatar_face1', Integer), # Optional
+    Column('avatar_face2', Integer), # Optional
+    Column('avatar_bodyall', Integer), # Optional
+    Column('avatar_wear', Integer), # Optional
+    Column('avatar_accessory', Integer), # Optional
+    Column('avatar_stamp', Integer), # Optional
+    Column('event_state', Integer),
+    Column('event_id', Integer),
+    Column('sp_bonus_category_id_1', Integer),
+    Column('sp_bonus_key_value_1', Integer),
+    Column('sp_bonus_category_id_2', Integer),
+    Column('sp_bonus_key_value_2', Integer),
+    Column('last_play_event_id', Integer), # Optional
+    mysql_charset="utf8mb4",
+)
+
+tutorial_progress = Table(
+    'pokken_tutorial_progress',
+    metadata,
+    Column('id', Integer, primary_key=True, nullable=False),
+    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column('flag', Integer),
+    UniqueConstraint('user', 'flag', name='pokken_tutorial_progress_uk'),
+    mysql_charset="utf8mb4",
+)
+
+rankmatch_progress = Table(
+    'pokken_rankmatch_progress',
+    metadata,
+    Column('id', Integer, primary_key=True, nullable=False),
+    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column('progress', Integer),
+    UniqueConstraint('user', 'progress', name='pokken_rankmatch_progress_uk'),
+    mysql_charset="utf8mb4",
+)
+
+achievement_flag = Table(
+    'pokken_achievement_flag',
+    metadata,
+    Column('id', Integer, primary_key=True, nullable=False),
+    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column('flag', Integer),
+    UniqueConstraint('user', 'flag', name='pokken_achievement_flag_uk'),
+    mysql_charset="utf8mb4",
+)
+
+event_achievement_flag = Table(
+    'pokken_event_achievement_flag',
+    metadata,
+    Column('id', Integer, primary_key=True, nullable=False),
+    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column('flag', Integer),
+    UniqueConstraint('user', 'flag', name='pokken_event_achievement_flag_uk'),
+    mysql_charset="utf8mb4",
+)
+
+event_achievement_param = Table(
+    'pokken_event_achievement_param',
+    metadata,
+    Column('id', Integer, primary_key=True, nullable=False),
+    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column('param', Integer),
+    UniqueConstraint('user', 'param', name='pokken_event_achievement_param_uk'),
+    mysql_charset="utf8mb4",
+)
+
+pokemon_data = Table(
+    'pokken_pokemon_data',
+    metadata,
+    Column('id', Integer, primary_key=True, nullable=False),
+    Column('user', ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
+    Column('char_id', Integer, nullable=False),
+    Column('illustration_book_no', Integer, nullable=False),
+    Column('pokemon_exp', Integer, nullable=False),
+    Column('battle_num_vs_wan', Integer, nullable=False),
+    Column('win_vs_wan', Integer, nullable=False),
+    Column('battle_num_vs_lan', Integer, nullable=False),
+    Column('win_vs_lan', Integer, nullable=False),
+    Column('battle_num_vs_cpu', Integer, nullable=False),
+    Column('win_cpu', Integer, nullable=False),
+    Column('battle_all_num_tutorial', Integer, nullable=False),
+    Column('battle_num_tutorial', Integer, nullable=False),
+    Column('bp_point_atk', Integer, nullable=False),
+    Column('bp_point_res', Integer, nullable=False),
+    Column('bp_point_def', Integer, nullable=False),
+    Column('bp_point_sp', Integer, nullable=False),
+)
+
+class PokkenProfileData(BaseData):
+    def create_profile(self, user_id: int) -> Optional[int]:
+        sql = insert(profile).values(user = user_id)
+        conflict = sql.on_duplicate_key_update(user = user_id)
+        
+        result = self.execute(conflict)
+        if result is None:
+            self.logger.error(f"Failed to create pokken profile for user {user_id}!")
+            return None
+        return result.lastrowid
+    
+    def set_profile_name(self, user_id: int, new_name: str) -> None:
+        sql = update(profile).where(profile.c.user == user_id).values(trainer_name = new_name)
+        result = self.execute(sql)
+        if result is None:
+            self.logger.error(f"Failed to update pokken profile name for user {user_id}!")
+    
+    def update_profile(self, user_id: int, profile_data: Dict) -> None:
+        """
+        TODO: Find out what comes in on the SaveUserRequestData protobuf and save it!
+        """
+        pass
+
+    def put_pokemon_data(self, user_id: int, pokemon_data: Dict) -> Optional[int]:
+        pass
+
+    def get_pokemon_data(self, user_id: int, pokemon_id: int) -> Optional[Row]:
+        pass
+
+    def get_all_pokemon_data(self, user_id: int) -> Optional[List[Row]]:
+        pass

titles/pokken/schema/static.py

@@ -0,0 +1,12 @@
+from typing import Optional, Dict, List
+from sqlalchemy import Table, Column, UniqueConstraint, PrimaryKeyConstraint, and_, case
+from sqlalchemy.types import Integer, String, TIMESTAMP, Boolean, JSON
+from sqlalchemy.schema import ForeignKey
+from sqlalchemy.sql import func, select, update, delete
+from sqlalchemy.engine import Row
+from sqlalchemy.dialects.mysql import insert
+
+from core.data.schema import BaseData, metadata
+
+class PokkenStaticData(BaseData):
+    pass