From 47325759e5d9abe9d882ec55d8fb1760591f2a3c Mon Sep 17 00:00:00 2001
From: Bruce Wayne <admin@bige0.com>
Date: Sat, 4 Jan 2020 20:59:37 +0800
Subject: [PATCH] Fix logic

---
 AutoSplitVideo.WPF/Model/Config.cs            |  2 +-
 AutoSplitVideo.WPF/View/MainWindow.xaml       | 14 ++++++++---
 AutoSplitVideo.WPF/View/MainWindow.xaml.cs    | 19 ++++++++-------
 .../ViewModel/MainWindowViewModel.cs          | 23 ++++++++++++++++++-
 BilibiliApi/BililiveAPI.cs                    | 17 ++++++++++----
 BilibiliApi/Passport/Passport.cs              | 14 ++++++++++-
 BilibiliApi/Utils.cs                          |  6 +++++
 7 files changed, 77 insertions(+), 18 deletions(-)

diff --git a/AutoSplitVideo.WPF/Model/Config.cs b/AutoSplitVideo.WPF/Model/Config.cs
index 2106f7c..a992de9 100644
--- a/AutoSplitVideo.WPF/Model/Config.cs
+++ b/AutoSplitVideo.WPF/Model/Config.cs
@@ -67,7 +67,7 @@ public bool FixTimestamp
 
 		public string Token
 		{
-			get => string.IsNullOrEmpty(_token) || _token.Length != 32 ? string.Empty : _token;
+			get => BilibiliApi.Utils.IsToken(_token) ? _token : string.Empty;
 			set => SetField(ref _token, value);
 		}
 
diff --git a/AutoSplitVideo.WPF/View/MainWindow.xaml b/AutoSplitVideo.WPF/View/MainWindow.xaml
index 12f8a5b..5baa877 100644
--- a/AutoSplitVideo.WPF/View/MainWindow.xaml
+++ b/AutoSplitVideo.WPF/View/MainWindow.xaml
@@ -318,12 +318,20 @@
 							<xctk:WatermarkTextBox Margin="3" Watermark="密码"
 							Text="{Binding Password,Mode=TwoWay,UpdateSourceTrigger=PropertyChanged}" />
 
-							<Button x:Name="LoginButton" Margin="3,3,3,6" Padding="2" Content="获取 Access Token" HorizontalAlignment="Left" Click="LoginButton_OnClick"/>
+							<Button x:Name="LoginButton" Margin="3,3,3,6" Padding="2" Content="获取 Token" HorizontalAlignment="Left" Click="LoginButton_OnClick"/>
 
-							<xctk:WatermarkTextBox Margin="3,0" Watermark="SESSDATA/Access Token"
+							<xctk:WatermarkTextBox Margin="3" Watermark="SESSDATA/Access Token"
 							Text="{Binding Token,Mode=TwoWay,UpdateSourceTrigger=PropertyChanged}"/>
 
-							<Button x:Name="ApplyToApiButton" Margin="3,3,3,6" Padding="2" Content="应用" HorizontalAlignment="Left" Click="ApplyToApiButton_OnClick"/>
+							<WrapPanel>
+								<Button x:Name="ApplyToApiButton"
+								Margin="3,3,3,6" Padding="2" 
+								Content="登录" Click="ApplyToApiButton_OnClick" />
+								<Button x:Name="RevokeButton"
+								Margin="3,3,3,6" Padding="2"
+								Content="注销" Click="RevokeButton_OnClick"/>
+							</WrapPanel>
+
 							<TextBlock Margin="3,0,0,0" Text="{Binding Status,Mode=TwoWay,UpdateSourceTrigger=PropertyChanged}" />
 						</StackPanel>
 					</GroupBox>
diff --git a/AutoSplitVideo.WPF/View/MainWindow.xaml.cs b/AutoSplitVideo.WPF/View/MainWindow.xaml.cs
index 85b45e4..68c02b4 100644
--- a/AutoSplitVideo.WPF/View/MainWindow.xaml.cs
+++ b/AutoSplitVideo.WPF/View/MainWindow.xaml.cs
@@ -538,10 +538,7 @@ private void LoginButton_OnClick(object sender, RoutedEventArgs e)
 			LoginButton.IsEnabled = false;
 			MainWindowViewModel.Login().ContinueWith(task =>
 			{
-				Dispatcher?.InvokeAsync(() =>
-				{
-					LoginButton.IsEnabled = true;
-				});
+				Dispatcher?.InvokeAsync(() => { LoginButton.IsEnabled = true; });
 			});
 		}
 
@@ -550,10 +547,16 @@ private void ApplyToApiButton_OnClick(object sender, RoutedEventArgs e)
 			ApplyToApiButton.IsEnabled = false;
 			MainWindowViewModel.ApplyToApi().ContinueWith(task =>
 			{
-				Dispatcher?.InvokeAsync(() =>
-				{
-					ApplyToApiButton.IsEnabled = true;
-				});
+				Dispatcher?.InvokeAsync(() => { ApplyToApiButton.IsEnabled = true; });
+			});
+		}
+
+		private void RevokeButton_OnClick(object sender, RoutedEventArgs e)
+		{
+			RevokeButton.IsEnabled = false;
+			MainWindowViewModel.Revoke().ContinueWith(task =>
+			{
+				Dispatcher?.InvokeAsync(() => { RevokeButton.IsEnabled = true; });
 			});
 		}
 
