sao: stub get_shop_resource_sales_data_list

titles/sao/base.py

@@ -832,3 +832,10 @@ class SaoBase:
         #card/scan_qr_quest_profile_card
         resp = SaoScanQrQuestProfileCardResponse(header.cmd +1)
         return resp.make()
+    
+    def handle_c700(self, header: SaoRequestHeader, request: bytes) -> bytes:
+        # shop/get_shop_resource_sales_data_list
+        # TODO: Shop data
+        req = GetShopResourceSalesDataListRequest(header, request)
+        resp = GetShopResourceSalesDataListResponse(header.cmd + 1)
+        return resp.make()

titles/sao/handlers/base.py

@@ -3095,3 +3095,24 @@ class TrialTowerPlayEndUnanalyzedLogFixed(SaoBaseRequest):
 
         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

titles/sao/handlers/helpers.py

@@ -1,6 +1,7 @@
-from typing import Tuple, List
+from typing import Tuple, List, Optional
 import struct
 import logging
+from datetime import datetime
 
 BIGINT_OFF = 16
 LONG_OFF = 8
@@ -8,6 +9,18 @@ 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')
@@ -40,6 +53,31 @@ def decode_str(data: bytes, offset: int) -> Tuple[str, int]:
         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)
 
@@ -64,13 +102,65 @@ def encode_str(s: str) -> bytes:
         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)
@@ -413,3 +503,111 @@ class MultiPlayEndRequestData(BaseHelper):
         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: int = "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
+