From 087905492601a55ff53541ef469526051c30aa0b Mon Sep 17 00:00:00 2001
From: Runxi Yu <me@runxiyu.org>
Date: Sun, 8 Dec 2024 10:40:08 +0800
Subject: Proper UID handling

---
 clients.go | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

(limited to 'clients.go')

diff --git a/clients.go b/clients.go
index 15512c0..a1e0641 100644
--- a/clients.go
+++ b/clients.go
@@ -1,7 +1,9 @@
 package main
 
 import (
+	"crypto/rand"
 	"log/slog"
+	"math/big"
 	"net"
 	"sync"
 )
@@ -58,6 +60,32 @@ func (client *Client) Teardown() {
 	}
 }
 
+func NewLocalClient(conn *net.Conn) (*Client, error) {
+	client := &Client{
+		conn:   conn,
+		Server: self,
+		State:  ClientStatePreRegistration,
+		Nick:   "*",
+	}
+	for _ = range 10 {
+		var uid_ = []byte(self.SID)
+		for _ = range 6 {
+			randint, err := rand.Int(rand.Reader, big.NewInt(26))
+			if err != nil {
+				return nil, err
+			}
+			uid_ = append(uid_, byte(65 + randint.Uint64()))
+		}
+		uid := string(uid_)
+		_, exists := uidToClient.LoadOrStore(uid, client)
+		if !exists {
+			client.UID = uid
+			return client, nil
+		}
+	}
+	return nil, ErrUIDBusy
+}
+
 func (client *Client) checkRegistration() {
 	if client.State != ClientStatePreRegistration {
 		slog.Error("spurious call to checkRegistration", "client", client)
-- 
cgit v1.2.3