forked from Hay1tsme/artemis
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
from operator import indexOf
|
|
import struct
|
|
import json
|
|
from random import choice, randrange
|
|
|
|
from .base import IDZHandlerBase
|
|
from core.config import CoreConfig
|
|
from ..config import IDZConfig
|
|
|
|
AUTO_TEAM_NAMES = ["スピードスターズ", "レッドサンズ", "ナイトキッズ"]
|
|
FULL_WIDTH_NUMS = ["\uff10", "\uff11", "\uff12", "\uff13", "\uff14", "\uff15", "\uff16", "\uff17", "\uff18", "\uff19"]
|
|
|
|
class IDZHandlerCreateAutoTeam(IDZHandlerBase):
|
|
cmd_codes = [0x007b, 0x007b, 0x0077, 0x0077]
|
|
rsp_codes = [0x007c, 0x007c, 0x0078, 0x0078]
|
|
name = "create_auto_team"
|
|
|
|
def __init__(self, core_cfg: CoreConfig, game_cfg: IDZConfig, version: int) -> None:
|
|
super().__init__(core_cfg, game_cfg, version)
|
|
self.size = 0x0ca0
|
|
|
|
def handle(self, data: bytes) -> bytearray:
|
|
ret = super().handle(data)
|
|
aime_id = struct.unpack_from("<I", data, 0x04)[0]
|
|
name = choice(AUTO_TEAM_NAMES)
|
|
bg = indexOf(AUTO_TEAM_NAMES, name)
|
|
number = choice(FULL_WIDTH_NUMS) + choice(FULL_WIDTH_NUMS) + choice(FULL_WIDTH_NUMS)
|
|
|
|
tdata = {
|
|
"id": aime_id,
|
|
"bg": bg,
|
|
"fx": 0,
|
|
}
|
|
|
|
tdata = {
|
|
"id": aime_id,
|
|
"name": (name + number),
|
|
"bg": bg,
|
|
"fx": 0,
|
|
}
|
|
tname = tdata['name'].encode("shift-jis")
|
|
|
|
struct.pack_into("<I", ret, 0x0C, tdata["id"])
|
|
struct.pack_into(f"{len(tname)}s", ret, 0x24, tname)
|
|
struct.pack_into("<I", ret, 0x80, tdata["id"])
|
|
struct.pack_into(f"<I", ret, 0xD8, tdata["bg"])
|
|
struct.pack_into(f"<I", ret, 0xDC, tdata["fx"])
|
|
|
|
return ret
|