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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using System;
using System.Linq;
using System.Threading.Tasks;
using Timeline.Entities;
using Timeline.Helpers.Cache;
using Timeline.Models;
using Timeline.Services.Data;
using Timeline.Services.Imaging;
namespace Timeline.Services.User.Avatar
{
public class UserAvatarService : IUserAvatarService
{
private readonly ILogger<UserAvatarService> _logger;
private readonly DatabaseContext _database;
private readonly IBasicUserService _basicUserService;
private readonly IDefaultUserAvatarProvider _defaultUserAvatarProvider;
private readonly IImageService _imageService;
private readonly IDataManager _dataManager;
private readonly IClock _clock;
public UserAvatarService(
ILogger<UserAvatarService> logger,
DatabaseContext database,
IBasicUserService basicUserService,
IDefaultUserAvatarProvider defaultUserAvatarProvider,
IImageService imageValidator,
IDataManager dataManager,
IClock clock)
{
_logger = logger;
_database = database;
_basicUserService = basicUserService;
_defaultUserAvatarProvider = defaultUserAvatarProvider;
_imageService = imageValidator;
_dataManager = dataManager;
_clock = clock;
}
public async Task<ICacheableDataDigest> GetAvatarDigest(long userId)
{
var usernameChangeTime = await _basicUserService.GetUsernameLastModifiedTimeAsync(userId);
var entity = await _database.UserAvatars.Where(a => a.UserId == userId).Select(a => new { a.DataTag, a.LastModified }).SingleOrDefaultAsync();
if (entity is null)
{
var defaultAvatarDigest = await _defaultUserAvatarProvider.GetDefaultAvatarDigest();
return new CacheableDataDigest(defaultAvatarDigest.ETag, new DateTime[] { usernameChangeTime, defaultAvatarDigest.LastModified }.Max());
}
else if (entity.DataTag is null)
{
var defaultAvatarDigest = await _defaultUserAvatarProvider.GetDefaultAvatarDigest();
return new CacheableDataDigest(defaultAvatarDigest.ETag, new DateTime[] { usernameChangeTime, defaultAvatarDigest.LastModified, entity.LastModified }.Max());
}
else
{
return new CacheableDataDigest(entity.DataTag, new DateTime[] { usernameChangeTime, entity.LastModified }.Max());
}
}
public async Task<ByteData> GetAvatar(long userId)
{
await _basicUserService.ThrowIfUserNotExist(userId);
var entity = await _database.UserAvatars.Where(a => a.UserId == userId).SingleOrDefaultAsync();
if (entity is null || entity.DataTag is null)
{
return await _defaultUserAvatarProvider.GetDefaultAvatar();
}
var data = await _dataManager.GetEntryAndCheck(entity.DataTag, $"This is required by avatar of {userId}.");
if (entity.Type is null)
{
var format = await _imageService.DetectFormatAsync(data);
entity.Type = format.DefaultMimeType;
await _database.SaveChangesAsync();
}
return new ByteData(data, entity.Type);
}
public async Task<ICacheableDataDigest> SetAvatar(long userId, ByteData avatar)
{
if (avatar is null)
throw new ArgumentNullException(nameof(avatar));
await _imageService.ValidateAsync(avatar.Data, avatar.ContentType, true);
await _basicUserService.ThrowIfUserNotExist(userId);
var entity = await _database.UserAvatars.Where(a => a.UserId == userId).SingleOrDefaultAsync();
await using var transaction = await _database.Database.BeginTransactionAsync();
var tag = await _dataManager.RetainEntryAsync(avatar.Data);
var now = _clock.GetCurrentTime();
if (entity is null)
{
var newEntity = new UserAvatarEntity
{
DataTag = tag,
Type = avatar.ContentType,
LastModified = now,
UserId = userId
};
_database.Add(newEntity);
}
else
{
if (entity.DataTag is not null)
await _dataManager.FreeEntryAsync(entity.DataTag);
entity.DataTag = tag;
entity.Type = avatar.ContentType;
entity.LastModified = now;
}
await _database.SaveChangesAsync();
await transaction.CommitAsync();
return new CacheableDataDigest(tag, now);
}
public async Task DeleteAvatar(long userId)
{
await _basicUserService.ThrowIfUserNotExist(userId);
var entity = await _database.UserAvatars.Where(a => a.UserId == userId).SingleOrDefaultAsync();
if (entity is null || entity.DataTag is null)
return;
await using var transaction = await _database.Database.BeginTransactionAsync();
await _dataManager.FreeEntryAsync(entity.DataTag);
entity.DataTag = null;
entity.Type = null;
entity.LastModified = _clock.GetCurrentTime();
await _database.SaveChangesAsync();
await transaction.CommitAsync();
}
}
}
|
