From e49c70b73806d0908706551780e2a7b5c5864df0 Mon Sep 17 00:00:00 2001
From: daydensteve <daydensteve@gmail.com>
Date: Sun, 3 Nov 2024 16:37:27 -0500
Subject: [PATCH] more enums!

---
 titles/chuni/base.py  |  6 +++---
 titles/chuni/const.py | 19 +++++++++++++++++++
 2 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/titles/chuni/base.py b/titles/chuni/base.py
index 0e0adac..2410ef0 100644
--- a/titles/chuni/base.py
+++ b/titles/chuni/base.py
@@ -7,7 +7,7 @@ import pytz
 from typing import Dict, Any, List
 
 from core.config import CoreConfig
-from titles.chuni.const import ChuniConstants
+from titles.chuni.const import ChuniConstants, ItemKind
 from titles.chuni.database import ChuniData
 from titles.chuni.config import ChuniConfig
 SCORE_BUFFER = {}
@@ -43,7 +43,7 @@ class ChuniBase:
                     user_id,
                     {
                         "itemId": ticket.strip(),
-                        "itemKind": 5,
+                        "itemKind": ItemKind.TICKET.value,
                         "stock": self.game_cfg.mods.stock_count,
                         "isValid": True,
                     },
@@ -116,7 +116,7 @@ class ChuniBase:
                         user_id,
                         {
                             "itemId": login_item["presentId"],
-                            "itemKind": 6,
+                            "itemKind": ItemKind.PRESENT.value,
                             "stock": login_item["itemNum"],
                             "isValid": True,
                         },
diff --git a/titles/chuni/const.py b/titles/chuni/const.py
index 003c618..68d7056 100644
--- a/titles/chuni/const.py
+++ b/titles/chuni/const.py
@@ -91,3 +91,22 @@ class MapAreaConditionType(Enum):
 class MapAreaConditionLogicalOperator(Enum):
     AND = 1
     OR = 2
+
+
+class AvatarCategory(Enum):
+    WEAR = 1
+    HEAD = 2
+    FACE = 3
+    SKIN = 4
+    ITEM = 5
+    FRONT = 6
+    BACK = 7
+
+class ItemKind(Enum):
+    NAMEPLATE = 1
+    TROPHY = 3
+    TICKET = 5
+    PRESENT = 6
+    MAP_ICON = 8
+    SYSTEM_VOICE = 9
+    AVATAR_ACCESSORY = 11
\ No newline at end of file