using AudibleApi; using AudibleUtilities; using Avalonia.Collections; using Avalonia.Controls; using Avalonia.Platform.Storage; using ReactiveUI; using System; using System.Collections.Generic; using System.Collections.Specialized; using System.IO; using System.Linq; using System.Threading.Tasks; namespace LibationAvalonia.Dialogs { public partial class AccountsDialog : DialogWindow { public AvaloniaList Accounts { get; } = new(); public class AccountDto : ViewModels.ViewModelBase { private string _accountId; public IReadOnlyList Locales => AccountsDialog.Locales; public bool LibraryScan { get; set; } = true; public string AccountId { get => _accountId; set { this.RaiseAndSetIfChanged(ref _accountId, value); this.RaisePropertyChanged(nameof(IsDefault)); } } public Locale SelectedLocale { get; set; } public string AccountName { get; set; } public bool IsDefault => string.IsNullOrEmpty(AccountId); public AccountDto() { } public AccountDto(Account account) { LibraryScan = account.LibraryScan; AccountId = account.AccountId; SelectedLocale = Locales.Single(l => l.Name == account.Locale.Name); AccountName = account.AccountName; } } private static string GetAudibleCliAppDataPath() => Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData), "Audible"); private static IReadOnlyList Locales => Localization.Locales.OrderBy(l => l.Name).ToList(); public AccountsDialog() { InitializeComponent(); // WARNING: accounts persister will write ANY EDIT to object immediately to file // here: copy strings and dispose of persister // only persist in 'save' step using var persister = AudibleApiStorage.GetAccountsSettingsPersister(); Accounts.CollectionChanged += Accounts_CollectionChanged; Accounts.AddRange(persister.AccountsSettings.Accounts.Select(a => new AccountDto(a))); DataContext = this; addBlankAccount(); } private void Accounts_CollectionChanged(object sender, NotifyCollectionChangedEventArgs e) { if (e.Action is NotifyCollectionChangedAction.Add && e.NewItems?.Count > 0) { foreach (var newItem in e.NewItems.OfType()) newItem.PropertyChanged += AccountDto_PropertyChanged; } else if (e.Action is NotifyCollectionChangedAction.Remove && e.OldItems?.Count > 0) { foreach (var oldItem in e.OldItems.OfType()) oldItem.PropertyChanged -= AccountDto_PropertyChanged; } } private void addBlankAccount() => Accounts.Insert(Accounts.Count, new AccountDto()); private void AccountDto_PropertyChanged(object sender, System.ComponentModel.PropertyChangedEventArgs e) { if (!Accounts.Any(a => a.IsDefault)) addBlankAccount(); } public void DeleteButton_Clicked(object sender, Avalonia.Interactivity.RoutedEventArgs e) { if (e.Source is Button expBtn && expBtn.DataContext is AccountDto acc) Accounts.Remove(acc); } public async void ImportButton_Clicked(object sender, Avalonia.Interactivity.RoutedEventArgs e) { var openFileDialogOptions = new FilePickerOpenOptions { Title = $"Select the audible-cli [account].json file", AllowMultiple = false, FileTypeFilter = new FilePickerFileType[] { new("JSON files (*.json)") { Patterns = new[] { "*.json" }, AppleUniformTypeIdentifiers = new[] { "public.json" } } } }; string audibleAppDataDir = GetAudibleCliAppDataPath(); if (Directory.Exists(audibleAppDataDir)) openFileDialogOptions.SuggestedStartLocation = await StorageProvider.TryGetFolderFromPathAsync(audibleAppDataDir); var selectedFiles = await StorageProvider.OpenFilePickerAsync(openFileDialogOptions); var selectedFile = selectedFiles.SingleOrDefault()?.TryGetLocalPath(); if (selectedFile is null) return; try { var jsonText = File.ReadAllText(selectedFile); var mkbAuth = Mkb79Auth.FromJson(jsonText); var account = await mkbAuth.ToAccountAsync(); // without transaction, accounts persister will write ANY EDIT immediately to file using var persister = AudibleApiStorage.GetAccountsSettingsPersister(); if (persister.AccountsSettings.Accounts.Any(a => a.AccountId == account.AccountId && a.IdentityTokens.Locale.Name == account.Locale.Name)) { await MessageBox.Show(this, $"An account with that account id and country already exists.\r\n\r\nAccount ID: {account.AccountId}\r\nCountry: {account.Locale.Name}", "Cannot Add Duplicate Account"); return; } persister.AccountsSettings.Add(account); Accounts.Add(new AccountDto(account)); } catch (Exception ex) { await MessageBox.ShowAdminAlert( this, $"An error occurred while importing an account from:\r\n{selectedFile}\r\n\r\nIs the file encrypted?", "Error Importing Account", ex); } } public void ExportButton_Clicked(object sender, Avalonia.Interactivity.RoutedEventArgs e) { if (e.Source is Button expBtn && expBtn.DataContext is AccountDto acc) Export(acc); } protected override async Task SaveAndCloseAsync() { try { accountsGrid.CommitEdit(); if (!await inputIsValid()) return; // without transaction, accounts persister will write ANY EDIT immediately to file using var persister = AudibleApiStorage.GetAccountsSettingsPersister(); persister.BeginTransation(); persist(persister.AccountsSettings); persister.CommitTransation(); base.SaveAndClose(); } catch (Exception ex) { await MessageBox.ShowAdminAlert(this, "Error attempting to save accounts", "Error saving accounts", ex); } } public async void SaveButton_Clicked(object sender, Avalonia.Interactivity.RoutedEventArgs e) => await SaveAndCloseAsync(); private void persist(AccountsSettings accountsSettings) { var existingAccounts = accountsSettings.Accounts; // editing account id is a special case. an account is defined by its account id, therefore this is really a different account. the user won't care about this distinction though. // these will be caught below by normal means and re-created minus the convenience of persisting identity tokens // delete for (var i = existingAccounts.Count - 1; i >= 0; i--) { var existing = existingAccounts[i]; if (!Accounts.Any(dto => dto.AccountId?.ToLower().Trim() == existing.AccountId.ToLower() && dto.SelectedLocale?.Name == existing.Locale?.Name)) { accountsSettings.Delete(existing); } } // upsert each. validation occurs through Account and AccountsSettings foreach (var dto in Accounts) { var acct = accountsSettings.Upsert(dto.AccountId, dto.SelectedLocale?.Name); acct.LibraryScan = dto.LibraryScan; acct.AccountName = string.IsNullOrWhiteSpace(dto.AccountName) ? $"{dto.AccountId} - {dto.SelectedLocale?.Name}" : dto.AccountName.Trim(); } } private async Task inputIsValid() { foreach (var dto in Accounts.ToList()) { if (dto.IsDefault) { Accounts.Remove(dto); continue; } if (string.IsNullOrWhiteSpace(dto.AccountId)) { await MessageBox.Show(this, "Account id cannot be blank. Please enter an account id for all accounts.", "Blank account", MessageBoxButtons.OK, MessageBoxIcon.Error); return false; } if (string.IsNullOrWhiteSpace(dto.SelectedLocale?.Name)) { await MessageBox.Show(this, "Please select a locale (i.e.: country or region) for all accounts.", "Blank region", MessageBoxButtons.OK, MessageBoxIcon.Error); return false; } } return true; } private async void Export(AccountDto acc) { // without transaction, accounts persister will write ANY EDIT immediately to file using var persister = AudibleApiStorage.GetAccountsSettingsPersister(); var account = persister.AccountsSettings.Accounts.FirstOrDefault(a => a.AccountId == acc.AccountId && a.Locale.Name == acc.SelectedLocale?.Name); if (account is null) return; if (account.IdentityTokens?.IsValid != true) { await MessageBox.Show(this, "This account hasn't been authenticated yet. First scan your library to log into your account, then try exporting again.", "Account Not Authenticated"); return; } var options = new FilePickerSaveOptions { Title = $"Save Sover Image", SuggestedFileName = $"{acc.AccountId}.json", DefaultExtension = "json", ShowOverwritePrompt = true, FileTypeChoices = new FilePickerFileType[] { new("JSON files (*.json)") { Patterns = new[] { "*.json" }, AppleUniformTypeIdentifiers = new[] { "public.json" } } } }; string audibleAppDataDir = GetAudibleCliAppDataPath(); if (Directory.Exists(audibleAppDataDir)) options.SuggestedStartLocation = await StorageProvider.TryGetFolderFromPathAsync(audibleAppDataDir); var selectedFile = (await StorageProvider.SaveFilePickerAsync(options))?.TryGetLocalPath(); if (selectedFile is null) return; try { var mkbAuth = Mkb79Auth.FromAccount(account); var jsonText = mkbAuth.ToJson(); File.WriteAllText(selectedFile, jsonText); await MessageBox.Show(this, $"Successfully exported {account.AccountName} to\r\n\r\n{selectedFile}", "Success!"); } catch (Exception ex) { await MessageBox.ShowAdminAlert( this, $"An error occurred while exporting account:\r\n{account.AccountName}", "Error Exporting Account", ex); } } } }