diff --git a/AutoSplitVideo.WPF/ViewModel/MainWindowViewModel.cs b/AutoSplitVideo.WPF/ViewModel/MainWindowViewModel.cs
index 80d8a72..651f647 100644
--- a/AutoSplitVideo.WPF/ViewModel/MainWindowViewModel.cs
+++ b/AutoSplitVideo.WPF/ViewModel/MainWindowViewModel.cs
@@ -386,6 +386,7 @@ public async Task Login()
 				{
 					Token = token.AccessToken;
 					UpdateStatus($@"获取 Access Token 成功，Token 有效期至 {token.Expires.AddHours(8)}");
+					Account = Password = string.Empty;
 				}
 			}
 			catch (Exception ex)
@@ -410,7 +411,7 @@ public async Task ApplyToApi()
 					CurrentConfig.Token = Token;
 					UpdateStatus(@"Cookie 应用成功");
 				}
-				else
+				else if (BilibiliApi.Utils.IsToken(Token))
 				{
 					try
 					{
@@ -431,6 +432,10 @@ public async Task ApplyToApi()
 						UpdateStatus($@"登录失败，{ex.Message}");
 					}
 				}
+				else
+				{
+					UpdateStatus(@"登录失败，Access Token 格式错误");
+				}
 			}
 			else
 			{
@@ -438,6 +443,22 @@ public async Task ApplyToApi()
 			}
 		}
 
+		public async Task Revoke()
+		{
+			if (BilibiliApi.Utils.IsToken(Token))
+			{
+				//TODO
+				await BilibiliApi.BililiveApi.RevokeToken(Token);
+				UpdateStatus(@"注销请求发送完成");
+				Token = string.Empty;
+				BilibiliApi.BililiveApi.Reload(null);
+			}
+			else
+			{
+				UpdateStatus(@"Access Token 格式错误");
+			}
+		}
+
 		#endregion
 	}
 }
diff --git a/BilibiliApi/BililiveAPI.cs b/BilibiliApi/BililiveAPI.cs
index f3a244f..f5e24be 100644
--- a/BilibiliApi/BililiveAPI.cs
+++ b/BilibiliApi/BililiveAPI.cs
@@ -1,4 +1,4 @@
-using BilibiliApi.Model;
+using BilibiliApi.Model;
 using System;
 using System.Diagnostics;
 using System.Linq;
@@ -147,11 +147,20 @@ public static async Task<BilibiliToken> LoginAsync(string userName, string passw
 		public static async Task<BilibiliToken> GetTokenInfo(string accessToken)
 		{
 			var token = new BilibiliToken(await Passport.Passport.GetTokenInfo(accessToken));
-			if (token.Code == 0)
+			return token.Code == 0 ? token : null;
+		}
+
+		public static async Task RevokeToken(string accessToken)
+		{
+			//TODO
+			try
 			{
-				return token;
+				await Passport.Passport.Revoke(accessToken);
+			}
+			catch
+			{
+				// ignored
 			}
-			return null;
 		}
 	}
 }
diff --git a/BilibiliApi/Passport/Passport.cs b/BilibiliApi/Passport/Passport.cs
index c39a87a..6c4b29a 100644
--- a/BilibiliApi/Passport/Passport.cs
+++ b/BilibiliApi/Passport/Passport.cs
@@ -1,4 +1,4 @@
-using System.Collections.Generic;
+using System.Collections.Generic;
 using System.Threading.Tasks;
 
 namespace BilibiliApi.Passport
@@ -32,6 +32,18 @@ public static async Task<string> Login(string hash, string publicKey, string use
 			return await PostAsync(BaseUri, requestUri, body);
 		}
 
+		public static async Task<string> Revoke(string accessToken)
+		{
+			const string requestUri = @"api/oauth2/revoke";
+			var pair = new Dictionary<string, string>
+			{
+					{@"platform", @"android"},
+					{@"access_token", accessToken}
+			};
+			using var body = await GetBody(pair, true);
+			return await PostAsync(BaseUri, requestUri, body);
+		}
+
 		public static async Task<string> GetTokenInfo(string accessToken)
 		{
 			var pair = new Dictionary<string, string>
diff --git a/BilibiliApi/Utils.cs b/BilibiliApi/Utils.cs
index 4d1ca10..6960411 100644
--- a/BilibiliApi/Utils.cs
+++ b/BilibiliApi/Utils.cs
@@ -3,6 +3,7 @@
 using System.Net.Sockets;
 using System.Security.Cryptography;
 using System.Text;
+using System.Text.RegularExpressions;
 
 namespace BilibiliApi
 {
@@ -84,5 +85,10 @@ public static DateTime GetTime(string timeStamp)
 			var toNow = new TimeSpan(lTime);
 			return dtStart.Add(toNow);
 		}
+
+		public static bool IsToken(string token)
+		{
+			return token.Length == 32 && Regex.IsMatch(token, @"^[a-f0-9]+$");
+		}
 	}
 }