about summary refs log tree commit diff
path: root/Tests/LibMatrix.HomeserverEmulator/Controllers/Users/UserController.cs
blob: 2be38960ff6c346900b78eaef358d885fc378ab4 (plain) (blame)
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
using System.Text.Json.Nodes;
using System.Text.Json.Serialization;
using ArcaneLibs.Extensions;
using LibMatrix.EventTypes.Spec.State;
using LibMatrix.Filters;
using LibMatrix.HomeserverEmulator.Services;
using LibMatrix.Responses;
using Microsoft.AspNetCore.Mvc;

namespace LibMatrix.HomeserverEmulator.Controllers;

[ApiController]
[Route("/_matrix/client/{version}/")]
public class UserController(ILogger<UserController> logger, TokenService tokenService, UserStore userStore, RoomStore roomStore) : ControllerBase {
    [HttpGet("account/whoami")]
    public async Task<WhoAmIResponse> Login() {
        var token = tokenService.GetAccessToken(HttpContext);
        var user = await userStore.GetUserByToken(token, Random.Shared.Next(101) <= 10, tokenService.GenerateServerName(HttpContext));
        if (user is null)
            throw new MatrixException() {
                ErrorCode = "M_UNKNOWN_TOKEN",
                Error = "Invalid token."
            };
        var whoAmIResponse = new WhoAmIResponse {
            UserId = user.UserId
        };
        return whoAmIResponse;
    }

    [HttpGet("joined_rooms")]
    public async Task<object> GetJoinedRooms() {
        var token = tokenService.GetAccessToken(HttpContext);
        var user = await userStore.GetUserByToken(token, false);
        if (user is null)
            throw new MatrixException() {
                ErrorCode = "M_UNAUTHORIZED",
                Error = "Invalid token."
            };
        // return user.JoinedRooms;

        return new {
            joined_rooms = roomStore._rooms.Where(r =>
                r.State.Any(s => s.StateKey == user.UserId && s.Type == RoomMemberEventContent.EventId && (s.TypedContent as RoomMemberEventContent).Membership == "join")
            ).Select(r => r.RoomId).ToList()
        };
    }
    
    [HttpGet("devices")]
    public async Task<DevicesResponse> GetDevices() {
        var token = tokenService.GetAccessToken(HttpContext);
        var user = await userStore.GetUserByToken(token, false);
        if (user is null)
            throw new MatrixException() {
                ErrorCode = "M_UNAUTHORIZED",
                Error = "Invalid token."
            };
        return new() {
            Devices = user.AccessTokens.Select(x=>new DevicesResponse.Device() {
                DeviceId = x.Value.DeviceId,
                DisplayName = x.Value.DeviceId
            }).ToList()
        };
    }

    public class DevicesResponse {
        [JsonPropertyName("devices")]
        public List<Device> Devices { get; set; }
        
        public class Device {
            [JsonPropertyName("device_id")]
            public string DeviceId { get; set; }
            [JsonPropertyName("display_name")]
            public string DisplayName { get; set; }
            [JsonPropertyName("last_seen_ip")]
            public string LastSeenIp { get; set; }
            [JsonPropertyName("last_seen_ts")]
            public long LastSeenTs { get; set; }
        }
    }
}