feat: get user info before run vip big point task

RayWangQvQ · Aug 17, 2022 · 712ed7c · 712ed7c
1 parent e8a8365
commit 712ed7c
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/Ray.BiliBiliTool.Application/VipBigPointAppService.cs b/src/Ray.BiliBiliTool.Application/VipBigPointAppService.cs
@@ -12,6 +12,7 @@
 using Ray.BiliBiliTool.Application.Attributes;
 using Ray.BiliBiliTool.Application.Contracts;
 using Ray.BiliBiliTool.Config.Options;
+using Ray.BiliBiliTool.DomainService.Interfaces;
 
 namespace Ray.BiliBiliTool.Application
 {
@@ -20,21 +21,26 @@ public class VipBigPointAppService : AppService, IVipBigPointAppService
         private readonly ILogger<VipBigPointAppService> _logger;
         private readonly IConfiguration _configuration;
         private readonly IVipBigPointApi _vipApi;
+        private readonly IAccountDomainService _loginDomainService;
 
         public VipBigPointAppService(
             IConfiguration configuration,
             ILogger<VipBigPointAppService> logger,
-            IVipBigPointApi vipApi
+            IVipBigPointApi vipApi,
+            IAccountDomainService loginDomainService
             )
         {
             _configuration = configuration;
             _logger = logger;
             _vipApi = vipApi;
+            _loginDomainService = loginDomainService;
         }
 
         [TaskInterceptor("大会员大积分", TaskLevel.One)]
         public override void DoTask()
         {
+            GetUserInfo();
+
             var re = _vipApi.GetTaskList().Result;
 
             if (re.Code != 0) throw new Exception(re.ToJson());
@@ -73,6 +79,13 @@ public override void DoTask()
             taskInfo.LogInfo(_logger);
         }
 
+        [TaskInterceptor("测试Cookie")]
+        private void GetUserInfo()
+        {
+            UserInfo userInfo = _loginDomainService.LoginByCookie();
+            if (userInfo == null) throw new Exception("登录失败，请检查Cookie");//终止流程
+        }
+
         [TaskInterceptor("签到", TaskLevel.Two, false)]
         private VipTaskInfo Sign(VipTaskInfo info)
         {