From add07cb84c14beb26c46026b3dc295e55aeb98f8 Mon Sep 17 00:00:00 2001
From: Rainnny7 <braydonrainnny@gmail.com>
Date: Sun, 7 Apr 2024 00:11:22 -0400
Subject: [PATCH] Make the skin part route dynamic

---
 .../mc/controller/PlayerController.java       | 40 +++++++++++--------
 1 file changed, 23 insertions(+), 17 deletions(-)

diff --git a/src/main/java/me/braydon/mc/controller/PlayerController.java b/src/main/java/me/braydon/mc/controller/PlayerController.java
index 056487a..a02af13 100644
--- a/src/main/java/me/braydon/mc/controller/PlayerController.java
+++ b/src/main/java/me/braydon/mc/controller/PlayerController.java
@@ -2,6 +2,7 @@ package me.braydon.mc.controller;
 
 import lombok.NonNull;
 import lombok.extern.log4j.Log4j2;
+import me.braydon.mc.common.EnumUtils;
 import me.braydon.mc.common.PlayerUtils;
 import me.braydon.mc.exception.impl.BadRequestException;
 import me.braydon.mc.exception.impl.ResourceNotFoundException;
@@ -24,8 +25,8 @@ import org.springframework.web.bind.annotation.*;
 @RequestMapping(value = "/player", produces = MediaType.APPLICATION_JSON_VALUE)
 @Log4j2(topic = "Player Controller")
 public final class PlayerController {
-    private static final int DEFAULT_HEAD_SIZE = 128;
-    private static final int MAX_HEAD_SIZE = 512;
+    private static final int DEFAULT_PART_TEXTURE_SIZE = 128;
+    private static final int MAX_PART_TEXTURE_SIZE = 512;
 
     /**
      * The Mojang service to use for player information.
@@ -42,7 +43,7 @@ public final class PlayerController {
      *
      * @param query the query to search for the player by
      * @return the player response
-     * @throws BadRequestException if the UUID is malformed
+     * @throws BadRequestException       if the UUID is malformed
      * @throws ResourceNotFoundException if the player is not found
      */
     @GetMapping("/{query}")
@@ -52,34 +53,39 @@ public final class PlayerController {
     }
 
     /**
-     * Get the head texture of a player
-     * by their username or UUID.
+     * Get the part of a skin texture for
+     * a player by their username or UUID.
      * <p>
      * If the player being searched is
      * invalid, the default Steve skin
      * will be used.
      * </p>
      *
-     * @param query the query to search for the player by
-     * @param extension the head image extension
-     * @param size the size of the head image
-     * @return the head texture
+     * @param partName  the part of the player's skin texture to get
+     * @param query     the query to search for the player by
+     * @param extension the skin part image extension
+     * @param size      the size of the skin part image
+     * @return the skin part texture
      * @throws BadRequestException if the extension is invalid
      */
-    @GetMapping("/head/{query}.{extension}")
+    @GetMapping("/{partName}/{query}.{extension}")
     @ResponseBody
-    public ResponseEntity<byte[]> getHead(@PathVariable @NonNull String query, @PathVariable @NonNull String extension,
-                                          @RequestParam(required = false) Integer size
+    public ResponseEntity<byte[]> getPartTexture(@PathVariable @NonNull String partName, @PathVariable @NonNull String query,
+                                                 @PathVariable @NonNull String extension, @RequestParam(required = false) Integer size
     ) throws BadRequestException {
+        Skin.Part part = EnumUtils.getEnumConstant(Skin.Part.class, partName.toUpperCase()); // The skin part to get
+        if (part == null) { // Default to the head part
+            part = Skin.Part.HEAD;
+        }
         if ((extension = extension.trim()).isBlank()) { // Invalid extension
             throw new BadRequestException("Invalid extension");
         }
         if (size == null || size <= 0) { // Invalid size
-            size = DEFAULT_HEAD_SIZE;
+            size = DEFAULT_PART_TEXTURE_SIZE;
         }
-        size = Math.min(size, MAX_HEAD_SIZE); // Limit the size to 512
+        size = Math.min(size, MAX_PART_TEXTURE_SIZE); // Limit the size to 512
 
-        Skin target = null; // The target skin to get the head of
+        Skin target = null; // The target skin to get the skin part of
         try {
             CachedPlayer player = mojangService.getPlayer(query, false); // Retrieve the player
             target = player.getSkin(); // Use the player's skin
@@ -89,9 +95,9 @@ public final class PlayerController {
         if (target == null) { // Fallback to the default skin
             target = Skin.DEFAULT_STEVE;
         }
-        // Return the head texture
+        // Return the skin part texture
         return ResponseEntity.ok()
                 .contentType(extension.equalsIgnoreCase("png") ? MediaType.IMAGE_PNG : MediaType.IMAGE_JPEG)
-                .body(PlayerUtils.getSkinPart(target, Skin.Part.HEAD, size));
+                .body(PlayerUtils.getSkinPart(target, part, size));
     }
 }
\ No newline at end of file