Fixes from upstream changes

goatcorp · Apr 17, 2022 · 4618607 · 4618607
1 parent 134c2a7
commit 4618607
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 15 deletions.
diff --git a/src/XIVLauncher/Windows/OtpUriSetupWindow.xaml.cs b/src/XIVLauncher/Windows/OtpUriSetupWindow.xaml.cs
@@ -2,12 +2,8 @@
 using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
-using System.Web.UI.WebControls;
 using System.Windows;
-using System.Windows.Controls;
 using System.Windows.Media;
-using NuGet;
-using XIVLauncher.Addon;
 using XIVLauncher.Windows.ViewModel;
 using CheckBox = System.Windows.Controls.CheckBox;
 

diff --git a/src/XIVLauncher/Windows/ViewModel/MainWindowViewModel.cs b/src/XIVLauncher/Windows/ViewModel/MainWindowViewModel.cs
@@ -214,14 +214,54 @@ private async Task Login(string username, string password, bool isOtp, bool isSt
 
             if (isOtp && (!hasValidCache || action == AfterLoginAction.Repair))
             {
-                otp = OtpInputDialog.AskForOtp((otpDialog, result) =>
+                if (!string.IsNullOrWhiteSpace(AccountManager?.CurrentAccount?.OtpUri) && !PreviousLoginFailure)
                 {
-                    if (AccountManager.CurrentAccount != null && result != null && AccountManager.CurrentAccount.LastSuccessfulOtp == result)
+                    try
                     {
-                        otpDialog.IgnoreCurrentResult(Loc.Localize("DuplicateOtpAfterSuccess",
-                            "This OTP has been already used.\nIt may take up to 30 seconds for a new one."));
+                        if (Uri.TryCreate(AccountManager.CurrentAccount.OtpUri, UriKind.Absolute, out var uri))
+                        {
+                            var query = System.Web.HttpUtility.ParseQueryString(uri.Query);
+
+                            if (!query.AllKeys.Contains("secret"))
+                            {
+                                throw new Exception("No Secret");
+                            }
+
+                            var secretKey = query["secret"];
+                            if (!query.AllKeys.Contains("period") || !int.TryParse(query["period"], out var period))
+                                period = 30;
+                            if (!query.AllKeys.Contains("digits") || !int.TryParse(query["digits"], out var digits))
+                                digits = 6;
+
+                            var algorithm = "sha1";
+                            if (query.AllKeys.Contains("algorithm") || new[] { "sha1", "sha256", "sha512" }.Any(a => a.Equals(query["algorithm"], StringComparison.OrdinalIgnoreCase)))
+                                algorithm = query["algorithm"].ToLowerInvariant();
+
+                            otp = Util.GetTotpToken(secretKey, algorithm, digits, period);
+                        }
+                        else
+                        {
+                            var secretKey = AccountManager.CurrentAccount.OtpUri;
+                            otp = Util.GetTotpToken(secretKey);
+                        }
                     }
-                }, _window);
+                    catch (Exception)
+                    {
+                        otp = "";
+                    }
+                }
+
+                if (string.IsNullOrWhiteSpace(otp))
+                {
+                    otp = OtpInputDialog.AskForOtp((otpDialog, result) =>
+                    {
+                        if (AccountManager.CurrentAccount != null && result != null && AccountManager.CurrentAccount.LastSuccessfulOtp == result)
+                        {
+                            otpDialog.IgnoreCurrentResult(Loc.Localize("DuplicateOtpAfterSuccess",
+                                "This OTP has been already used.\nIt may take up to 30 seconds for a new one."));
+                        }
+                    }, _window);
+                }
             }
 
             if (otp == null)

diff --git a/src/XIVLauncher/Windows/ViewModel/OtpUriSetupWindowViewModel.cs b/src/XIVLauncher/Windows/ViewModel/OtpUriSetupWindowViewModel.cs
@@ -1,9 +1,4 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
-using CheapLoc;
+using CheapLoc;
 
 namespace XIVLauncher.Windows.ViewModel
 {