Bug fixed: All the gifts had before same image link

Client/pom.xml

@@ -5,7 +5,7 @@
     <parent>
         <artifactId>TikTokLiveJava</artifactId>
         <groupId>io.github.jwdeveloper.tiktok</groupId>
-        <version>0.0.25-Release</version>
+        <version>1.0.0-Release</version>
     </parent>
     <modelVersion>4.0.0</modelVersion>
 

Client/src/main/java/io/github/jwdeveloper/tiktok/handlers/events/TikTokGiftEventHandler.java

@@ -28,9 +28,11 @@ import io.github.jwdeveloper.tiktok.data.events.gift.TikTokGiftEvent;
 import io.github.jwdeveloper.tiktok.data.models.Picture;
 import io.github.jwdeveloper.tiktok.data.models.gifts.Gift;
 import io.github.jwdeveloper.tiktok.data.models.gifts.GiftSendType;
+import io.github.jwdeveloper.tiktok.exceptions.TikTokLiveException;
 import io.github.jwdeveloper.tiktok.live.GiftManager;
 import io.github.jwdeveloper.tiktok.messages.webcast.WebcastGiftMessage;
 import lombok.SneakyThrows;
+import sun.misc.Unsafe;
 
 import java.util.HashMap;
 import java.util.List;
@@ -45,9 +47,8 @@ public class TikTokGiftEventHandler {
         giftsMessages = new HashMap<>();
     }
 
-    @SneakyThrows
-    public List<TikTokEvent> handleGift(byte[] msg) {
-        var currentMessage = WebcastGiftMessage.parseFrom(msg);
+    public List<TikTokEvent> handleGift(WebcastGiftMessage currentMessage)
+    {
         var userId = currentMessage.getUser().getId();
         var currentType = GiftSendType.fromNumber(currentMessage.getSendType());
         var containsPreviousMessage = giftsMessages.containsKey(userId);
@@ -80,6 +81,13 @@ public class TikTokGiftEventHandler {
 
         return List.of();
     }
+    @SneakyThrows
+    public List<TikTokEvent> handleGift(byte[] msg) {
+        var currentMessage = WebcastGiftMessage.parseFrom(msg);
+        return handleGift(currentMessage);
+    }
+
+
 
     private TikTokGiftEvent getGiftEvent(WebcastGiftMessage message) {
         var gift = getGiftObject(message);
@@ -103,6 +111,24 @@ public class TikTokGiftEventHandler {
                     giftMessage.getGift().getDiamondCount(),
                     Picture.map(giftMessage.getGift().getImage()));
         }
+
+        if (gift.getPicture().getLink().endsWith(".webp")) {
+            updatePicture(gift, giftMessage);
+        }
         return gift;
     }
+
+
+    private void updatePicture(Gift gift, WebcastGiftMessage webcastGiftMessage) {
+        try {
+            var picture = Picture.map(webcastGiftMessage.getGift().getImage());
+            var constructor = Unsafe.class.getDeclaredConstructors()[0];
+            constructor.setAccessible(true);
+            var field = Gift.class.getDeclaredField("picture");
+            field.setAccessible(true);
+            field.set(gift, picture);
+        } catch (Exception e) {
+            throw new TikTokLiveException("Unable to update picture in gift: " + gift.toString());
+        }
+    }
 }

Client/src/test/java/io/github/jwdeveloper/tiktok/handlers/events/TikTokGiftEventHandlerTest.java

@@ -0,0 +1,65 @@
+package io.github.jwdeveloper.tiktok.handlers.events;
+
+import io.github.jwdeveloper.tiktok.data.events.gift.TikTokGiftEvent;
+import io.github.jwdeveloper.tiktok.data.models.Picture;
+import io.github.jwdeveloper.tiktok.gifts.TikTokGiftManager;
+import io.github.jwdeveloper.tiktok.messages.data.GiftStruct;
+import io.github.jwdeveloper.tiktok.messages.data.Image;
+import io.github.jwdeveloper.tiktok.messages.data.User;
+import io.github.jwdeveloper.tiktok.messages.webcast.WebcastGiftMessage;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.TestInstance;
+
+
+@TestInstance(TestInstance.Lifecycle.PER_CLASS)
+class TikTokGiftEventHandlerTest {
+
+    public static TikTokGiftEventHandler handler;
+
+
+    @BeforeAll
+    public void before() {
+        var manager = new TikTokGiftManager();
+        manager.registerGift(123, "example", 123, new Picture("image.webp"));
+        handler = new TikTokGiftEventHandler(manager);
+    }
+
+    @Test
+    void shouldHandleGifts() {
+        var message = getGiftMessage("example-new-name", 123, "image-new.png", 0, 1);
+        var result = handler.handleGift(message);
+
+        Assertions.assertEquals(1, result.size());
+
+        var event = (TikTokGiftEvent) result.get(0);
+        var gift = event.getGift();
+        Assertions.assertEquals("image-new.png",gift.getPicture().getLink());
+        Assertions.assertEquals(123,gift.getId());
+    }
+
+
+    public WebcastGiftMessage getGiftMessage(String giftName,
+                                             int giftId,
+                                             String giftImage,
+                                             int sendType,
+                                             int userId) {
+        var builder = WebcastGiftMessage.newBuilder();
+        var giftBuilder = GiftStruct.newBuilder();
+        var userBuilder = User.newBuilder();
+
+
+        giftBuilder.setId(giftId);
+        giftBuilder.setName(giftName);
+        giftBuilder.setImage(Image.newBuilder().addUrlList(giftImage).build());
+        userBuilder.setId(userId);
+
+        builder.setGiftId(giftId);
+        builder.setUser(userBuilder);
+        builder.setSendType(sendType);
+        builder.setGift(giftBuilder);
+        return builder.build();
+    }
+
+}