BackEnd/Timeline/Services/User/RegisterCode/RegisterCodeService.cs


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

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 IUserService _userService;

        private readonly RandomNumberGenerator _randomNumberGenerator;

        public RegisterCodeService(DatabaseContext databaseContext, IUserService userService)
        {
            _databaseContext = databaseContext;
            _userService = userService;

            _randomNumberGenerator = RandomNumberGenerator.Create();
        }

        public async Task<string> CreateNewCodeAsync(long userId)
        {
            await _userService.CheckUserExistenceAsync(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 async Task<UserRegisterInfo> CreateRegisterInfoAsync(long userId, string registerCode, DateTime registerTime)
        {
            await _userService.CheckUserExistenceAsync(userId);

            var introducerId = await GetCodeOwnerAsync(registerCode, false);

            var entity = new UserRegisterInfo()
            {
                UserId = userId,
                IntroducerId = introducerId,
                RegisterCode = registerCode,
                RegisterTime = registerTime
            };

            _databaseContext.UserRegisterInfos.Add(entity);
            await _databaseContext.SaveChangesAsync();

            return entity;
        }

        public void Dispose()
        {
            _randomNumberGenerator.Dispose();
        }

        public async Task<long?> GetCodeOwnerAsync(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?> GetCurrentCodeAsync(long userId)
        {
            await _userService.CheckUserExistenceAsync(userId);

            var entity = await _databaseContext.RegisterCodes.Where(r => r.OwnerId == userId && r.Enabled).SingleOrDefaultAsync();
            return entity?.Code;
        }

        public async Task<UserRegisterInfo?> GetUserRegisterInfoAsync(long userId)
        {
            await _userService.CheckUserExistenceAsync(userId);
            return await _databaseContext.UserRegisterInfos.Where(i => i.UserId == userId).SingleOrDefaultAsync();
        }

        public Task<List<UserRegisterInfo>> GetUserRegisterInfoOfIntroducerAsync(long introducerId)
        {
            throw new NotImplementedException();
        }
    }
}