diff options
Diffstat (limited to 'Timeline')
| -rw-r--r-- | Timeline/Entities/DatabaseContext.cs | 3 | ||||
| -rw-r--r-- | Timeline/Entities/UserAvatar.cs | 18 | ||||
| -rw-r--r-- | Timeline/Services/UserAvatarService.cs | 181 | ||||
| -rw-r--r-- | Timeline/Timeline.csproj | 6 | ||||
| -rw-r--r-- | Timeline/appsettings.Development.json | 2 | ||||
| -rw-r--r-- | Timeline/default-avatar.png | bin | 0 -> 26442 bytes |
6 files changed, 203 insertions, 7 deletions
diff --git a/Timeline/Entities/DatabaseContext.cs b/Timeline/Entities/DatabaseContext.cs index 3629e821..f32e5992 100644 --- a/Timeline/Entities/DatabaseContext.cs +++ b/Timeline/Entities/DatabaseContext.cs @@ -27,6 +27,8 @@ namespace Timeline.Entities [Column("version"), Required]
public long Version { get; set; }
+
+ public UserAvatar Avatar { get; set; }
}
public class DatabaseContext : DbContext
@@ -44,5 +46,6 @@ namespace Timeline.Entities }
public DbSet<User> Users { get; set; }
+ public DbSet<UserAvatar> UserAvatars { get; set; }
}
}
diff --git a/Timeline/Entities/UserAvatar.cs b/Timeline/Entities/UserAvatar.cs new file mode 100644 index 00000000..d7c24403 --- /dev/null +++ b/Timeline/Entities/UserAvatar.cs @@ -0,0 +1,18 @@ +using System.ComponentModel.DataAnnotations;
+using System.ComponentModel.DataAnnotations.Schema;
+
+namespace Timeline.Entities
+{
+ [Table("user_avatars")]
+ public class UserAvatar
+ {
+ [Column("id"), Key, DatabaseGenerated(DatabaseGeneratedOption.Identity)]
+ public long Id { get; set; }
+
+ [Column("data"), Required]
+ public byte[] Data { get; set; }
+
+ [Column("type"), Required]
+ public string Type { get; set; }
+ }
+}
diff --git a/Timeline/Services/UserAvatarService.cs b/Timeline/Services/UserAvatarService.cs new file mode 100644 index 00000000..21153575 --- /dev/null +++ b/Timeline/Services/UserAvatarService.cs @@ -0,0 +1,181 @@ +using Microsoft.AspNetCore.Hosting;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.Extensions.Logging;
+using System;
+using System.IO;
+using System.Linq;
+using System.Threading.Tasks;
+using Timeline.Entities;
+
+namespace Timeline.Services
+{
+ public class Avatar
+ {
+ public string Type { get; set; }
+ public byte[] Data { get; set; }
+ }
+
+ /// <summary>
+ /// Thrown when avatar is of bad format.
+ /// </summary>
+ [Serializable]
+ public class AvatarDataException : Exception
+ {
+ public AvatarDataException(Avatar avatar, string message) : base(message) { Avatar = avatar; }
+ public AvatarDataException(Avatar avatar, string message, Exception inner) : base(message, inner) { Avatar = avatar; }
+ protected AvatarDataException(
+ System.Runtime.Serialization.SerializationInfo info,
+ System.Runtime.Serialization.StreamingContext context) : base(info, context) { }
+
+ public Avatar Avatar { get; set; }
+ }
+
+ /// <summary>
+ /// Provider for default user avatar.
+ /// </summary>
+ /// <remarks>
+ /// Mainly for unit tests.
+ /// </remarks>
+ public interface IDefaultUserAvatarProvider
+ {
+ /// <summary>
+ /// Get the default avatar.
+ /// </summary>
+ Task<Avatar> GetDefaultAvatar();
+ }
+
+ public interface IUserAvatarService
+ {
+ /// <summary>
+ /// Get avatar of a user. If the user has no avatar, a default one is returned.
+ /// </summary>
+ /// <param name="username">The username of the user to get avatar of.</param>
+ /// <returns>The avatar.</returns>
+ /// <exception cref="ArgumentException">Thrown if <paramref name="username"/> is null or empty.</exception>
+ /// <exception cref="UserNotExistException">Thrown if the user does not exist.</exception>
+ Task<Avatar> GetAvatar(string username);
+
+ /// <summary>
+ /// Set avatar for a user.
+ /// </summary>
+ /// <param name="username">The username of the user to set avatar for.</param>
+ /// <param name="avatar">The avatar. Can be null to delete the saved avatar.</param>
+ /// <exception cref="ArgumentException">Throw if <paramref name="username"/> is null or empty.
+ /// Or thrown if <paramref name="avatar"/> is not null but <see cref="Avatar.Type"/> is null or empty or <see cref="Avatar.Data"/> is null.</exception>
+ /// <exception cref="UserNotExistException">Thrown if the user does not exist.</exception>
+ /// <exception cref="AvatarDataException">Thrown if avatar is of bad format.</exception>
+ Task SetAvatar(string username, Avatar avatar);
+ }
+
+ public class DefaultUserAvatarProvider : IDefaultUserAvatarProvider
+ {
+ private readonly IHostingEnvironment _environment;
+
+ public DefaultUserAvatarProvider(IHostingEnvironment environment)
+ {
+ _environment = environment;
+ }
+
+ public async Task<Avatar> GetDefaultAvatar()
+ {
+ return new Avatar
+ {
+ Type = "image/png",
+ Data = await File.ReadAllBytesAsync(Path.Combine(_environment.ContentRootPath, "default-avatar.png"))
+ };
+ }
+ }
+
+ public class UserAvatarService : IUserAvatarService
+ {
+
+ private readonly ILogger<UserAvatarService> _logger;
+
+ private readonly DatabaseContext _database;
+
+ private readonly IDefaultUserAvatarProvider _defaultUserAvatarProvider;
+
+ public UserAvatarService(ILogger<UserAvatarService> logger, DatabaseContext database, IDefaultUserAvatarProvider defaultUserAvatarProvider)
+ {
+ _logger = logger;
+ _database = database;
+ _defaultUserAvatarProvider = defaultUserAvatarProvider;
+ }
+
+ public async Task<Avatar> GetAvatar(string username)
+ {
+ if (string.IsNullOrEmpty(username))
+ throw new ArgumentException("Username is null or empty.", nameof(username));
+
+ var user = await _database.Users.Where(u => u.Name == username).SingleOrDefaultAsync();
+ if (user == null)
+ throw new UserNotExistException(username);
+
+ await _database.Entry(user).Reference(u => u.Avatar).LoadAsync();
+ var avatar = user.Avatar;
+
+ if (avatar == null)
+ {
+ return await _defaultUserAvatarProvider.GetDefaultAvatar();
+ }
+ else
+ {
+ return new Avatar
+ {
+ Type = avatar.Type,
+ Data = avatar.Data
+ };
+ }
+ }
+
+ public async Task SetAvatar(string username, Avatar avatar)
+ {
+ if (string.IsNullOrEmpty(username))
+ throw new ArgumentException("Username is null or empty.", nameof(username));
+
+ if (avatar != null)
+ {
+ if (string.IsNullOrEmpty(avatar.Type))
+ throw new ArgumentException("Type of avatar is null or empty.", nameof(avatar));
+ if (avatar.Data == null)
+ throw new ArgumentException("Data of avatar is null.", nameof(avatar));
+ }
+
+ var user = await _database.Users.Where(u => u.Name == username).SingleOrDefaultAsync();
+ if (user == null)
+ throw new UserNotExistException(username);
+
+ await _database.Entry(user).Reference(u => u.Avatar).LoadAsync();
+ var avatarEntity = user.Avatar;
+
+ if (avatar == null)
+ {
+ if (avatarEntity == null)
+ return;
+ else
+ {
+ _database.UserAvatars.Remove(avatarEntity);
+ await _database.SaveChangesAsync();
+ }
+ }
+ else
+ {
+ // TODO: Use image library to check the format to prohibit bad data.
+ if (avatarEntity == null)
+ {
+ user.Avatar = new UserAvatar
+ {
+ Type = avatar.Type,
+ Data = avatar.Data
+ };
+ }
+ else
+ {
+ avatarEntity.Type = avatar.Type;
+ avatarEntity.Data = avatar.Data;
+ }
+ await _database.SaveChangesAsync();
+ }
+ }
+ }
+}
diff --git a/Timeline/Timeline.csproj b/Timeline/Timeline.csproj index 1f70c634..29ff3354 100644 --- a/Timeline/Timeline.csproj +++ b/Timeline/Timeline.csproj @@ -5,12 +5,6 @@ <UserSecretsId>1f6fb74d-4277-4bc0-aeea-b1fc5ffb0b43</UserSecretsId>
<Authors>crupest</Authors>
</PropertyGroup>
- <ItemGroup>
- <Compile Remove="Migrations\20190817094126_Enhance.cs" />
- <Compile Remove="Migrations\20190817094126_Enhance.Designer.cs" />
- <Compile Remove="Migrations\20190817094254_Enhance.cs" />
- <Compile Remove="Migrations\20190817094254_Enhance.Designer.cs" />
- </ItemGroup>
<ItemGroup>
<PackageReference Include="Microsoft.AspNetCore.App" />
diff --git a/Timeline/appsettings.Development.json b/Timeline/appsettings.Development.json index db4b074a..424b3885 100644 --- a/Timeline/appsettings.Development.json +++ b/Timeline/appsettings.Development.json @@ -7,6 +7,6 @@ }
},
"JwtConfig": {
- "SigningKey": "crupest hahahahahahahhahahahahaha"
+ "SigningKey": "this is very very very very very long secret"
}
}
diff --git a/Timeline/default-avatar.png b/Timeline/default-avatar.png Binary files differnew file mode 100644 index 00000000..4086e1d2 --- /dev/null +++ b/Timeline/default-avatar.png |
