summaryrefslogtreecommitdiff
path: root/NetClient.cs
blob: da287df5b7df4ec65017a910fcead6eb33124026 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
using System.Collections.ObjectModel;
using LiteNetLib;
using LiteNetLib.Utils;

namespace OnekoOnline.Net;

class Client
{
    public static readonly string UserName = OnekoOnline.Config.GetValue("UserName", "Oneko");
    public int Id {get; private set;} = -1;
    public bool Connected => NetClient?.ConnectedPeersCount > 0 && Id != -1;

    readonly EventBasedNetListener Listener;
    readonly NetManager NetClient;
    public NetPeer ConnectedServer => NetClient.FirstPeer;

    readonly Dictionary<int, User> users = [];
    public ReadOnlyDictionary<int, User> Users => users.AsReadOnly();

    static public Action<User>? UserConnected;
    static public Action<User>? UserDisconnected;
    static public Action<NetDataReader, User, PacketType>? PacketRecived;
    static public Action? ServerDisconnected;

    public Client(string ServerAddress, int port, string ServerPassword)
    {
        if (ServerAddress == "") throw new Exception("Server Address invalid!");

        Listener = new();
        NetClient = new(Listener);

        NetClient.Start();
        NetClient.Connect(ServerAddress, port, ServerPassword);

        Listener.PeerConnectedEvent += peer => {
            Console.WriteLine("Connected to the Server!");
            NetDataWriter writer = new();
            
            writer.Put(new PacketInfo(PacketType.UserInfo, Id));
            writer.Put(UserName);
            writer.PutBytesWithLength(OnekoLocal.Instance!.SpriteSheet.Serialize());
            peer.Send(writer, DeliveryMethod.ReliableUnordered);
        };

        Listener.PeerDisconnectedEvent += (peer, info) => {
            ServerDisconnected?.Invoke();
            users.Clear();
            Console.WriteLine("Server Disconnected! You're offline!");
        };

        Listener.NetworkReceiveEvent += (fromPeer, reader, channel, DeliveryMethod) => {
            if (reader.AvailableBytes < PacketInfo.SizeOf) return;
            PacketInfo info = reader.GetPacketInfo();

            if (info.Type == PacketType.UserId) {
                Id = reader.GetInt();
                return;
            }

            User? from; 
            if (!users.TryGetValue(info.FromId, out from)) {
                from = new(info.FromId);
                users.Add(from.Id, from);
            }

            if (info.Type == PacketType.Disconnect) {
                //Disconnect user that has disconnected
                if (from.Username != null) Console.WriteLine($"User {from.Username} left.");
                users.Remove(info.FromId);
                UserDisconnected?.Invoke(from);
                return;
            }

            else if (info.Type == PacketType.UserInfo) {
                from.Username = reader.GetString();
                from.SpriteSheet = reader.GetBytesWithLength();
                Console.WriteLine($"User {from.Username} joined!");
                from.Initialized = true;
                UserConnected?.Invoke(from);
                return;
            }

            NetDataReader newReader = new(reader.GetRemainingBytes());
            PacketRecived?.Invoke(newReader, from, info.Type);
        };
    }

    public void Poll()
    {
        NetClient.PollEvents();
    }

    public void Disconnect()
    {
        NetClient.DisconnectAll();
    }
}