131 lines
3.4 KiB
C#
131 lines
3.4 KiB
C#
using System.Text.Json;
|
|
using RelayServer.Services;
|
|
using WebSocketSharp.Server;
|
|
using RelayServer.Models;
|
|
|
|
var surrealService = new SurrealService();
|
|
var coreClient = new CoreClientService();
|
|
var cryptoService = new ChannelCryptoService();
|
|
|
|
await using var db = await surrealService.ConnectAsync();
|
|
|
|
ChatTest.ClientKeyService = new ClientKeyService(db);
|
|
ChatTest.Db = db;
|
|
|
|
var wssv = new WebSocketServer("ws://localhost:1337");
|
|
wssv.AddWebSocketService<ChatTest>("/");
|
|
wssv.Start();
|
|
Console.WriteLine("WebSocket server started");
|
|
|
|
var keeper = await coreClient.GetUserByUsernameAsync("Keeper317");
|
|
var kira = await coreClient.GetUserByUsernameAsync("Ru_Kira");
|
|
var test = await coreClient.GetUserByUsernameAsync("Test");
|
|
|
|
if (keeper is null || kira is null || test is null)
|
|
{
|
|
Console.WriteLine("One or more required users do not exist in RelayCore.");
|
|
return;
|
|
}
|
|
|
|
if (!keeper.Licensed || !kira.Licensed || !test.Licensed)
|
|
{
|
|
Console.WriteLine("One or more required users are not licensed.");
|
|
return;
|
|
}
|
|
|
|
Console.WriteLine($"Core verified user: {keeper.Username}");
|
|
Console.WriteLine($"Core verified user: {kira.Username}");
|
|
Console.WriteLine($"Core verified user: {test.Username}");
|
|
|
|
var server = await db.Create("servers", new Servers
|
|
{
|
|
Name = "Test Server",
|
|
OwnerUserId = keeper.Id,
|
|
CreatedAt = DateTime.UtcNow
|
|
});
|
|
|
|
Console.WriteLine($"Server created: {ToJsonString(server)}");
|
|
|
|
var keeperMember = await db.Create("server_members", new ServerMembers
|
|
{
|
|
UserId = keeper.Id,
|
|
JoinedAt = DateTime.UtcNow,
|
|
IsOwner = true
|
|
});
|
|
|
|
var kiraMember = await db.Create("server_members", new ServerMembers
|
|
{
|
|
UserId = kira.Id,
|
|
JoinedAt = DateTime.UtcNow,
|
|
IsOwner = false
|
|
});
|
|
|
|
var testMember = await db.Create("server_members", new ServerMembers
|
|
{
|
|
UserId = test.Id,
|
|
JoinedAt = DateTime.UtcNow,
|
|
IsOwner = false
|
|
});
|
|
|
|
Console.WriteLine("Server members created.");
|
|
|
|
var channel = await db.Create("channels", new Channels
|
|
{
|
|
Name = "general",
|
|
CreatedAt = DateTime.UtcNow
|
|
});
|
|
|
|
Console.WriteLine($"Channel created: {ToJsonString(channel)}");
|
|
|
|
var channelId = GetRecordId(channel.Id);
|
|
Console.WriteLine($"Resolved channelId: {channelId}");
|
|
ChatTest.DefaultChannelId = channelId;
|
|
|
|
var keyBase64 = cryptoService.GenerateKey();
|
|
var serverKeys = E2EeHelper.GenerateRsaKeyPair();
|
|
|
|
var serverKey = await db.Create("server_encryption_keys", new ServerEncryptionKeys
|
|
{
|
|
KeyBase64 = keyBase64,
|
|
PublicKey = serverKeys.publicKey,
|
|
PrivateKey = serverKeys.privateKey,
|
|
CreatedAt = DateTime.UtcNow,
|
|
UpdatedAt = DateTime.UtcNow
|
|
});
|
|
|
|
ChatTest.ServerPublicKey = serverKeys.publicKey;
|
|
ChatTest.ServerPrivateKey = serverKeys.privateKey;
|
|
ChatTest.ChannelDbKey = keyBase64;
|
|
|
|
Console.WriteLine("Server encryption key created.");
|
|
|
|
Console.ReadKey(true);
|
|
wssv.Stop();
|
|
|
|
return;
|
|
|
|
static string ToJsonString(object? obj)
|
|
{
|
|
return JsonSerializer.Serialize(obj, new JsonSerializerOptions
|
|
{
|
|
WriteIndented = true,
|
|
Encoder = System.Text.Encodings.Web.JavaScriptEncoder.UnsafeRelaxedJsonEscaping
|
|
});
|
|
}
|
|
|
|
static string GetRecordId(object? id)
|
|
{
|
|
if (id is null)
|
|
return string.Empty;
|
|
|
|
var json = JsonSerializer.Serialize(id);
|
|
|
|
using var doc = JsonDocument.Parse(json);
|
|
|
|
var root = doc.RootElement;
|
|
|
|
var recordId = root.GetProperty("Id").GetString() ?? string.Empty;
|
|
var table = root.GetProperty("Table").GetString() ?? string.Empty;
|
|
|
|
return $"{table}:{recordId}";
|
|
} |