Implement Client
This commit is contained in:
@@ -1,22 +1,134 @@
|
||||
using System;
|
||||
using System.IO;
|
||||
using System.Threading.Tasks;
|
||||
using Akari.Prototype.Shared.Protos;
|
||||
using Avalonia;
|
||||
using Avalonia.Controls;
|
||||
using Avalonia.Interactivity;
|
||||
using Avalonia.Markup.Xaml;
|
||||
using Google.Protobuf;
|
||||
using Grpc.Net.Client;
|
||||
using MessageBox.Avalonia;
|
||||
|
||||
namespace Akari.Prototype.Client
|
||||
{
|
||||
public partial class MainWindow : Window
|
||||
public partial class MainWindow : Window, IDisposable
|
||||
{
|
||||
public const string Hostname = "https://localhost:5001";
|
||||
|
||||
private readonly GrpcChannel _channel;
|
||||
|
||||
private string? _file = null;
|
||||
|
||||
public MainWindow()
|
||||
{
|
||||
InitializeComponent();
|
||||
#if DEBUG
|
||||
this.AttachDevTools();
|
||||
#endif
|
||||
|
||||
_channel = GrpcChannel.ForAddress(Hostname);
|
||||
}
|
||||
|
||||
private void InitializeComponent()
|
||||
private async void OnPickFile(object sender, RoutedEventArgs e)
|
||||
{
|
||||
AvaloniaXamlLoader.Load(this);
|
||||
var mode = GetMode();
|
||||
|
||||
var dialog = new OpenFileDialog()
|
||||
{
|
||||
AllowMultiple = false,
|
||||
Title = $"Pick a file to {mode}"
|
||||
};
|
||||
|
||||
if (await dialog.ShowAsync(this) is string[] { Length: 1 } res)
|
||||
{
|
||||
_file = res[0];
|
||||
TxtFile.Text = Path.GetFileName(_file);
|
||||
}
|
||||
}
|
||||
|
||||
private async void OnSubmit(object sender, RoutedEventArgs e)
|
||||
{
|
||||
if (_file is null)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
if (string.IsNullOrWhiteSpace(BoxName.Text))
|
||||
{
|
||||
await MessageBoxManager.GetMessageBoxStandardWindow("Error", $"Invalid application name", icon: MessageBox.Avalonia.Enums.Icon.Error).Show();
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
if (string.IsNullOrWhiteSpace(BoxToken.Text))
|
||||
{
|
||||
await MessageBoxManager.GetMessageBoxStandardWindow("Error", $"Invalid token", icon: MessageBox.Avalonia.Enums.Icon.Error).Show();
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
var client = new AkariApi.AkariApiClient(_channel);
|
||||
|
||||
// Decrypt
|
||||
if (GetMode() == Mode.Decrypt)
|
||||
{
|
||||
var response = await client.DecryptAsync(new DecryptRequest()
|
||||
{
|
||||
Application = BoxName.Text,
|
||||
Token = BoxToken.Text,
|
||||
Encrypted = ByteString.CopyFrom(File.ReadAllBytes(_file))
|
||||
});
|
||||
|
||||
if (response.ResponseCase == DecryptResponse.ResponseOneofCase.ErrorMessage)
|
||||
{
|
||||
await MessageBoxManager.GetMessageBoxStandardWindow("Error", $"Error during decryption, server response: {response.ErrorMessage}", icon: MessageBox.Avalonia.Enums.Icon.Error).Show();
|
||||
}
|
||||
else
|
||||
{
|
||||
string outPath = Path.ChangeExtension(_file, null);
|
||||
|
||||
File.WriteAllBytes(outPath, response.Plain.ToByteArray());
|
||||
|
||||
await MessageBoxManager.GetMessageBoxStandardWindow("Success", $"Saved decrypted file to {outPath}\n").Show();
|
||||
|
||||
}
|
||||
}
|
||||
// Encrypt
|
||||
else
|
||||
{
|
||||
var response = await client.EncryptAsync(new EncryptRequest()
|
||||
{
|
||||
Application = BoxName.Text,
|
||||
Token = BoxToken.Text,
|
||||
Plain = ByteString.CopyFrom(File.ReadAllBytes(_file))
|
||||
});
|
||||
|
||||
if (response.ResponseCase == EncryptResponse.ResponseOneofCase.ErrorMessage)
|
||||
{
|
||||
await MessageBoxManager.GetMessageBoxStandardWindow("Error", $"Error during encryption, server response: {response.ErrorMessage}", icon: MessageBox.Avalonia.Enums.Icon.Error).Show();
|
||||
}
|
||||
else
|
||||
{
|
||||
string outPath = $"{_file}.enc";
|
||||
|
||||
File.WriteAllBytes(outPath, response.Encrypted.ToByteArray());
|
||||
|
||||
await MessageBoxManager.GetMessageBoxStandardWindow("Success", $"Saved encrypted file to {outPath}").Show();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private Mode GetMode()
|
||||
{
|
||||
if (CbxMode.SelectedItem is ComboBoxItem { Content: string mode })
|
||||
{
|
||||
return Enum.Parse<Mode>(mode);
|
||||
}
|
||||
|
||||
throw new Exception("Invalid mode in CbxMode");
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
_channel.Dispose();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user