Add user data exporting

Author: Aux

src/Dogey.Core/Data/Models/Guild.cs

@@ -0,0 +1,13 @@
+using System.Collections.Generic;
+
+namespace Dogey
+{
+    public class Guild
+    {
+        public ulong Id { get; set; }
+        public string StringPrefix { get; set; }
+        public bool AllowMentionPrefix { get; set; } = true;
+
+        public List<GuildUser> Users { get; set; }
+    }
+}

src/Dogey.Core/Data/Models/GuildUser.cs

@@ -0,0 +1,12 @@
+namespace Dogey
+{
+    public class GuildUser
+    {
+        public ulong GuildId { get; set; }
+        public ulong UserId { get; set; }
+        public bool IsBlocked { get; set; }
+
+        public Guild Guild { get; set; }
+        public User User { get; set; }
+    }
+}

src/Dogey.Core/Data/Models/User.cs

@@ -0,0 +1,12 @@
+using System.Collections.Generic;
+
+namespace Dogey
+{
+    public class User
+    {
+        public ulong Id { get; set; }
+        public string Locale { get; set; }
+
+        public List<GuildUser> Guilds { get; set; }
+    }
+}

src/Dogey.Core/Data/RootDatabase.cs

@@ -0,0 +1,49 @@
+using Microsoft.EntityFrameworkCore;
+using System;
+using System.IO;
+
+namespace Dogey
+{
+    public class RootDatabase : DbContext
+    {
+        public DbSet<User> Users { get; set; }
+        public DbSet<GuildUser> GuildUsers { get; set; }
+        public DbSet<Guild> Guilds { get; set; }
+
+        public RootDatabase()
+        {
+            Database.EnsureCreated();
+        }
+
+        protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
+        {
+            string baseDir = Path.Combine(AppContext.BaseDirectory, "common/data");
+            if (!Directory.Exists(baseDir))
+                Directory.CreateDirectory(baseDir);
+
+            string datadir = Path.Combine(baseDir, "root.sqlite.db");
+            optionsBuilder.UseSqlite($"Filename={datadir}");
+        }
+
+        protected override void OnModelCreating(ModelBuilder modelBuilder)
+        {
+            modelBuilder.Entity<User>()
+                .HasKey(x => x.Id);
+            modelBuilder.Entity<User>()
+                .HasMany(x => x.Guilds)
+                .WithOne(x => x.User);
+
+            modelBuilder.Entity<GuildUser>()
+                .HasKey(x => new { x.GuildId, x.UserId });
+            modelBuilder.Entity<GuildUser>()
+                .HasOne(x => x.Guild)
+                .WithMany(x => x.Users);
+
+            modelBuilder.Entity<Guild>()
+                .HasKey(x => x.Id);
+            modelBuilder.Entity<Guild>()
+                .HasMany(x => x.Users)
+                .WithOne(x => x.Guild);
+        }
+    }
+}

src/Dogey.Core/Dogey.Core.csproj

@@ -8,6 +8,8 @@
   <ItemGroup>
     <PackageReference Include="Colorful.Console" Version="1.2.10" />
     <PackageReference Include="Discord.Net" Version="2.2.0-dev-20200331.2" />
+    <PackageReference Include="FormatWith" Version="2.2.1" />
+    <PackageReference Include="Microsoft.EntityFrameworkCore.Sqlite" Version="3.1.3" />
     <PackageReference Include="Microsoft.Extensions.Configuration.CommandLine" Version="3.1.3" />
     <PackageReference Include="Microsoft.Extensions.DependencyInjection" Version="3.1.3" />
     <PackageReference Include="Microsoft.Extensions.Logging" Version="3.1.3" />
@@ -33,6 +35,9 @@
     <None Update="common\locale\zh-CN.yml">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
     </None>
+    <None Update="common\permissions.yml">
+      <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
+    </None>
   </ItemGroup>
 
 </Project>

src/Dogey/Commands/GDPRModule.cs

@@ -0,0 +1,59 @@
+using Discord;
+using Discord.Commands;
+using Dogey.Services;
+using Microsoft.EntityFrameworkCore;
+using Newtonsoft.Json;
+using System;
+using System.IO;
+using System.Linq;
+using System.Threading.Tasks;
+
+namespace Dogey.Commands
+{
+    public class GDPRModule : DogeyModuleBase
+    {
+        private readonly RootDatabase _db;
+        private User _currentUser;
+
+        public GDPRModule(LocaleService locale, RootDatabase db) : base(locale) 
+        {
+            _db = db;
+        }
+
+        protected override void BeforeExecute(CommandInfo command)
+        {
+            _currentUser = _db.Users.SingleOrDefault(x => x.Id == Context.User.Id);
+        }
+
+        [Command("exportuser")]
+        public async Task ExportUserDataAsync()
+        {
+            var user = await _db.Users.Include(x => x.Guilds).SingleOrDefaultAsync(x => x.Id == Context.User.Id);
+            if (user == null)
+            {
+                await ReplyAsync(_locale.GetString("gdpr:user_data_null", Context.User, _currentUser.Locale));
+                return;
+            }
+
+            var directory = Path.Combine(AppContext.BaseDirectory, $"common/temp");
+            if (!Directory.Exists(directory))
+                Directory.CreateDirectory(directory);
+
+            var filePath = Path.Combine(directory, $"{Context.User.Id}-data.json");
+            using (var file = File.CreateText(filePath))
+            {
+                var serializer = new JsonSerializer()
+                {
+                    Formatting = Formatting.Indented,
+                    NullValueHandling = NullValueHandling.Ignore,
+                    ReferenceLoopHandling = ReferenceLoopHandling.Ignore
+                };
+                serializer.Serialize(file, user);
+            }
+
+            await Context.User.SendFileAsync(filePath);
+            await ReplyAsync(_locale.GetString("gdpr:user_data_get", Context.User, _currentUser.Locale));
+            File.Delete(filePath);
+        }
+    }
+}