mirror of
https://gitea.tendokyu.moe/Hay1tsme/artemis.git
synced 2026-08-28 14:54:16 -05:00
Merge branch 'develop' into diva_handler_classes
This commit is contained in:
@@ -15,6 +15,7 @@ COPY dbutils.py dbutils.py
|
||||
COPY read.py read.py
|
||||
ADD core core
|
||||
ADD titles titles
|
||||
ADD config config
|
||||
ADD logs logs
|
||||
ADD cert cert
|
||||
|
||||
|
||||
12
changelog.md
12
changelog.md
@@ -8,6 +8,18 @@ Documenting updates to ARTEMiS, to be updated every time the master branch is pu
|
||||
### Card Maker
|
||||
+ Added support for maimai DX FESTiVAL PLUS
|
||||
|
||||
## 20231001
|
||||
### Initial D THE ARCADE
|
||||
+ Added support for Initial D THE ARCADE S2
|
||||
+ Story mode progress added
|
||||
+ Bunta Challenge/Touhou Project modes added
|
||||
+ Time Trials added
|
||||
+ Leaderboards added, but doesn't refresh sometimes
|
||||
+ Theory of Street mode added (with CPUs)
|
||||
+ Play Stamp/Timetrial events added
|
||||
+ Frontend to download profile added
|
||||
+ Importer to import profiles added
|
||||
|
||||
## 20230716
|
||||
### General
|
||||
+ Docker files added (#19)
|
||||
|
||||
22
core/data/schema/versions/SDDT_5_rollback.sql
Normal file
22
core/data/schema/versions/SDDT_5_rollback.sql
Normal file
@@ -0,0 +1,22 @@
|
||||
SET FOREIGN_KEY_CHECKS=0;
|
||||
|
||||
ALTER TABLE ongeki_user_event_point DROP COLUMN version;
|
||||
ALTER TABLE ongeki_user_event_point DROP COLUMN rank;
|
||||
ALTER TABLE ongeki_user_event_point DROP COLUMN type;
|
||||
ALTER TABLE ongeki_user_event_point DROP COLUMN date;
|
||||
|
||||
ALTER TABLE ongeki_user_tech_event DROP COLUMN version;
|
||||
|
||||
ALTER TABLE ongeki_user_mission_point DROP COLUMN version;
|
||||
|
||||
ALTER TABLE ongeki_static_events DROP COLUMN endDate;
|
||||
|
||||
DROP TABLE ongeki_tech_event_ranking;
|
||||
DROP TABLE ongeki_static_music_ranking_list;
|
||||
DROP TABLE ongeki_static_rewards;
|
||||
DROP TABLE ongeki_static_present_list;
|
||||
DROP TABLE ongeki_static_tech_music;
|
||||
DROP TABLE ongeki_static_client_testmode;
|
||||
DROP TABLE ongeki_static_game_point;
|
||||
|
||||
SET FOREIGN_KEY_CHECKS=1;
|
||||
98
core/data/schema/versions/SDDT_6_upgrade.sql
Normal file
98
core/data/schema/versions/SDDT_6_upgrade.sql
Normal file
@@ -0,0 +1,98 @@
|
||||
SET FOREIGN_KEY_CHECKS=0;
|
||||
|
||||
ALTER TABLE ongeki_user_event_point ADD COLUMN version INTEGER NOT NULL;
|
||||
ALTER TABLE ongeki_user_event_point ADD COLUMN rank INTEGER;
|
||||
ALTER TABLE ongeki_user_event_point ADD COLUMN type INTEGER NOT NULL;
|
||||
ALTER TABLE ongeki_user_event_point ADD COLUMN date VARCHAR(25);
|
||||
|
||||
ALTER TABLE ongeki_user_tech_event ADD COLUMN version INTEGER NOT NULL;
|
||||
|
||||
ALTER TABLE ongeki_user_mission_point ADD COLUMN version INTEGER NOT NULL;
|
||||
|
||||
ALTER TABLE ongeki_static_events ADD COLUMN endDate TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP;
|
||||
|
||||
CREATE TABLE ongeki_tech_event_ranking (
|
||||
id INT PRIMARY KEY NOT NULL AUTO_INCREMENT,
|
||||
user INT NOT NULL,
|
||||
version INT NOT NULL,
|
||||
date VARCHAR(25),
|
||||
eventId INT NOT NULL,
|
||||
rank INT,
|
||||
totalPlatinumScore INT NOT NULL,
|
||||
totalTechScore INT NOT NULL,
|
||||
UNIQUE KEY ongeki_tech_event_ranking_uk (user, eventId),
|
||||
CONSTRAINT ongeki_tech_event_ranking_ibfk1 FOREIGN KEY (user) REFERENCES aime_user(id) ON DELETE CASCADE ON UPDATE CASCADE
|
||||
);
|
||||
|
||||
CREATE TABLE ongeki_static_music_ranking_list (
|
||||
id INT PRIMARY KEY NOT NULL AUTO_INCREMENT,
|
||||
version INT NOT NULL,
|
||||
musicId INT NOT NULL,
|
||||
point INT NOT NULL,
|
||||
userName VARCHAR(255),
|
||||
UNIQUE KEY ongeki_static_music_ranking_list_uk (version, musicId)
|
||||
);
|
||||
|
||||
CREATE TABLE ongeki_static_rewards (
|
||||
id INT PRIMARY KEY NOT NULL AUTO_INCREMENT,
|
||||
version INT NOT NULL,
|
||||
rewardId INT NOT NULL,
|
||||
rewardName VARCHAR(255) NOT NULL,
|
||||
itemKind INT NOT NULL,
|
||||
itemId INT NOT NULL,
|
||||
UNIQUE KEY ongeki_tech_event_ranking_uk (version, rewardId)
|
||||
);
|
||||
|
||||
CREATE TABLE ongeki_static_present_list (
|
||||
id INT PRIMARY KEY NOT NULL AUTO_INCREMENT,
|
||||
version INT NOT NULL,
|
||||
presentId INT NOT NULL,
|
||||
presentName VARCHAR(255) NOT NULL,
|
||||
rewardId INT NOT NULL,
|
||||
stock INT NOT NULL,
|
||||
message VARCHAR(255),
|
||||
startDate VARCHAR(25) NOT NULL,
|
||||
endDate VARCHAR(25) NOT NULL,
|
||||
UNIQUE KEY ongeki_static_present_list_uk (version, presentId, rewardId)
|
||||
);
|
||||
|
||||
CREATE TABLE ongeki_static_tech_music (
|
||||
id INT PRIMARY KEY NOT NULL AUTO_INCREMENT,
|
||||
version INT NOT NULL,
|
||||
eventId INT NOT NULL,
|
||||
musicId INT NOT NULL,
|
||||
level INT NOT NULL,
|
||||
UNIQUE KEY ongeki_static_tech_music_uk (version, musicId, eventId)
|
||||
);
|
||||
|
||||
CREATE TABLE ongeki_static_client_testmode (
|
||||
id INT PRIMARY KEY NOT NULL AUTO_INCREMENT,
|
||||
regionId INT NOT NULL,
|
||||
placeId INT NOT NULL,
|
||||
clientId VARCHAR(11) NOT NULL,
|
||||
updateDate TIMESTAMP NOT NULL,
|
||||
isDelivery BOOLEAN NOT NULL,
|
||||
groupId INT NOT NULL,
|
||||
groupRole INT NOT NULL,
|
||||
continueMode INT NOT NULL,
|
||||
selectMusicTime INT NOT NULL,
|
||||
advertiseVolume INT NOT NULL,
|
||||
eventMode INT NOT NULL,
|
||||
eventMusicNum INT NOT NULL,
|
||||
patternGp INT NOT NULL,
|
||||
limitGp INT NOT NULL,
|
||||
maxLeverMovable INT NOT NULL,
|
||||
minLeverMovable INT NOT NULL,
|
||||
UNIQUE KEY ongeki_static_client_testmode_uk (clientId)
|
||||
);
|
||||
|
||||
CREATE TABLE ongeki_static_game_point (
|
||||
id INT PRIMARY KEY NOT NULL AUTO_INCREMENT,
|
||||
type INT NOT NULL,
|
||||
cost INT NOT NULL,
|
||||
startDate VARCHAR(25) NOT NULL DEFAULT "2000-01-01 05:00:00.0",
|
||||
endDate VARCHAR(25) NOT NULL DEFAULT "2099-01-01 05:00:00.0",
|
||||
UNIQUE KEY ongeki_static_game_point_uk (type)
|
||||
);
|
||||
|
||||
SET FOREIGN_KEY_CHECKS=1;
|
||||
@@ -10,11 +10,11 @@ This step-by-step guide assumes that you are using a fresh install of Windows 10
|
||||
3. Make sure that you enable "Create shortcuts for installed applications" and "Add Python to environment variables" and hit Install
|
||||
|
||||
## Install MySQL 8.0
|
||||
1. Download MySQL 8.0 Server : [Link](https://cdn.mysql.com//Downloads/MySQLInstaller/mysql-installer-web-community-8.0.31.0.msi)
|
||||
2. Install mysql-installer-web-community-8.0.31.0.msi
|
||||
1. Download MySQL 8.0 Server : [Link](https://dev.mysql.com/get/Downloads/MySQLInstaller/mysql-installer-community-8.0.34.0.msi)
|
||||
2. Install mysql-installer-web-community-8.0.34.0.msi
|
||||
1. Click on "Add ..." on the side
|
||||
2. Click on the "+" next to MySQL Servers
|
||||
3. Make sure MySQL Server 8.0.29 - X64 is under the products to be installed.
|
||||
3. Make sure MySQL Server 8.0.34 - X64 is under the products to be installed.
|
||||
4. Hit Next and Next once installed
|
||||
5. Select the configuration type "Development Computer"
|
||||
6. Hit Next
|
||||
@@ -23,9 +23,10 @@ This step-by-step guide assumes that you are using a fresh install of Windows 10
|
||||
9. Leave everything under Windows Service as default and hit Next >
|
||||
10. Click on Execute and for it to finish and hit Next> and then Finish
|
||||
3. Open MySQL 8.0 Command Line Client and login as your root user
|
||||
4. Type those commands to create your user and the database
|
||||
```
|
||||
CREATE USER 'aime'@'localhost' IDENTIFIED BY 'MyStrongPass.';
|
||||
4. Change `<Enter Password Here>` to a new password for the user aime, type those commands to create your user and the database
|
||||
|
||||
```sql
|
||||
CREATE USER 'aime'@'localhost' IDENTIFIED BY '<Enter Password Here>';
|
||||
CREATE DATABASE aime;
|
||||
GRANT Alter,Create,Delete,Drop,Index,Insert,References,Select,Update ON aime.* TO 'aime'@'localhost';
|
||||
FLUSH PRIVILEGES;
|
||||
@@ -34,33 +35,50 @@ exit;
|
||||
|
||||
## Install Python modules
|
||||
1. Change your work path to the artemis-master folder using 'cd' and install the requirements:
|
||||
> pip install -r requirements.txt
|
||||
|
||||
## Copy/Rename the folder example_config to config
|
||||
```shell
|
||||
pip install -r requirements.txt
|
||||
```
|
||||
|
||||
## Adjust /config/core.yaml
|
||||
## Copy/Rename the folder `example_config` to `config`
|
||||
|
||||
1. Make sure to change the server listen_address to be set to your local machine IP (ex.: 192.168.1.xxx)
|
||||
## Adjust `config/core.yaml`
|
||||
|
||||
1. Make sure to change the server `hostname` to be set to your local machine IP (ex.: 192.168.xxx.xxx)
|
||||
- In case you want to run this only locally, set the following values:
|
||||
```
|
||||
|
||||
```yaml
|
||||
server:
|
||||
listen_address: 0.0.0.0
|
||||
title:
|
||||
hostname: localhost
|
||||
hostname: 192.168.xxx.xxx
|
||||
```
|
||||
|
||||
1. Adjust the proper MySQL information you created earlier
|
||||
```yaml
|
||||
database:
|
||||
host: "localhost"
|
||||
username: "aime"
|
||||
password: "<Enter Password Here>"
|
||||
name: "aime"
|
||||
```
|
||||
2. Adjust the proper MySQL information you created earlier
|
||||
3. Add the AimeDB key at the bottom of the file
|
||||
4. If the webui is needed, change the flag from False to True
|
||||
|
||||
## Create the database tables for ARTEMiS
|
||||
> python dbutils.py create
|
||||
|
||||
```shell
|
||||
python dbutils.py create
|
||||
```
|
||||
|
||||
## Firewall Adjustements
|
||||
Make sure the following ports are open both on your router and local Windows firewall in case you want to use this for public use (NOT recommended):
|
||||
> Port 80 (TCP), 443 (TCP), 8443 (TCP), 22345 (TCP), 8080 (TCP), 8090 (TCP) **webui, 8444 (TCP) **mucha
|
||||
|
||||
## Running the ARTEMiS instance
|
||||
> python index.py
|
||||
```shell
|
||||
python index.py
|
||||
```
|
||||
|
||||
# Troubleshooting
|
||||
|
||||
@@ -78,6 +96,7 @@ Make sure the following ports are open both on your router and local Windows fir
|
||||
## AttributeError: module 'collections' has no attribute 'Hashable'
|
||||
1. This means the pyYAML module is obsolete, simply rerun pip with the -U (force update) flag, as shown below.
|
||||
- Change your work path to the artemis-master (or artemis-develop) folder using 'cd' and run the following commands:
|
||||
```
|
||||
|
||||
```shell
|
||||
pip install -r requirements.txt -U
|
||||
```
|
||||
|
||||
@@ -6,6 +6,12 @@ the corresponding importer and database upgrades.
|
||||
**Important: The described database upgrades are only required if you are using an old database schema, f.e. still
|
||||
using the megaime database. Clean installations always create the latest database structure!**
|
||||
|
||||
To upgrade the core database and the database for every game, execute:
|
||||
|
||||
```shell
|
||||
python dbutils.py autoupgrade
|
||||
```
|
||||
|
||||
# Table of content
|
||||
|
||||
- [Supported Games](#supported-games)
|
||||
@@ -16,6 +22,7 @@ using the megaime database. Clean installations always create the latest databas
|
||||
- [Card Maker](#card-maker)
|
||||
- [WACCA](#wacca)
|
||||
- [Sword Art Online Arcade](#sao)
|
||||
- [Initial D THE ARCADE](#initial-d-the-arcade)
|
||||
|
||||
|
||||
# Supported Games
|
||||
@@ -27,7 +34,7 @@ Games listed below have been tested and confirmed working.
|
||||
### SDBT
|
||||
|
||||
| Version ID | Version Name |
|
||||
|------------|-----------------------|
|
||||
| ---------- | --------------------- |
|
||||
| 0 | CHUNITHM |
|
||||
| 1 | CHUNITHM PLUS |
|
||||
| 2 | CHUNITHM AIR |
|
||||
@@ -43,7 +50,7 @@ Games listed below have been tested and confirmed working.
|
||||
### SDHD/SDBT
|
||||
|
||||
| Version ID | Version Name |
|
||||
|------------|---------------------|
|
||||
| ---------- | ------------------- |
|
||||
| 11 | CHUNITHM NEW!! |
|
||||
| 12 | CHUNITHM NEW PLUS!! |
|
||||
| 13 | CHUNITHM SUN |
|
||||
@@ -83,9 +90,7 @@ crypto:
|
||||
|
||||
### Database upgrade
|
||||
|
||||
Always make sure your database (tables) are up-to-date, to do so go to the `core/data/schema/versions` folder and see
|
||||
which version is the latest, f.e. `SDBT_4_upgrade.sql`. In order to upgrade to version 4 in this case you need to
|
||||
perform all previous updates as well:
|
||||
Always make sure your database (tables) are up-to-date:
|
||||
|
||||
```shell
|
||||
python dbutils.py --game SDBT upgrade
|
||||
@@ -146,7 +151,7 @@ The songId is based on the actual ID within your version of Chunithm.
|
||||
### SDCA
|
||||
|
||||
| Version ID | Version Name |
|
||||
|------------|------------------------------------|
|
||||
| ---------- | ---------------------------------- |
|
||||
| 0 | crossbeats REV. |
|
||||
| 1 | crossbeats REV. SUNRISE |
|
||||
| 2 | crossbeats REV. SUNRISE S2 |
|
||||
@@ -166,26 +171,26 @@ The importer for crossbeats REV. will import Music.
|
||||
|
||||
Config file is located in `config/cxb.yaml`.
|
||||
|
||||
| Option | Info |
|
||||
|------------------------|------------------------------------------------------------|
|
||||
| `hostname` | Requires a proper `hostname` (not localhost!) to run |
|
||||
| `ssl_enable` | Enables/Disables the use of the `ssl_cert` and `ssl_key` |
|
||||
| `port` | Set your unsecure port number |
|
||||
| `port_secure` | Set your secure/SSL port number |
|
||||
| `ssl_cert`, `ssl_key` | Enter your SSL certificate (requires not self signed cert) |
|
||||
| Option | Info |
|
||||
| --------------------- | ---------------------------------------------------------- |
|
||||
| `hostname` | Requires a proper `hostname` (not localhost!) to run |
|
||||
| `ssl_enable` | Enables/Disables the use of the `ssl_cert` and `ssl_key` |
|
||||
| `port` | Set your unsecure port number |
|
||||
| `port_secure` | Set your secure/SSL port number |
|
||||
| `ssl_cert`, `ssl_key` | Enter your SSL certificate (requires not self signed cert) |
|
||||
|
||||
|
||||
## maimai DX
|
||||
|
||||
### SDEZ
|
||||
|
||||
| Game Code | Version ID | Version Name |
|
||||
|-----------|------------|-------------------------|
|
||||
| Game Code | Version ID | Version Name |
|
||||
| --------- | ---------- | ------------ |
|
||||
|
||||
|
||||
For versions pre-dx
|
||||
| Game Code | Version ID | Version Name |
|
||||
|-----------|------------|-------------------------|
|
||||
| --------- | ---------- | ----------------------- |
|
||||
| SBXL | 0 | maimai |
|
||||
| SBXL | 1 | maimai PLUS |
|
||||
| SBZF | 2 | maimai GreeN |
|
||||
@@ -227,11 +232,12 @@ The importer for maimai Pre-DX will import Events and Music. Not all games will
|
||||
|
||||
### Database upgrade
|
||||
|
||||
Always make sure your database (tables) are up-to-date, to do so go to the `core/data/schema/versions` folder and see which version is the latest, f.e. `SDEZ_2_upgrade.sql`. In order to upgrade to version 2 in this case you need to perform all previous updates as well:
|
||||
Always make sure your database (tables) are up-to-date:
|
||||
|
||||
```shell
|
||||
python dbutils.py --game SDEZ upgrade
|
||||
```
|
||||
|
||||
Pre-Dx uses the same database as DX, so only upgrade using the SDEZ game code!
|
||||
|
||||
## Hatsune Miku Project Diva
|
||||
@@ -239,7 +245,7 @@ Pre-Dx uses the same database as DX, so only upgrade using the SDEZ game code!
|
||||
### SBZV
|
||||
|
||||
| Version ID | Version Name |
|
||||
|------------|---------------------------------|
|
||||
| ---------- | ------------------------------- |
|
||||
| 0 | Project Diva Arcade |
|
||||
| 1 | Project Diva Arcade Future Tone |
|
||||
|
||||
@@ -260,7 +266,7 @@ the Shop, Modules and Customizations.
|
||||
Config file is located in `config/diva.yaml`.
|
||||
|
||||
| Option | Info |
|
||||
|----------------------|-------------------------------------------------------------------------------------------------|
|
||||
| -------------------- | ----------------------------------------------------------------------------------------------- |
|
||||
| `unlock_all_modules` | Unlocks all modules (costumes) by default, if set to `False` all modules need to be purchased |
|
||||
| `unlock_all_items` | Unlocks all items (customizations) by default, if set to `False` all items need to be purchased |
|
||||
|
||||
@@ -270,9 +276,7 @@ In order to use custom PV Lists, simply drop in your .dat files inside of /title
|
||||
|
||||
### Database upgrade
|
||||
|
||||
Always make sure your database (tables) are up-to-date, to do so go to the `core/data/schema/versions` folder and see
|
||||
which version is the latest, f.e. `SBZV_4_upgrade.sql`. In order to upgrade to version 4 in this case you need to
|
||||
perform all previous updates as well:
|
||||
Always make sure your database (tables) are up-to-date:
|
||||
|
||||
```shell
|
||||
python dbutils.py --game SBZV upgrade
|
||||
@@ -283,7 +287,7 @@ python dbutils.py --game SBZV upgrade
|
||||
### SDDT
|
||||
|
||||
| Version ID | Version Name |
|
||||
|------------|----------------------------|
|
||||
| ---------- | -------------------------- |
|
||||
| 0 | O.N.G.E.K.I. |
|
||||
| 1 | O.N.G.E.K.I. + |
|
||||
| 2 | O.N.G.E.K.I. SUMMER |
|
||||
@@ -311,7 +315,7 @@ The importer for O.N.G.E.K.I. will all all Cards, Music and Events.
|
||||
Config file is located in `config/ongeki.yaml`.
|
||||
|
||||
| Option | Info |
|
||||
|------------------|----------------------------------------------------------------------------------------------------------------|
|
||||
| ---------------- | -------------------------------------------------------------------------------------------------------------- |
|
||||
| `enabled_gachas` | Enter all gacha IDs for Card Maker to work, other than default may not work due to missing cards added to them |
|
||||
| `crypto` | This option is used to enable the TLS Encryption |
|
||||
|
||||
@@ -328,20 +332,80 @@ crypto:
|
||||
|
||||
### Database upgrade
|
||||
|
||||
Always make sure your database (tables) are up-to-date, to do so go to the `core/data/schema/versions` folder and see
|
||||
which version is the latest, f.e. `SDDT_4_upgrade.sql`. In order to upgrade to version 4 in this case you need to
|
||||
perform all previous updates as well:
|
||||
Always make sure your database (tables) are up-to-date:
|
||||
|
||||
```shell
|
||||
python dbutils.py --game SDDT upgrade
|
||||
```
|
||||
|
||||
### Controlling Events (Ranking Event, Technical Challenge Event, Mission Event)
|
||||
|
||||
Events are controlled by 2 types of enabled events:
|
||||
- RankingEvent (type 6), TechChallengeEvent (type 17)
|
||||
- AcceptRankingEvent (type 7), AcceptTechChallengeEvent (type 18)
|
||||
|
||||
Both Ranking and Accept must be enabled for event to function properly
|
||||
|
||||
Event will run for the time specified in startDate and endDate
|
||||
|
||||
AcceptRankingEvent and AcceptTechChallengeEvent are reward period for events, which specify from what startDate until endDate you can collect the rewards for attending the event, so the reward period must start in the future, e.g. :
|
||||
|
||||
- RankingEvent startDate 2023-12-01 - endDate 2023-12-30 - period in which whole event is running
|
||||
- AcceptRankingEvent startDate 2023-12-23 - endDate 2023-12-30 - period in which you can collect rewards for the event
|
||||
|
||||
If player misses the AcceptRankingEvent period - ranking will be invalidated and receive lowest reward from the event (typically 500x money)
|
||||
|
||||
Technical Challenge Song List:
|
||||
|
||||
Songs that are used for Technical Challenge are not stored anywhere in data files, so you need to fill the database table by yourself, you can gather all songs that should be in Technical Challenges from ONGEKI japanese wikis, or, you can create your own sets:
|
||||
|
||||
Database table : `ongeki_static_tech_music`
|
||||
|
||||
```
|
||||
id: Id in table, just increment for each entry
|
||||
version: version of the game you want the tech challenge to be in (from RED and up)
|
||||
eventId: Id of the event in ongeki_static_events, insert the Id of the TechChallengeEvent (type 17) you want the song be assigned to
|
||||
musicId: Id of the song you want to add, use songId from ongeki_static_music table
|
||||
level: Difficulty of the song you want to track during the event, from 0(basic) to 3(master)
|
||||
|
||||
```
|
||||
|
||||
Current implementation of Ranking and Technical Challenge Events are updated on every profile save to the Network, and Ranked on each player login, in official specification, calculation for current rank on the network should be done in the maintenance window
|
||||
|
||||
Mission Event (type 13) is a monthly type of event, which is used when another event doesn't have it's own Ranking or Technical Challenge Event running, only one Mission Event should be running at a time, so enable only the specific Mission you want to run currently on the Network
|
||||
|
||||
If you're often trying fresh cards, registering new profiles etc., you can also consider disabling all Announcement Events (type 1), as it will disable all the banners that pop up on login (they show up only once though, so if you click through them once they won't show again)
|
||||
|
||||
Event type 2 in Database are Advertisement Movies, enable only 1 you want to currently play, and disable others
|
||||
|
||||
|
||||
Present and Reward List - populate reward list using read.py
|
||||
|
||||
Create present for players by adding an entry in `ongeki_static_present_list`
|
||||
```
|
||||
id: unique for each entry
|
||||
version: game version you want the present be in
|
||||
presentId: id of the present - starts with 1001 and go up from that, must be unique for each reward(don't set multiple rewardIds with same presentId)
|
||||
presentName: present name which will be shown on the bottom when received
|
||||
rewardId: ID of item from ongeki_static_rewards
|
||||
stock: how many you want to give (like 5 copies of same card, or 10000 money, etc.)
|
||||
message: no idea, can be left empty
|
||||
startDate: date when to start giving out
|
||||
endDate: date when ends
|
||||
```
|
||||
|
||||
After inserting present to the table, add the presentId into players `ongeki_static_item`, where itemKind is 9, itemId is the presentId, and stock set 1 and isValid to 1
|
||||
|
||||
After that, on next login the present should be received (or whenever it supposed to happen)
|
||||
|
||||
|
||||
|
||||
## Card Maker
|
||||
|
||||
### SDED
|
||||
|
||||
| Version ID | Version Name |
|
||||
|------------|-----------------|
|
||||
| ---------- | --------------- |
|
||||
| 0 | Card Maker 1.30 |
|
||||
| 1 | Card Maker 1.35 |
|
||||
|
||||
@@ -463,7 +527,7 @@ Gacha IDs up to 1140 will be loaded for CM 1.34 and all gachas will be loaded fo
|
||||
### SDFE
|
||||
|
||||
| Version ID | Version Name |
|
||||
|------------|---------------|
|
||||
| ---------- | ------------- |
|
||||
| 0 | WACCA |
|
||||
| 1 | WACCA S |
|
||||
| 2 | WACCA Lily |
|
||||
@@ -486,7 +550,7 @@ The importer for WACCA will import all Music data.
|
||||
Config file is located in `config/wacca.yaml`.
|
||||
|
||||
| Option | Info |
|
||||
|--------------------|-----------------------------------------------------------------------------|
|
||||
| ------------------ | --------------------------------------------------------------------------- |
|
||||
| `always_vip` | Enables/Disables VIP, if disabled it needs to be purchased manually in game |
|
||||
| `infinite_tickets` | Always set the "unlock expert" tickets to 5 |
|
||||
| `infinite_wp` | Sets the user WP to `999999` |
|
||||
@@ -495,7 +559,7 @@ Config file is located in `config/wacca.yaml`.
|
||||
|
||||
### Database upgrade
|
||||
|
||||
Always make sure your database (tables) are up-to-date, to do so go to the `core/data/schema/versions` folder and see which version is the latest, f.e. `SDFE_3_upgrade.sql`. In order to upgrade to version 3 in this case you need to perform all previous updates as well:
|
||||
Always make sure your database (tables) are up-to-date:
|
||||
|
||||
```shell
|
||||
python dbutils.py --game SDFE upgrade
|
||||
@@ -540,9 +604,9 @@ Below is a list of VIP rewards. Currently, VIP is not implemented, and thus thes
|
||||
|
||||
### SDEW
|
||||
|
||||
| Version ID | Version Name |
|
||||
|------------|---------------|
|
||||
| 0 | SAO |
|
||||
| Version ID | Version Name |
|
||||
| ---------- | ------------ |
|
||||
| 0 | SAO |
|
||||
|
||||
|
||||
### Importer
|
||||
@@ -559,16 +623,16 @@ The importer for SAO will import all items, heroes, support skills and titles da
|
||||
|
||||
Config file is located in `config/sao.yaml`.
|
||||
|
||||
| Option | Info |
|
||||
|--------------------|-----------------------------------------------------------------------------|
|
||||
| `hostname` | Changes the server listening address for Mucha |
|
||||
| `port` | Changes the listing port |
|
||||
| `auto_register` | Allows the game to handle the automatic registration of new cards |
|
||||
| Option | Info |
|
||||
| --------------- | ----------------------------------------------------------------- |
|
||||
| `hostname` | Changes the server listening address for Mucha |
|
||||
| `port` | Changes the listing port |
|
||||
| `auto_register` | Allows the game to handle the automatic registration of new cards |
|
||||
|
||||
|
||||
### Database upgrade
|
||||
|
||||
Always make sure your database (tables) are up-to-date, to do so go to the `core/data/schema/versions` folder and see which version is the latest, f.e. `SDEW_1_upgrade.sql`. In order to upgrade to version 3 in this case you need to perform all previous updates as well:
|
||||
Always make sure your database (tables) are up-to-date:
|
||||
|
||||
```shell
|
||||
python dbutils.py --game SDEW upgrade
|
||||
@@ -587,4 +651,135 @@ python dbutils.py --game SDEW upgrade
|
||||
|
||||
- Midorica - Limited Network Support
|
||||
- Dniel97 - Helping with network base
|
||||
- tungnotpunk - Source
|
||||
- tungnotpunk - Source
|
||||
|
||||
## Initial D THE ARCADE
|
||||
|
||||
### SDGT
|
||||
|
||||
| Version ID | Version Name |
|
||||
| ---------- | ----------------------------- |
|
||||
| 0 | Initial D THE ARCADE Season 1 |
|
||||
| 1 | Initial D THE ARCADE Season 2 |
|
||||
|
||||
**Important: Only version 1.50.00 (Season 2) is currently working and actively supported!**
|
||||
|
||||
### Profile Importer
|
||||
|
||||
In order to use the profile importer download the `idac_profile.json` file from the frontend
|
||||
and either directly use the folder path with `idac_profile.json` in it or specify the complete
|
||||
path to the `.json` file
|
||||
|
||||
```shell
|
||||
python read.py --game SDGT --version <Version ID> --optfolder /path/to/game/download/folder
|
||||
```
|
||||
|
||||
The importer for SDGT will import the complete profile data with personal high scores as well.
|
||||
|
||||
### Config
|
||||
|
||||
Config file is located in `config/idac.yaml`.
|
||||
|
||||
| Option | Info |
|
||||
| ----------------------------- | ----------------------------------------------------------------------------------------------------------- |
|
||||
| `ssl` | Enables/Disables the use of the `ssl_cert` and `ssl_key` (currently unsuported) |
|
||||
| `matching_host` | IPv4 address of your PC for the Online Battle (currently unsupported) |
|
||||
| `port_matching` | Port number for the Online Battle Matching |
|
||||
| `port_echo1/2` | Port numbers for Echos |
|
||||
| `port_matching_p2p` | Port number for Online Battle (currently unsupported) |
|
||||
| `stamp.enable` | Enables/Disabled the play stamp events |
|
||||
| `stamp.enabled_stamps` | Define up to 3 play stamp events (without `.json` extension, which are placed in `titles/idac/data/stamps`) |
|
||||
| `timetrial.enable` | Enables/Disables the time trial event |
|
||||
| `timetrial.enabled_timetrial` | Define one! trial event (without `.json` extension, which are placed in `titles/idac/data/timetrial`) |
|
||||
|
||||
|
||||
### Database upgrade
|
||||
|
||||
Always make sure your database (tables) are up-to-date:
|
||||
|
||||
```shell
|
||||
python dbutils.py --game SDGT upgrade
|
||||
```
|
||||
|
||||
### Notes
|
||||
- Online Battle is not supported
|
||||
- Online Battle Matching is not supported
|
||||
|
||||
### Item categories
|
||||
|
||||
| Category ID | Category Name |
|
||||
| ----------- | ------------------------ |
|
||||
| 1 | D Coin |
|
||||
| 3 | Car Dressup Token |
|
||||
| 5 | Avatar Dressup Token |
|
||||
| 6 | Tachometer |
|
||||
| 7 | Aura |
|
||||
| 8 | Aura Color |
|
||||
| 9 | Avatar Face |
|
||||
| 10 | Avatar Eye |
|
||||
| 11 | Avatar Mouth |
|
||||
| 12 | Avatar Hair |
|
||||
| 13 | Avatar Glasses |
|
||||
| 14 | Avatar Face accessories |
|
||||
| 15 | Avatar Body |
|
||||
| 18 | Avatar Background |
|
||||
| 21 | Chat Stamp |
|
||||
| 22 | Keychain |
|
||||
| 24 | Title |
|
||||
| 25 | FullTune Ticket |
|
||||
| 26 | Paper Cup |
|
||||
| 27 | BGM |
|
||||
| 28 | Drifting Text |
|
||||
| 31 | Start Menu BG |
|
||||
| 32 | Car Color/Paint |
|
||||
| 33 | Aura Level |
|
||||
| 34 | FullTune Ticket Fragment |
|
||||
| 35 | Underneon Lights |
|
||||
|
||||
### TimeRelease Chapter:
|
||||
|
||||
1. Story: 1, 2, 3, 4, 5, 6, 7, 8, 9, 19 (Chapter 10), (29 Chapter 11?)
|
||||
2. MF Ghost: 10, 11, 12, 13, 14, 15
|
||||
3. Bunta: 15, 16, 17, 18, 19, 20, (21, 21, 22?)
|
||||
4. Special Event: 23, 24, 25, 26, 27, 28 (Touhou Project)
|
||||
|
||||
### TimeRelease Courses:
|
||||
|
||||
|
||||
| Course ID | Course Name | Direction |
|
||||
| --------- | ------------------------- | ------------------------ |
|
||||
| 0 | Akina Lake(秋名湖) | CounterClockwise(左周り) |
|
||||
| 2 | Akina Lake(秋名湖) | Clockwise(右周り) |
|
||||
| 52 | Hakone(箱根) | Downhill(下り) |
|
||||
| 54 | Hakone(箱根) | Hillclimb(上り) |
|
||||
| 36 | Usui(碓氷) | CounterClockwise(左周り) |
|
||||
| 38 | Usui(碓氷) | Clockwise(右周り) |
|
||||
| 4 | Myogi(妙義) | Downhill(下り) |
|
||||
| 6 | Myogi(妙義) | Hillclimb(上り) |
|
||||
| 8 | Akagi(赤城) | Downhill(下り) |
|
||||
| 10 | Akagi(赤城) | Hillclimb(上り) |
|
||||
| 12 | Akina(秋名) | Downhill(下り) |
|
||||
| 14 | Akina(秋名) | Hillclimb(上り) |
|
||||
| 16 | Irohazaka(いろは坂) | Downhill(下り) |
|
||||
| 18 | Irohazaka(いろは坂) | Reverse(逆走) |
|
||||
| 56 | Momiji Line(もみじライン) | Downhill(下り) |
|
||||
| 58 | Momiji Line(もみじライン) | Hillclimb(上り) |
|
||||
| 20 | Tsukuba(筑波) | Outbound(往路) |
|
||||
| 22 | Tsukuba(筑波) | Inbound(復路) |
|
||||
| 24 | Happogahara(八方ヶ原) | Outbound(往路) |
|
||||
| 26 | Happogahara(八方ヶ原) | Inbound(復路) |
|
||||
| 40 | Sadamine(定峰) | Downhill(下り) |
|
||||
| 42 | Sadamine(定峰) | Hillclimb(上り) |
|
||||
| 44 | Tsuchisaka(土坂) | Outbound(往路) |
|
||||
| 46 | Tsuchisaka(土坂) | Inbound(復路) |
|
||||
| 48 | Akina Snow(秋名雪) | Downhill(下り) |
|
||||
| 50 | Akina Snow(秋名雪) | Hillclimb(上り) |
|
||||
| 68 | Odawara(小田原) | Forward(順走) |
|
||||
| 70 | Odawara(小田原) | Reverse(逆走) |
|
||||
|
||||
### Credits
|
||||
- Bottersnike: For the HUGE Reverse Engineering help
|
||||
- Kinako: For helping with the timeRelease unlocking of courses and special mode
|
||||
|
||||
A huge thanks to all people who helped shaping this project to what it is now and don't want to be mentioned here.
|
||||
|
||||
|
||||
@@ -1,9 +1,3 @@
|
||||
server:
|
||||
enable: True
|
||||
loglevel: "info"
|
||||
hostname: "localhost"
|
||||
ssl_enable: False
|
||||
port: 8082
|
||||
port_secure: 443
|
||||
ssl_cert: "cert/title.crt"
|
||||
ssl_key: "cert/title.key"
|
||||
loglevel: "info"
|
||||
22
example_config/idac.yaml
Normal file
22
example_config/idac.yaml
Normal file
@@ -0,0 +1,22 @@
|
||||
server:
|
||||
enable: True
|
||||
loglevel: "info"
|
||||
ssl: False
|
||||
ssl_key: "cert/idac.key"
|
||||
ssl_cert: "cert/idac.crt"
|
||||
matching_host: "127.0.0.1"
|
||||
port_matching: 20000
|
||||
port_echo1: 20001
|
||||
port_echo2: 20002
|
||||
port_matching_p2p: 20003
|
||||
|
||||
stamp:
|
||||
enable: True
|
||||
enabled_stamps: # max 3 play stamps
|
||||
- "touhou_remilia_scarlet"
|
||||
- "touhou_flandre_scarlet"
|
||||
- "touhou_sakuya_izayoi"
|
||||
|
||||
timetrial:
|
||||
enable: True
|
||||
enabled_timetrial: "touhou_remilia_scarlet"
|
||||
@@ -1,6 +1,7 @@
|
||||
server:
|
||||
enable: True
|
||||
loglevel: "info"
|
||||
use_https: False
|
||||
|
||||
gachas:
|
||||
enabled_gachas:
|
||||
|
||||
@@ -1,6 +1,13 @@
|
||||
server:
|
||||
hostname: "localhost"
|
||||
enable: True
|
||||
loglevel: "info"
|
||||
port: 9000
|
||||
auto_register: True
|
||||
auto_register: True
|
||||
|
||||
crypt:
|
||||
enable: False
|
||||
key: ""
|
||||
iv: ""
|
||||
|
||||
hash:
|
||||
verify_hash: False
|
||||
hash_base: ""
|
||||
@@ -33,6 +33,9 @@ Games listed below have been tested and confirmed working. Only game versions ol
|
||||
+ Sword Art Online Arcade (partial support)
|
||||
+ Final
|
||||
|
||||
+ Initial D THE ARCADE
|
||||
+ Season 2
|
||||
|
||||
## Requirements
|
||||
- python 3 (tested working with 3.9 and 3.10, other versions YMMV)
|
||||
- pip
|
||||
|
||||
@@ -198,14 +198,6 @@ class CxbBase:
|
||||
).decode("utf-8")
|
||||
)
|
||||
|
||||
def task_generateIndexData(self, versionindex: List[str], uid: int):
|
||||
try:
|
||||
v_profile = self.data.profile.get_profile_index(0, uid, self.version)
|
||||
v_profile_data = v_profile["data"]
|
||||
versionindex.append(int(v_profile_data["appVersion"]))
|
||||
except Exception:
|
||||
versionindex.append("10400")
|
||||
|
||||
def handle_action_loadrange_request(self, data: Dict) -> Dict:
|
||||
range_start = data["loadrange"]["range"][0]
|
||||
range_end = data["loadrange"]["range"][1]
|
||||
@@ -279,9 +271,14 @@ class CxbBase:
|
||||
thread_ScoreData = Thread(target=CxbBase.task_generateScoreData(song, index, data1))
|
||||
thread_ScoreData.start()
|
||||
|
||||
for v in index:
|
||||
thread_IndexData = Thread(target=CxbBase.task_generateIndexData(self, versionindex, uid))
|
||||
thread_IndexData.start()
|
||||
v_profile = self.data.profile.get_profile_index(0, uid, self.version)
|
||||
v_profile_data = v_profile["data"]
|
||||
|
||||
for _, data in enumerate(profile):
|
||||
if v_profile_data:
|
||||
versionindex.append(int(v_profile_data["appVersion"]))
|
||||
else:
|
||||
versionindex.append("10400")
|
||||
|
||||
return {"index": index, "data": data1, "version": versionindex}
|
||||
|
||||
|
||||
@@ -19,42 +19,6 @@ class CxbServerConfig:
|
||||
)
|
||||
)
|
||||
|
||||
@property
|
||||
def hostname(self) -> str:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "cxb", "server", "hostname", default="localhost"
|
||||
)
|
||||
|
||||
@property
|
||||
def ssl_enable(self) -> bool:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "cxb", "server", "ssl_enable", default=False
|
||||
)
|
||||
|
||||
@property
|
||||
def port(self) -> int:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "cxb", "server", "port", default=8082
|
||||
)
|
||||
|
||||
@property
|
||||
def port_secure(self) -> int:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "cxb", "server", "port_secure", default=443
|
||||
)
|
||||
|
||||
@property
|
||||
def ssl_cert(self) -> str:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "cxb", "server", "ssl_cert", default="cert/title.crt"
|
||||
)
|
||||
|
||||
@property
|
||||
def ssl_key(self) -> str:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "cxb", "server", "ssl_key", default="cert/title.key"
|
||||
)
|
||||
|
||||
|
||||
class CxbConfig(dict):
|
||||
def __init__(self) -> None:
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
from twisted.web.http import Request
|
||||
import traceback
|
||||
from twisted.web import resource, server
|
||||
from twisted.internet import reactor, endpoints
|
||||
import sys
|
||||
import yaml
|
||||
import json
|
||||
import re
|
||||
@@ -77,7 +76,7 @@ class CxbServlet(BaseServlet):
|
||||
return True
|
||||
|
||||
def get_allnet_info(self, game_code: str, game_ver: int, keychip: str) -> Tuple[str, str]:
|
||||
if not self.core_cfg.server.is_using_proxy and Utils.get_title_port_ssl(self.core_cfg) != 443:
|
||||
if not self.core_cfg.server.is_using_proxy and Utils.get_title_port_ssl(self.core_cfg):
|
||||
return (
|
||||
f"https://{self.core_cfg.title.hostname}:{self.core_cfg.title.port_ssl}",
|
||||
"",
|
||||
@@ -124,35 +123,42 @@ class CxbServlet(BaseServlet):
|
||||
func_to_find = "handle_data_"
|
||||
version_string = "Base"
|
||||
internal_ver = 0
|
||||
version = 0
|
||||
|
||||
if req_json == {}:
|
||||
self.logger.warning(f"Empty json request to /data")
|
||||
return b""
|
||||
|
||||
subcmd = list(req_json.keys())[0]
|
||||
if subcmd == "dldate":
|
||||
|
||||
if (
|
||||
not type(req_json["dldate"]) is dict
|
||||
or "filetype" not in req_json["dldate"]
|
||||
):
|
||||
self.logger.warning(f"Malformed dldate request: {req_json}")
|
||||
return b""
|
||||
if (
|
||||
not type(req_json["dldate"]) is dict
|
||||
or "filetype" not in req_json["dldate"]
|
||||
):
|
||||
self.logger.warning(f"Malformed dldate request: {req_json}")
|
||||
return b""
|
||||
|
||||
filetype = req_json["dldate"]["filetype"]
|
||||
filetype_split = filetype.split("/")
|
||||
filetype = req_json["dldate"]["filetype"]
|
||||
filetype_split = filetype.split("/")
|
||||
|
||||
if len(filetype_split) < 2 or not filetype_split[0].isnumeric():
|
||||
self.logger.warning(f"Malformed dldate request: {req_json}")
|
||||
return b""
|
||||
if len(filetype_split) < 2 or not filetype_split[0].isnumeric():
|
||||
self.logger.warning(f"Malformed dldate request: {req_json}")
|
||||
return b""
|
||||
|
||||
version = int(filetype_split[0])
|
||||
filename = filetype_split[len(filetype_split) - 1]
|
||||
version = int(filetype_split[0])
|
||||
filename = filetype_split[len(filetype_split) - 1]
|
||||
|
||||
match = re.match(
|
||||
"^([A-Za-z]*)(\d\d\d\d)$", filetype_split[len(filetype_split) - 1]
|
||||
)
|
||||
if match:
|
||||
func_to_find += f"{inflection.underscore(match.group(1))}xxxx"
|
||||
match = re.match(
|
||||
"^([A-Za-z]*)(\d\d\d\d)$", filetype_split[len(filetype_split) - 1]
|
||||
)
|
||||
if match:
|
||||
func_to_find += f"{inflection.underscore(match.group(1))}xxxx"
|
||||
else:
|
||||
func_to_find += f"{inflection.underscore(filename)}"
|
||||
else:
|
||||
func_to_find += f"{inflection.underscore(filename)}"
|
||||
filetype = subcmd
|
||||
func_to_find += filetype
|
||||
|
||||
func_to_find += "_request"
|
||||
|
||||
@@ -183,9 +189,10 @@ class CxbServlet(BaseServlet):
|
||||
except Exception as e:
|
||||
self.logger.error(f"Error handling request for file {filetype} - {e}")
|
||||
if self.logger.level == logging.DEBUG:
|
||||
traceback.print_exception(e, limit=1)
|
||||
tp, val, tb = sys.exc_info()
|
||||
traceback.print_exception(tp, val, tb, limit=1)
|
||||
with open("{0}/{1}.log".format(self.core_cfg.server.log_dir, "cxb"), "a") as f:
|
||||
traceback.print_exception(e, limit=1, file=f)
|
||||
traceback.print_exception(tp, val, tb, limit=1, file=f)
|
||||
return ""
|
||||
|
||||
self.logger.debug(f"{version_string} Response {resp}")
|
||||
@@ -211,9 +218,10 @@ class CxbServlet(BaseServlet):
|
||||
except Exception as e:
|
||||
self.logger.error(f"Error handling action {subcmd} request - {e}")
|
||||
if self.logger.level == logging.DEBUG:
|
||||
traceback.print_exception(e, limit=1)
|
||||
tp, val, tb = sys.exc_info()
|
||||
traceback.print_exception(tp, val, tb, limit=1)
|
||||
with open("{0}/{1}.log".format(self.core_cfg.server.log_dir, "cxb"), "a") as f:
|
||||
traceback.print_exception(e, limit=1, file=f)
|
||||
traceback.print_exception(tp, val, tb, limit=1, file=f)
|
||||
return ""
|
||||
|
||||
self.logger.debug(f"Response {resp}")
|
||||
@@ -239,9 +247,10 @@ class CxbServlet(BaseServlet):
|
||||
except Exception as e:
|
||||
self.logger.error(f"Error handling auth {subcmd} request - {e}")
|
||||
if self.logger.level == logging.DEBUG:
|
||||
traceback.print_exception(e, limit=1)
|
||||
tp, val, tb = sys.exc_info()
|
||||
traceback.print_exception(tp, val, tb, limit=1)
|
||||
with open("{0}/{1}.log".format(self.core_cfg.server.log_dir, "cxb"), "a") as f:
|
||||
traceback.print_exception(e, limit=1, file=f)
|
||||
traceback.print_exception(tp, val, tb, limit=1, file=f)
|
||||
return ""
|
||||
|
||||
self.logger.debug(f"Response {resp}")
|
||||
|
||||
12
titles/idac/__init__.py
Normal file
12
titles/idac/__init__.py
Normal file
@@ -0,0 +1,12 @@
|
||||
from titles.idac.index import IDACServlet
|
||||
from titles.idac.const import IDACConstants
|
||||
from titles.idac.database import IDACData
|
||||
from titles.idac.read import IDACReader
|
||||
from titles.idac.frontend import IDACFrontend
|
||||
|
||||
index = IDACServlet
|
||||
database = IDACData
|
||||
reader = IDACReader
|
||||
frontend = IDACFrontend
|
||||
game_codes = [IDACConstants.GAME_CODE]
|
||||
current_schema_version = 1
|
||||
16
titles/idac/base.py
Normal file
16
titles/idac/base.py
Normal file
@@ -0,0 +1,16 @@
|
||||
import logging
|
||||
|
||||
from core.config import CoreConfig
|
||||
from titles.idac.config import IDACConfig
|
||||
from titles.idac.const import IDACConstants
|
||||
from titles.idac.database import IDACData
|
||||
|
||||
|
||||
class IDACBase:
|
||||
def __init__(self, core_cfg: CoreConfig, game_cfg: IDACConfig) -> None:
|
||||
self.core_cfg = core_cfg
|
||||
self.game_config = game_cfg
|
||||
self.game = IDACConstants.GAME_CODE
|
||||
self.version = IDACConstants.VER_IDAC_SEASON_1
|
||||
self.data = IDACData(core_cfg)
|
||||
self.logger = logging.getLogger("idac")
|
||||
121
titles/idac/config.py
Normal file
121
titles/idac/config.py
Normal file
@@ -0,0 +1,121 @@
|
||||
from core.config import CoreConfig
|
||||
|
||||
|
||||
class IDACServerConfig:
|
||||
def __init__(self, parent: "IDACConfig") -> None:
|
||||
self.__config = parent
|
||||
|
||||
@property
|
||||
def enable(self) -> bool:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "idac", "server", "enable", default=True
|
||||
)
|
||||
|
||||
@property
|
||||
def loglevel(self) -> int:
|
||||
return CoreConfig.str_to_loglevel(
|
||||
CoreConfig.get_config_field(
|
||||
self.__config, "idac", "server", "loglevel", default="info"
|
||||
)
|
||||
)
|
||||
|
||||
@property
|
||||
def ssl(self) -> bool:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "idac", "server", "ssl", default=False
|
||||
)
|
||||
|
||||
@property
|
||||
def ssl_cert(self) -> str:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "idac", "server", "ssl_cert", default="cert/title.crt"
|
||||
)
|
||||
|
||||
@property
|
||||
def ssl_key(self) -> str:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "idac", "server", "ssl_key", default="cert/title.key"
|
||||
)
|
||||
|
||||
@property
|
||||
def matching_host(self) -> str:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "idac", "server", "matching_host", default="127.0.0.1"
|
||||
)
|
||||
|
||||
@property
|
||||
def matching(self) -> int:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "idac", "server", "port_matching", default=20000
|
||||
)
|
||||
|
||||
@property
|
||||
def echo1(self) -> int:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "idac", "server", "port_echo1", default=20001
|
||||
)
|
||||
|
||||
@property
|
||||
def echo2(self) -> int:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "idac", "server", "port_echo2", default=20002
|
||||
)
|
||||
|
||||
@property
|
||||
def matching_p2p(self) -> int:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "idac", "server", "port_matching_p2p", default=20003
|
||||
)
|
||||
|
||||
|
||||
class IDACStampConfig:
|
||||
def __init__(self, parent: "IDACConfig") -> None:
|
||||
self.__config = parent
|
||||
|
||||
@property
|
||||
def enable(self) -> bool:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "idac", "stamp", "enable", default=True
|
||||
)
|
||||
|
||||
@property
|
||||
def enabled_stamps(self) -> list:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config,
|
||||
"idac",
|
||||
"stamp",
|
||||
"enabled_stamps",
|
||||
default=[
|
||||
"touhou_remilia_scarlet",
|
||||
"touhou_flandre_scarlet",
|
||||
"touhou_sakuya_izayoi",
|
||||
],
|
||||
)
|
||||
|
||||
|
||||
class IDACTimetrialConfig:
|
||||
def __init__(self, parent: "IDACConfig") -> None:
|
||||
self.__config = parent
|
||||
|
||||
@property
|
||||
def enable(self) -> bool:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "idac", "timetrial", "enable", default=True
|
||||
)
|
||||
|
||||
@property
|
||||
def enabled_timetrial(self) -> str:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config,
|
||||
"idac",
|
||||
"timetrial",
|
||||
"enabled_timetrial",
|
||||
default="touhou_remilia_scarlet",
|
||||
)
|
||||
|
||||
|
||||
class IDACConfig(dict):
|
||||
def __init__(self) -> None:
|
||||
self.server = IDACServerConfig(self)
|
||||
self.stamp = IDACStampConfig(self)
|
||||
self.timetrial = IDACTimetrialConfig(self)
|
||||
16
titles/idac/const.py
Normal file
16
titles/idac/const.py
Normal file
@@ -0,0 +1,16 @@
|
||||
class IDACConstants():
|
||||
GAME_CODE = "SDGT"
|
||||
|
||||
CONFIG_NAME = "idac.yaml"
|
||||
|
||||
VER_IDAC_SEASON_1 = 0
|
||||
VER_IDAC_SEASON_2 = 1
|
||||
|
||||
VERSION_STRING = (
|
||||
"Initial D THE ARCADE Season 1",
|
||||
"Initial D THE ARCADE Season 2",
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def game_ver_to_string(cls, ver: int):
|
||||
return cls.VERSION_STRING[ver]
|
||||
BIN
titles/idac/data/avatarGacha.json
Normal file
BIN
titles/idac/data/avatarGacha.json
Normal file
Binary file not shown.
38
titles/idac/data/create_delivery_images.py
Normal file
38
titles/idac/data/create_delivery_images.py
Normal file
@@ -0,0 +1,38 @@
|
||||
import os
|
||||
import hashlib
|
||||
|
||||
|
||||
def prepare_images(image_folder="titles/idac/data/images"):
|
||||
print(f"Preparing image delivery files in {image_folder}...")
|
||||
|
||||
for file in os.listdir(image_folder):
|
||||
if file.endswith(".png") or file.endswith(".jpg"):
|
||||
# dpg_name = "adv-" + file[:-4].upper()
|
||||
dpg_name = file[:-4]
|
||||
if file.endswith(".png"):
|
||||
dpg_name += ".dpg"
|
||||
else:
|
||||
dpg_name += ".djg"
|
||||
|
||||
if os.path.exists(os.path.join(image_folder, dpg_name)):
|
||||
continue
|
||||
else:
|
||||
with open(
|
||||
os.path.join(image_folder, file), "rb"
|
||||
) as original_image_file:
|
||||
original_image = original_image_file.read()
|
||||
image_hash = hashlib.md5(original_image).hexdigest()
|
||||
print(
|
||||
f"DPG for {file} not found, creating with hash {image_hash.upper()} ..."
|
||||
)
|
||||
md5_buf = bytes.fromhex(image_hash)
|
||||
dpg_buf = md5_buf + original_image
|
||||
dpg_name = "adv-" + image_hash.upper() + dpg_name[:-4]
|
||||
with open(os.path.join(image_folder, dpg_name), "wb") as dpg_file:
|
||||
dpg_file.write(dpg_buf)
|
||||
|
||||
print(f"Created {dpg_name}.")
|
||||
|
||||
|
||||
# Call the function to execute it
|
||||
prepare_images()
|
||||
BIN
titles/idac/data/stamps/touhou_flandre_scarlet.json
Normal file
BIN
titles/idac/data/stamps/touhou_flandre_scarlet.json
Normal file
Binary file not shown.
BIN
titles/idac/data/stamps/touhou_remilia_scarlet.json
Normal file
BIN
titles/idac/data/stamps/touhou_remilia_scarlet.json
Normal file
Binary file not shown.
BIN
titles/idac/data/stamps/touhou_sakuya_izayoi.json
Normal file
BIN
titles/idac/data/stamps/touhou_sakuya_izayoi.json
Normal file
Binary file not shown.
BIN
titles/idac/data/timeRelease_v0100.json
Normal file
BIN
titles/idac/data/timeRelease_v0100.json
Normal file
Binary file not shown.
BIN
titles/idac/data/timeRelease_v0131.json
Normal file
BIN
titles/idac/data/timeRelease_v0131.json
Normal file
Binary file not shown.
BIN
titles/idac/data/timeRelease_v0141.json
Normal file
BIN
titles/idac/data/timeRelease_v0141.json
Normal file
Binary file not shown.
BIN
titles/idac/data/timeRelease_v0150.json
Normal file
BIN
titles/idac/data/timeRelease_v0150.json
Normal file
Binary file not shown.
BIN
titles/idac/data/timetrial/touhou_flandre_scarlet.json
Normal file
BIN
titles/idac/data/timetrial/touhou_flandre_scarlet.json
Normal file
Binary file not shown.
BIN
titles/idac/data/timetrial/touhou_remilia_scarlet.json
Normal file
BIN
titles/idac/data/timetrial/touhou_remilia_scarlet.json
Normal file
Binary file not shown.
BIN
titles/idac/data/timetrial/touhou_sakuya_izayoi.json
Normal file
BIN
titles/idac/data/timetrial/touhou_sakuya_izayoi.json
Normal file
Binary file not shown.
12
titles/idac/database.py
Normal file
12
titles/idac/database.py
Normal file
@@ -0,0 +1,12 @@
|
||||
from core.data import Data
|
||||
from core.config import CoreConfig
|
||||
from titles.idac.schema.profile import IDACProfileData
|
||||
from titles.idac.schema.item import IDACItemData
|
||||
|
||||
|
||||
class IDACData(Data):
|
||||
def __init__(self, cfg: CoreConfig) -> None:
|
||||
super().__init__(cfg)
|
||||
|
||||
self.profile = IDACProfileData(cfg, self.session)
|
||||
self.item = IDACItemData(cfg, self.session)
|
||||
63
titles/idac/echo.py
Normal file
63
titles/idac/echo.py
Normal file
@@ -0,0 +1,63 @@
|
||||
import logging
|
||||
import socket
|
||||
|
||||
from twisted.internet.protocol import DatagramProtocol
|
||||
from socketserver import BaseRequestHandler, TCPServer
|
||||
from typing import Tuple
|
||||
|
||||
from core.config import CoreConfig
|
||||
from titles.idac.config import IDACConfig
|
||||
from titles.idac.database import IDACData
|
||||
|
||||
|
||||
class IDACEchoUDP(DatagramProtocol):
|
||||
def __init__(self, cfg: CoreConfig, game_cfg: IDACConfig, port: int) -> None:
|
||||
super().__init__()
|
||||
self.port = port
|
||||
self.core_config = cfg
|
||||
self.game_config = game_cfg
|
||||
self.logger = logging.getLogger("idac")
|
||||
|
||||
def datagramReceived(self, data, addr):
|
||||
self.logger.info(
|
||||
f"UDP Ping from from {addr[0]}:{addr[1]} -> {self.port} - {data.hex()}"
|
||||
)
|
||||
self.transport.write(data, addr)
|
||||
|
||||
|
||||
class IDACEchoTCP(BaseRequestHandler):
|
||||
def __init__(
|
||||
self, request, client_address, server, cfg: CoreConfig, game_cfg: IDACConfig
|
||||
) -> None:
|
||||
self.core_config = cfg
|
||||
self.game_config = game_cfg
|
||||
self.logger = logging.getLogger("idac")
|
||||
self.data = IDACData(cfg)
|
||||
super().__init__(request, client_address, server)
|
||||
|
||||
def handle(self):
|
||||
data = self.request.recv(1024).strip()
|
||||
self.logger.debug(
|
||||
f"TCP Ping from {self.client_address[0]}:{self.client_address[1]} -> {self.server.server_address[1]}: {data.hex()}"
|
||||
)
|
||||
self.request.sendall(data)
|
||||
self.request.shutdown(socket.SHUT_WR)
|
||||
|
||||
|
||||
class IDACEchoTCPFactory(TCPServer):
|
||||
def __init__(
|
||||
self,
|
||||
server_address: Tuple[str, int],
|
||||
RequestHandlerClass,
|
||||
cfg: CoreConfig,
|
||||
game_cfg: IDACConfig,
|
||||
bind_and_activate: bool = ...,
|
||||
) -> None:
|
||||
super().__init__(server_address, RequestHandlerClass, bind_and_activate)
|
||||
self.core_config = cfg
|
||||
self.game_config = game_cfg
|
||||
|
||||
def finish_request(self, request, client_address):
|
||||
self.RequestHandlerClass(
|
||||
request, client_address, self, self.core_config, self.game_config
|
||||
)
|
||||
142
titles/idac/frontend.py
Normal file
142
titles/idac/frontend.py
Normal file
@@ -0,0 +1,142 @@
|
||||
import json
|
||||
import yaml
|
||||
import jinja2
|
||||
from os import path
|
||||
from twisted.web.util import redirectTo
|
||||
from twisted.web.http import Request
|
||||
from twisted.web.server import Session
|
||||
|
||||
from core.frontend import FE_Base, IUserSession
|
||||
from core.config import CoreConfig
|
||||
from titles.idac.database import IDACData
|
||||
from titles.idac.schema.profile import *
|
||||
from titles.idac.schema.item import *
|
||||
from titles.idac.config import IDACConfig
|
||||
from titles.idac.const import IDACConstants
|
||||
|
||||
|
||||
class IDACFrontend(FE_Base):
|
||||
def __init__(
|
||||
self, cfg: CoreConfig, environment: jinja2.Environment, cfg_dir: str
|
||||
) -> None:
|
||||
super().__init__(cfg, environment)
|
||||
self.data = IDACData(cfg)
|
||||
self.game_cfg = IDACConfig()
|
||||
if path.exists(f"{cfg_dir}/{IDACConstants.CONFIG_NAME}"):
|
||||
self.game_cfg.update(
|
||||
yaml.safe_load(open(f"{cfg_dir}/{IDACConstants.CONFIG_NAME}"))
|
||||
)
|
||||
self.nav_name = "頭文字D THE ARCADE"
|
||||
# TODO: Add version list
|
||||
self.version = IDACConstants.VER_IDAC_SEASON_2
|
||||
|
||||
self.ticket_names = {
|
||||
3: "car_dressup_points",
|
||||
5: "avatar_points",
|
||||
25: "full_tune_tickets",
|
||||
34: "full_tune_fragments",
|
||||
}
|
||||
|
||||
def generate_all_tables_json(self, user_id: int):
|
||||
json_export = {}
|
||||
|
||||
idac_tables = {
|
||||
profile,
|
||||
config,
|
||||
avatar,
|
||||
rank,
|
||||
stock,
|
||||
theory,
|
||||
car,
|
||||
ticket,
|
||||
story,
|
||||
episode,
|
||||
difficulty,
|
||||
course,
|
||||
trial,
|
||||
challenge,
|
||||
theory_course,
|
||||
theory_partner,
|
||||
theory_running,
|
||||
vs_info,
|
||||
stamp,
|
||||
timetrial_event
|
||||
}
|
||||
|
||||
for table in idac_tables:
|
||||
sql = select(table).where(
|
||||
table.c.user == user_id,
|
||||
)
|
||||
|
||||
# check if the table has a version column
|
||||
if "version" in table.c:
|
||||
sql = sql.where(table.c.version == self.version)
|
||||
|
||||
# lol use the profile connection for items, dirty hack
|
||||
result = self.data.profile.execute(sql)
|
||||
data_list = result.fetchall()
|
||||
|
||||
# add the list to the json export with the correct table name
|
||||
json_export[table.name] = []
|
||||
for data in data_list:
|
||||
tmp = data._asdict()
|
||||
tmp.pop("id")
|
||||
tmp.pop("user")
|
||||
json_export[table.name].append(tmp)
|
||||
|
||||
return json.dumps(json_export, indent=4, default=str, ensure_ascii=False)
|
||||
|
||||
def render_GET(self, request: Request) -> bytes:
|
||||
uri: str = request.uri.decode()
|
||||
|
||||
template = self.environment.get_template(
|
||||
"titles/idac/frontend/idac_index.jinja"
|
||||
)
|
||||
sesh: Session = request.getSession()
|
||||
usr_sesh = IUserSession(sesh)
|
||||
user_id = usr_sesh.userId
|
||||
# user_id = usr_sesh.user_id
|
||||
|
||||
# profile export
|
||||
if uri.startswith("/game/idac/export"):
|
||||
if user_id == 0:
|
||||
return redirectTo(b"/game/idac", request)
|
||||
|
||||
# set the file name, content type and size to download the json
|
||||
content = self.generate_all_tables_json(user_id).encode("utf-8")
|
||||
request.responseHeaders.addRawHeader(
|
||||
b"content-type", b"application/octet-stream"
|
||||
)
|
||||
request.responseHeaders.addRawHeader(
|
||||
b"content-disposition", b"attachment; filename=idac_profile.json"
|
||||
)
|
||||
request.responseHeaders.addRawHeader(
|
||||
b"content-length", str(len(content)).encode("utf-8")
|
||||
)
|
||||
|
||||
self.logger.info(f"User {user_id} exported their IDAC data")
|
||||
return content
|
||||
|
||||
profile_data, tickets, rank = None, None, None
|
||||
if user_id > 0:
|
||||
profile_data = self.data.profile.get_profile(user_id, self.version)
|
||||
ticket_data = self.data.item.get_tickets(user_id)
|
||||
rank = self.data.profile.get_profile_rank(user_id, self.version)
|
||||
|
||||
tickets = {
|
||||
self.ticket_names[ticket["ticket_id"]]: ticket["ticket_cnt"]
|
||||
for ticket in ticket_data
|
||||
}
|
||||
|
||||
return template.render(
|
||||
title=f"{self.core_config.server.name} | {self.nav_name}",
|
||||
game_list=self.environment.globals["game_list"],
|
||||
profile=profile_data,
|
||||
tickets=tickets,
|
||||
rank=rank,
|
||||
sesh=vars(usr_sesh),
|
||||
active_page="idac",
|
||||
).encode("utf-16")
|
||||
|
||||
def render_POST(self, request: Request) -> bytes:
|
||||
pass
|
||||
134
titles/idac/frontend/idac_index.jinja
Normal file
134
titles/idac/frontend/idac_index.jinja
Normal file
@@ -0,0 +1,134 @@
|
||||
{% extends "core/frontend/index.jinja" %}
|
||||
{% block content %}
|
||||
<h1 class="mb-3">頭文字D THE ARCADE</h1>
|
||||
|
||||
{% if sesh is defined and sesh["userId"] > 0 %}
|
||||
<div class="card mb-3">
|
||||
<div class="card-body">
|
||||
<div class="card-title">
|
||||
<div class="d-flex justify-content-between flex-wrap flex-md-nowrap align-items-center">
|
||||
<h3>{{ sesh["username"] }}'s Profile</h3>
|
||||
<div class="btn-toolbar mb-2 mb-md-0">
|
||||
<div class="btn-group me-2">
|
||||
<!--<button type="button" class="btn btn-sm btn-outline-secondary">Share</button>-->
|
||||
<button type="button" data-bs-toggle="modal" data-bs-target="#export"
|
||||
class="btn btn-sm btn-outline-primary">Export</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
<!--<h4 class="card-subtitle mb-2 text-body-secondary">Card subtitle</h4>-->
|
||||
{% if profile is defined and profile is not none %}
|
||||
<div class="row d-flex justify-content-center h-100">
|
||||
<div class="col col-lg-3 col-12">
|
||||
<div class="card mb-3">
|
||||
<div class="card-body p-4">
|
||||
<h5>Information</h5>
|
||||
<hr class="mt-0 mb-4">
|
||||
<h6>Username</h6>
|
||||
<p class="text-muted">{{ profile.username }}</p>
|
||||
<h6>Cash</h6>
|
||||
<p class="text-muted">{{ profile.cash }} D</p>
|
||||
<h6>Grade</h6>
|
||||
<h4>
|
||||
{% set grade = rank.grade %}
|
||||
{% if grade >= 1 and grade <= 72 %}
|
||||
{% set grade_number = (grade - 1) // 9 %}
|
||||
{% set grade_letters = ['E', 'D', 'C', 'B', 'A', 'S', 'SS', 'X'] %}
|
||||
{{ grade_letters[grade_number] }}{{ 9 - ((grade-1) % 9) }}
|
||||
{% else %}
|
||||
Unknown
|
||||
{% endif %}
|
||||
</h4>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
<div class="col col-lg-9 col-12">
|
||||
<div class="card mb-3">
|
||||
|
||||
<div class="card-body p-4">
|
||||
<h5>Statistics</h5>
|
||||
<hr class="mt-0 mb-4">
|
||||
<div class="row pt-1">
|
||||
<div class="col-lg-4 col-md-6 mb-3">
|
||||
<h6>Total Plays</h6>
|
||||
<p class="text-muted">{{ profile.total_play }}</p>
|
||||
</div>
|
||||
<div class="col-lg-4 col-md-6 mb-3">
|
||||
<h6>Last Played</h6>
|
||||
<p class="text-muted">{{ profile.last_play_date }}</p>
|
||||
</div>
|
||||
<div class="col-lg-4 col-md-6 mb-3">
|
||||
<h6>Mileage</h6>
|
||||
<p class="text-muted">{{ profile.mileage / 1000}} km</p>
|
||||
</div>
|
||||
</div>
|
||||
{% if tickets is defined and tickets|length > 0 %}
|
||||
<h5>Tokens/Tickets</h5>
|
||||
<hr class="mt-0 mb-4">
|
||||
<div class="row pt-1">
|
||||
<div class="col-lg-3 col-md-6 mb-3">
|
||||
<h6>Avatar Tokens</h6>
|
||||
<p class="text-muted">{{ tickets.avatar_points }}/30</p>
|
||||
</div>
|
||||
<div class="col-lg-3 col-md-6 mb-3">
|
||||
<h6>Car Dressup Tokens</h6>
|
||||
<p class="text-muted">{{ tickets.car_dressup_points }}/30</p>
|
||||
</div>
|
||||
<div class="col-lg-3 col-md-6 mb-3">
|
||||
<h6>FullTune Tickets</h6>
|
||||
<p class="text-muted">{{ tickets.full_tune_tickets }}/99</p>
|
||||
</div>
|
||||
<div class="col-lg-3 col-md-6 mb-3">
|
||||
<h6>FullTune Fragments</h6>
|
||||
<p class="text-muted">{{ tickets.full_tune_fragments }}/10</p>
|
||||
</div>
|
||||
</div>
|
||||
{% endif %}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
{% else %}
|
||||
<div class="alert alert-warning" role="alert">
|
||||
You need to play 頭文字D THE ARCADE first to view your profile.
|
||||
</div>
|
||||
{% endif %}
|
||||
<!--<a href="#" data-bs-toggle="modal" data-bs-target="#card-add" class="card-link">Add Card</a>-->
|
||||
</div>
|
||||
</div>
|
||||
{% else %}
|
||||
<div class="alert alert-info" role="alert">
|
||||
You need to be logged in to view this page. <a href="/gate">Login</a></a>
|
||||
</div>
|
||||
{% endif %}
|
||||
|
||||
<div class="modal fade" id="export" tabindex="-1" aria-labelledby="export-label" aria-hidden="true">
|
||||
<div class="modal-dialog">
|
||||
<div class="modal-content">
|
||||
<div class="modal-header">
|
||||
<h1 class="modal-title fs-5" id="exort-label">Export Profile</h1>
|
||||
<button type="button" class="btn-close" data-bs-dismiss="modal" aria-label="Close"></button>
|
||||
</div>
|
||||
<div class="modal-body">
|
||||
Download your profile as a <strong>.json</strong> file in order to import it into your local ARTEMiS
|
||||
database.
|
||||
<div class="alert alert-warning mt-3" role="alert">
|
||||
{% if profile is defined and profile is not none %}
|
||||
Are you sure you want to export your profile with the username {{ profile.username }}?
|
||||
{% endif %}
|
||||
</div>
|
||||
</div>
|
||||
<div class="modal-footer">
|
||||
<button type="button" class="btn btn-secondary" data-bs-dismiss="modal">Close</button>
|
||||
<button type="button" class="btn btn-primary" id="exportBtn">Download Profile</a>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<script type="text/javascript">
|
||||
{% include "titles/idac/frontend/js/idac_scripts.js" %}
|
||||
</script>
|
||||
|
||||
{% endblock content %}
|
||||
10
titles/idac/frontend/js/idac_scripts.js
Normal file
10
titles/idac/frontend/js/idac_scripts.js
Normal file
@@ -0,0 +1,10 @@
|
||||
$(document).ready(function () {
|
||||
$('#exportBtn').click(function () {
|
||||
window.location = "/game/idac/export";
|
||||
|
||||
// appendAlert('Successfully exported the profile', 'success');
|
||||
|
||||
// Close the modal on success
|
||||
$('#export').modal('hide');
|
||||
});
|
||||
});
|
||||
167
titles/idac/index.py
Normal file
167
titles/idac/index.py
Normal file
@@ -0,0 +1,167 @@
|
||||
import json
|
||||
import traceback
|
||||
import inflection
|
||||
import yaml
|
||||
import logging
|
||||
import coloredlogs
|
||||
|
||||
from os import path
|
||||
from typing import Dict, List, Tuple
|
||||
from logging.handlers import TimedRotatingFileHandler
|
||||
from twisted.web import server
|
||||
from twisted.web.http import Request
|
||||
from twisted.internet import reactor, endpoints
|
||||
|
||||
from core.config import CoreConfig
|
||||
from core.utils import Utils
|
||||
from titles.idac.base import IDACBase
|
||||
from titles.idac.season2 import IDACSeason2
|
||||
from titles.idac.config import IDACConfig
|
||||
from titles.idac.const import IDACConstants
|
||||
from titles.idac.echo import IDACEchoUDP
|
||||
from titles.idac.matching import IDACMatching
|
||||
|
||||
|
||||
class IDACServlet:
|
||||
def __init__(self, core_cfg: CoreConfig, cfg_dir: str) -> None:
|
||||
self.core_cfg = core_cfg
|
||||
self.game_cfg = IDACConfig()
|
||||
if path.exists(f"{cfg_dir}/{IDACConstants.CONFIG_NAME}"):
|
||||
self.game_cfg.update(
|
||||
yaml.safe_load(open(f"{cfg_dir}/{IDACConstants.CONFIG_NAME}"))
|
||||
)
|
||||
|
||||
self.versions = [
|
||||
IDACBase(core_cfg, self.game_cfg),
|
||||
IDACSeason2(core_cfg, self.game_cfg)
|
||||
]
|
||||
|
||||
self.logger = logging.getLogger("idac")
|
||||
log_fmt_str = "[%(asctime)s] IDAC | %(levelname)s | %(message)s"
|
||||
log_fmt = logging.Formatter(log_fmt_str)
|
||||
fileHandler = TimedRotatingFileHandler(
|
||||
"{0}/{1}.log".format(self.core_cfg.server.log_dir, "idac"),
|
||||
encoding="utf8",
|
||||
when="d",
|
||||
backupCount=10,
|
||||
)
|
||||
|
||||
fileHandler.setFormatter(log_fmt)
|
||||
|
||||
consoleHandler = logging.StreamHandler()
|
||||
consoleHandler.setFormatter(log_fmt)
|
||||
|
||||
self.logger.addHandler(fileHandler)
|
||||
self.logger.addHandler(consoleHandler)
|
||||
|
||||
self.logger.setLevel(self.game_cfg.server.loglevel)
|
||||
coloredlogs.install(
|
||||
level=self.game_cfg.server.loglevel, logger=self.logger, fmt=log_fmt_str
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def is_game_enabled(cls, game_code: str, core_cfg: CoreConfig, cfg_dir: str) -> bool:
|
||||
game_cfg = IDACConfig()
|
||||
|
||||
if path.exists(f"{cfg_dir}/{IDACConstants.CONFIG_NAME}"):
|
||||
game_cfg.update(
|
||||
yaml.safe_load(open(f"{cfg_dir}/{IDACConstants.CONFIG_NAME}"))
|
||||
)
|
||||
|
||||
if not game_cfg.server.enable:
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
def get_endpoint_matchers(self) -> Tuple[List[Tuple[str, str, Dict]], List[Tuple[str, str, Dict]]]:
|
||||
return (
|
||||
[],
|
||||
[("render_POST", "/SDGT/{version}/initiald/{category}/{endpoint}", {})]
|
||||
)
|
||||
|
||||
def get_allnet_info(
|
||||
self, game_code: str, game_ver: int, keychip: str
|
||||
) -> Tuple[bool, str, str]:
|
||||
title_port_int = Utils.get_title_port(self.core_cfg)
|
||||
t_port = f":{title_port_int}" if title_port_int and not self.core_cfg.server.is_using_proxy else ""
|
||||
|
||||
return (
|
||||
f"",
|
||||
# requires http or else it defaults to https
|
||||
f"http://{self.core_cfg.title.hostname}{t_port}/{game_code}/{game_ver}/",
|
||||
)
|
||||
|
||||
def render_POST(self, request: Request, game_code: int, matchers: Dict) -> bytes:
|
||||
req_raw = request.content.getvalue()
|
||||
internal_ver = 0
|
||||
version = int(matchers['version'])
|
||||
category = matchers['category']
|
||||
endpoint = matchers['endpoint']
|
||||
client_ip = Utils.get_ip_addr(request)
|
||||
|
||||
if version >= 100 and version < 140: # IDAC Season 1
|
||||
internal_ver = IDACConstants.VER_IDAC_SEASON_1
|
||||
elif version >= 140 and version < 171: # IDAC Season 2
|
||||
internal_ver = IDACConstants.VER_IDAC_SEASON_2
|
||||
|
||||
header_application = self.decode_header(request.getAllHeaders())
|
||||
|
||||
req_data = json.loads(req_raw)
|
||||
|
||||
self.logger.info(f"v{version} {endpoint} request from {client_ip}")
|
||||
self.logger.debug(f"Headers: {header_application}")
|
||||
self.logger.debug(req_data)
|
||||
|
||||
# func_to_find = "handle_" + inflection.underscore(endpoint) + "_request"
|
||||
func_to_find = "handle_"
|
||||
func_to_find += f"{category.lower()}_" if not category == "" else ""
|
||||
func_to_find += f"{endpoint.lower()}_request"
|
||||
|
||||
if not hasattr(self.versions[internal_ver], func_to_find):
|
||||
self.logger.warning(f"Unhandled v{version} request {endpoint}")
|
||||
return '{"status_code": "0"}'.encode("utf-8")
|
||||
|
||||
resp = None
|
||||
try:
|
||||
handler = getattr(self.versions[internal_ver], func_to_find)
|
||||
resp = handler(req_data, header_application)
|
||||
|
||||
except Exception as e:
|
||||
traceback.print_exc()
|
||||
self.logger.error(f"Error handling v{version} method {endpoint} - {e}")
|
||||
return '{"status_code": "0"}'.encode("utf-8")
|
||||
|
||||
if resp is None:
|
||||
resp = {"status_code": "0"}
|
||||
|
||||
self.logger.debug(f"Response {resp}")
|
||||
return json.dumps(resp, ensure_ascii=False).encode("utf-8")
|
||||
|
||||
|
||||
def decode_header(self, data: Dict) -> Dict:
|
||||
app: str = data[b"application"].decode()
|
||||
ret = {}
|
||||
|
||||
for x in app.split(", "):
|
||||
y = x.split("=")
|
||||
ret[y[0]] = y[1].replace('"', "")
|
||||
|
||||
return ret
|
||||
|
||||
def setup(self):
|
||||
if self.game_cfg.server.enable:
|
||||
endpoints.serverFromString(
|
||||
reactor,
|
||||
f"tcp:{self.game_cfg.server.matching}:interface={self.core_cfg.server.listen_address}",
|
||||
).listen(server.Site(IDACMatching(self.core_cfg, self.game_cfg)))
|
||||
|
||||
reactor.listenUDP(
|
||||
self.game_cfg.server.echo1,
|
||||
IDACEchoUDP(self.core_cfg, self.game_cfg, self.game_cfg.server.echo1),
|
||||
)
|
||||
reactor.listenUDP(
|
||||
self.game_cfg.server.echo2,
|
||||
IDACEchoUDP(self.core_cfg, self.game_cfg, self.game_cfg.server.echo2),
|
||||
)
|
||||
|
||||
self.logger.info(f"Matching listening on {self.game_cfg.server.matching} with echos on {self.game_cfg.server.echo1} and {self.game_cfg.server.echo2}")
|
||||
72
titles/idac/matching.py
Normal file
72
titles/idac/matching.py
Normal file
@@ -0,0 +1,72 @@
|
||||
import json
|
||||
import logging
|
||||
|
||||
from typing import Dict
|
||||
from twisted.web import resource
|
||||
|
||||
from core import CoreConfig
|
||||
from titles.idac.season2 import IDACBase
|
||||
from titles.idac.config import IDACConfig
|
||||
|
||||
|
||||
class IDACMatching(resource.Resource):
|
||||
isLeaf = True
|
||||
|
||||
def __init__(self, cfg: CoreConfig, game_cfg: IDACConfig) -> None:
|
||||
self.core_config = cfg
|
||||
self.game_config = game_cfg
|
||||
self.base = IDACBase(cfg, game_cfg)
|
||||
self.logger = logging.getLogger("idac")
|
||||
|
||||
self.queue = 0
|
||||
|
||||
def get_matching_state(self):
|
||||
if self.queue >= 1:
|
||||
self.queue -= 1
|
||||
return 0
|
||||
else:
|
||||
return 1
|
||||
|
||||
def render_POST(self, req) -> bytes:
|
||||
url = req.uri.decode()
|
||||
req_data = json.loads(req.content.getvalue().decode())
|
||||
header_application = self.decode_header(req.getAllHeaders())
|
||||
user_id = int(header_application["session"])
|
||||
|
||||
# self.getMatchingStatus(user_id)
|
||||
|
||||
self.logger.info(
|
||||
f"IDAC Matching request from {req.getClientIP()}: {url} - {req_data}"
|
||||
)
|
||||
|
||||
resp = {"status_code": "0"}
|
||||
if url == "/regist":
|
||||
self.queue = self.queue + 1
|
||||
elif url == "/status":
|
||||
if req_data.get("cancel_flag"):
|
||||
self.queue = self.queue - 1
|
||||
self.logger.info(
|
||||
f"IDAC Matching endpoint {req.getClientIP()} had quited"
|
||||
)
|
||||
|
||||
resp = {
|
||||
"status_code": "0",
|
||||
# Only IPv4 is supported
|
||||
"host": self.game_config.server.matching_host,
|
||||
"port": self.game_config.server.matching_p2p,
|
||||
"room_name": "INDTA",
|
||||
"state": 1,
|
||||
}
|
||||
|
||||
self.logger.debug(f"Response {resp}")
|
||||
return json.dumps(resp, ensure_ascii=False).encode("utf-8")
|
||||
|
||||
def decode_header(self, data: Dict) -> Dict:
|
||||
app: str = data[b"application"].decode()
|
||||
ret = {}
|
||||
|
||||
for x in app.split(", "):
|
||||
y = x.split("=")
|
||||
ret[y[0]] = y[1].replace('"', "")
|
||||
|
||||
return ret
|
||||
161
titles/idac/read.py
Normal file
161
titles/idac/read.py
Normal file
@@ -0,0 +1,161 @@
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
from read import BaseReader
|
||||
from core.data import Data
|
||||
from core.config import CoreConfig
|
||||
from titles.idac.const import IDACConstants
|
||||
from titles.idac.database import IDACData
|
||||
from titles.idac.schema.profile import *
|
||||
from titles.idac.schema.item import *
|
||||
|
||||
|
||||
class IDACReader(BaseReader):
|
||||
def __init__(
|
||||
self,
|
||||
config: CoreConfig,
|
||||
version: int,
|
||||
bin_dir: Optional[str],
|
||||
opt_dir: Optional[str],
|
||||
extra: Optional[str],
|
||||
) -> None:
|
||||
super().__init__(config, version, bin_dir, opt_dir, extra)
|
||||
self.card_data = Data(config).card
|
||||
self.data = IDACData(config)
|
||||
|
||||
try:
|
||||
self.logger.info(
|
||||
f"Start importer for {IDACConstants.game_ver_to_string(version)}"
|
||||
)
|
||||
except IndexError:
|
||||
self.logger.error(f"Invalid Initial D THE ARCADE version {version}")
|
||||
exit(1)
|
||||
|
||||
def read(self) -> None:
|
||||
if self.bin_dir is None and self.opt_dir is None:
|
||||
self.logger.error(
|
||||
(
|
||||
"To import your profile specify the '--optfolder'",
|
||||
" path to your idac_profile.json file, exiting",
|
||||
)
|
||||
)
|
||||
exit(1)
|
||||
|
||||
if self.opt_dir is not None:
|
||||
if not os.path.exists(self.opt_dir):
|
||||
self.logger.error(
|
||||
f"Path to idac_profile.json does not exist: {self.opt_dir}"
|
||||
)
|
||||
exit(1)
|
||||
|
||||
if os.path.isdir(self.opt_dir):
|
||||
self.opt_dir = os.path.join(self.opt_dir, "idac_profile.json")
|
||||
|
||||
if not os.path.isfile(self.opt_dir) or self.opt_dir[-5:] != ".json":
|
||||
self.logger.error(
|
||||
f"Path to idac_profile.json does not exist: {self.opt_dir}"
|
||||
)
|
||||
exit(1)
|
||||
|
||||
self.read_idac_profile(self.opt_dir)
|
||||
|
||||
def read_idac_profile(self, file_path: str) -> None:
|
||||
self.logger.info(f"Reading profile from {file_path}...")
|
||||
|
||||
# read it as binary to avoid encoding issues
|
||||
profile_data: Dict[str, Any] = {}
|
||||
with open(file_path, "rb") as f:
|
||||
profile_data = json.loads(f.read().decode("utf-8"))
|
||||
|
||||
if not profile_data:
|
||||
self.logger.error("Profile could not be parsed, exiting")
|
||||
exit(1)
|
||||
|
||||
access_code = None
|
||||
while access_code is None:
|
||||
access_code = input("Enter your 20 digits access code: ")
|
||||
if len(access_code) != 20 or not access_code.isdigit():
|
||||
access_code = None
|
||||
self.logger.warning("Invalid access code, please try again.")
|
||||
|
||||
# check if access code already exists, if not create a new profile
|
||||
user_id = self.card_data.get_user_id_from_card(access_code)
|
||||
if user_id is None:
|
||||
choice = input("Access code does not exist, do you want to create a new profile? (Y/n): ")
|
||||
if choice.lower() == "n":
|
||||
self.logger.info("Exiting...")
|
||||
exit(0)
|
||||
|
||||
user_id = self.data.user.create_user()
|
||||
|
||||
if user_id is None:
|
||||
self.logger.error("Failed to register user!")
|
||||
user_id = -1
|
||||
|
||||
else:
|
||||
card_id = self.data.card.create_card(user_id, access_code)
|
||||
|
||||
if card_id is None:
|
||||
self.logger.error("Failed to register card!")
|
||||
user_id = -1
|
||||
|
||||
if user_id == -1:
|
||||
self.logger.error("Failed to create profile, exiting")
|
||||
exit(1)
|
||||
|
||||
# table mapping to insert the data properly
|
||||
tables = {
|
||||
"idac_profile": profile,
|
||||
"idac_profile_config": config,
|
||||
"idac_profile_avatar": avatar,
|
||||
"idac_profile_rank": rank,
|
||||
"idac_profile_stock": stock,
|
||||
"idac_profile_theory": theory,
|
||||
"idac_user_car": car,
|
||||
"idac_user_ticket": ticket,
|
||||
"idac_user_story": story,
|
||||
"idac_user_story_episode": episode,
|
||||
"idac_user_story_episode_difficulty": difficulty,
|
||||
"idac_user_course": course,
|
||||
"idac_user_time_trial": trial,
|
||||
"idac_user_challenge": challenge,
|
||||
"idac_user_theory_course": theory_course,
|
||||
"idac_user_theory_partner": theory_partner,
|
||||
"idac_user_theory_running": theory_running,
|
||||
"idac_user_vs_info": vs_info,
|
||||
"idac_user_stamp": stamp,
|
||||
"idac_user_timetrial_event": timetrial_event,
|
||||
}
|
||||
|
||||
for name, data_list in profile_data.items():
|
||||
# get the SQLAlchemy table object from the name
|
||||
table = tables.get(name)
|
||||
if table is None:
|
||||
self.logger.warning(f"Unknown table {name}, skipping")
|
||||
continue
|
||||
|
||||
for data in data_list:
|
||||
# add user to the data
|
||||
data["user"] = user_id
|
||||
|
||||
# check if the table has a version column
|
||||
if "version" in table.c:
|
||||
data["version"] = self.version
|
||||
|
||||
sql = insert(table).values(
|
||||
**data
|
||||
)
|
||||
|
||||
# lol use the profile connection for items, dirty hack
|
||||
conflict = sql.on_duplicate_key_update(**data)
|
||||
result = self.data.profile.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.error(f"Failed to insert data into table {name}")
|
||||
exit(1)
|
||||
|
||||
self.logger.info(f"Inserted data into table {name}")
|
||||
|
||||
self.logger.info("Profile import complete!")
|
||||
983
titles/idac/schema/item.py
Normal file
983
titles/idac/schema/item.py
Normal file
@@ -0,0 +1,983 @@
|
||||
from typing import Dict, Optional, List
|
||||
from sqlalchemy import (
|
||||
Table,
|
||||
Column,
|
||||
UniqueConstraint,
|
||||
PrimaryKeyConstraint,
|
||||
and_,
|
||||
update,
|
||||
)
|
||||
from sqlalchemy.types import Integer, String, TIMESTAMP, Boolean, JSON
|
||||
from sqlalchemy.schema import ForeignKey
|
||||
from sqlalchemy.engine import Row
|
||||
from sqlalchemy.sql import func, select
|
||||
from sqlalchemy.dialects.mysql import insert
|
||||
|
||||
from core.data.schema import BaseData, metadata
|
||||
|
||||
car = Table(
|
||||
"idac_user_car",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("car_id", Integer),
|
||||
Column("style_car_id", Integer),
|
||||
Column("color", Integer),
|
||||
Column("bureau", Integer),
|
||||
Column("kana", Integer),
|
||||
Column("s_no", Integer),
|
||||
Column("l_no", Integer),
|
||||
Column("car_flag", Integer),
|
||||
Column("tune_point", Integer),
|
||||
Column("tune_level", Integer, server_default="1"),
|
||||
Column("tune_parts", Integer),
|
||||
Column("infinity_tune", Integer, server_default="0"),
|
||||
Column("online_vs_win", Integer, server_default="0"),
|
||||
Column(
|
||||
"pickup_seq", Integer, server_default="1"
|
||||
), # the order in which the car was picked up
|
||||
Column(
|
||||
"purchase_seq", Integer, server_default="1"
|
||||
), # the order in which the car was purchased
|
||||
Column("color_stock_list", String(32)),
|
||||
Column("color_stock_new_list", String(32)),
|
||||
Column("parts_stock_list", String(48)),
|
||||
Column("parts_stock_new_list", String(48)),
|
||||
Column("parts_set_equip_list", String(48)),
|
||||
Column("parts_list", JSON),
|
||||
Column("equip_parts_count", Integer, server_default="0"),
|
||||
Column("total_car_parts_count", Integer, server_default="0"),
|
||||
Column("use_count", Integer, server_default="0"),
|
||||
Column("story_use_count", Integer, server_default="0"),
|
||||
Column("timetrial_use_count", Integer, server_default="0"),
|
||||
Column("vs_use_count", Integer, server_default="0"),
|
||||
Column("net_vs_use_count", Integer, server_default="0"),
|
||||
Column("theory_use_count", Integer, server_default="0"),
|
||||
Column("car_mileage", Integer, server_default="0"),
|
||||
UniqueConstraint("user", "version", "style_car_id", name="idac_user_car_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
ticket = Table(
|
||||
"idac_user_ticket",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("ticket_id", Integer),
|
||||
Column("ticket_cnt", Integer),
|
||||
UniqueConstraint("user", "ticket_id", name="idac_user_ticket_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
story = Table(
|
||||
"idac_user_story",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("story_type", Integer),
|
||||
Column("chapter", Integer),
|
||||
Column("loop_count", Integer, server_default="1"),
|
||||
UniqueConstraint("user", "chapter", name="idac_user_story_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
episode = Table(
|
||||
"idac_user_story_episode",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("chapter", Integer),
|
||||
Column("episode", Integer),
|
||||
Column("play_status", Integer),
|
||||
UniqueConstraint("user", "chapter", "episode", name="idac_user_story_episode_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
difficulty = Table(
|
||||
"idac_user_story_episode_difficulty",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("episode", Integer),
|
||||
Column("difficulty", Integer),
|
||||
Column("play_count", Integer),
|
||||
Column("clear_count", Integer),
|
||||
Column("play_status", Integer),
|
||||
Column("play_score", Integer),
|
||||
UniqueConstraint(
|
||||
"user", "episode", "difficulty", name="idac_user_story_episode_difficulty_uk"
|
||||
),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
course = Table(
|
||||
"idac_user_course",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("course_id", Integer),
|
||||
Column("run_counts", Integer, server_default="1"),
|
||||
Column("skill_level_exp", Integer, server_default="0"),
|
||||
UniqueConstraint("user", "course_id", name="idac_user_course_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
trial = Table(
|
||||
"idac_user_time_trial",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("style_car_id", Integer),
|
||||
Column("course_id", Integer),
|
||||
Column("eval_id", Integer, server_default="0"),
|
||||
Column("goal_time", Integer),
|
||||
Column("section_time_1", Integer),
|
||||
Column("section_time_2", Integer),
|
||||
Column("section_time_3", Integer),
|
||||
Column("section_time_4", Integer),
|
||||
Column("mission", Integer),
|
||||
Column("play_dt", TIMESTAMP, server_default=func.now()),
|
||||
UniqueConstraint(
|
||||
"user", "version", "course_id", "style_car_id", name="idac_user_time_trial_uk"
|
||||
),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
challenge = Table(
|
||||
"idac_user_challenge",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("vs_type", Integer),
|
||||
Column("play_difficulty", Integer),
|
||||
Column("cleared_difficulty", Integer),
|
||||
Column("story_type", Integer),
|
||||
Column("play_count", Integer, server_default="1"),
|
||||
Column("weak_difficulty", Integer, server_default="0"),
|
||||
Column("eval_id", Integer),
|
||||
Column("advantage", Integer),
|
||||
Column("sec1_advantage_avg", Integer),
|
||||
Column("sec2_advantage_avg", Integer),
|
||||
Column("sec3_advantage_avg", Integer),
|
||||
Column("sec4_advantage_avg", Integer),
|
||||
Column("nearby_advantage_rate", Integer),
|
||||
Column("win_flag", Integer),
|
||||
Column("result", Integer),
|
||||
Column("record", Integer),
|
||||
Column("course_id", Integer),
|
||||
Column("last_play_course_id", Integer),
|
||||
Column("style_car_id", Integer),
|
||||
Column("course_day", Integer),
|
||||
UniqueConstraint(
|
||||
"user", "vs_type", "play_difficulty", name="idac_user_challenge_uk"
|
||||
),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
theory_course = Table(
|
||||
"idac_user_theory_course",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("course_id", Integer),
|
||||
Column("max_victory_grade", Integer, server_default="0"),
|
||||
Column("run_count", Integer, server_default="1"),
|
||||
Column("powerhouse_lv", Integer),
|
||||
Column("powerhouse_exp", Integer),
|
||||
Column("played_powerhouse_lv", Integer),
|
||||
Column("update_dt", TIMESTAMP, server_default=func.now()),
|
||||
UniqueConstraint("user", "course_id", name="idac_user_theory_course_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
theory_partner = Table(
|
||||
"idac_user_theory_partner",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("partner_id", Integer),
|
||||
Column("fellowship_lv", Integer),
|
||||
Column("fellowship_exp", Integer),
|
||||
UniqueConstraint("user", "partner_id", name="idac_user_theory_partner_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
theory_running = Table(
|
||||
"idac_user_theory_running",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("course_id", Integer),
|
||||
Column("attack", Integer),
|
||||
Column("defense", Integer),
|
||||
Column("safety", Integer),
|
||||
Column("runaway", Integer),
|
||||
Column("trick_flag", Integer),
|
||||
UniqueConstraint("user", "course_id", name="idac_user_theory_running_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
vs_info = Table(
|
||||
"idac_user_vs_info",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("group_key", String(25)),
|
||||
Column("win_flg", Integer),
|
||||
Column("style_car_id", Integer),
|
||||
Column("course_id", Integer),
|
||||
Column("course_day", Integer),
|
||||
Column("players_num", Integer),
|
||||
Column("winning", Integer),
|
||||
Column("advantage_1", Integer),
|
||||
Column("advantage_2", Integer),
|
||||
Column("advantage_3", Integer),
|
||||
Column("advantage_4", Integer),
|
||||
Column("select_course_id", Integer),
|
||||
Column("select_course_day", Integer),
|
||||
Column("select_course_random", Integer),
|
||||
Column("matching_success_sec", Integer),
|
||||
Column("boost_flag", Integer),
|
||||
Column("vs_history", Integer),
|
||||
Column("break_count", Integer),
|
||||
Column("break_penalty_flag", Integer),
|
||||
UniqueConstraint("user", "group_key", name="idac_user_vs_info_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
stamp = Table(
|
||||
"idac_user_stamp",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column(
|
||||
"user",
|
||||
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
||||
nullable=False,
|
||||
),
|
||||
Column("m_stamp_event_id", Integer),
|
||||
Column("select_flag", Integer),
|
||||
Column("stamp_masu", Integer),
|
||||
Column("daily_bonus", Integer),
|
||||
Column("weekly_bonus", Integer),
|
||||
Column("weekday_bonus", Integer),
|
||||
Column("weekend_bonus", Integer),
|
||||
Column("total_bonus", Integer),
|
||||
Column("day_total_bonus", Integer),
|
||||
Column("store_battle_bonus", Integer),
|
||||
Column("story_bonus", Integer),
|
||||
Column("online_battle_bonus", Integer),
|
||||
Column("timetrial_bonus", Integer),
|
||||
Column("fasteststreetlegaltheory_bonus", Integer),
|
||||
Column("collaboration_bonus", Integer),
|
||||
Column("add_bonus_daily_flag_1", Integer),
|
||||
Column("add_bonus_daily_flag_2", Integer),
|
||||
Column("add_bonus_daily_flag_3", Integer),
|
||||
Column("create_date_daily", TIMESTAMP, server_default=func.now()),
|
||||
Column("create_date_weekly", TIMESTAMP, server_default=func.now()),
|
||||
UniqueConstraint("user", "m_stamp_event_id", name="idac_user_stamp_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
timetrial_event = Table(
|
||||
"idac_user_timetrial_event",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column(
|
||||
"user",
|
||||
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
||||
nullable=False,
|
||||
),
|
||||
Column("timetrial_event_id", Integer),
|
||||
Column("point", Integer),
|
||||
UniqueConstraint("user", "timetrial_event_id", name="idac_user_timetrial_event_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
|
||||
class IDACItemData(BaseData):
|
||||
def get_random_user_car(self, aime_id: int, version: int) -> Optional[List[Row]]:
|
||||
sql = (
|
||||
select(car)
|
||||
.where(and_(car.c.user == aime_id, car.c.version == version))
|
||||
.order_by(func.rand())
|
||||
.limit(1)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_random_car(self, version: int) -> Optional[List[Row]]:
|
||||
sql = select(car).where(car.c.version == version).order_by(func.rand()).limit(1)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_car(
|
||||
self, aime_id: int, version: int, style_car_id: int
|
||||
) -> Optional[List[Row]]:
|
||||
sql = select(car).where(
|
||||
and_(
|
||||
car.c.user == aime_id,
|
||||
car.c.version == version,
|
||||
car.c.style_car_id == style_car_id,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_cars(
|
||||
self, version: int, aime_id: int, only_pickup: bool = False
|
||||
) -> Optional[List[Row]]:
|
||||
if only_pickup:
|
||||
sql = select(car).where(
|
||||
and_(
|
||||
car.c.user == aime_id,
|
||||
car.c.version == version,
|
||||
car.c.pickup_seq != 0,
|
||||
)
|
||||
)
|
||||
else:
|
||||
sql = select(car).where(
|
||||
and_(car.c.user == aime_id, car.c.version == version)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_ticket(self, aime_id: int, ticket_id: int) -> Optional[Row]:
|
||||
sql = select(ticket).where(
|
||||
ticket.c.user == aime_id, ticket.c.ticket_id == ticket_id
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_tickets(self, aime_id: int) -> Optional[List[Row]]:
|
||||
sql = select(ticket).where(ticket.c.user == aime_id)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_story(self, aime_id: int, chapter_id: int) -> Optional[Row]:
|
||||
sql = select(story).where(
|
||||
and_(story.c.user == aime_id, story.c.chapter == chapter_id)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_stories(self, aime_id: int) -> Optional[List[Row]]:
|
||||
sql = select(story).where(story.c.user == aime_id)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_story_episodes(self, aime_id: int, chapter_id: int) -> Optional[List[Row]]:
|
||||
sql = select(episode).where(
|
||||
and_(episode.c.user == aime_id, episode.c.chapter == chapter_id)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_story_episode(self, aime_id: int, episode_id: int) -> Optional[Row]:
|
||||
sql = select(episode).where(
|
||||
and_(episode.c.user == aime_id, episode.c.episode == episode_id)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_story_episode_difficulties(
|
||||
self, aime_id: int, episode_id: int
|
||||
) -> Optional[List[Row]]:
|
||||
sql = select(difficulty).where(
|
||||
and_(difficulty.c.user == aime_id, difficulty.c.episode == episode_id)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_courses(self, aime_id: int) -> Optional[List[Row]]:
|
||||
sql = select(course).where(course.c.user == aime_id)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_course(self, aime_id: int, course_id: int) -> Optional[Row]:
|
||||
sql = select(course).where(
|
||||
and_(course.c.user == aime_id, course.c.course_id == course_id)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_time_trial_courses(self, version: int) -> Optional[List[Row]]:
|
||||
sql = select(trial.c.course_id).where(trial.c.version == version).distinct()
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_time_trial_user_best_time_by_course_car(
|
||||
self, version: int, aime_id: int, course_id: int, style_car_id: int
|
||||
) -> Optional[Row]:
|
||||
sql = select(trial).where(
|
||||
and_(
|
||||
trial.c.user == aime_id,
|
||||
trial.c.version == version,
|
||||
trial.c.course_id == course_id,
|
||||
trial.c.style_car_id == style_car_id,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_time_trial_user_best_courses(
|
||||
self, version: int, aime_id: int
|
||||
) -> Optional[List[Row]]:
|
||||
# get for a given aime_id the best time for each course
|
||||
subquery = (
|
||||
select(
|
||||
trial.c.version,
|
||||
func.min(trial.c.goal_time).label("min_goal_time"),
|
||||
trial.c.course_id,
|
||||
)
|
||||
.where(and_(trial.c.version == version, trial.c.user == aime_id))
|
||||
.group_by(trial.c.course_id)
|
||||
.subquery()
|
||||
)
|
||||
|
||||
# now get the full row for each best time
|
||||
sql = select(trial).where(
|
||||
and_(
|
||||
trial.c.version == subquery.c.version,
|
||||
trial.c.goal_time == subquery.c.min_goal_time,
|
||||
trial.c.course_id == subquery.c.course_id,
|
||||
trial.c.user == aime_id,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_time_trial_best_cars_by_course(
|
||||
self, version: int, course_id: int, aime_id: Optional[int] = None
|
||||
) -> Optional[List[Row]]:
|
||||
subquery = (
|
||||
select(
|
||||
trial.c.version,
|
||||
func.min(trial.c.goal_time).label("min_goal_time"),
|
||||
trial.c.style_car_id,
|
||||
)
|
||||
.where(
|
||||
and_(
|
||||
trial.c.version == version,
|
||||
trial.c.course_id == course_id,
|
||||
)
|
||||
)
|
||||
)
|
||||
|
||||
if aime_id is not None:
|
||||
subquery = subquery.where(trial.c.user == aime_id)
|
||||
|
||||
subquery = subquery.group_by(trial.c.style_car_id).subquery()
|
||||
|
||||
sql = select(trial).where(
|
||||
and_(
|
||||
trial.c.version == subquery.c.version,
|
||||
trial.c.goal_time == subquery.c.min_goal_time,
|
||||
trial.c.style_car_id == subquery.c.style_car_id,
|
||||
trial.c.course_id == course_id,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_time_trial_ranking_by_course(
|
||||
self,
|
||||
version: int,
|
||||
course_id: int,
|
||||
style_car_id: Optional[int] = None,
|
||||
limit: Optional[int] = 10,
|
||||
) -> Optional[List[Row]]:
|
||||
# get the top 10 ranking by goal_time for a given course which is grouped by user
|
||||
subquery = select(
|
||||
trial.c.version,
|
||||
trial.c.user,
|
||||
func.min(trial.c.goal_time).label("min_goal_time"),
|
||||
).where(and_(trial.c.version == version, trial.c.course_id == course_id))
|
||||
|
||||
# if wantd filter only by style_car_id
|
||||
if style_car_id is not None:
|
||||
subquery = subquery.where(trial.c.style_car_id == style_car_id)
|
||||
|
||||
subquery = subquery.group_by(trial.c.user).subquery()
|
||||
|
||||
sql = (
|
||||
select(trial)
|
||||
.where(
|
||||
and_(
|
||||
trial.c.version == subquery.c.version,
|
||||
trial.c.user == subquery.c.user,
|
||||
trial.c.goal_time == subquery.c.min_goal_time,
|
||||
),
|
||||
)
|
||||
.order_by(trial.c.goal_time)
|
||||
)
|
||||
|
||||
# limit the result if needed
|
||||
if limit is not None:
|
||||
sql = sql.limit(limit)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_time_trial_best_ranking_by_course(
|
||||
self, version: int, aime_id: int, course_id: int
|
||||
) -> Optional[Row]:
|
||||
sql = (
|
||||
select(trial)
|
||||
.where(
|
||||
and_(
|
||||
trial.c.version == version,
|
||||
trial.c.user == aime_id,
|
||||
trial.c.course_id == course_id,
|
||||
),
|
||||
)
|
||||
.order_by(trial.c.goal_time)
|
||||
.limit(1)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_challenge(
|
||||
self, aime_id: int, vs_type: int, play_difficulty: int
|
||||
) -> Optional[Row]:
|
||||
sql = select(challenge).where(
|
||||
and_(
|
||||
challenge.c.user == aime_id,
|
||||
challenge.c.vs_type == vs_type,
|
||||
challenge.c.play_difficulty == play_difficulty,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_challenges(self, aime_id: int) -> Optional[List[Row]]:
|
||||
sql = select(challenge).where(challenge.c.user == aime_id)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_best_challenges_by_vs_type(
|
||||
self, aime_id: int, story_type: int = 4
|
||||
) -> Optional[List[Row]]:
|
||||
subquery = (
|
||||
select(
|
||||
challenge.c.story_type,
|
||||
challenge.c.user,
|
||||
challenge.c.vs_type,
|
||||
func.max(challenge.c.play_difficulty).label("last_play_lv"),
|
||||
)
|
||||
.where(
|
||||
and_(challenge.c.user == aime_id, challenge.c.story_type == story_type)
|
||||
)
|
||||
.group_by(challenge.c.vs_type)
|
||||
)
|
||||
|
||||
sql = (
|
||||
select(
|
||||
challenge.c.story_type,
|
||||
challenge.c.vs_type,
|
||||
challenge.c.cleared_difficulty.label("max_clear_lv"),
|
||||
challenge.c.play_difficulty.label("last_play_lv"),
|
||||
challenge.c.course_id,
|
||||
challenge.c.play_count,
|
||||
)
|
||||
.where(
|
||||
and_(
|
||||
challenge.c.user == subquery.c.user,
|
||||
challenge.c.vs_type == subquery.c.vs_type,
|
||||
challenge.c.play_difficulty == subquery.c.last_play_lv,
|
||||
),
|
||||
)
|
||||
.order_by(challenge.c.vs_type)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_theory_courses(self, aime_id: int) -> Optional[List[Row]]:
|
||||
sql = select(theory_course).where(theory_course.c.user == aime_id)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_theory_course_by_powerhouse_lv(
|
||||
self, aime_id: int, course_id: int, powerhouse_lv: int, count: int = 3
|
||||
) -> Optional[List[Row]]:
|
||||
sql = (
|
||||
select(theory_course)
|
||||
.where(
|
||||
and_(
|
||||
theory_course.c.user != aime_id,
|
||||
theory_course.c.course_id == course_id,
|
||||
theory_course.c.powerhouse_lv == powerhouse_lv,
|
||||
)
|
||||
)
|
||||
.order_by(func.rand())
|
||||
.limit(count)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_theory_course(self, aime_id: int, course_id: int) -> Optional[List[Row]]:
|
||||
sql = select(theory_course).where(
|
||||
and_(
|
||||
theory_course.c.user == aime_id, theory_course.c.course_id == course_id
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_theory_partners(self, aime_id: int) -> Optional[List[Row]]:
|
||||
sql = select(theory_partner).where(theory_partner.c.user == aime_id)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_theory_running(self, aime_id: int) -> Optional[List[Row]]:
|
||||
sql = select(theory_running).where(theory_running.c.user == aime_id)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_theory_running_by_course(
|
||||
self, aime_id: int, course_id: int
|
||||
) -> Optional[Row]:
|
||||
sql = select(theory_running).where(
|
||||
and_(
|
||||
theory_running.c.user == aime_id,
|
||||
theory_running.c.course_id == course_id,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_vs_infos(self, aime_id: int) -> Optional[List[Row]]:
|
||||
sql = select(vs_info).where(vs_info.c.user == aime_id)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_stamps(self, aime_id: int) -> Optional[List[Row]]:
|
||||
sql = select(stamp).where(
|
||||
and_(
|
||||
stamp.c.user == aime_id,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_timetrial_event(self, aime_id: int, timetrial_event_id: int) -> Optional[Row]:
|
||||
sql = select(timetrial_event).where(
|
||||
and_(
|
||||
timetrial_event.c.user == aime_id,
|
||||
timetrial_event.c.timetrial_event_id == timetrial_event_id,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def put_car(self, aime_id: int, version: int, car_data: Dict) -> Optional[int]:
|
||||
car_data["user"] = aime_id
|
||||
car_data["version"] = version
|
||||
|
||||
sql = insert(car).values(**car_data)
|
||||
conflict = sql.on_duplicate_key_update(**car_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_car: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_ticket(self, aime_id: int, ticket_data: Dict) -> Optional[int]:
|
||||
ticket_data["user"] = aime_id
|
||||
|
||||
sql = insert(ticket).values(**ticket_data)
|
||||
conflict = sql.on_duplicate_key_update(**ticket_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_ticket: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_story(self, aime_id: int, story_data: Dict) -> Optional[int]:
|
||||
story_data["user"] = aime_id
|
||||
|
||||
sql = insert(story).values(**story_data)
|
||||
conflict = sql.on_duplicate_key_update(**story_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_story: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_story_episode_play_status(
|
||||
self, aime_id: int, chapter_id: int, play_status: int = 1
|
||||
) -> Optional[int]:
|
||||
sql = (
|
||||
update(episode)
|
||||
.where(and_(episode.c.user == aime_id, episode.c.chapter == chapter_id))
|
||||
.values(play_status=play_status)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
self.logger.warn(
|
||||
f"put_story_episode_play_status: Failed to update! aime_id: {aime_id}"
|
||||
)
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_story_episode(
|
||||
self, aime_id: int, chapter_id: int, episode_data: Dict
|
||||
) -> Optional[int]:
|
||||
episode_data["user"] = aime_id
|
||||
episode_data["chapter"] = chapter_id
|
||||
|
||||
sql = insert(episode).values(**episode_data)
|
||||
conflict = sql.on_duplicate_key_update(**episode_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_story_episode: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_story_episode_difficulty(
|
||||
self, aime_id: int, episode_id: int, difficulty_data: Dict
|
||||
) -> Optional[int]:
|
||||
difficulty_data["user"] = aime_id
|
||||
difficulty_data["episode"] = episode_id
|
||||
|
||||
sql = insert(difficulty).values(**difficulty_data)
|
||||
conflict = sql.on_duplicate_key_update(**difficulty_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(
|
||||
f"put_story_episode_difficulty: Failed to update! aime_id: {aime_id}"
|
||||
)
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_course(self, aime_id: int, course_data: Dict) -> Optional[int]:
|
||||
course_data["user"] = aime_id
|
||||
|
||||
sql = insert(course).values(**course_data)
|
||||
conflict = sql.on_duplicate_key_update(**course_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_course: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_time_trial(
|
||||
self, version: int, aime_id: int, time_trial_data: Dict
|
||||
) -> Optional[int]:
|
||||
time_trial_data["user"] = aime_id
|
||||
time_trial_data["version"] = version
|
||||
|
||||
sql = insert(trial).values(**time_trial_data)
|
||||
conflict = sql.on_duplicate_key_update(**time_trial_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_time_trial: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_challenge(self, aime_id: int, challenge_data: Dict) -> Optional[int]:
|
||||
challenge_data["user"] = aime_id
|
||||
|
||||
sql = insert(challenge).values(**challenge_data)
|
||||
conflict = sql.on_duplicate_key_update(**challenge_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_challenge: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_theory_course(
|
||||
self, aime_id: int, theory_course_data: Dict
|
||||
) -> Optional[int]:
|
||||
theory_course_data["user"] = aime_id
|
||||
|
||||
sql = insert(theory_course).values(**theory_course_data)
|
||||
conflict = sql.on_duplicate_key_update(**theory_course_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_theory_course: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_theory_partner(
|
||||
self, aime_id: int, theory_partner_data: Dict
|
||||
) -> Optional[int]:
|
||||
theory_partner_data["user"] = aime_id
|
||||
|
||||
sql = insert(theory_partner).values(**theory_partner_data)
|
||||
conflict = sql.on_duplicate_key_update(**theory_partner_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(
|
||||
f"put_theory_partner: Failed to update! aime_id: {aime_id}"
|
||||
)
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_theory_running(
|
||||
self, aime_id: int, theory_running_data: Dict
|
||||
) -> Optional[int]:
|
||||
theory_running_data["user"] = aime_id
|
||||
|
||||
sql = insert(theory_running).values(**theory_running_data)
|
||||
conflict = sql.on_duplicate_key_update(**theory_running_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(
|
||||
f"put_theory_running: Failed to update! aime_id: {aime_id}"
|
||||
)
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_vs_info(self, aime_id: int, vs_info_data: Dict) -> Optional[int]:
|
||||
vs_info_data["user"] = aime_id
|
||||
|
||||
sql = insert(vs_info).values(**vs_info_data)
|
||||
conflict = sql.on_duplicate_key_update(**vs_info_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_vs_info: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_stamp(
|
||||
self, aime_id: int, stamp_data: Dict
|
||||
) -> Optional[int]:
|
||||
stamp_data["user"] = aime_id
|
||||
|
||||
sql = insert(stamp).values(**stamp_data)
|
||||
conflict = sql.on_duplicate_key_update(**stamp_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(
|
||||
f"putstamp: Failed to update! aime_id: {aime_id}"
|
||||
)
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_timetrial_event(
|
||||
self, aime_id: int, time_trial_event_id: int, point: int
|
||||
) -> Optional[int]:
|
||||
timetrial_event_data = {
|
||||
"user": aime_id,
|
||||
"timetrial_event_id": time_trial_event_id,
|
||||
"point": point,
|
||||
}
|
||||
|
||||
sql = insert(timetrial_event).values(**timetrial_event_data)
|
||||
conflict = sql.on_duplicate_key_update(**timetrial_event_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(
|
||||
f"put_timetrial_event: Failed to update! aime_id: {aime_id}"
|
||||
)
|
||||
return None
|
||||
return result.lastrowid
|
||||
440
titles/idac/schema/profile.py
Normal file
440
titles/idac/schema/profile.py
Normal file
@@ -0,0 +1,440 @@
|
||||
from typing import Dict, List, Optional
|
||||
from sqlalchemy import Table, Column, UniqueConstraint, PrimaryKeyConstraint, and_
|
||||
from sqlalchemy.types import Integer, String, TIMESTAMP, Boolean, JSON, BigInteger
|
||||
from sqlalchemy.engine.base import Connection
|
||||
from sqlalchemy.schema import ForeignKey
|
||||
from sqlalchemy.sql import func, select
|
||||
from sqlalchemy.engine import Row
|
||||
from sqlalchemy.dialects.mysql import insert
|
||||
|
||||
from core.data.schema import BaseData, metadata
|
||||
from core.config import CoreConfig
|
||||
|
||||
profile = Table(
|
||||
"idac_profile",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column(
|
||||
"user",
|
||||
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
||||
nullable=False,
|
||||
),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("username", String(8)),
|
||||
Column("country", Integer),
|
||||
Column("store", Integer),
|
||||
Column("team_id", Integer, server_default="0"),
|
||||
Column("total_play", Integer, server_default="0"),
|
||||
Column("daily_play", Integer, server_default="0"),
|
||||
Column("day_play", Integer, server_default="0"),
|
||||
Column("mileage", Integer, server_default="0"),
|
||||
Column("asset_version", Integer, server_default="1"),
|
||||
Column("last_play_date", TIMESTAMP, server_default=func.now()),
|
||||
Column("mytitle_id", Integer, server_default="0"),
|
||||
Column("mytitle_efffect_id", Integer, server_default="0"),
|
||||
Column("sticker_id", Integer, server_default="0"),
|
||||
Column("sticker_effect_id", Integer, server_default="0"),
|
||||
Column("papercup_id", Integer, server_default="0"),
|
||||
Column("tachometer_id", Integer, server_default="0"),
|
||||
Column("aura_id", Integer, server_default="0"),
|
||||
Column("aura_color_id", Integer, server_default="0"),
|
||||
Column("aura_line_id", Integer, server_default="0"),
|
||||
Column("bgm_id", Integer, server_default="0"),
|
||||
Column("keyholder_id", Integer, server_default="0"),
|
||||
Column("start_menu_bg_id", Integer, server_default="0"),
|
||||
Column("use_car_id", Integer, server_default="1"),
|
||||
Column("use_style_car_id", Integer, server_default="1"),
|
||||
Column("bothwin_count", Integer, server_default="0"),
|
||||
Column("bothwin_score", Integer, server_default="0"),
|
||||
Column("subcard_count", Integer, server_default="0"),
|
||||
Column("vs_history", Integer, server_default="0"),
|
||||
Column("stamp_key_assign_0", Integer),
|
||||
Column("stamp_key_assign_1", Integer),
|
||||
Column("stamp_key_assign_2", Integer),
|
||||
Column("stamp_key_assign_3", Integer),
|
||||
Column("name_change_category", Integer, server_default="0"),
|
||||
Column("factory_disp", Integer, server_default="0"),
|
||||
Column("create_date", TIMESTAMP, server_default=func.now()),
|
||||
Column("cash", Integer, server_default="0"),
|
||||
Column("dressup_point", Integer, server_default="0"),
|
||||
Column("avatar_point", Integer, server_default="0"),
|
||||
Column("total_cash", Integer, server_default="0"),
|
||||
UniqueConstraint("user", "version", name="idac_profile_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
# No point setting defaults since the game sends everything on profile creation anyway
|
||||
config = Table(
|
||||
"idac_profile_config",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column(
|
||||
"user",
|
||||
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
||||
nullable=False,
|
||||
),
|
||||
Column("config_id", Integer),
|
||||
Column("steering_intensity", Integer),
|
||||
Column("transmission_type", Integer),
|
||||
Column("default_viewpoint", Integer),
|
||||
Column("favorite_bgm", Integer),
|
||||
Column("bgm_volume", Integer),
|
||||
Column("se_volume", Integer),
|
||||
Column("master_volume", Integer),
|
||||
Column("store_battle_policy", Integer),
|
||||
Column("battle_onomatope_display", Integer),
|
||||
Column("cornering_guide", Integer),
|
||||
Column("minimap", Integer),
|
||||
Column("line_guide", Integer),
|
||||
Column("ghost", Integer),
|
||||
Column("race_exit", Integer),
|
||||
Column("result_skip", Integer),
|
||||
Column("stamp_select_skip", Integer),
|
||||
UniqueConstraint("user", name="idac_profile_config_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
# No point setting defaults since the game sends everything on profile creation anyway
|
||||
avatar = Table(
|
||||
"idac_profile_avatar",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column(
|
||||
"user",
|
||||
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
||||
nullable=False,
|
||||
),
|
||||
Column("sex", Integer),
|
||||
Column("face", Integer),
|
||||
Column("eye", Integer),
|
||||
Column("mouth", Integer),
|
||||
Column("hair", Integer),
|
||||
Column("glasses", Integer),
|
||||
Column("face_accessory", Integer),
|
||||
Column("body", Integer),
|
||||
Column("body_accessory", Integer),
|
||||
Column("behind", Integer),
|
||||
Column("bg", Integer),
|
||||
Column("effect", Integer),
|
||||
Column("special", Integer),
|
||||
UniqueConstraint("user", name="idac_profile_avatar_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
rank = Table(
|
||||
"idac_profile_rank",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column(
|
||||
"user",
|
||||
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
||||
nullable=False,
|
||||
),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("story_rank_exp", Integer, server_default="0"),
|
||||
Column("story_rank", Integer, server_default="1"),
|
||||
Column("time_trial_rank_exp", Integer, server_default="0"),
|
||||
Column("time_trial_rank", Integer, server_default="1"),
|
||||
Column("online_battle_rank_exp", Integer, server_default="0"),
|
||||
Column("online_battle_rank", Integer, server_default="1"),
|
||||
Column("store_battle_rank_exp", Integer, server_default="0"),
|
||||
Column("store_battle_rank", Integer, server_default="1"),
|
||||
Column("theory_exp", Integer, server_default="0"),
|
||||
Column("theory_rank", Integer, server_default="1"),
|
||||
Column("pride_group_id", Integer, server_default="0"),
|
||||
Column("pride_point", Integer, server_default="0"),
|
||||
Column("grade_exp", Integer, server_default="0"),
|
||||
Column("grade", Integer, server_default="1"),
|
||||
Column("grade_reward_dist", Integer, server_default="0"),
|
||||
Column("story_rank_reward_dist", Integer, server_default="0"),
|
||||
Column("time_trial_rank_reward_dist", Integer, server_default="0"),
|
||||
Column("online_battle_rank_reward_dist", Integer, server_default="0"),
|
||||
Column("store_battle_rank_reward_dist", Integer, server_default="0"),
|
||||
Column("theory_rank_reward_dist", Integer, server_default="0"),
|
||||
Column("max_attained_online_battle_rank", Integer, server_default="1"),
|
||||
Column("max_attained_pride_point", Integer, server_default="0"),
|
||||
Column("is_last_max", Integer, server_default="0"),
|
||||
UniqueConstraint("user", "version", name="idac_profile_rank_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
stock = Table(
|
||||
"idac_profile_stock",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column(
|
||||
"user",
|
||||
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
||||
nullable=False,
|
||||
),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("mytitle_list", String(1024), server_default=""),
|
||||
Column("mytitle_new_list", String(1024), server_default=""),
|
||||
Column("avatar_face_list", String(255), server_default=""),
|
||||
Column("avatar_face_new_list", String(255), server_default=""),
|
||||
Column("avatar_eye_list", String(255), server_default=""),
|
||||
Column("avatar_eye_new_list", String(255), server_default=""),
|
||||
Column("avatar_hair_list", String(255), server_default=""),
|
||||
Column("avatar_hair_new_list", String(255), server_default=""),
|
||||
Column("avatar_body_list", String(255), server_default=""),
|
||||
Column("avatar_body_new_list", String(255), server_default=""),
|
||||
Column("avatar_mouth_list", String(255), server_default=""),
|
||||
Column("avatar_mouth_new_list", String(255), server_default=""),
|
||||
Column("avatar_glasses_list", String(255), server_default=""),
|
||||
Column("avatar_glasses_new_list", String(255), server_default=""),
|
||||
Column("avatar_face_accessory_list", String(255), server_default=""),
|
||||
Column("avatar_face_accessory_new_list", String(255), server_default=""),
|
||||
Column("avatar_body_accessory_list", String(255), server_default=""),
|
||||
Column("avatar_body_accessory_new_list", String(255), server_default=""),
|
||||
Column("avatar_behind_list", String(255), server_default=""),
|
||||
Column("avatar_behind_new_list", String(255), server_default=""),
|
||||
Column("avatar_bg_list", String(255), server_default=""),
|
||||
Column("avatar_bg_new_list", String(255), server_default=""),
|
||||
Column("avatar_effect_list", String(255), server_default=""),
|
||||
Column("avatar_effect_new_list", String(255), server_default=""),
|
||||
Column("avatar_special_list", String(255), server_default=""),
|
||||
Column("avatar_special_new_list", String(255), server_default=""),
|
||||
Column("stamp_list", String(255), server_default=""),
|
||||
Column("stamp_new_list", String(255), server_default=""),
|
||||
Column("keyholder_list", String(256), server_default=""),
|
||||
Column("keyholder_new_list", String(256), server_default=""),
|
||||
Column("papercup_list", String(255), server_default=""),
|
||||
Column("papercup_new_list", String(255), server_default=""),
|
||||
Column("tachometer_list", String(255), server_default=""),
|
||||
Column("tachometer_new_list", String(255), server_default=""),
|
||||
Column("aura_list", String(255), server_default=""),
|
||||
Column("aura_new_list", String(255), server_default=""),
|
||||
Column("aura_color_list", String(255), server_default=""),
|
||||
Column("aura_color_new_list", String(255), server_default=""),
|
||||
Column("aura_line_list", String(255), server_default=""),
|
||||
Column("aura_line_new_list", String(255), server_default=""),
|
||||
Column("bgm_list", String(255), server_default=""),
|
||||
Column("bgm_new_list", String(255), server_default=""),
|
||||
Column("dx_color_list", String(255), server_default=""),
|
||||
Column("dx_color_new_list", String(255), server_default=""),
|
||||
Column("start_menu_bg_list", String(255), server_default=""),
|
||||
Column("start_menu_bg_new_list", String(255), server_default=""),
|
||||
Column("under_neon_list", String(255), server_default=""),
|
||||
UniqueConstraint("user", "version", name="idac_profile_stock_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
theory = Table(
|
||||
"idac_profile_theory",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column(
|
||||
"user",
|
||||
ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"),
|
||||
nullable=False,
|
||||
),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("play_count", Integer, server_default="0"),
|
||||
Column("play_count_multi", Integer, server_default="0"),
|
||||
Column("partner_id", Integer),
|
||||
Column("partner_progress", Integer),
|
||||
Column("partner_progress_score", Integer),
|
||||
Column("practice_start_rank", Integer, server_default="0"),
|
||||
Column("general_flag", Integer, server_default="0"),
|
||||
Column("vs_history", Integer, server_default="0"),
|
||||
Column("vs_history_multi", Integer, server_default="0"),
|
||||
Column("win_count", Integer, server_default="0"),
|
||||
Column("win_count_multi", Integer, server_default="0"),
|
||||
UniqueConstraint("user", "version", name="idac_profile_theory_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
|
||||
class IDACProfileData(BaseData):
|
||||
def __init__(self, cfg: CoreConfig, conn: Connection) -> None:
|
||||
super().__init__(cfg, conn)
|
||||
self.date_time_format_ext = (
|
||||
"%Y-%m-%d %H:%M:%S.%f" # needs to be lopped off at [:-5]
|
||||
)
|
||||
self.date_time_format_short = "%Y-%m-%d"
|
||||
|
||||
def get_profile(self, aime_id: int, version: int) -> Optional[Row]:
|
||||
sql = select(profile).where(
|
||||
and_(
|
||||
profile.c.user == aime_id,
|
||||
profile.c.version == version,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_different_random_profiles(
|
||||
self, aime_id: int, version: int, count: int = 9
|
||||
) -> Optional[Row]:
|
||||
sql = (
|
||||
select(profile)
|
||||
.where(
|
||||
and_(
|
||||
profile.c.user != aime_id,
|
||||
profile.c.version == version,
|
||||
)
|
||||
)
|
||||
.order_by(func.rand())
|
||||
.limit(count)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_profile_config(self, aime_id: int) -> Optional[Row]:
|
||||
sql = select(config).where(
|
||||
and_(
|
||||
config.c.user == aime_id,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_profile_avatar(self, aime_id: int) -> Optional[Row]:
|
||||
sql = select(avatar).where(
|
||||
and_(
|
||||
avatar.c.user == aime_id,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_profile_rank(self, aime_id: int, version: int) -> Optional[Row]:
|
||||
sql = select(rank).where(
|
||||
and_(
|
||||
rank.c.user == aime_id,
|
||||
rank.c.version == version,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_profile_stock(self, aime_id: int, version: int) -> Optional[Row]:
|
||||
sql = select(stock).where(
|
||||
and_(
|
||||
stock.c.user == aime_id,
|
||||
stock.c.version == version,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_profile_theory(self, aime_id: int, version: int) -> Optional[Row]:
|
||||
sql = select(theory).where(
|
||||
and_(
|
||||
theory.c.user == aime_id,
|
||||
theory.c.version == version,
|
||||
)
|
||||
)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def put_profile(
|
||||
self, aime_id: int, version: int, profile_data: Dict
|
||||
) -> Optional[int]:
|
||||
profile_data["user"] = aime_id
|
||||
profile_data["version"] = version
|
||||
|
||||
sql = insert(profile).values(**profile_data)
|
||||
conflict = sql.on_duplicate_key_update(**profile_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_profile: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_profile_config(self, aime_id: int, config_data: Dict) -> Optional[int]:
|
||||
config_data["user"] = aime_id
|
||||
|
||||
sql = insert(config).values(**config_data)
|
||||
conflict = sql.on_duplicate_key_update(**config_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(
|
||||
f"put_profile_config: Failed to update! aime_id: {aime_id}"
|
||||
)
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_profile_avatar(self, aime_id: int, avatar_data: Dict) -> Optional[int]:
|
||||
avatar_data["user"] = aime_id
|
||||
|
||||
sql = insert(avatar).values(**avatar_data)
|
||||
conflict = sql.on_duplicate_key_update(**avatar_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(
|
||||
f"put_profile_avatar: Failed to update! aime_id: {aime_id}"
|
||||
)
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_profile_rank(
|
||||
self, aime_id: int, version: int, rank_data: Dict
|
||||
) -> Optional[int]:
|
||||
rank_data["user"] = aime_id
|
||||
rank_data["version"] = version
|
||||
|
||||
sql = insert(rank).values(**rank_data)
|
||||
conflict = sql.on_duplicate_key_update(**rank_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_profile_rank: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_profile_stock(
|
||||
self, aime_id: int, version: int, stock_data: Dict
|
||||
) -> Optional[int]:
|
||||
stock_data["user"] = aime_id
|
||||
stock_data["version"] = version
|
||||
|
||||
sql = insert(stock).values(**stock_data)
|
||||
conflict = sql.on_duplicate_key_update(**stock_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(f"put_profile_stock: Failed to update! aime_id: {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_profile_theory(
|
||||
self, aime_id: int, version: int, theory_data: Dict
|
||||
) -> Optional[int]:
|
||||
theory_data["user"] = aime_id
|
||||
theory_data["version"] = version
|
||||
|
||||
sql = insert(theory).values(**theory_data)
|
||||
conflict = sql.on_duplicate_key_update(**theory_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warn(
|
||||
f"put_profile_theory: Failed to update! aime_id: {aime_id}"
|
||||
)
|
||||
return None
|
||||
return result.lastrowid
|
||||
2505
titles/idac/season2.py
Normal file
2505
titles/idac/season2.py
Normal file
File diff suppressed because it is too large
Load Diff
@@ -1,5 +1,6 @@
|
||||
import struct
|
||||
|
||||
from core.utils import Utils
|
||||
from .base import IDZHandlerBase
|
||||
from core.config import CoreConfig
|
||||
from ..config import IDZConfig
|
||||
@@ -20,9 +21,11 @@ class IDZHandlerLoadServerInfo(IDZHandlerBase):
|
||||
offset = 0
|
||||
if self.version >= IDZConstants.VER_IDZ_210:
|
||||
offset = 2
|
||||
|
||||
t_port = Utils.get_title_port(self.core_config)
|
||||
|
||||
news_str = f"http://{self.core_config.title.hostname}:{self.core_config.title.port}/SDDF/230/news/news80**.txt"
|
||||
err_str = f"http://{self.core_config.title.hostname}:{self.core_config.title.port}/SDDF/230/error"
|
||||
news_str = f"http://{self.core_config.title.hostname}:{t_port}/idz/news/news80**.txt"
|
||||
err_str = f"http://{self.core_config.title.hostname}:{t_port}/idz/error"
|
||||
|
||||
len_hostname = len(self.core_config.title.hostname)
|
||||
len_news = len(news_str)
|
||||
|
||||
@@ -85,7 +85,8 @@ class IDZServlet(BaseServlet):
|
||||
|
||||
def get_endpoint_matchers(self) -> Tuple[List[Tuple[str, str, Dict]], List[Tuple[str, str, Dict]]]:
|
||||
return[
|
||||
[("render_GET", "/{game}/{version}/{endpoint:.*?}", {'game': R'S...'})], # TODO: Slim this down to only the news stuff
|
||||
[("render_GET", "/idz/news/{endpoint:.*?}", {}),
|
||||
("render_GET", "/idz/error", {})],
|
||||
[]
|
||||
]
|
||||
|
||||
|
||||
@@ -157,54 +157,44 @@ class OngekiBase:
|
||||
return {"type": data["type"], "length": 0, "gameIdlistList": []}
|
||||
|
||||
def handle_get_game_ranking_api_request(self, data: Dict) -> Dict:
|
||||
return {"length": 0, "gameRankingList": []}
|
||||
game_ranking_list = self.data.static.get_ranking_list(self.version)
|
||||
|
||||
ranking_list = []
|
||||
for music in game_ranking_list:
|
||||
tmp = music._asdict()
|
||||
ranking_list.append(tmp)
|
||||
|
||||
def handle_get_game_point_api_request(self, data: Dict) -> Dict:
|
||||
"""
|
||||
Sets the GP amount for A and B sets for 1 - 3 credits
|
||||
"""
|
||||
if ranking_list is None:
|
||||
return {"length": 0, "gameRankingList": []}
|
||||
return {
|
||||
"length": 6,
|
||||
"gamePointList": [
|
||||
{
|
||||
"type": 0,
|
||||
"cost": 100,
|
||||
"startDate": "2000-01-01 05:00:00.0",
|
||||
"endDate": "2099-01-01 05:00:00.0",
|
||||
},
|
||||
{
|
||||
"type": 1,
|
||||
"cost": 230,
|
||||
"startDate": "2000-01-01 05:00:00.0",
|
||||
"endDate": "2099-01-01 05:00:00.0",
|
||||
},
|
||||
{
|
||||
"type": 2,
|
||||
"cost": 370,
|
||||
"startDate": "2000-01-01 05:00:00.0",
|
||||
"endDate": "2099-01-01 05:00:00.0",
|
||||
},
|
||||
{
|
||||
"type": 3,
|
||||
"cost": 120,
|
||||
"startDate": "2000-01-01 05:00:00.0",
|
||||
"endDate": "2099-01-01 05:00:00.0",
|
||||
},
|
||||
{
|
||||
"type": 4,
|
||||
"cost": 240,
|
||||
"startDate": "2000-01-01 05:00:00.0",
|
||||
"endDate": "2099-01-01 05:00:00.0",
|
||||
},
|
||||
{
|
||||
"type": 5,
|
||||
"cost": 360,
|
||||
"startDate": "2000-01-01 05:00:00.0",
|
||||
"endDate": "2099-01-01 05:00:00.0",
|
||||
},
|
||||
],
|
||||
"type": data["type"],
|
||||
"gameRankingList": ranking_list,
|
||||
}
|
||||
|
||||
|
||||
def handle_get_game_point_api_request(self, data: Dict) -> Dict:
|
||||
get_game_point = self.data.static.get_static_game_point()
|
||||
game_point = []
|
||||
|
||||
if not get_game_point:
|
||||
self.logger.info(f"GP table is empty, inserting defaults")
|
||||
self.data.static.put_static_game_point_defaults()
|
||||
get_game_point = self.data.static.get_static_game_point()
|
||||
for gp in get_game_point:
|
||||
tmp = gp._asdict()
|
||||
game_point.append(tmp)
|
||||
return {
|
||||
"length": len(game_point),
|
||||
"gamePointList": game_point,
|
||||
}
|
||||
for gp in get_game_point:
|
||||
tmp = gp._asdict()
|
||||
game_point.append(tmp)
|
||||
return {
|
||||
"length": len(game_point),
|
||||
"gamePointList": game_point,
|
||||
}
|
||||
|
||||
def handle_game_login_api_request(self, data: Dict) -> Dict:
|
||||
return {"returnCode": 1, "apiName": "gameLogin"}
|
||||
|
||||
@@ -215,11 +205,39 @@ class OngekiBase:
|
||||
return {"returnCode": 1, "apiName": "ExtendLockTimeApi"}
|
||||
|
||||
def handle_get_game_reward_api_request(self, data: Dict) -> Dict:
|
||||
# TODO: reward list
|
||||
return {"length": 0, "gameRewardList": []}
|
||||
get_game_rewards = self.data.static.get_reward_list(self.version)
|
||||
|
||||
reward_list = []
|
||||
for reward in get_game_rewards:
|
||||
tmp = reward._asdict()
|
||||
tmp.pop("id")
|
||||
tmp.pop("version")
|
||||
tmp.pop("rewardname")
|
||||
reward_list.append(tmp)
|
||||
|
||||
if reward_list is None:
|
||||
return {"length": 0, "gameRewardList": []}
|
||||
return {
|
||||
"length": len(reward_list),
|
||||
"gameRewardList": reward_list,
|
||||
}
|
||||
|
||||
def handle_get_game_present_api_request(self, data: Dict) -> Dict:
|
||||
return {"length": 0, "gamePresentList": []}
|
||||
get_present = self.data.static.get_present_list(self.version)
|
||||
|
||||
present_list = []
|
||||
for present in get_present:
|
||||
tmp = present._asdict()
|
||||
tmp.pop("id")
|
||||
tmp.pop("version")
|
||||
present_list.append(tmp)
|
||||
|
||||
if present_list is None:
|
||||
return {"length": 0, "gamePresentList": []}
|
||||
return {
|
||||
"length": len(present_list),
|
||||
"gamePresentList": present_list,
|
||||
}
|
||||
|
||||
def handle_get_game_message_api_request(self, data: Dict) -> Dict:
|
||||
return {"length": 0, "gameMessageList": []}
|
||||
@@ -228,12 +246,39 @@ class OngekiBase:
|
||||
return {"length": 0, "gameSaleList": []}
|
||||
|
||||
def handle_get_game_tech_music_api_request(self, data: Dict) -> Dict:
|
||||
return {"length": 0, "gameTechMusicList": []}
|
||||
music_list = self.data.static.get_tech_music(self.version)
|
||||
|
||||
prep_music_list = []
|
||||
for music in music_list:
|
||||
tmp = music._asdict()
|
||||
tmp.pop("id")
|
||||
tmp.pop("version")
|
||||
prep_music_list.append(tmp)
|
||||
|
||||
if prep_music_list is None:
|
||||
return {"length": 0, "gameTechMusicList": []}
|
||||
|
||||
return {
|
||||
"length": len(prep_music_list),
|
||||
"gameTechMusicList": prep_music_list,
|
||||
}
|
||||
|
||||
def handle_upsert_client_setting_api_request(self, data: Dict) -> Dict:
|
||||
if self.core_cfg.server.is_develop:
|
||||
return {"returnCode": 1, "apiName": "UpsertClientSettingApi"}
|
||||
|
||||
client_id = data["clientId"]
|
||||
client_setting_data = data["clientSetting"]
|
||||
self.data.static.put_client_setting_data(client_id, client_setting_data)
|
||||
return {"returnCode": 1, "apiName": "UpsertClientSettingApi"}
|
||||
|
||||
def handle_upsert_client_testmode_api_request(self, data: Dict) -> Dict:
|
||||
if self.core_cfg.server.is_develop:
|
||||
return {"returnCode": 1, "apiName": "UpsertClientTestmodeApi"}
|
||||
|
||||
region_id = data["regionId"]
|
||||
client_testmode_data = data["clientTestmode"]
|
||||
self.data.static.put_client_testmode_data(region_id, client_testmode_data)
|
||||
return {"returnCode": 1, "apiName": "UpsertClientTestmodeApi"}
|
||||
|
||||
def handle_upsert_client_bookkeeping_api_request(self, data: Dict) -> Dict:
|
||||
@@ -277,13 +322,12 @@ class OngekiBase:
|
||||
"id": event["eventId"],
|
||||
# actually use the startDate from the import so it
|
||||
# properly shows all the events when new ones are imported
|
||||
"startDate": datetime.strftime(
|
||||
event["startDate"], "%Y-%m-%d %H:%M:%S.0"
|
||||
),
|
||||
"endDate": "2099-12-31 00:00:00.0",
|
||||
"startDate": datetime.strftime(event["startDate"], "%Y-%m-%d %H:%M:%S.0"),
|
||||
#"endDate": "2099-12-31 00:00:00.0",
|
||||
"endDate": datetime.strftime(event["endDate"], "%Y-%m-%d %H:%M:%S.0"),
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
return {
|
||||
"type": data["type"],
|
||||
"length": len(evt_list),
|
||||
@@ -385,7 +429,7 @@ class OngekiBase:
|
||||
}
|
||||
|
||||
def handle_get_user_tech_event_api_request(self, data: Dict) -> Dict:
|
||||
user_tech_event_list = self.data.item.get_tech_event(data["userId"])
|
||||
user_tech_event_list = self.data.item.get_tech_event(self.version, data["userId"])
|
||||
if user_tech_event_list is None:
|
||||
return {}
|
||||
|
||||
@@ -394,6 +438,7 @@ class OngekiBase:
|
||||
tmp = evt._asdict()
|
||||
tmp.pop("id")
|
||||
tmp.pop("user")
|
||||
tmp.pop("version")
|
||||
tech_evt.append(tmp)
|
||||
|
||||
return {
|
||||
@@ -403,15 +448,24 @@ class OngekiBase:
|
||||
}
|
||||
|
||||
def handle_get_user_tech_event_ranking_api_request(self, data: Dict) -> Dict:
|
||||
# user_event_ranking_list = self.data.item.get_tech_event_ranking(data["userId"])
|
||||
# if user_event_ranking_list is None: return {}
|
||||
user_tech_event_ranks = self.data.item.get_tech_event_ranking(self.version, data["userId"])
|
||||
if user_tech_event_ranks is None:
|
||||
return {
|
||||
"userId": data["userId"],
|
||||
"length": 0,
|
||||
"userTechEventRankingList": [],
|
||||
}
|
||||
|
||||
# collect the whole table and clear other players, to preserve proper ranking
|
||||
evt_ranking = []
|
||||
# for evt in user_event_ranking_list:
|
||||
# tmp = evt._asdict()
|
||||
# tmp.pop("id")
|
||||
# tmp.pop("user")
|
||||
# evt_ranking.append(tmp)
|
||||
for evt in user_tech_event_ranks:
|
||||
tmp = evt._asdict()
|
||||
if tmp["user"] != data["userId"]:
|
||||
tmp.clear()
|
||||
else:
|
||||
tmp.pop("id")
|
||||
tmp.pop("user")
|
||||
evt_ranking.append(tmp)
|
||||
|
||||
return {
|
||||
"userId": data["userId"],
|
||||
@@ -533,20 +587,26 @@ class OngekiBase:
|
||||
return {"userId": data["userId"], "userData": user_data}
|
||||
|
||||
def handle_get_user_event_ranking_api_request(self, data: Dict) -> Dict:
|
||||
# user_event_ranking_list = self.data.item.get_event_ranking(data["userId"])
|
||||
# if user_event_ranking_list is None: return {}
|
||||
user_event_ranking_list = self.data.item.get_ranking_event_ranks(self.version, data["userId"])
|
||||
if user_event_ranking_list is None:
|
||||
return {}
|
||||
|
||||
evt_ranking = []
|
||||
# for evt in user_event_ranking_list:
|
||||
# tmp = evt._asdict()
|
||||
# tmp.pop("id")
|
||||
# tmp.pop("user")
|
||||
# evt_ranking.append(tmp)
|
||||
# We collect the whole ranking table, and clear out any not needed data, this way we preserve the proper ranking
|
||||
# In official spec this should be done server side, in maintenance period
|
||||
prep_event_ranking = []
|
||||
for evt in user_event_ranking_list:
|
||||
tmp = evt._asdict()
|
||||
if tmp["user"] != data["userId"]:
|
||||
tmp.clear()
|
||||
else:
|
||||
tmp.pop("id")
|
||||
tmp.pop("user")
|
||||
prep_event_ranking.append(tmp)
|
||||
|
||||
return {
|
||||
"userId": data["userId"],
|
||||
"length": len(evt_ranking),
|
||||
"userEventRankingList": evt_ranking,
|
||||
"length": len(prep_event_ranking),
|
||||
"userEventRankingList": prep_event_ranking,
|
||||
}
|
||||
|
||||
def handle_get_user_login_bonus_api_request(self, data: Dict) -> Dict:
|
||||
@@ -777,7 +837,7 @@ class OngekiBase:
|
||||
}
|
||||
|
||||
def handle_get_user_mission_point_api_request(self, data: Dict) -> Dict:
|
||||
user_mission_point_list = self.data.item.get_mission_points(data["userId"])
|
||||
user_mission_point_list = self.data.item.get_mission_points(self.version, data["userId"])
|
||||
if user_mission_point_list is None:
|
||||
return {}
|
||||
|
||||
@@ -786,8 +846,10 @@ class OngekiBase:
|
||||
tmp = evt_music._asdict()
|
||||
tmp.pop("id")
|
||||
tmp.pop("user")
|
||||
tmp.pop("version")
|
||||
mission_point_list.append(tmp)
|
||||
|
||||
|
||||
return {
|
||||
"userId": data["userId"],
|
||||
"length": len(mission_point_list),
|
||||
@@ -804,6 +866,10 @@ class OngekiBase:
|
||||
tmp = evt_music._asdict()
|
||||
tmp.pop("id")
|
||||
tmp.pop("user")
|
||||
# pop other stuff event_point doesn't want
|
||||
tmp.pop("rank")
|
||||
tmp.pop("type")
|
||||
tmp.pop("date")
|
||||
event_point_list.append(tmp)
|
||||
|
||||
return {
|
||||
@@ -951,11 +1017,11 @@ class OngekiBase:
|
||||
|
||||
if "userEventPointList" in upsert:
|
||||
for x in upsert["userEventPointList"]:
|
||||
self.data.item.put_event_point(user_id, x)
|
||||
self.data.item.put_event_point(user_id, self.version, x)
|
||||
|
||||
if "userMissionPointList" in upsert:
|
||||
for x in upsert["userMissionPointList"]:
|
||||
self.data.item.put_mission_point(user_id, x)
|
||||
self.data.item.put_mission_point(user_id, self.version, x)
|
||||
|
||||
if "userRatinglogList" in upsert:
|
||||
for x in upsert["userRatinglogList"]:
|
||||
@@ -985,7 +1051,10 @@ class OngekiBase:
|
||||
|
||||
if "userTechEventList" in upsert:
|
||||
for x in upsert["userTechEventList"]:
|
||||
self.data.item.put_tech_event(user_id, x)
|
||||
self.data.item.put_tech_event(user_id, self.version, x)
|
||||
|
||||
# This should be updated once a day in maintenance window, but for time being we will push the update on each upsert
|
||||
self.data.item.put_tech_event_ranking(user_id, self.version, x)
|
||||
|
||||
if "userKopList" in upsert:
|
||||
for x in upsert["userKopList"]:
|
||||
|
||||
@@ -21,6 +21,12 @@ class OngekiServerConfig:
|
||||
self.__config, "ongeki", "server", "loglevel", default="info"
|
||||
)
|
||||
)
|
||||
|
||||
@property
|
||||
def use_https(self) -> bool:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "ongeki", "server", "use_https", default=False
|
||||
)
|
||||
|
||||
|
||||
class OngekiGachaConfig:
|
||||
|
||||
@@ -19,7 +19,6 @@ class OngekiConstants:
|
||||
EVT_TYPES: Enum = Enum(
|
||||
"EVT_TYPES",
|
||||
[
|
||||
"None",
|
||||
"Announcement",
|
||||
"Movie",
|
||||
"AddMyList",
|
||||
@@ -39,6 +38,36 @@ class OngekiConstants:
|
||||
"TechChallengeEvent",
|
||||
"AcceptTechChallengeEvent",
|
||||
"SilverJewelEvent",
|
||||
"Max",
|
||||
"None",
|
||||
],
|
||||
)
|
||||
|
||||
REWARD_TYPES: Enum = Enum(
|
||||
"REWARD_TYPES",
|
||||
[
|
||||
"Card",
|
||||
"NamePlate",
|
||||
"Trophy",
|
||||
"LimitBreakItem",
|
||||
"AlmightyJewel",
|
||||
"Money",
|
||||
"Music",
|
||||
"ProfileVoice",
|
||||
"Present",
|
||||
"ChapterJewel",
|
||||
"GachaTicket",
|
||||
"KaikaItem",
|
||||
"ExpUpItem",
|
||||
"IntimateUpItem",
|
||||
"BookItem",
|
||||
"SystemVoice",
|
||||
"Costume",
|
||||
"Medal",
|
||||
"Attachment",
|
||||
"UnlockItem",
|
||||
"Max",
|
||||
"None",
|
||||
],
|
||||
)
|
||||
|
||||
|
||||
@@ -127,15 +127,19 @@ class OngekiServlet(BaseServlet):
|
||||
)
|
||||
|
||||
def get_allnet_info(self, game_code: str, game_ver: int, keychip: str) -> Tuple[str, str]:
|
||||
if not self.core_cfg.server.is_using_proxy and Utils.get_title_port(self.core_cfg) != 80:
|
||||
return (
|
||||
f"http://{self.core_cfg.title.hostname}:{Utils.get_title_port(self.core_cfg)}/{game_code}/{game_ver}/",
|
||||
f"{self.core_cfg.title.hostname}:{Utils.get_title_port(self.core_cfg)}/",
|
||||
)
|
||||
title_port_int = Utils.get_title_port(self.core_cfg)
|
||||
title_port_ssl_int = Utils.get_title_port_ssl(self.core_cfg)
|
||||
proto = "https" if self.game_cfg.server.use_https and game_ver >= 120 else "http"
|
||||
|
||||
if proto == "https":
|
||||
t_port = f":{title_port_ssl_int}" if title_port_ssl_int and not self.core_cfg.server.is_using_proxy else ""
|
||||
|
||||
else:
|
||||
t_port = f":{title_port_int}" if title_port_int and not self.core_cfg.server.is_using_proxy else ""
|
||||
|
||||
return (
|
||||
f"http://{self.core_cfg.title.hostname}/{game_code}/{game_ver}/",
|
||||
f"{self.core_cfg.title.hostname}/",
|
||||
f"{proto}://{self.core_cfg.title.hostname}{t_port}/{game_code}/{game_ver}/",
|
||||
f"{self.core_cfg.title.hostname}{t_port}/",
|
||||
)
|
||||
|
||||
def render_POST(self, request: Request, game_code: str, matchers: Dict) -> bytes:
|
||||
@@ -204,6 +208,7 @@ class OngekiServlet(BaseServlet):
|
||||
if (
|
||||
not encrtped
|
||||
and self.game_cfg.crypto.encrypted_only
|
||||
and version >= 120
|
||||
):
|
||||
self.logger.error(
|
||||
f"Unencrypted v{version} {endpoint} request, but config is set to encrypted only: {req_raw}"
|
||||
@@ -247,7 +252,7 @@ class OngekiServlet(BaseServlet):
|
||||
|
||||
zipped = zlib.compress(json.dumps(resp, ensure_ascii=False).encode("utf-8"))
|
||||
|
||||
if not encrtped:
|
||||
if not encrtped or version < 120:
|
||||
return zipped
|
||||
|
||||
padded = pad(zipped, 16)
|
||||
|
||||
@@ -44,6 +44,7 @@ class OngekiReader(BaseReader):
|
||||
self.read_events(f"{dir}/event")
|
||||
self.read_music(f"{dir}/music")
|
||||
self.read_card(f"{dir}/card")
|
||||
self.read_reward(f"{dir}/reward")
|
||||
|
||||
def read_card(self, base_dir: str) -> None:
|
||||
self.logger.info(f"Reading cards from {base_dir}...")
|
||||
@@ -171,3 +172,28 @@ class OngekiReader(BaseReader):
|
||||
self.version, song_id, chart_id, title, artist, genre, level
|
||||
)
|
||||
self.logger.info(f"Added song {song_id} chart {chart_id}")
|
||||
|
||||
def read_reward(self, base_dir: str) -> None:
|
||||
self.logger.info(f"Reading rewards from {base_dir}...")
|
||||
|
||||
for root, dirs, files in os.walk(base_dir):
|
||||
for dir in dirs:
|
||||
if os.path.exists(f"{root}/{dir}/Reward.xml"):
|
||||
strdata = ""
|
||||
|
||||
with open(f"{root}/{dir}/Reward.xml", "r", encoding="utf-8") as f:
|
||||
strdata = f.read()
|
||||
|
||||
troot = ET.fromstring(strdata)
|
||||
|
||||
if root is None:
|
||||
continue
|
||||
|
||||
name = troot.find("Name")
|
||||
rewardId = name.find("id").text
|
||||
rewardname = name.find("str").text
|
||||
itemKind = OngekiConstants.REWARD_TYPES[troot.find("ItemType").text].value
|
||||
itemId = troot.find("RewardItem").find("ItemName").find("id").text
|
||||
|
||||
self.data.static.put_reward(self.version, rewardId, rewardname, itemKind, itemId)
|
||||
self.logger.info(f"Added reward {rewardId}")
|
||||
|
||||
@@ -1,3 +1,4 @@
|
||||
from datetime import date, datetime, timedelta
|
||||
from typing import Dict, Optional, List
|
||||
from sqlalchemy import Table, Column, UniqueConstraint, PrimaryKeyConstraint, and_
|
||||
from sqlalchemy.types import Integer, String, TIMESTAMP, Boolean, JSON
|
||||
@@ -170,8 +171,12 @@ event_point = Table(
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("eventId", Integer),
|
||||
Column("point", Integer),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("eventId", Integer, nullable=False),
|
||||
Column("point", Integer, nullable=False),
|
||||
Column("rank", Integer),
|
||||
Column("type", Integer, nullable=False),
|
||||
Column("date", String(25)),
|
||||
Column("isRankingRewarded", Boolean),
|
||||
UniqueConstraint("user", "eventId", name="ongeki_user_event_point_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
@@ -182,6 +187,7 @@ mission_point = Table(
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("version", Integer),
|
||||
Column("eventId", Integer),
|
||||
Column("point", Integer),
|
||||
UniqueConstraint("user", "eventId", name="ongeki_user_mission_point_uk"),
|
||||
@@ -237,9 +243,10 @@ tech_event = Table(
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade")),
|
||||
Column("eventId", Integer),
|
||||
Column("totalTechScore", Integer),
|
||||
Column("totalPlatinumScore", Integer),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("eventId", Integer, nullable=False),
|
||||
Column("totalTechScore", Integer, nullable=False),
|
||||
Column("totalPlatinumScore", Integer, nullable=False),
|
||||
Column("techRecordDate", String(25)),
|
||||
Column("isRankingRewarded", Boolean),
|
||||
Column("isTotalTechNewRecord", Boolean),
|
||||
@@ -247,6 +254,21 @@ tech_event = Table(
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
tech_ranking = Table(
|
||||
"ongeki_tech_event_ranking",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("user", ForeignKey("aime_user.id", ondelete="cascade", onupdate="cascade"), nullable=False),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("date", String(25)),
|
||||
Column("eventId", Integer, nullable=False),
|
||||
Column("rank", Integer),
|
||||
Column("totalPlatinumScore", Integer, nullable=False),
|
||||
Column("totalTechScore", Integer, nullable=False),
|
||||
UniqueConstraint("user", "eventId", name="ongeki_tech_event_ranking_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
gacha = Table(
|
||||
"ongeki_user_gacha",
|
||||
metadata,
|
||||
@@ -316,7 +338,6 @@ print_detail = Table(
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
|
||||
class OngekiItemData(BaseData):
|
||||
def put_card(self, aime_id: int, card_data: Dict) -> Optional[int]:
|
||||
card_data["user"] = aime_id
|
||||
@@ -511,9 +532,8 @@ class OngekiItemData(BaseData):
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def put_mission_point(
|
||||
self, aime_id: int, mission_point_data: Dict
|
||||
) -> Optional[int]:
|
||||
def put_mission_point(self, aime_id: int, version: int, mission_point_data: Dict) -> Optional[int]:
|
||||
mission_point_data["version"] = version
|
||||
mission_point_data["user"] = aime_id
|
||||
|
||||
sql = insert(mission_point).values(**mission_point_data)
|
||||
@@ -525,16 +545,22 @@ class OngekiItemData(BaseData):
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def get_mission_points(self, aime_id: int) -> Optional[List[Dict]]:
|
||||
sql = select(mission_point).where(mission_point.c.user == aime_id)
|
||||
def get_mission_points(self, version: int, aime_id: int) -> Optional[List[Dict]]:
|
||||
sql = select(mission_point).where(and_(mission_point.c.user == aime_id, mission_point.c.version == version))
|
||||
result = self.execute(sql)
|
||||
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def put_event_point(self, aime_id: int, event_point_data: Dict) -> Optional[int]:
|
||||
def put_event_point(self, aime_id: int, version: int, event_point_data: Dict) -> Optional[int]:
|
||||
# We update only the newest (type: 1) entry, in official spec game watches for both latest(type:1) and previous (type:2) entries to give an additional info how many ranks has player moved up or down
|
||||
# This fully featured is on TODO list, at the moment we just update the tables as data comes and give out rank as request comes
|
||||
event_point_data["user"] = aime_id
|
||||
event_point_data["version"] = version
|
||||
event_point_data["type"] = 1
|
||||
event_point_time = datetime.now()
|
||||
event_point_data["date"] = datetime.strftime(event_point_time, "%Y-%m-%d %H:%M")
|
||||
|
||||
sql = insert(event_point).values(**event_point_data)
|
||||
conflict = sql.on_duplicate_key_update(**event_point_data)
|
||||
@@ -625,8 +651,25 @@ class OngekiItemData(BaseData):
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def get_tech_event(self, aime_id: int) -> Optional[List[Dict]]:
|
||||
sql = select(tech_event).where(tech_event.c.user == aime_id)
|
||||
def put_tech_event_ranking(self, version: int, aime_id: int, tech_event_data: Dict) -> Optional[int]:
|
||||
tech_event_data["user"] = aime_id
|
||||
tech_event_data["version"] = version
|
||||
tech_event_data.pop("isRankingRewarded")
|
||||
tech_event_data.pop("isTotalTechNewRecord")
|
||||
tech_event_data["date"] = tech_event_data.pop("techRecordDate")
|
||||
tech_event_data["rank"] = 0
|
||||
|
||||
sql = insert(tech_ranking).values(**tech_event_data)
|
||||
conflict = sql.on_duplicate_key_update(**tech_event_data)
|
||||
result = self.execute(conflict)
|
||||
|
||||
if result is None:
|
||||
self.logger.warning(f"put_tech_event_ranking: Failed to update ranking! aime_id {aime_id}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def get_tech_event(self, version: int, aime_id: int) -> Optional[List[Dict]]:
|
||||
sql = select(tech_event).where(and_(tech_event.c.user == aime_id, tech_event.c.version == version))
|
||||
result = self.execute(sql)
|
||||
|
||||
if result is None:
|
||||
@@ -714,3 +757,21 @@ class OngekiItemData(BaseData):
|
||||
)
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
|
||||
def get_ranking_event_ranks(self, version: int, aime_id: int) -> Optional[List[Dict]]:
|
||||
# Calculates player rank on GameRequest from server, and sends it back, official spec would rank players in maintenance period, on TODO list
|
||||
sql = select(event_point.c.id, event_point.c.user, event_point.c.eventId, event_point.c.type, func.row_number().over(partition_by=event_point.c.eventId, order_by=event_point.c.point.desc()).label('rank'), event_point.c.date, event_point.c.point).where(event_point.c.version == version)
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
self.logger.error(f"failed to rank aime_id: {aime_id} ranking event positions")
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_tech_event_ranking(self, version: int, aime_id: int) -> Optional[List[Dict]]:
|
||||
sql = select(tech_ranking.c.id, tech_ranking.c.user, tech_ranking.c.date, tech_ranking.c.eventId, func.row_number().over(partition_by=tech_ranking.c.eventId, order_by=[tech_ranking.c.totalTechScore.desc(),tech_ranking.c.totalPlatinumScore.desc()]).label('rank'), tech_ranking.c.totalTechScore, tech_ranking.c.totalPlatinumScore).where(tech_ranking.c.version == version)
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
self.logger.warning(f"aime_id: {aime_id} has no tech ranking ranks")
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
@@ -17,6 +17,7 @@ events = Table(
|
||||
Column("type", Integer),
|
||||
Column("name", String(255)),
|
||||
Column("startDate", TIMESTAMP, server_default=func.now()),
|
||||
Column("endDate", TIMESTAMP, server_default=func.now()),
|
||||
Column("enabled", Boolean, server_default="1"),
|
||||
UniqueConstraint("version", "eventId", "type", name="ongeki_static_events_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
@@ -96,6 +97,94 @@ cards = Table(
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
music_ranking = Table(
|
||||
"ongeki_static_music_ranking_list",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("musicId", Integer, nullable=False),
|
||||
Column("point", Integer, nullable=False),
|
||||
Column("userName", String(255)),
|
||||
UniqueConstraint("version", "musicId", name="ongeki_static_music_ranking_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
rewards = Table(
|
||||
"ongeki_static_rewards",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("rewardId", Integer, nullable=False),
|
||||
Column("rewardname", String(255), nullable=False),
|
||||
Column("itemKind", Integer, nullable=False),
|
||||
Column("itemId", Integer, nullable=False),
|
||||
UniqueConstraint("version", "rewardId", name="ongeki_static_rewards_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
present = Table(
|
||||
"ongeki_static_present_list",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("presentId", Integer, nullable=False),
|
||||
Column("presentName", String(255), nullable=False),
|
||||
Column("rewardId", Integer, nullable=False),
|
||||
Column("stock", Integer, nullable=False),
|
||||
Column("message", String(255)),
|
||||
Column("startDate", String(25), nullable=False),
|
||||
Column("endDate", String(25), nullable=False),
|
||||
UniqueConstraint("version", "presentId", name="ongeki_static_present_list_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
tech_music = Table(
|
||||
"ongeki_static_tech_music",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("version", Integer, nullable=False),
|
||||
Column("eventId", Integer, nullable=False),
|
||||
Column("musicId", Integer, nullable=False),
|
||||
Column("level", Integer, nullable=False),
|
||||
UniqueConstraint("version", "musicId", name="ongeki_static_tech_music_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
client_testmode = Table(
|
||||
"ongeki_static_client_testmode",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("regionId", Integer, nullable=False),
|
||||
Column("placeId", Integer, nullable=False),
|
||||
Column("clientId", String(11), nullable=False),
|
||||
Column("updateDate", TIMESTAMP, nullable=False),
|
||||
Column("isDelivery", Boolean, nullable=False),
|
||||
Column("groupId", Integer, nullable=False),
|
||||
Column("groupRole", Integer, nullable=False),
|
||||
Column("continueMode", Integer, nullable=False),
|
||||
Column("selectMusicTime", Integer, nullable=False),
|
||||
Column("advertiseVolume", Integer, nullable=False),
|
||||
Column("eventMode", Integer, nullable=False),
|
||||
Column("eventMusicNum", Integer, nullable=False),
|
||||
Column("patternGp", Integer, nullable=False),
|
||||
Column("limitGp", Integer, nullable=False),
|
||||
Column("maxLeverMovable", Integer, nullable=False),
|
||||
Column("minLeverMovable", Integer, nullable=False),
|
||||
UniqueConstraint("clientId", name="ongeki_static_client_testmode_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
game_point = Table(
|
||||
"ongeki_static_game_point",
|
||||
metadata,
|
||||
Column("id", Integer, primary_key=True, nullable=False),
|
||||
Column("type", Integer, nullable=False),
|
||||
Column("cost", Integer, nullable=False),
|
||||
Column("startDate", String(25), nullable=False, server_default="2000-01-01 05:00:00.0"),
|
||||
Column("endDate", String(25), nullable=False, server_default="2099-01-01 05:00:00.0"),
|
||||
UniqueConstraint("type", name="ongeki_static_game_point_uk"),
|
||||
mysql_charset="utf8mb4",
|
||||
)
|
||||
|
||||
class OngekiStaticData(BaseData):
|
||||
def put_card(self, version: int, card_id: int, **card_data) -> Optional[int]:
|
||||
@@ -235,6 +324,7 @@ class OngekiStaticData(BaseData):
|
||||
eventId=event_id,
|
||||
type=event_type,
|
||||
name=event_name,
|
||||
endDate=f"2038-01-01 00:00:00",
|
||||
)
|
||||
|
||||
conflict = sql.on_duplicate_key_update(
|
||||
@@ -333,3 +423,90 @@ class OngekiStaticData(BaseData):
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchone()
|
||||
|
||||
def get_ranking_list(self, version: int) -> Optional[List[Dict]]:
|
||||
sql = select(music_ranking.c.musicId.label('id'), music_ranking.c.point, music_ranking.c.userName).where(music_ranking.c.version == version)
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def put_reward(self, version: int, rewardId: int, rewardname: str, itemKind: int, itemId: int) -> Optional[int]:
|
||||
sql = insert(rewards).values(
|
||||
version=version,
|
||||
rewardId=rewardId,
|
||||
rewardname=rewardname,
|
||||
itemKind=itemKind,
|
||||
itemId=itemId,
|
||||
)
|
||||
conflict = sql.on_duplicate_key_update(
|
||||
rewardname=rewardname,
|
||||
)
|
||||
result = self.execute(conflict)
|
||||
if result is None:
|
||||
self.logger.warning(f"Failed to insert reward! reward_id: {rewardId}")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def get_reward_list(self, version: int) -> Optional[List[Dict]]:
|
||||
sql = select(rewards).where(rewards.c.version == version)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
self.logger.warning(f"Failed to load reward list")
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_present_list(self, version: int) -> Optional[List[Dict]]:
|
||||
sql = select(present).where(present.c.version == version)
|
||||
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
self.logger.warning(f"Failed to load present list")
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def get_tech_music(self, version: int) -> Optional[List[Dict]]:
|
||||
sql = select(tech_music).where(tech_music.c.version == version)
|
||||
|
||||
result = self.execute(sql)
|
||||
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
def put_client_testmode_data(self, region_id: int, client_testmode_data: Dict) -> Optional[List[Dict]]:
|
||||
sql = insert(client_testmode).values(regionId=region_id, **client_testmode_data)
|
||||
conflict = sql.on_duplicate_key_update(regionId=region_id, **client_testmode_data)
|
||||
|
||||
result = self.execute(conflict)
|
||||
if result is None:
|
||||
self.logger.warning(f"clientId: {clientId} Failed to update ClientTestMode data"),
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_client_setting_data(self, client_id: str, client_setting_data: Dict) -> Optional[List[Dict]]:
|
||||
sql = insert(machine).values(data=client_setting_data)
|
||||
conflict = sql.on_duplicate_key_update(serial=client_id)
|
||||
|
||||
result = self.execute(conflict)
|
||||
if result is None:
|
||||
self.logger.warning(f"clientId: {clientId} Failed to update ClientSetting data"),
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def put_static_game_point_defaults(self) -> Optional[List[Dict]]:
|
||||
game_point_defaults = [{"type": 0, "cost": 100},{"type": 1, "cost": 230},{"type": 2, "cost": 370},{"type": 3, "cost": 120},{"type": 4, "cost": 240},{"type": 5, "cost": 360}]
|
||||
sql = insert(game_point).values(game_point_defaults)
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
self.logger.warning(f"Failed to insert default GP table!")
|
||||
return None
|
||||
return result.lastrowid
|
||||
|
||||
def get_static_game_point(self) -> Optional[List[Dict]]:
|
||||
sql = select(game_point.c.type, game_point.c.cost, game_point.c.startDate, game_point.c.endDate)
|
||||
result = self.execute(sql)
|
||||
if result is None:
|
||||
return None
|
||||
return result.fetchall()
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@@ -5,12 +5,6 @@ class SaoServerConfig:
|
||||
def __init__(self, parent_config: "SaoConfig"):
|
||||
self.__config = parent_config
|
||||
|
||||
@property
|
||||
def hostname(self) -> str:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "sao", "server", "hostname", default="localhost"
|
||||
)
|
||||
|
||||
@property
|
||||
def enable(self) -> bool:
|
||||
return CoreConfig.get_config_field(
|
||||
@@ -25,12 +19,6 @@ class SaoServerConfig:
|
||||
)
|
||||
)
|
||||
|
||||
@property
|
||||
def port(self) -> int:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "sao", "server", "port", default=9000
|
||||
)
|
||||
|
||||
@property
|
||||
def auto_register(self) -> bool:
|
||||
"""
|
||||
@@ -41,7 +29,47 @@ class SaoServerConfig:
|
||||
self.__config, "sao", "server", "auto_register", default=True
|
||||
)
|
||||
|
||||
class SaoCryptConfig:
|
||||
def __init__(self, parent_config: "SaoConfig"):
|
||||
self.__config = parent_config
|
||||
|
||||
@property
|
||||
def enable(self) -> bool:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "sao", "crypt", "enable", default=False
|
||||
)
|
||||
|
||||
@property
|
||||
def key(self) -> str:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "sao", "crypt", "key", default=""
|
||||
)
|
||||
|
||||
@property
|
||||
def iv(self) -> str:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "sao", "crypt", "iv", default=""
|
||||
)
|
||||
|
||||
class SaoHashConfig:
|
||||
def __init__(self, parent_config: "SaoConfig"):
|
||||
self.__config = parent_config
|
||||
|
||||
@property
|
||||
def verify_hash(self) -> bool:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "sao", "hash", "verify_hash", default=False
|
||||
)
|
||||
|
||||
@property
|
||||
def hash_base(self) -> str:
|
||||
return CoreConfig.get_config_field(
|
||||
self.__config, "sao", "hash", "hash_base", default=""
|
||||
)
|
||||
|
||||
|
||||
class SaoConfig(dict):
|
||||
def __init__(self) -> None:
|
||||
self.server = SaoServerConfig(self)
|
||||
self.crypt = SaoCryptConfig(self)
|
||||
self.hash = SaoHashConfig(self)
|
||||
|
||||
BIN
titles/sao/data/GashaMedalShopItems.csv
Normal file
BIN
titles/sao/data/GashaMedalShopItems.csv
Normal file
Binary file not shown.
|
BIN
titles/sao/data/GashaMedalShops.csv
Normal file
BIN
titles/sao/data/GashaMedalShops.csv
Normal file
Binary file not shown.
|
BIN
titles/sao/data/ResEarnCampaignShopItems.csv
Normal file
BIN
titles/sao/data/ResEarnCampaignShopItems.csv
Normal file
Binary file not shown.
|
BIN
titles/sao/data/ResEarnCampaignShops.csv
Normal file
BIN
titles/sao/data/ResEarnCampaignShops.csv
Normal file
Binary file not shown.
|
BIN
titles/sao/data/YuiMedalShopItems.csv
Normal file
BIN
titles/sao/data/YuiMedalShopItems.csv
Normal file
Binary file not shown.
|
BIN
titles/sao/data/YuiMedalShops.csv
Normal file
BIN
titles/sao/data/YuiMedalShops.csv
Normal file
Binary file not shown.
|
@@ -1 +1,2 @@
|
||||
from titles.sao.handlers.base import *
|
||||
from .base import *
|
||||
from .helpers import *
|
||||
@@ -1,16 +1,29 @@
|
||||
import struct
|
||||
from datetime import datetime
|
||||
from typing import List
|
||||
from construct import *
|
||||
import sys
|
||||
from .helpers import *
|
||||
import csv
|
||||
from csv import *
|
||||
|
||||
class SaoBaseRequest:
|
||||
class SaoRequestHeader:
|
||||
def __init__(self, data: bytes) -> None:
|
||||
self.cmd = struct.unpack_from("!H", bytes)[0]
|
||||
# TODO: The rest of the request header
|
||||
collection = struct.unpack_from("!HHIIII16sI", data)
|
||||
self.cmd: int = collection[0]
|
||||
self.err_status = collection[1]
|
||||
self.error_type = collection[2]
|
||||
self.vendor_id: int = collection[3]
|
||||
self.game_id: int = collection[4]
|
||||
self.version_id: int = collection[5]
|
||||
self.hash: str = collection[6]
|
||||
self.data_len: str = collection[7]
|
||||
|
||||
class SaoBaseResponse:
|
||||
class SaoBaseRequest:
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
self.header = header
|
||||
# TODO: Length check
|
||||
|
||||
class SaoResponseHeader:
|
||||
def __init__(self, cmd_id: int) -> None:
|
||||
self.cmd = cmd_id
|
||||
self.err_status = 0
|
||||
@@ -23,6 +36,13 @@ class SaoBaseResponse:
|
||||
def make(self) -> bytes:
|
||||
return struct.pack("!HHIIIII", self.cmd, self.err_status, self.error_type, self.vendor_id, self.game_id, self.version_id, self.length)
|
||||
|
||||
class SaoBaseResponse:
|
||||
def __init__(self, cmd_id: int) -> None:
|
||||
self.header = SaoResponseHeader(cmd_id)
|
||||
|
||||
def make(self) -> bytes:
|
||||
return self.header.make()
|
||||
|
||||
class SaoNoopResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd: int) -> None:
|
||||
super().__init__(cmd)
|
||||
@@ -33,8 +53,8 @@ class SaoNoopResponse(SaoBaseResponse):
|
||||
return super().make() + struct.pack("!bI", self.result, 0)
|
||||
|
||||
class SaoGetMaintRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
# TODO: The rest of the mait info request
|
||||
|
||||
class SaoGetMaintResponse(SaoBaseResponse):
|
||||
@@ -68,8 +88,8 @@ class SaoGetMaintResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoCommonAcCabinetBootNotificationRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoCommonAcCabinetBootNotificationResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -90,8 +110,8 @@ class SaoCommonAcCabinetBootNotificationResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoMasterDataVersionCheckRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoMasterDataVersionCheckResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -118,8 +138,8 @@ class SaoMasterDataVersionCheckResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoCommonGetAppVersionsRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoCommonGetAppVersionsRequest(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -154,8 +174,8 @@ class SaoCommonGetAppVersionsRequest(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoCommonPayingPlayStartRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoCommonPayingPlayStartRequest(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -181,8 +201,30 @@ class SaoCommonPayingPlayStartRequest(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetAuthCardDataRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
self.cabinet_type = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
self.auth_type = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
store_id = decode_str(data, off)
|
||||
self.store_id = store_id[0]
|
||||
off += store_id[1]
|
||||
|
||||
serial_no = decode_str(data, off)
|
||||
self.serial_no = serial_no[0]
|
||||
off += serial_no[1]
|
||||
|
||||
access_code = decode_str(data, off)
|
||||
self.access_code = access_code[0]
|
||||
off += access_code[1]
|
||||
|
||||
chip_id = decode_str(data, off)
|
||||
self.chip_id = chip_id[0]
|
||||
off += chip_id[1]
|
||||
|
||||
class SaoGetAuthCardDataResponse(SaoBaseResponse): #GssSite.dll / GssSiteSystem / GameConnectProt / public class get_auth_card_data_R : GameConnect.GssProtocolBase
|
||||
def __init__(self, cmd, profile_data) -> None:
|
||||
@@ -225,8 +267,8 @@ class SaoGetAuthCardDataResponse(SaoBaseResponse): #GssSite.dll / GssSiteSystem
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoHomeCheckAcLoginBonusRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoHomeCheckAcLoginBonusResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -262,8 +304,8 @@ class SaoHomeCheckAcLoginBonusResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetQuestSceneMultiPlayPhotonServerRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoGetQuestSceneMultiPlayPhotonServerResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -289,8 +331,8 @@ class SaoGetQuestSceneMultiPlayPhotonServerResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoTicketRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoTicketResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -318,8 +360,37 @@ class SaoTicketResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoCommonLoginRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
self.cabinet_type = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
self.auth_type = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
store_id = decode_str(data, off)
|
||||
self.store_id = store_id[0]
|
||||
off += store_id[1]
|
||||
|
||||
store_name = decode_str(data, off)
|
||||
self.store_name = store_name[0]
|
||||
off += store_name[1]
|
||||
|
||||
serial_no = decode_str(data, off)
|
||||
self.serial_no = serial_no[0]
|
||||
off += serial_no[1]
|
||||
|
||||
access_code = decode_str(data, off)
|
||||
self.access_code = access_code[0]
|
||||
off += access_code[1]
|
||||
|
||||
chip_id = decode_str(data, off)
|
||||
self.chip_id = chip_id[0]
|
||||
off += chip_id[1]
|
||||
|
||||
self.free_ticket_distribution_target_flag = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
class SaoCommonLoginResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, profile_data) -> None:
|
||||
@@ -362,8 +433,8 @@ class SaoCommonLoginResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoCheckComebackEventRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoCheckComebackEventRequest(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -392,8 +463,9 @@ class SaoCheckComebackEventRequest(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetUserBasicDataRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
self.user_id = decode_str(data, 0)[0]
|
||||
|
||||
class SaoGetUserBasicDataResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, profile_data) -> None:
|
||||
@@ -476,8 +548,9 @@ class SaoGetUserBasicDataResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetHeroLogUserDataListRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
self.user_id = decode_str(data, 0)[0]
|
||||
|
||||
class SaoGetHeroLogUserDataListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, hero_data) -> None:
|
||||
@@ -647,6 +720,8 @@ class SaoGetHeroLogUserDataListResponse(SaoBaseResponse):
|
||||
|
||||
resp_data.hero_log_user_data_list.append(hero_data)
|
||||
|
||||
resp_data["hero_log_user_data_list_size"] = len(resp_data.hero_log_user_data_list)
|
||||
|
||||
# finally, rebuild the resp_data
|
||||
resp_data = resp_struct.build(resp_data)
|
||||
|
||||
@@ -654,8 +729,9 @@ class SaoGetHeroLogUserDataListResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetEquipmentUserDataListRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
self.user_id = decode_str(data, 0)[0]
|
||||
|
||||
class SaoGetEquipmentUserDataListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, equipment_data) -> None:
|
||||
@@ -803,6 +879,8 @@ class SaoGetEquipmentUserDataListResponse(SaoBaseResponse):
|
||||
|
||||
resp_data.equipment_user_data_list.append(equipment_data)
|
||||
|
||||
resp_data["equipment_user_data_list_size"] = len(resp_data.equipment_user_data_list)
|
||||
|
||||
# finally, rebuild the resp_data
|
||||
resp_data = resp_struct.build(resp_data)
|
||||
|
||||
@@ -810,8 +888,9 @@ class SaoGetEquipmentUserDataListResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetItemUserDataListRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
self.user_id = decode_str(data, 0)[0]
|
||||
|
||||
class SaoGetItemUserDataListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, item_data) -> None:
|
||||
@@ -867,6 +946,8 @@ class SaoGetItemUserDataListResponse(SaoBaseResponse):
|
||||
|
||||
resp_data.item_user_data_list.append(item_data)
|
||||
|
||||
resp_data["item_user_data_list_size"] = len(resp_data.item_user_data_list)
|
||||
|
||||
# finally, rebuild the resp_data
|
||||
resp_data = resp_struct.build(resp_data)
|
||||
|
||||
@@ -874,8 +955,8 @@ class SaoGetItemUserDataListResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetSupportLogUserDataListRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoGetSupportLogUserDataListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, supportIdsData) -> None:
|
||||
@@ -939,14 +1020,16 @@ class SaoGetSupportLogUserDataListResponse(SaoBaseResponse):
|
||||
|
||||
resp_data.support_log_user_data_list.append(support_data)
|
||||
|
||||
resp_data["support_log_user_data_list_size"] = len(resp_data.support_log_user_data_list)
|
||||
|
||||
resp_data = resp_struct.build(resp_data)
|
||||
|
||||
self.length = len(resp_data)
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetTitleUserDataListRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoGetTitleUserDataListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, titleIdsData) -> None:
|
||||
@@ -986,6 +1069,8 @@ class SaoGetTitleUserDataListResponse(SaoBaseResponse):
|
||||
|
||||
resp_data.title_user_data_list.append(title_data)
|
||||
|
||||
resp_data["title_user_data_list_size"] = len(resp_data.title_user_data_list)
|
||||
|
||||
# finally, rebuild the resp_data
|
||||
resp_data = resp_struct.build(resp_data)
|
||||
|
||||
@@ -993,8 +1078,9 @@ class SaoGetTitleUserDataListResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetEpisodeAppendDataListRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
self.user_id = decode_str(data, 0)[0]
|
||||
|
||||
class SaoGetEpisodeAppendDataListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, profile_data) -> None:
|
||||
@@ -1044,6 +1130,8 @@ class SaoGetEpisodeAppendDataListResponse(SaoBaseResponse):
|
||||
own_num=self.own_num_list[i],
|
||||
))
|
||||
|
||||
resp_data["episode_append_data_list_size"] = len(resp_data.episode_append_data_list)
|
||||
|
||||
# finally, rebuild the resp_data
|
||||
resp_data = resp_struct.build(resp_data)
|
||||
|
||||
@@ -1051,8 +1139,9 @@ class SaoGetEpisodeAppendDataListResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetPartyDataListRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
self.user_id = decode_str(data, 0)[0]
|
||||
|
||||
class SaoGetPartyDataListResponse(SaoBaseResponse): # Default party
|
||||
def __init__(self, cmd, hero1_data, hero2_data, hero3_data) -> None:
|
||||
@@ -1215,8 +1304,8 @@ class SaoGetPartyDataListResponse(SaoBaseResponse): # Default party
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetQuestScenePrevScanProfileCardRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoGetQuestScenePrevScanProfileCardResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -1352,8 +1441,8 @@ class SaoGetQuestScenePrevScanProfileCardResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetResourcePathInfoRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoGetResourcePathInfoResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -1394,8 +1483,40 @@ class SaoGetResourcePathInfoResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoEpisodePlayStartRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
ticket_id = decode_str(data, off)
|
||||
self.ticket_id = ticket_id[0]
|
||||
off += ticket_id[1]
|
||||
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
self.episode_id = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.play_mode = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
self.play_start_request_data_count = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.play_start_request_data: List[PlayStartRequestData] = []
|
||||
for _ in range(self.play_start_request_data_count):
|
||||
tmp = PlayStartRequestData(data, off)
|
||||
self.play_start_request_data.append(tmp)
|
||||
off += tmp.get_size()
|
||||
|
||||
self.multi_play_start_request_data_count = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.multi_play_start_request_data: List[MultiPlayStartRequestData] = []
|
||||
for _ in range(self.multi_play_start_request_data_count):
|
||||
tmp = MultiPlayStartRequestData(data, off)
|
||||
off += tmp.get_size()
|
||||
self.multi_play_start_request_data.append(tmp)
|
||||
|
||||
class SaoEpisodePlayStartResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, profile_data) -> None:
|
||||
@@ -1441,8 +1562,37 @@ class SaoEpisodePlayStartResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoEpisodePlayEndRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
ticket_id = decode_str(data, off)
|
||||
self.ticket_id = ticket_id[0]
|
||||
off += ticket_id[1]
|
||||
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
self.episode_id = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.play_end_request_data_count = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.play_end_request_data_list: List[PlayEndRequestData] = []
|
||||
for _ in range(self.play_end_request_data_count):
|
||||
tmp = PlayEndRequestData(data, off)
|
||||
off += tmp.get_size()
|
||||
self.play_end_request_data_list.append(tmp)
|
||||
|
||||
self.multi_play_end_request_data_count = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.multi_play_end_request_data_list: List[MultiPlayEndRequestData] = []
|
||||
for _ in range(self.multi_play_end_request_data_count):
|
||||
tmp = MultiPlayEndRequestData(data, off)
|
||||
off += tmp.get_size()
|
||||
self.multi_play_end_request_data_list.append(tmp)
|
||||
|
||||
class SaoEpisodePlayEndResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -1528,9 +1678,74 @@ class SaoEpisodePlayEndResponse(SaoBaseResponse):
|
||||
self.length = len(resp_data)
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoTrialTowerPlayStartRequest(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
ticket_id = decode_str(data, off)
|
||||
self.ticket_id = ticket_id[0]
|
||||
off += ticket_id[1]
|
||||
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
self.trial_tower_id = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.play_mode = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
self.play_start_request_data_count = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.play_start_request_data: List[PlayStartRequestData] = []
|
||||
for _ in range(self.play_start_request_data_count):
|
||||
tmp = PlayStartRequestData(data, off)
|
||||
self.play_start_request_data.append(tmp)
|
||||
off += tmp.get_size()
|
||||
|
||||
self.multi_play_start_request_data_count = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.multi_play_start_request_data: List[MultiPlayStartRequestData] = []
|
||||
for _ in range(self.multi_play_start_request_data_count):
|
||||
tmp = MultiPlayStartRequestData(data, off)
|
||||
off += tmp.get_size()
|
||||
self.multi_play_start_request_data.append(tmp)
|
||||
|
||||
class SaoTrialTowerPlayEndRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
ticket_id = decode_str(data, off)
|
||||
self.ticket_id = ticket_id[0]
|
||||
off += ticket_id[1]
|
||||
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
self.trial_tower_id = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.play_end_request_data_count = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.play_end_request_data_list: List[PlayEndRequestData] = []
|
||||
for _ in range(self.play_end_request_data_count):
|
||||
tmp = PlayEndRequestData(data, off)
|
||||
off += tmp.get_size()
|
||||
self.play_end_request_data_list.append(tmp)
|
||||
|
||||
self.multi_play_end_request_data_count = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.multi_play_end_request_data_list: List[MultiPlayEndRequestData] = []
|
||||
for _ in range(self.multi_play_end_request_data_count):
|
||||
tmp = MultiPlayEndRequestData(data, off)
|
||||
off += tmp.get_size()
|
||||
self.multi_play_end_request_data_list.append(tmp)
|
||||
|
||||
class SaoTrialTowerPlayEndResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -1655,8 +1870,22 @@ class SaoTrialTowerPlayEndResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoEpisodePlayEndUnanalyzedLogFixedRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
ticket_id = decode_str(data, off)
|
||||
self.ticket_id = ticket_id[0]
|
||||
off += ticket_id[1]
|
||||
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
self.episode_id = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.rarity_up_exec_flag = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
class SaoEpisodePlayEndUnanalyzedLogFixedResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, end_session_data) -> None:
|
||||
@@ -1728,6 +1957,10 @@ class SaoEpisodePlayEndUnanalyzedLogFixedResponse(SaoBaseResponse):
|
||||
|
||||
resp_data.play_end_unanalyzed_log_reward_data_list.append(reward_resp_data)
|
||||
|
||||
resp_data["play_end_unanalyzed_log_reward_data_list_size"] = len(resp_data.play_end_unanalyzed_log_reward_data_list)
|
||||
for i in range(len(resp_data.play_end_unanalyzed_log_reward_data_list)):
|
||||
resp_data.play_end_unanalyzed_log_reward_data_list[i]["common_reward_data_size"] = len(resp_data.play_end_unanalyzed_log_reward_data_list[i]["common_reward_data"])
|
||||
|
||||
# finally, rebuild the resp_data
|
||||
resp_data = resp_struct.build(resp_data)
|
||||
|
||||
@@ -1735,12 +1968,14 @@ class SaoEpisodePlayEndUnanalyzedLogFixedResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetQuestSceneUserDataListRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
self.user_id = decode_str(data, 0)[0]
|
||||
|
||||
class SaoGetQuestSceneUserDataListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, quest_data) -> None:
|
||||
super().__init__(cmd)
|
||||
self.length = None
|
||||
self.result = 1
|
||||
|
||||
# quest_scene_user_data_list_size
|
||||
@@ -1837,6 +2072,10 @@ class SaoGetQuestSceneUserDataListResponse(SaoBaseResponse):
|
||||
|
||||
resp_data.quest_scene_user_data_list.append(quest_resp_data)
|
||||
|
||||
resp_data["quest_scene_user_data_list_size"] = len(resp_data.quest_scene_user_data_list)
|
||||
for i in range(len(resp_data.quest_scene_user_data_list)):
|
||||
resp_data.quest_scene_user_data_list[i]["quest_scene_best_score_user_data_size"] = len(resp_data.quest_scene_user_data_list[i]["quest_scene_best_score_user_data"])
|
||||
|
||||
# finally, rebuild the resp_data
|
||||
resp_data = resp_struct.build(resp_data)
|
||||
|
||||
@@ -1844,8 +2083,8 @@ class SaoGetQuestSceneUserDataListResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoCheckYuiMedalGetConditionRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoCheckYuiMedalGetConditionResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -1875,8 +2114,8 @@ class SaoCheckYuiMedalGetConditionResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetYuiMedalBonusUserDataRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoGetYuiMedalBonusUserDataResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -1920,8 +2159,8 @@ class SaoGetYuiMedalBonusUserDataResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoCheckProfileCardUsedRewardRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoCheckProfileCardUsedRewardResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -1952,8 +2191,30 @@ class SaoCheckProfileCardUsedRewardResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoSynthesizeEnhancementHeroLogRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
ticket_id = decode_str(data, off)
|
||||
self.ticket_id = ticket_id[0]
|
||||
off += ticket_id[1]
|
||||
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
origin_user_hero_log_id = decode_str(data, off)
|
||||
self.origin_user_hero_log_id = origin_user_hero_log_id[0]
|
||||
off += origin_user_hero_log_id[1]
|
||||
|
||||
self.material_common_reward_user_data_list: List[MaterialCommonRewardUserData] = []
|
||||
|
||||
self.material_common_reward_user_data_count = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
for _ in range(self.material_common_reward_user_data_count):
|
||||
mat = MaterialCommonRewardUserData(data, off)
|
||||
off += mat.get_size()
|
||||
self.material_common_reward_user_data_list.append(mat)
|
||||
|
||||
class SaoSynthesizeEnhancementHeroLogResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, hero_data) -> None:
|
||||
@@ -2100,15 +2361,39 @@ class SaoSynthesizeEnhancementHeroLogResponse(SaoBaseResponse):
|
||||
|
||||
resp_data.hero_log_user_data_list.append(hero_data)
|
||||
|
||||
resp_data["hero_log_user_data_list_size"] = len(resp_data.hero_log_user_data_list)
|
||||
|
||||
# finally, rebuild the resp_data
|
||||
resp_data = resp_struct.build(resp_data)
|
||||
|
||||
self.length = len(resp_data)
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoSynthesizeEnhancementEquipment(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
class SaoSynthesizeEnhancementEquipmentRequest(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
ticket_id = decode_str(data, off)
|
||||
self.ticket_id = ticket_id[0]
|
||||
off += ticket_id[1]
|
||||
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
origin_user_equipment_id = decode_str(data, off)
|
||||
self.origin_user_equipment_id = origin_user_equipment_id[0]
|
||||
off += origin_user_equipment_id[1]
|
||||
|
||||
self.material_common_reward_user_data_list: List[MaterialCommonRewardUserData] = []
|
||||
|
||||
self.material_common_reward_user_data_count = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
for _ in range(self.material_common_reward_user_data_count):
|
||||
mat = MaterialCommonRewardUserData(data, off)
|
||||
off += mat.get_size()
|
||||
self.material_common_reward_user_data_list.append(mat)
|
||||
|
||||
class SaoSynthesizeEnhancementEquipmentResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd, synthesize_equipment_data) -> None:
|
||||
@@ -2237,6 +2522,8 @@ class SaoSynthesizeEnhancementEquipmentResponse(SaoBaseResponse):
|
||||
|
||||
resp_data.after_equipment_user_data.append(synthesize_equipment_data)
|
||||
|
||||
resp_data["after_equipment_user_data_size"] = len(resp_data.after_equipment_user_data)
|
||||
|
||||
# finally, rebuild the resp_data
|
||||
resp_data = resp_struct.build(resp_data)
|
||||
|
||||
@@ -2244,8 +2531,8 @@ class SaoSynthesizeEnhancementEquipmentResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetDefragMatchBasicDataRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoGetDefragMatchBasicDataResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -2291,8 +2578,8 @@ class SaoGetDefragMatchBasicDataResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetDefragMatchRankingUserDataRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoGetDefragMatchRankingUserDataResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -2351,8 +2638,8 @@ class SaoGetDefragMatchRankingUserDataResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetDefragMatchLeaguePointRankingListRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoGetDefragMatchLeaguePointRankingListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -2423,8 +2710,8 @@ class SaoGetDefragMatchLeaguePointRankingListResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoGetDefragMatchLeagueScoreRankingListRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoGetDefragMatchLeagueScoreRankingListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -2495,8 +2782,8 @@ class SaoGetDefragMatchLeagueScoreRankingListResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoBnidSerialCodeCheckRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoBnidSerialCodeCheckResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -2526,8 +2813,8 @@ class SaoBnidSerialCodeCheckResponse(SaoBaseResponse):
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoScanQrQuestProfileCardRequest(SaoBaseRequest):
|
||||
def __init__(self, data: bytes) -> None:
|
||||
super().__init__(data)
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
|
||||
class SaoScanQrQuestProfileCardResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd) -> None:
|
||||
@@ -2737,8 +3024,209 @@ class SaoScanQrQuestProfileCardResponse(SaoBaseResponse):
|
||||
|
||||
resp_data.read_profile_card_data.append(hero_data)
|
||||
|
||||
resp_data["read_profile_card_data_size"] = len(resp_data.read_profile_card_data)
|
||||
|
||||
# finally, rebuild the resp_data
|
||||
resp_data = resp_struct.build(resp_data)
|
||||
|
||||
self.length = len(resp_data)
|
||||
return super().make() + resp_data
|
||||
return super().make() + resp_data
|
||||
|
||||
class SaoConsumeCreditGuestRequest(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
shop_id = decode_str(data, off)
|
||||
self.shop_id = shop_id[0]
|
||||
off += shop_id[1]
|
||||
|
||||
serial_num = decode_str(data, off)
|
||||
self.serial_num = serial_num[0]
|
||||
off += serial_num[1]
|
||||
|
||||
self.cab_type = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
self.act_type = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
self.consume_num = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
class SaoChangePartyRequest(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
ticket_id = decode_str(data, off)
|
||||
self.ticket_id = ticket_id[0]
|
||||
off += ticket_id[1]
|
||||
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
self.act_type = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
self.party_data_count = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.party_data_list: List[PartyData] = []
|
||||
|
||||
for _ in range(self.party_data_count):
|
||||
tmp = PartyData(data, off)
|
||||
self.party_data_list.append(tmp)
|
||||
off += tmp.get_size()
|
||||
|
||||
class TrialTowerPlayEndUnanalyzedLogFixed(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
ticket_id = decode_str(data, off)
|
||||
self.ticket_id = ticket_id[0]
|
||||
off += ticket_id[1]
|
||||
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
self.trial_tower_id = decode_int(data, off)
|
||||
off += INT_OFF
|
||||
|
||||
self.rarity_up_exec_flag = decode_byte(data, off)
|
||||
off += BYTE_OFF
|
||||
|
||||
class GetShopResourceSalesDataListRequest(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
class GetShopResourceSalesDataListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd_id: int) -> None:
|
||||
super().__init__(cmd_id)
|
||||
self.result = 1 # byte
|
||||
self.shop_resource_sales_data: List[ShopResourceSalesData] = []
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_byte(self.result)
|
||||
ret += encode_arr_cls(self.shop_resource_sales_data)
|
||||
|
||||
self.header.length = len(ret)
|
||||
return super().make() + ret
|
||||
|
||||
class GetYuiMedalShopUserDataListRequest(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
class GetYuiMedalShopUserDataListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd_id: int) -> None:
|
||||
super().__init__(cmd_id)
|
||||
self.result = 1 # byte
|
||||
self.user_data_list: List[YuiMedalShopUserData] = []
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_byte(self.result)
|
||||
ret += encode_arr_cls(self.user_data_list)
|
||||
|
||||
self.header.length = len(ret)
|
||||
return super().make() + ret
|
||||
|
||||
class GetGashaMedalShopUserDataListRequest(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
off = 0
|
||||
user_id = decode_str(data, off)
|
||||
self.user_id = user_id[0]
|
||||
off += user_id[1]
|
||||
|
||||
class GetGashaMedalShopUserDataListResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd_id: int) -> None:
|
||||
super().__init__(cmd_id)
|
||||
self.result = 1 # byte
|
||||
self.data_list: List[GashaMedalShopUserData] = []
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_byte(self.result)
|
||||
ret += encode_arr_cls(self.data_list)
|
||||
|
||||
self.header.length = len(ret)
|
||||
return super().make() + ret
|
||||
|
||||
class GetMYuiMedalShopDataRequest(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
self.dummy = decode_byte(data, 0)
|
||||
|
||||
class GetMYuiMedalShopDataResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd_id: int) -> None:
|
||||
super().__init__(cmd_id)
|
||||
self.result = 1 # byte
|
||||
self.data_list: List[YuiMedalShopData] = []
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_byte(self.result)
|
||||
ret += encode_arr_cls(self.data_list)
|
||||
|
||||
self.header.length = len(ret)
|
||||
return super().make() + ret
|
||||
|
||||
class GetMYuiMedalShopItemsRequest(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
self.dummy = decode_byte(data, 0)
|
||||
|
||||
class GetMYuiMedalShopItemsResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd_id: int) -> None:
|
||||
super().__init__(cmd_id)
|
||||
self.result = 1 # byte
|
||||
self.data_list: List[YuiMedalShopItemData] = []
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_byte(self.result)
|
||||
ret += encode_arr_cls(self.data_list)
|
||||
|
||||
self.header.length = len(ret)
|
||||
return super().make() + ret
|
||||
|
||||
class GetMGashaMedalShopsRequest(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
self.dummy = decode_byte(data, 0)
|
||||
|
||||
class GetMGashaMedalShopsResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd_id: int) -> None:
|
||||
super().__init__(cmd_id)
|
||||
self.result = 1 # byte
|
||||
self.data_list: List[GashaMedalShop] = []
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_byte(self.result)
|
||||
ret += encode_arr_cls(self.data_list)
|
||||
|
||||
self.header.length = len(ret)
|
||||
return super().make() + ret
|
||||
|
||||
class GetMResEarnCampaignShopsRequest(SaoBaseRequest):
|
||||
def __init__(self, header: SaoRequestHeader, data: bytes) -> None:
|
||||
super().__init__(header, data)
|
||||
self.dummy = decode_byte(data, 0)
|
||||
|
||||
class GetMResEarnCampaignShopsResponse(SaoBaseResponse):
|
||||
def __init__(self, cmd_id: int) -> None:
|
||||
super().__init__(cmd_id)
|
||||
self.result = 1 # byte
|
||||
self.data_list: List[ResEarnCampaignShop] = []
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_byte(self.result)
|
||||
ret += encode_arr_cls(self.data_list)
|
||||
|
||||
self.header.length = len(ret)
|
||||
return super().make() + ret
|
||||
|
||||
917
titles/sao/handlers/helpers.py
Normal file
917
titles/sao/handlers/helpers.py
Normal file
@@ -0,0 +1,917 @@
|
||||
from typing import Tuple, List, Optional
|
||||
import struct
|
||||
import logging
|
||||
from datetime import datetime
|
||||
|
||||
BIGINT_OFF = 16
|
||||
LONG_OFF = 8
|
||||
INT_OFF = 4
|
||||
SHORT_OFF = 2
|
||||
BYTE_OFF = 1
|
||||
|
||||
DT_FMT = "%Y%m%d%H%M%S"
|
||||
|
||||
def fmt_dt(d: Optional[datetime] = None) -> str:
|
||||
if d is None:
|
||||
d = datetime.fromtimestamp(0)
|
||||
return d.strftime(DT_FMT)
|
||||
|
||||
def prs_dt(s: Optional[str] = None) -> datetime:
|
||||
if not s:
|
||||
s = "19691231190000"
|
||||
return datetime.strptime(s, DT_FMT)
|
||||
|
||||
def decode_num(data: bytes, offset: int, size: int) -> int:
|
||||
try:
|
||||
return int.from_bytes(data[offset:offset + size], 'big')
|
||||
except:
|
||||
logging.getLogger('sao').error(f"Failed to parse {data[offset:offset + size]} as BE number of width {size}")
|
||||
return 0
|
||||
|
||||
def decode_byte(data: bytes, offset: int) -> int:
|
||||
return decode_num(data, offset, BYTE_OFF)
|
||||
|
||||
def decode_short(data: bytes, offset: int) -> int:
|
||||
return decode_num(data, offset, SHORT_OFF)
|
||||
|
||||
def decode_int(data: bytes, offset: int) -> int:
|
||||
return decode_num(data, offset, INT_OFF)
|
||||
|
||||
def decode_long(data: bytes, offset: int) -> int:
|
||||
return decode_num(data, offset, LONG_OFF)
|
||||
|
||||
def decode_bigint(data: bytes, offset: int) -> int:
|
||||
return decode_num(data, offset, BIGINT_OFF)
|
||||
|
||||
def decode_str(data: bytes, offset: int) -> Tuple[str, int]:
|
||||
try:
|
||||
str_len = decode_int(data, offset)
|
||||
num_bytes_decoded = INT_OFF + str_len
|
||||
str_out = data[offset + INT_OFF:offset + num_bytes_decoded].decode("utf-16-le", errors="replace")
|
||||
return (str_out, num_bytes_decoded)
|
||||
except:
|
||||
logging.getLogger('sao').error(f"Failed to parse {data[offset:]} as string!")
|
||||
return ("", 0)
|
||||
|
||||
def decode_arr_num(data: bytes, offset:int, element_size: int) -> Tuple[List[int], int]:
|
||||
size = 0
|
||||
num_obj = decode_int(data, offset + size)
|
||||
size += INT_OFF
|
||||
|
||||
ret: List[int] = []
|
||||
for _ in range(num_obj):
|
||||
ret.append(decode_num(data, offset + size, element_size))
|
||||
size += element_size
|
||||
|
||||
return (ret, size)
|
||||
|
||||
def decode_arr_str(data: bytes, offset: int) -> Tuple[List[str], int]:
|
||||
size = 0
|
||||
num_obj = decode_int(data, offset + size)
|
||||
size += INT_OFF
|
||||
|
||||
ret: List[str] = []
|
||||
for _ in range(num_obj):
|
||||
tmp = decode_str(data, offset + size)
|
||||
ret.append(tmp[0])
|
||||
size += tmp[1]
|
||||
|
||||
return (ret, size)
|
||||
|
||||
def encode_byte(data: int) -> bytes:
|
||||
return struct.pack("!b", data)
|
||||
|
||||
def encode_short(data: int) -> bytes:
|
||||
return struct.pack("!h", data)
|
||||
|
||||
def encode_int(data: int) -> bytes:
|
||||
return struct.pack("!i", data)
|
||||
|
||||
def encode_long(data: int) -> bytes:
|
||||
return struct.pack("!l", data)
|
||||
|
||||
def encode_bigint(data: int) -> bytes:
|
||||
return struct.pack("!q", data)
|
||||
|
||||
def encode_str(s: str) -> bytes:
|
||||
try:
|
||||
str_bytes = s.encode("utf-16-le", errors="replace")
|
||||
str_len_bytes = struct.pack("!I", len(str_bytes))
|
||||
return str_len_bytes + str_bytes
|
||||
except:
|
||||
logging.getLogger('sao').error(f"Failed to encode {s} as bytes!")
|
||||
return b""
|
||||
|
||||
def encode_arr_num(data: List[int], element_size: int) -> bytes:
|
||||
ret = encode_int(len(data))
|
||||
|
||||
if element_size == BYTE_OFF:
|
||||
for x in data:
|
||||
ret += encode_byte(x)
|
||||
elif element_size == SHORT_OFF:
|
||||
for x in data:
|
||||
ret += encode_short(x)
|
||||
elif element_size == INT_OFF:
|
||||
for x in data:
|
||||
ret += encode_int(x)
|
||||
elif element_size == LONG_OFF:
|
||||
for x in data:
|
||||
ret += encode_long(x)
|
||||
elif element_size == BIGINT_OFF:
|
||||
for x in data:
|
||||
ret += encode_bigint(x)
|
||||
else:
|
||||
logging.getLogger('sao').error(f"Unknown element size {element_size}")
|
||||
return b"\x00" * INT_OFF
|
||||
|
||||
return ret
|
||||
|
||||
class BaseHelper:
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
self._sz = 0
|
||||
|
||||
@classmethod
|
||||
def from_args(cls) -> "BaseHelper":
|
||||
return cls(b"", 0)
|
||||
|
||||
def get_size(self) -> int:
|
||||
return self._sz
|
||||
|
||||
def make(self) -> bytes:
|
||||
return b""
|
||||
|
||||
def decode_arr_cls(data: bytes, offset: int, cls: BaseHelper):
|
||||
size = 0
|
||||
num_cls = decode_int(data, offset + size)
|
||||
cls_type = type(cls)
|
||||
|
||||
ret: List[cls_type] = []
|
||||
for _ in range(num_cls):
|
||||
tmp = cls(data, offset + size)
|
||||
size += tmp.get_size()
|
||||
ret.append(tmp)
|
||||
|
||||
return (ret, size)
|
||||
|
||||
def encode_arr_cls(data: List[BaseHelper]) -> bytes:
|
||||
ret = encode_int(len(data))
|
||||
|
||||
for x in data:
|
||||
ret += x.make()
|
||||
|
||||
return ret
|
||||
|
||||
class MaterialCommonRewardUserData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
self.common_reward_type = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
|
||||
user_common_reward_id = decode_str(data, offset + self._sz)
|
||||
self.user_common_reward_id = user_common_reward_id[0]
|
||||
self._sz += user_common_reward_id[1]
|
||||
|
||||
class PartyTeamData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
sz = 0
|
||||
user_party_team_id = decode_str(data, offset + sz)
|
||||
self.user_party_team_id = user_party_team_id[0]
|
||||
sz += user_party_team_id[1]
|
||||
|
||||
self.arrangement_num = decode_byte(data, offset + sz)
|
||||
sz += BYTE_OFF
|
||||
|
||||
user_hero_log_id = decode_str(data, offset + sz)
|
||||
self.user_hero_log_id = user_hero_log_id[0]
|
||||
sz += user_hero_log_id[1]
|
||||
|
||||
main_weapon_user_equipment_id = decode_str(data, offset + sz)
|
||||
self.main_weapon_user_equipment_id = main_weapon_user_equipment_id[0]
|
||||
sz += main_weapon_user_equipment_id[1]
|
||||
|
||||
sub_equipment_user_equipment_id = decode_str(data, offset + sz)
|
||||
self.sub_equipment_user_equipment_id = sub_equipment_user_equipment_id[0]
|
||||
sz += sub_equipment_user_equipment_id[1]
|
||||
|
||||
self.skill_slot1_skill_id = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.skill_slot2_skill_id = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.skill_slot3_skill_id = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.skill_slot4_skill_id = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.skill_slot5_skill_id = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self._sz = sz
|
||||
|
||||
class PartyData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
sz = 0
|
||||
user_party_id = decode_str(data, offset + sz)
|
||||
self.user_party_id = user_party_id[0]
|
||||
sz += user_party_id[1]
|
||||
|
||||
self.team_no = decode_byte(data, offset + sz)
|
||||
sz += BYTE_OFF
|
||||
|
||||
self.party_team_data_count = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.party_team_data_list: List[PartyTeamData] = []
|
||||
for _ in range(self.party_team_data_count):
|
||||
tmp = PartyTeamData(data, offset + sz)
|
||||
self.party_team_data_list.append(tmp)
|
||||
sz += tmp.get_size()
|
||||
|
||||
self._sz = sz
|
||||
|
||||
class PlayStartRequestData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
sz = 0
|
||||
user_party_id = decode_str(data, offset + sz)
|
||||
self.user_party_id = user_party_id[0]
|
||||
sz += user_party_id[1]
|
||||
|
||||
appoint_leader_resource_card_code = decode_str(data, offset + sz)
|
||||
self.appoint_leader_resource_card_code = appoint_leader_resource_card_code[0]
|
||||
sz += appoint_leader_resource_card_code[1]
|
||||
|
||||
use_profile_card_code = decode_str(data, offset + sz)
|
||||
self.use_profile_card_code = use_profile_card_code[0]
|
||||
sz += use_profile_card_code[1]
|
||||
|
||||
self.quest_drop_boost_apply_flag = decode_byte(data, offset + sz)
|
||||
sz += BYTE_OFF
|
||||
|
||||
self._sz = sz
|
||||
|
||||
class GetPlayerTraceData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
user_quest_scene_player_trace_id = decode_str(data, offset)
|
||||
self.user_quest_scene_player_trace_id = user_quest_scene_player_trace_id[0]
|
||||
self._sz = user_quest_scene_player_trace_id[1]
|
||||
|
||||
class BaseGetData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
self.get_hero_log_exp = decode_int(data, offset)
|
||||
offset += INT_OFF
|
||||
|
||||
self.get_col = decode_int(data, offset)
|
||||
|
||||
self._sz = INT_OFF + INT_OFF
|
||||
|
||||
class RareDropData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
self.quest_rare_drop_id = decode_int(data, offset)
|
||||
|
||||
self._sz = INT_OFF
|
||||
|
||||
class UnanalyzedLogTmpRewardData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
self.unanalyzed_log_grade_id = decode_int(data, offset)
|
||||
|
||||
self._sz = INT_OFF
|
||||
|
||||
class SpecialRareDropData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
self.quest_special_rare_drop_id = decode_int(data, offset)
|
||||
|
||||
self._sz = INT_OFF
|
||||
|
||||
class EventItemData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
self.event_item_id = decode_int(data, offset)
|
||||
offset += INT_OFF
|
||||
self.get_num = decode_short(data, offset)
|
||||
|
||||
self._sz = INT_OFF + SHORT_OFF
|
||||
|
||||
class DiscoveryEnemyData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
self.enemy_kind_id = decode_int(data, offset)
|
||||
offset += INT_OFF
|
||||
self.destroy_num = decode_short(data, offset)
|
||||
|
||||
self._sz = INT_OFF + SHORT_OFF
|
||||
|
||||
class DestroyBossData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
self.boss_type = decode_byte(data, offset)
|
||||
offset += BYTE_OFF
|
||||
self.enemy_kind_id = decode_int(data, offset)
|
||||
offset += INT_OFF
|
||||
self.mission_difficulty_id = decode_short(data, offset)
|
||||
|
||||
self._sz = INT_OFF + SHORT_OFF + BYTE_OFF
|
||||
|
||||
class MissionData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
self.mission_id = decode_int(data, offset)
|
||||
offset += INT_OFF
|
||||
self.clear_flag = decode_byte(data, offset)
|
||||
offset += BYTE_OFF
|
||||
self.destroy_num = decode_short(data, offset)
|
||||
|
||||
self._sz = INT_OFF + SHORT_OFF + BYTE_OFF
|
||||
|
||||
class ScoreData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
self.clear_time = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.combo_num = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
total_damage = decode_str(data, offset + self._sz)
|
||||
self.total_damage = total_damage[0]
|
||||
self._sz += total_damage[1]
|
||||
|
||||
self.concurrent_destroying_num = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
|
||||
self.reaching_skill_level = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
|
||||
self.ko_chara_num = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
|
||||
self.acceleration_invocation_num = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
|
||||
self.boss_destroying_num = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
|
||||
self.synchro_skill_used_flag = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
|
||||
self.used_friend_skill_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.friend_skill_used_flag = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
|
||||
self.continue_cnt = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
|
||||
self.total_loss_num = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
|
||||
class PlayEndRequestData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
sz = 0
|
||||
self.play_result_flag = decode_byte(data, offset + sz)
|
||||
sz += BYTE_OFF
|
||||
|
||||
self.base_get_data_count = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.base_get_data_list: List[BaseGetData] = []
|
||||
for _ in range(self.base_get_data_count):
|
||||
tmp = BaseGetData(data, offset + sz)
|
||||
sz += tmp.get_size()
|
||||
self.base_get_data_list.append(tmp)
|
||||
|
||||
self.get_player_trace_data_count = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.get_player_trace_data: List[GetPlayerTraceData] = []
|
||||
for _ in range(self.get_player_trace_data_count):
|
||||
tmp = GetPlayerTraceData(data, offset + sz)
|
||||
sz += tmp.get_size()
|
||||
self.get_player_trace_data.append(tmp)
|
||||
|
||||
self.get_rare_drop_data_count = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.get_rare_drop_data_list: List[RareDropData] = []
|
||||
for _ in range(self.get_rare_drop_data_count):
|
||||
tmp = RareDropData(data, offset + sz)
|
||||
sz += tmp.get_size()
|
||||
self.get_rare_drop_data_list.append(tmp)
|
||||
|
||||
self.get_special_rare_drop_data_count = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.get_special_rare_drop_data_list: List[SpecialRareDropData] = []
|
||||
for _ in range(self.get_special_rare_drop_data_count):
|
||||
tmp = SpecialRareDropData(data, offset + sz)
|
||||
sz += tmp.get_size()
|
||||
self.get_special_rare_drop_data_list.append(tmp)
|
||||
|
||||
self.get_unanalyzed_log_tmp_reward_data_count = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.get_unanalyzed_log_tmp_reward_data_list: List[UnanalyzedLogTmpRewardData] = []
|
||||
for _ in range(self.get_unanalyzed_log_tmp_reward_data_count):
|
||||
tmp = UnanalyzedLogTmpRewardData(data, offset + sz)
|
||||
sz += tmp.get_size()
|
||||
self.get_unanalyzed_log_tmp_reward_data_list.append(tmp)
|
||||
|
||||
self.get_event_item_data_count = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.get_event_item_data_list: List[EventItemData] = []
|
||||
for _ in range(self.get_event_item_data_count):
|
||||
tmp = EventItemData(data, offset + sz)
|
||||
sz += tmp.get_size()
|
||||
self.get_event_item_data_list.append(tmp)
|
||||
|
||||
self.discovery_enemy_data_count = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.discovery_enemy_data_list: List[DiscoveryEnemyData] = []
|
||||
for _ in range(self.discovery_enemy_data_count):
|
||||
tmp = DiscoveryEnemyData(data, offset + sz)
|
||||
sz += tmp.get_size()
|
||||
self.discovery_enemy_data_list.append(tmp)
|
||||
|
||||
self.destroy_boss_data_count = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.destroy_boss_data_list: List[DestroyBossData] = []
|
||||
for _ in range(self.destroy_boss_data_count):
|
||||
tmp = DestroyBossData(data, offset + sz)
|
||||
sz += tmp.get_size()
|
||||
self.destroy_boss_data_list.append(tmp)
|
||||
|
||||
self.mission_data_count = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.mission_data_list: List[MissionData] = []
|
||||
for _ in range(self.mission_data_count):
|
||||
tmp = MissionData(data, offset + sz)
|
||||
sz += tmp.get_size()
|
||||
self.mission_data_list.append(tmp)
|
||||
|
||||
self.score_data_count = decode_int(data, offset + sz)
|
||||
sz += INT_OFF
|
||||
|
||||
self.score_data_list: List[ScoreData] = []
|
||||
for _ in range(self.score_data_count):
|
||||
tmp = ScoreData(data, offset + sz)
|
||||
sz += tmp.get_size()
|
||||
self.score_data_list.append(tmp)
|
||||
|
||||
self._sz = sz
|
||||
|
||||
class EntryUserData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
store_id = decode_str(data, offset + self._sz)
|
||||
self.store_id = store_id[0]
|
||||
self._sz += store_id[1]
|
||||
|
||||
user_id = decode_str(data, offset + self._sz)
|
||||
self.user_id = user_id[0]
|
||||
self._sz += user_id[1]
|
||||
|
||||
self.host_flag = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
|
||||
class MultiPlayStartRequestData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
room_id = decode_str(data, offset + self._sz)
|
||||
self.room_id = room_id[0]
|
||||
self._sz += room_id[1]
|
||||
|
||||
self.matching_mode = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
|
||||
self.entry_user_data_count = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.entry_user_data_list: List[EntryUserData] = []
|
||||
for _ in range(self.entry_user_data_count):
|
||||
tmp = EntryUserData(data, offset + self._sz)
|
||||
self._sz += tmp.get_size()
|
||||
self.entry_user_data_list.append(tmp)
|
||||
|
||||
class MultiPlayEndRequestData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
self.dummy_1 = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
self.dummy_2 = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
self.dummy_3 = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
|
||||
class SalesResourceData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
self.common_reward_type = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
self.common_reward_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.property1_property_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property1_value1 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property1_value2 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.property2_property_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property2_value1 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property2_value2 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.property3_property_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property3_value1 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property3_value2 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.property4_property_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property4_value1 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property4_value2 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
@classmethod
|
||||
def from_args(cls, reward_type: int = 0, reward_id: int = 0) -> "SalesResourceData":
|
||||
ret = cls(b"\x00" * 54, 0)
|
||||
ret.common_reward_type = reward_type # short
|
||||
ret.common_reward_id = reward_id # int
|
||||
|
||||
return ret
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = b""
|
||||
ret += encode_short(self.common_reward_type)
|
||||
ret += encode_int(self.common_reward_id)
|
||||
|
||||
ret += encode_int(self.property1_property_id)
|
||||
ret += encode_int(self.property1_value1)
|
||||
ret += encode_int(self.property1_value2)
|
||||
|
||||
ret += encode_int(self.property2_property_id)
|
||||
ret += encode_int(self.property2_value1)
|
||||
ret += encode_int(self.property2_value2)
|
||||
|
||||
ret += encode_int(self.property3_property_id)
|
||||
ret += encode_int(self.property3_value1)
|
||||
ret += encode_int(self.property3_value2)
|
||||
|
||||
ret += encode_int(self.property4_property_id)
|
||||
ret += encode_int(self.property4_value1)
|
||||
ret += encode_int(self.property4_value2)
|
||||
|
||||
class ShopResourceSalesData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
user_shop_resource_id = decode_str(data, offset + self._sz)
|
||||
self.user_shop_resource_id = user_shop_resource_id[0]
|
||||
self._sz = user_shop_resource_id[1]
|
||||
|
||||
discharge_user_id = decode_str(data, offset + self._sz)
|
||||
self.discharge_user_id = discharge_user_id[0]
|
||||
self._sz = discharge_user_id[1]
|
||||
|
||||
self.remaining_num = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
self.purchase_num = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
|
||||
sales_start_date = decode_str(data, offset + self._sz)
|
||||
self.sales_start_date = prs_dt(sales_start_date[0])
|
||||
self._sz = sales_start_date[1]
|
||||
|
||||
sales_resource_data_list = decode_arr_cls(data, offset + self._sz, SalesResourceData)
|
||||
self.sales_resource_data_list: List[SalesResourceData] = sales_resource_data_list[0]
|
||||
self._sz += sales_resource_data_list[1]
|
||||
|
||||
@classmethod
|
||||
def from_args(cls, resource_id: str = "0", discharge_id: str = "0", remaining: int = 0, purchased: int = 0) -> "ShopResourceSalesData":
|
||||
ret = cls(b"\x00" * 20, 0)
|
||||
ret.user_shop_resource_id = resource_id
|
||||
ret.discharge_user_id = discharge_id
|
||||
ret.remaining_num = remaining # short
|
||||
ret.purchase_num = purchased # short
|
||||
ret.sales_start_date = prs_dt()
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_str(self.user_shop_resource_id)
|
||||
ret += encode_str(self.discharge_user_id)
|
||||
ret += encode_short(self.remaining_num)
|
||||
ret += encode_short(self.purchase_num)
|
||||
ret += encode_str(fmt_dt(self.sales_start_date))
|
||||
ret += encode_arr_cls(self.sales_resource_data_list)
|
||||
return ret
|
||||
|
||||
class YuiMedalShopUserData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
self.yui_medal_shop_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.purchase_num = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
last_purchase_date = decode_str(data, offset + self._sz)
|
||||
self.last_purchase_date = last_purchase_date[0]
|
||||
self._sz += last_purchase_date[1]
|
||||
|
||||
@classmethod
|
||||
def from_args(cls, yui_medal_shop_id: int = 0, purchase_num: int = 0, last_purchase_date: datetime = datetime.fromtimestamp(0)) -> "YuiMedalShopUserData":
|
||||
ret = cls(b"\x00" * 20, 0)
|
||||
ret.yui_medal_shop_id = yui_medal_shop_id
|
||||
ret.purchase_num = purchase_num
|
||||
ret.last_purchase_date = last_purchase_date
|
||||
return ret
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_int(self.yui_medal_shop_id)
|
||||
ret += encode_int(self.purchase_num)
|
||||
ret += encode_str(fmt_dt(self.last_purchase_date))
|
||||
return ret
|
||||
|
||||
class GashaMedalShopUserData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
self.gasha_medal_shop_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.purchase_num = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
@classmethod
|
||||
def from_args(cls, gasha_medal_shop_id: int = 0, purchase_num: int = 0) -> "GashaMedalShopUserData":
|
||||
ret = cls(b"\x00" * 20, 0)
|
||||
ret.gasha_medal_shop_id = gasha_medal_shop_id
|
||||
ret.purchase_num = purchase_num
|
||||
return ret
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_int(self.gasha_medal_shop_id)
|
||||
ret += encode_int(self.purchase_num)
|
||||
return ret
|
||||
|
||||
class YuiMedalShopData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
self.yui_medal_shop_id = decode_int(data, offset + self._sz)
|
||||
|
||||
name = decode_str(data, offset + self._sz)
|
||||
self.name = name[0]
|
||||
self._sz += name[1]
|
||||
|
||||
description = decode_str(data, offset + self._sz)
|
||||
self.description = description[0]
|
||||
self._sz += description[1]
|
||||
|
||||
self.selling_yui_medal = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
self.selling_col = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.selling_event_item_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.selling_event_item_num = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.selling_ticket_num = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.purchase_limit = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
self.pick_up_flag = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
self.product_category = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
self.sales_type = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
self.target_days = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
self.target_hour = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
self.interval_hour = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
|
||||
sales_start_date = decode_str(data, offset + self._sz)
|
||||
self.sales_start_date = prs_dt(sales_start_date[0])
|
||||
self._sz += sales_start_date[1]
|
||||
|
||||
sales_end_date = decode_str(data, offset + self._sz)
|
||||
self.sales_end_date = prs_dt(sales_end_date[0])
|
||||
self._sz += sales_end_date[1]
|
||||
|
||||
self.sort = decode_byte(data, offset + self._sz)
|
||||
|
||||
@classmethod
|
||||
def from_args(cls, shop_id: int = 0, name: str = "", desc: str = "") -> "YuiMedalShopData":
|
||||
ret = cls(b"\x00" * 43, 0)
|
||||
ret.yui_medal_shop_id = shop_id
|
||||
ret.name = name
|
||||
ret.description = desc
|
||||
return ret
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_int(self.yui_medal_shop_id)
|
||||
ret += encode_str(self.name)
|
||||
ret += encode_str(self.description)
|
||||
ret += encode_short(self.selling_yui_medal)
|
||||
ret += encode_int(self.selling_col)
|
||||
ret += encode_int(self.selling_event_item_id)
|
||||
ret += encode_int(self.selling_event_item_num)
|
||||
ret += encode_int(self.selling_ticket_num)
|
||||
ret += encode_short(self.purchase_limit)
|
||||
ret += encode_byte(self.pick_up_flag)
|
||||
ret += encode_byte(self.product_category)
|
||||
ret += encode_byte(self.sales_type)
|
||||
ret += encode_byte(self.target_days)
|
||||
ret += encode_byte(self.target_hour)
|
||||
ret += encode_byte(self.interval_hour)
|
||||
ret += encode_str(fmt_dt(self.sales_start_date))
|
||||
ret += encode_str(fmt_dt(self.sales_end_date))
|
||||
ret += encode_byte(self.sort)
|
||||
return ret
|
||||
|
||||
class YuiMedalShopItemData(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
self.yui_medal_shop_item_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.yui_medal_shop_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.common_reward_type = decode_byte(data, offset + self._sz)
|
||||
self._sz += BYTE_OFF
|
||||
self.common_reward_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.common_reward_num = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
self.strength = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.property1_property_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property1_value1 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property1_value2 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.property2_property_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property2_value1 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property2_value2 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.property3_property_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property3_value1 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property3_value2 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
self.property4_property_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property4_value1 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.property4_value2 = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
@classmethod
|
||||
def from_args(cls, item_id: int = 0, shop_id: int = 0, reward_type: int = 0, reward_id: int = 0, reward_num: int = 0, strength: int = 0) -> "YuiMedalShopItemData":
|
||||
ret = cls(b"\x00" * 67, 0)
|
||||
ret.yui_medal_shop_item_id = item_id
|
||||
ret.yui_medal_shop_id = shop_id
|
||||
ret.common_reward_type = reward_type
|
||||
ret.common_reward_id = reward_id
|
||||
ret.common_reward_num = reward_num
|
||||
ret.strength = strength
|
||||
return ret
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_int(self.yui_medal_shop_item_id)
|
||||
ret += encode_int(self.yui_medal_shop_id)
|
||||
ret += encode_byte(self.common_reward_type)
|
||||
ret += encode_int(self.common_reward_id)
|
||||
ret += encode_short(self.common_reward_num)
|
||||
ret += encode_int(self.strength)
|
||||
|
||||
ret += encode_int(self.property1_property_id)
|
||||
ret += encode_int(self.property1_value1)
|
||||
ret += encode_int(self.property1_value2)
|
||||
|
||||
ret += encode_int(self.property2_property_id)
|
||||
ret += encode_int(self.property2_value1)
|
||||
ret += encode_int(self.property2_value2)
|
||||
|
||||
ret += encode_int(self.property3_property_id)
|
||||
ret += encode_int(self.property3_value1)
|
||||
ret += encode_int(self.property3_value2)
|
||||
|
||||
ret += encode_int(self.property4_property_id)
|
||||
ret += encode_int(self.property4_value1)
|
||||
ret += encode_int(self.property4_value2)
|
||||
return ret
|
||||
|
||||
class ResEarnCampaignShop(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
self.res_earn_campaign_shop_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.res_earn_campaign_application_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
name = decode_str(data, offset + self._sz)
|
||||
self.name = name[0]
|
||||
self._sz += name[1]
|
||||
|
||||
self.selling_yui_medal = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
self.selling_col = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.selling_event_item_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.selling_event_item_num = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.purchase_limit = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
self.get_application_point = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
|
||||
sales_start_date = decode_str(data, offset + self._sz)
|
||||
self.sales_start_date = prs_dt(sales_start_date[0])
|
||||
self._sz += sales_start_date[1]
|
||||
|
||||
sales_end_date = decode_str(data, offset + self._sz)
|
||||
self.sales_end_date = prs_dt(sales_end_date[0])
|
||||
self._sz += sales_end_date[1]
|
||||
|
||||
@classmethod
|
||||
def from_args(cls, shop_id: int = 0, app_id: int = 0, name: str = "") -> "ResEarnCampaignShop":
|
||||
ret = cls(b"\x00" * 26, 0)
|
||||
ret.res_earn_campaign_shop_id = shop_id
|
||||
ret.res_earn_campaign_application_id = app_id
|
||||
ret.name = name
|
||||
return ret
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_int(self.res_earn_campaign_shop_id)
|
||||
ret = encode_int(self.res_earn_campaign_application_id)
|
||||
ret += encode_str(self.name)
|
||||
ret += encode_short(self.selling_yui_medal)
|
||||
ret += encode_int(self.selling_col)
|
||||
ret += encode_int(self.selling_event_item_id)
|
||||
ret += encode_int(self.selling_event_item_num)
|
||||
ret += encode_short(self.purchase_limit)
|
||||
ret += encode_short(self.get_application_point)
|
||||
ret += encode_str(fmt_dt(self.sales_start_date))
|
||||
ret += encode_str(fmt_dt(self.sales_end_date))
|
||||
return ret
|
||||
|
||||
class GashaMedalShop(BaseHelper):
|
||||
def __init__(self, data: bytes, offset: int) -> None:
|
||||
super().__init__(data, offset)
|
||||
self.gasha_medal_shop_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
|
||||
name = decode_str(data, offset + self._sz)
|
||||
self.name = name[0]
|
||||
self._sz += name[1]
|
||||
|
||||
self.gasha_medal_id = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.use_gasha_medal_num = decode_int(data, offset + self._sz)
|
||||
self._sz += INT_OFF
|
||||
self.purchase_limit = decode_short(data, offset + self._sz)
|
||||
self._sz += SHORT_OFF
|
||||
|
||||
sales_start_date = decode_str(data, offset + self._sz)
|
||||
self.sales_start_date = prs_dt(sales_start_date[0])
|
||||
self._sz += sales_start_date[1]
|
||||
|
||||
sales_end_date = decode_str(data, offset + self._sz)
|
||||
self.sales_end_date = prs_dt(sales_end_date[0])
|
||||
self._sz += sales_end_date[1]
|
||||
|
||||
@classmethod
|
||||
def from_args(cls, shop_id: int = 0, name: str = "", medal_id: int = 0, medal_num: int = 0, purchase_limit: int = 0) -> "GashaMedalShop":
|
||||
ret = cls(b"\x00" * 26, 0)
|
||||
ret.gasha_medal_shop_id = shop_id
|
||||
ret.name = name
|
||||
ret.gasha_medal_id = medal_id
|
||||
ret.use_gasha_medal_num = medal_num
|
||||
ret.purchase_limit = purchase_limit
|
||||
return ret
|
||||
|
||||
def make(self) -> bytes:
|
||||
ret = encode_int(self.gasha_medal_shop_id)
|
||||
ret += encode_str(self.name)
|
||||
ret += encode_int(self.gasha_medal_id)
|
||||
ret += encode_int(self.use_gasha_medal_num)
|
||||
ret += encode_short(self.purchase_limit)
|
||||
ret += encode_str(fmt_dt(self.sales_start_date))
|
||||
ret += encode_str(fmt_dt(self.sales_end_date))
|
||||
return ret
|
||||
@@ -4,6 +4,9 @@ import yaml
|
||||
import logging, coloredlogs
|
||||
from logging.handlers import TimedRotatingFileHandler
|
||||
from os import path
|
||||
from Crypto.Cipher import Blowfish
|
||||
from hashlib import md5
|
||||
import secrets
|
||||
|
||||
from core import CoreConfig, Utils
|
||||
from core.title import BaseServlet
|
||||
@@ -47,11 +50,15 @@ class SaoServlet(BaseServlet):
|
||||
self.logger.inited = True
|
||||
|
||||
self.base = SaoBase(core_cfg, self.game_cfg)
|
||||
self.static_hash = None
|
||||
|
||||
if self.game_cfg.hash.verify_hash:
|
||||
self.static_hash = md5(self.game_cfg.hash.hash_base.encode()).digest() # Greate hashing guys, really validates the data
|
||||
|
||||
def get_endpoint_matchers(self) -> Tuple[List[Tuple[str, str, Dict]], List[Tuple[str, str, Dict]]]:
|
||||
return (
|
||||
[],
|
||||
[("render_POST", "/SaoServlet/{datecode}/proto/if/{endpoint}", {})]
|
||||
[("render_POST", "/{datecode}/proto/if/{category}/{endpoint}", {})]
|
||||
)
|
||||
|
||||
@classmethod
|
||||
@@ -69,10 +76,13 @@ class SaoServlet(BaseServlet):
|
||||
return True
|
||||
|
||||
def get_allnet_info(self, game_code: str, game_ver: int, keychip: str) -> Tuple[str, str]:
|
||||
return (
|
||||
f"http://{self.game_cfg.server.hostname}:{self.game_cfg.server.port}/SaoServlet/",
|
||||
f"{self.game_cfg.server.hostname}/SaoServlet/",
|
||||
)
|
||||
if not self.core_cfg.server.is_using_proxy and Utils.get_title_port_ssl(self.core_cfg):
|
||||
return (
|
||||
f"https://{self.core_cfg.title.hostname}:{self.core_cfg.title.port_ssl}/",
|
||||
f"{self.core_cfg.title.hostname}/",
|
||||
)
|
||||
|
||||
return (f"http://{self.core_cfg.title.hostname}:{self.core_cfg.title.port}/", "")
|
||||
|
||||
def get_mucha_info(self, core_cfg: CoreConfig, cfg_dir: str) -> Tuple[bool, str]:
|
||||
if not self.game_cfg.server.enable:
|
||||
@@ -80,23 +90,62 @@ class SaoServlet(BaseServlet):
|
||||
|
||||
return (True, "SAO1")
|
||||
|
||||
|
||||
def render_POST(self, request: Request, game_code: str, matchers: Dict) -> bytes:
|
||||
endpoint = matchers.get('endpoint', '')
|
||||
request.responseHeaders.addRawHeader(b"content-type", b"text/html; charset=utf-8")
|
||||
iv = b""
|
||||
|
||||
sao_request = request.content.getvalue().hex()
|
||||
req_raw = request.content.read()
|
||||
sao_request = req_raw.hex()
|
||||
req_header = SaoRequestHeader(req_raw)
|
||||
|
||||
cmd_str = f"{req_header.cmd:04x}"
|
||||
|
||||
if self.game_cfg.hash.verify_hash and self.static_hash != req_header.hash:
|
||||
self.logger.error(f"Hash mismatch! Expecting {self.static_hash} but recieved {req_header.hash}")
|
||||
return b""
|
||||
|
||||
if self.game_cfg.crypt.enable:
|
||||
iv = req_raw[40:48]
|
||||
cipher = Blowfish.new(self.game_cfg.crypt.key.encode(), Blowfish.MODE_CBC, iv)
|
||||
crypt_data = req_raw[48:]
|
||||
req_data = cipher.decrypt(crypt_data)
|
||||
self.logger.debug(f"Decrypted {req_data.hex()} with IV {iv.hex()}")
|
||||
|
||||
else:
|
||||
req_data = req_raw[40:]
|
||||
|
||||
handler = getattr(self.base, f"handle_{sao_request[:4]}", None)
|
||||
if handler is None:
|
||||
self.logger.info(f"Generic Handler for {endpoint} - {sao_request[:4]}")
|
||||
self.logger.debug(f"Request: {request.content.getvalue().hex()}")
|
||||
resp = SaoNoopResponse(int.from_bytes(bytes.fromhex(sao_request[:4]), "big")+1)
|
||||
self.logger.debug(f"Response: {resp.make().hex()}")
|
||||
return resp.make()
|
||||
handler = getattr(self.base, f"handle_{cmd_str}", self.base.handle_noop)
|
||||
self.logger.info(f"{endpoint} - {cmd_str} request")
|
||||
self.logger.debug(f"Request: {req_raw.hex()}")
|
||||
resp = handler(req_header, req_data)
|
||||
|
||||
self.logger.info(f"Handler {endpoint} - {sao_request[:4]} request")
|
||||
self.logger.debug(f"Request: {request.content.getvalue().hex()}")
|
||||
resp = handler(sao_request)
|
||||
if resp is None:
|
||||
resp = SaoNoopResponse(req_header.cmd + 1).make()
|
||||
|
||||
if type(resp) == bytes:
|
||||
pass
|
||||
|
||||
elif issubclass(resp, SaoBaseResponse):
|
||||
resp = resp.make()
|
||||
|
||||
else:
|
||||
self.logger.error(f"Unknown response type {type(resp)}")
|
||||
return b""
|
||||
|
||||
self.logger.debug(f"Response: {resp.hex()}")
|
||||
|
||||
if self.game_cfg.crypt.enable:
|
||||
iv = secrets.token_bytes(8)
|
||||
data_to_crypt = resp[24:]
|
||||
while len(data_to_crypt) % 8 != 0:
|
||||
data_to_crypt += b"\x00"
|
||||
|
||||
cipher = Blowfish.new(self.game_cfg.crypt.key.encode(), Blowfish.MODE_CBC, iv)
|
||||
data_crypt = cipher.encrypt(data_to_crypt)
|
||||
crypt_data_len = len(data_crypt) + len(iv)
|
||||
tmp = struct.pack("!I", crypt_data_len) # does it want the length of the encrypted response??
|
||||
resp = resp[:20] + tmp + iv + data_crypt
|
||||
self.logger.debug(f"Encrypted Response: {resp.hex()}")
|
||||
|
||||
return resp
|
||||
@@ -96,6 +96,7 @@ class WaccaBase:
|
||||
|
||||
def handle_housing_start_request(self, data: Dict) -> Dict:
|
||||
req = HousingStartRequestV1(data)
|
||||
allnet_region_id = None
|
||||
|
||||
machine = self.data.arcade.get_machine(req.chipId)
|
||||
if machine is not None:
|
||||
@@ -235,6 +236,11 @@ class WaccaBase:
|
||||
|
||||
if profileId is None:
|
||||
return BaseResponse().make()
|
||||
|
||||
if profileId == 0:
|
||||
# We've already made this profile, just return success
|
||||
new_user = self.data.profile.get_profile(aime_id=req.aimeId)
|
||||
profileId = new_user['id']
|
||||
|
||||
# Insert starting items
|
||||
self.data.item.put_item(req.aimeId, WaccaConstants.ITEM_TYPES["title"], 104001)
|
||||
@@ -255,25 +261,11 @@ class WaccaBase:
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["note_sound"], 105001
|
||||
)
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["note_sound"], 205005
|
||||
) # Added lily
|
||||
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["navigator"], 210001
|
||||
)
|
||||
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["user_plate"], 211001
|
||||
) # Added lily
|
||||
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["touch_effect"], 312000
|
||||
) # Added reverse
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["touch_effect"], 312001
|
||||
) # Added reverse
|
||||
|
||||
return UserStatusCreateResponseV2(profileId, req.username).make()
|
||||
|
||||
def handle_user_status_getDetail_request(self, data: Dict) -> Dict:
|
||||
|
||||
@@ -32,6 +32,8 @@ class HousingStartRequestV1(BaseRequest):
|
||||
class HousingStartRequestV2(HousingStartRequestV1):
|
||||
def __init__(self, data: Dict) -> None:
|
||||
super(HousingStartRequestV1, self).__init__(data)
|
||||
self.unknown0: str = self.params[0]
|
||||
self.errorLog: str = self.params[1]
|
||||
self.creditLog: str = self.params[2]
|
||||
self.info: List[HousingInfo] = []
|
||||
|
||||
|
||||
@@ -7,6 +7,8 @@ from hashlib import md5
|
||||
from twisted.web.http import Request
|
||||
from typing import Dict, Tuple, List
|
||||
from os import path
|
||||
import traceback
|
||||
import sys
|
||||
|
||||
from core import CoreConfig, Utils
|
||||
from .config import WaccaConfig
|
||||
@@ -17,7 +19,7 @@ from .lilyr import WaccaLilyR
|
||||
from .lily import WaccaLily
|
||||
from .s import WaccaS
|
||||
from .base import WaccaBase
|
||||
from .handlers.base import BaseResponse
|
||||
from .handlers.base import BaseResponse, BaseRequest
|
||||
from .handlers.helpers import Version
|
||||
|
||||
|
||||
@@ -114,10 +116,20 @@ class WaccaServlet:
|
||||
try:
|
||||
req_json = json.loads(request.content.getvalue())
|
||||
version_full = Version(req_json["appVersion"])
|
||||
req = BaseRequest(req_json)
|
||||
|
||||
except Exception:
|
||||
except KeyError as e:
|
||||
self.logger.error(
|
||||
f"Failed to parse request to {url_path} -> {request.content.getvalue()}"
|
||||
f"Failed to parse request to {request.content.getvalue()} -> Missing required value {e}"
|
||||
)
|
||||
resp = BaseResponse()
|
||||
resp.status = 1
|
||||
resp.message = "不正なリクエスト エラーです"
|
||||
return end(resp.make())
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(
|
||||
f"Failed to parse request to {url_path} -> {request.content.getvalue()} -> {e}"
|
||||
)
|
||||
resp = BaseResponse()
|
||||
resp.status = 1
|
||||
@@ -143,7 +155,7 @@ class WaccaServlet:
|
||||
|
||||
else:
|
||||
self.logger.warning(
|
||||
f"Unsupported version ({req_json['appVersion']}) request {url_path} - {req_json}"
|
||||
f"Unsupported version ({req.appVersion}) request {url_path} - {req_json}"
|
||||
)
|
||||
resp = BaseResponse()
|
||||
resp.status = 1
|
||||
@@ -151,7 +163,7 @@ class WaccaServlet:
|
||||
return end(resp.make())
|
||||
|
||||
self.logger.info(
|
||||
f"v{req_json['appVersion']} {url_path} request from {client_ip} with chipId {req_json['chipId']}"
|
||||
f"v{req.appVersion} {url_path} request from {client_ip} with chipId {req.chipId}"
|
||||
)
|
||||
self.logger.debug(req_json)
|
||||
|
||||
@@ -166,17 +178,20 @@ class WaccaServlet:
|
||||
handler = getattr(self.versions[internal_ver], func_to_find)
|
||||
resp = handler(req_json)
|
||||
|
||||
self.logger.debug(f"{req_json['appVersion']} response {resp}")
|
||||
self.logger.debug(f"{req.appVersion} response {resp}")
|
||||
return end(resp)
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(
|
||||
f"{req_json['appVersion']} Error handling method {url_path} -> {e}"
|
||||
f"{req.appVersion} Error handling method {url_path} -> {e}"
|
||||
)
|
||||
if self.core_cfg.server.is_develop:
|
||||
raise
|
||||
if self.logger.level == logging.DEBUG:
|
||||
tp, val, tb = sys.exc_info()
|
||||
traceback.print_exception(tp, val, tb, limit=3)
|
||||
with open("{0}/{1}.log".format(self.core_cfg.server.log_dir, "wacca"), "a") as f:
|
||||
traceback.print_exception(tp, val, tb, limit=3, file=f)
|
||||
|
||||
resp = BaseResponse()
|
||||
resp.status = 1
|
||||
resp.message = "A server error occoured."
|
||||
return end(resp)
|
||||
return end(resp.make())
|
||||
|
||||
@@ -41,28 +41,28 @@ class WaccaLily(WaccaS):
|
||||
resp = GetNewsResponseV3()
|
||||
return resp.make()
|
||||
|
||||
def handle_housing_start_request(self, data: Dict) -> Dict:
|
||||
req = HousingStartRequestV2(data)
|
||||
|
||||
if req.appVersion.country != "JPN" and req.appVersion.country in [
|
||||
region.name for region in WaccaConstants.Region
|
||||
]:
|
||||
region_id = WaccaConstants.Region[req.appVersion.country]
|
||||
else:
|
||||
region_id = self.region_id
|
||||
|
||||
resp = HousingStartResponseV1(region_id)
|
||||
return resp.make()
|
||||
|
||||
def handle_user_status_create_request(self, data: Dict) -> Dict:
|
||||
req = UserStatusCreateRequest(data)
|
||||
resp = super().handle_user_status_create_request(data)
|
||||
ret = super().handle_user_status_create_request(data)
|
||||
|
||||
new_user = self.data.profile.get_profile(aime_id=req.aimeId)
|
||||
|
||||
if new_user is None:
|
||||
return BaseResponse().make()
|
||||
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["user_plate"], 211001
|
||||
) # Added lily
|
||||
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["note_sound"], 205005
|
||||
) # Added lily
|
||||
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["navigator"], 210002
|
||||
) # Lily, Added Lily
|
||||
|
||||
return resp
|
||||
return ret
|
||||
|
||||
def handle_user_status_get_request(self, data: Dict) -> Dict:
|
||||
req = UserStatusGetRequest(data)
|
||||
@@ -426,7 +426,7 @@ class WaccaLily(WaccaS):
|
||||
|
||||
except Exception:
|
||||
self.logger.error(
|
||||
f"{__name__} Failed to load item {item['item_id']} for user {profile['user']}"
|
||||
f"Failed to load item {item['item_id']} for user {profile['user']}"
|
||||
)
|
||||
|
||||
resp.seasonInfo.level = profile["xp"]
|
||||
|
||||
@@ -315,5 +315,12 @@ class WaccaReverse(WaccaLilyR):
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["navigator"], 310002
|
||||
) # Added reverse
|
||||
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["touch_effect"], 312000
|
||||
) # Added reverse
|
||||
self.data.item.put_item(
|
||||
req.aimeId, WaccaConstants.ITEM_TYPES["touch_effect"], 312001
|
||||
) # Added reverse
|
||||
|
||||
return resp
|
||||
|
||||
Reference in New Issue
Block a user