From 5b94cfc93db81d1f0388081feabc0addec580ae1 Mon Sep 17 00:00:00 2001
From: Azalea Gui <me@hydev.org>
Date: Tue, 7 Mar 2023 23:32:59 -0500
Subject: [PATCH] [+] Update snow particles

---
 tngame/telnet.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/tngame/telnet.py b/tngame/telnet.py
index ac6d521..00b3cc7 100644
--- a/tngame/telnet.py
+++ b/tngame/telnet.py
@@ -75,6 +75,33 @@ async def shell(reader: TelnetReaderUnicode, writer: TelnetWriterUnicode):
             snow.append(SnowParticle(x, y, xv, yv, color))
 
         return snow
+
+    # Update snow particles
+    def update_snow():
+        nonlocal snow
+        # Erase snow particles
+        for particle in snow:
+            # Erase snow particle
+            writer.write(f'\x1b[{particle.y};{particle.x}H\x1b[0m ')
+
+        # Update snow particles
+        for particle in snow:
+            # Update position
+            particle.x += particle.xv
+            particle.y += particle.yv
+
+            # Wrap around the screen
+            if particle.x >= width:
+                particle.x = 0
+            elif particle.x < 0:
+                particle.x = width - 1
+
+            if particle.y >= height:
+                particle.y = 0
+
+            # Draw snow particle
+            writer.write(f'\x1b[{particle.y};{particle.x}H\x1b[38;2;{particle.color.r};{particle.color.g};{particle.color.b}m*')
+
     # Get the size of the terminal
     async def get_size() -> tuple[int, int]:
         # Get the size of the terminal
@@ -109,6 +136,9 @@ async def shell(reader: TelnetReaderUnicode, writer: TelnetWriterUnicode):
     # Position the cat at center bottom by default
     x, y = (width - ASCII_WIDTH) // 2, height - ASCII_HEIGHT
 
+    # Create snow particles
+    snow = create_snow(100)
+
     # Draw the cat function
     def draw_cat():
         log.info(f"Drawing cat at {x}, {y}")
@@ -126,6 +156,8 @@ async def shell(reader: TelnetReaderUnicode, writer: TelnetWriterUnicode):
     async def update():
         # Move the cat
         draw_cat()
+        # Update snow
+        update_snow()
         # Move the cursor to the bottom
         writer.write('\x1b[9999;9999H')
         # Flush the output