From 84cb99d087f62dca89fa682feae6738b3350fed4 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Sun, 17 Apr 2022 22:55:32 +0800
Subject: ...

---
 .../User/RegisterCode/RegisterCodeService.cs       | 80 ++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 BackEnd/Timeline/Services/User/RegisterCode/RegisterCodeService.cs

(limited to 'BackEnd/Timeline/Services')

diff --git a/BackEnd/Timeline/Services/User/RegisterCode/RegisterCodeService.cs b/BackEnd/Timeline/Services/User/RegisterCode/RegisterCodeService.cs
new file mode 100644
index 00000000..eb8bc37a
--- /dev/null
+++ b/BackEnd/Timeline/Services/User/RegisterCode/RegisterCodeService.cs
@@ -0,0 +1,80 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Security.Cryptography;
+using System.Threading.Tasks;
+using Microsoft.EntityFrameworkCore;
+using Timeline.Entities;
+using Timeline.Helpers;
+
+namespace Timeline.Services.User.RegisterCode
+{
+    public class RegisterCodeService : IRegisterCodeService, IDisposable
+    {
+        private readonly DatabaseContext _databaseContext;
+        private readonly RandomNumberGenerator _randomNumberGenerator;
+
+        public RegisterCodeService(DatabaseContext databaseContext)
+        {
+            _databaseContext = databaseContext;
+            _randomNumberGenerator = RandomNumberGenerator.Create();
+        }
+
+        public async Task<string> CreateNewCode(long userId)
+        {
+            var oldEntity = await _databaseContext.RegisterCodes.Where(r => r.OwnerId == userId && r.Enabled).SingleOrDefaultAsync();
+
+            if (oldEntity is not null)
+            {
+                oldEntity.Enabled = false;
+            }
+
+            var newEntity = new Entities.RegisterCode()
+            {
+                Code = _randomNumberGenerator.GenerateAlphaDigitString(6),
+                OwnerId = userId,
+                Enabled = true
+            };
+            _databaseContext.RegisterCodes.Add(newEntity);
+
+            await _databaseContext.SaveChangesAsync();
+
+            return newEntity.Code;
+        }
+
+        public Task<UserRegisterInfo> CreateRegisterInfo(long userId, long introducerId, string registerCode, DateTime registerTime)
+        {
+            throw new NotImplementedException();
+        }
+
+        public void Dispose()
+        {
+            _randomNumberGenerator.Dispose();
+        }
+
+        public async Task<long?> GetCodeOwner(string code, bool onlyEnabled = true)
+        {
+            var entity = await _databaseContext.RegisterCodes.Where(r => r.Code == code).SingleOrDefaultAsync();
+            if (entity is null) return null;
+            if (onlyEnabled && !entity.Enabled) return null;
+            return entity.OwnerId;
+        }
+
+        public async Task<string?> GetCurrentCode(long userId)
+        {
+            var entity = await _databaseContext.RegisterCodes.Where(r => r.OwnerId == userId && r.Enabled).SingleOrDefaultAsync();
+            return entity?.Code;
+        }
+
+        public Task<UserRegisterInfo?> GetUserRegisterInfo(long userId)
+        {
+            throw new NotImplementedException();
+        }
+
+        public Task<List<UserRegisterInfo>> GetUserRegisterInfoOfIntroducer(long introducerId)
+        {
+            throw new NotImplementedException();
+        }
+    }
+}
+
-- 
cgit v1.2.3