feat: Add query command

1 files changed, 44 insertions, 0 deletions

diff --git a/src/main/kotlin/moe/nea/ledger/ItemChange.kt b/src/main/kotlin/moe/nea/ledger/ItemChange.kt
index a8bb7e8..834cd2b 100644
--- a/src/main/kotlin/moe/nea/ledger/ItemChange.kt
+++ b/src/main/kotlin/moe/nea/ledger/ItemChange.kt
@@ -1,16 +1,52 @@
 package moe.nea.ledger
 
+import moe.nea.ledger.database.DBItemEntry
+import moe.nea.ledger.database.ResultRow
+import net.minecraft.event.HoverEvent
+import net.minecraft.util.ChatComponentText
+import net.minecraft.util.ChatStyle
+import net.minecraft.util.EnumChatFormatting
+import net.minecraft.util.IChatComponent
+
 data class ItemChange(
 	val itemId: ItemId,
 	val count: Double,
 	val direction: ChangeDirection,
 ) {
+	fun formatChat(): IChatComponent {
+		return ChatComponentText(" ")
+			.appendSibling(direction.chatFormat)
+			.appendText(" ")
+			.appendSibling(ChatComponentText("$count").setChatStyle(ChatStyle().setColor(EnumChatFormatting.WHITE)))
+			.appendSibling(ChatComponentText("x").setChatStyle(ChatStyle().setColor(EnumChatFormatting.DARK_GRAY)))
+			.appendText(" ")
+			.appendSibling(ChatComponentText(itemId.string).setChatStyle(ChatStyle().setParentStyle(ChatStyle().setColor(
+				EnumChatFormatting.WHITE))))
+	}
+
 	enum class ChangeDirection {
 		GAINED,
 		TRANSFORM,
 		SYNC,
 		CATALYST,
 		LOST;
+
+		val chatFormat by lazy { formatChat0() }
+		private fun formatChat0(): IChatComponent {
+			val (text, color) = when (this) {
+				GAINED -> "+" to EnumChatFormatting.GREEN
+				TRANSFORM -> "~" to EnumChatFormatting.YELLOW
+				SYNC -> "=" to EnumChatFormatting.BLUE
+				CATALYST -> "*" to EnumChatFormatting.DARK_PURPLE
+				LOST -> "-" to EnumChatFormatting.RED
+			}
+			return ChatComponentText(text)
+				.setChatStyle(
+					ChatStyle()
+						.setColor(color)
+						.setChatHoverEvent(HoverEvent(HoverEvent.Action.SHOW_TEXT,
+						                              ChatComponentText(name).setChatStyle(ChatStyle().setColor(color)))))
+		}
 	}
 
 	companion object {
@@ -29,5 +65,13 @@ data class ItemChange(
 		fun loseCoins(number: Double): ItemChange {
 			return lose(ItemId.COINS, number)
 		}
+
+		fun from(result: ResultRow): ItemChange {
+			return ItemChange(
+				result[DBItemEntry.itemId],
+				result[DBItemEntry.size],
+				result[DBItemEntry.mode],
+			)
+		}
 	}
 }
\ No newline at end of file