From 521f38468b641e75ab747574ced87a0705765891 Mon Sep 17 00:00:00 2001
From: Steven Tracey <steven@nevets.tech>
Date: Thu, 11 Jan 2024 06:32:18 -0600
Subject: [PATCH] Fix issue #1

---
 .idea/gradle.xml                                            | 1 +
 .../java/tech/nevets/dliteserver/MessageDistributer.java    | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/.idea/gradle.xml b/.idea/gradle.xml
index 14746e7..2a65317 100644
--- a/.idea/gradle.xml
+++ b/.idea/gradle.xml
@@ -1,5 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
+  <component name="GradleMigrationSettings" migrationVersion="1" />
   <component name="GradleSettings">
     <option name="linkedExternalProjectsSettings">
       <GradleProjectSettings>
diff --git a/src/main/java/tech/nevets/dliteserver/MessageDistributer.java b/src/main/java/tech/nevets/dliteserver/MessageDistributer.java
index 3786bb9..c07e6da 100644
--- a/src/main/java/tech/nevets/dliteserver/MessageDistributer.java
+++ b/src/main/java/tech/nevets/dliteserver/MessageDistributer.java
@@ -5,6 +5,7 @@ import java.io.OutputStream;
 import java.nio.charset.StandardCharsets;
 import java.util.HashMap;
 import java.util.Map;
+import java.util.Set;
 
 public class MessageDistributer extends OutputStream {
     private static Map<String, OutputStream> userStreams = new HashMap<>();
@@ -30,6 +31,11 @@ public class MessageDistributer extends OutputStream {
     }
 
     public void registerStream(String username, OutputStream os) {
+        Set<String> usernames = userStreams.keySet();
+        while (usernames.contains(username)) {
+            username += ".imposter";
+        }
+
         userStreams.put(username, os);
         try {
             this.write(("SERVER> Welcome, " + username + "!\n").getBytes(StandardCharsets.UTF_8));