// SpeedTestWindow.xaml.cs
// SpeedOf.Me API - Windows/WPF Integration
//
// This file demonstrates how to integrate the SpeedOf.Me speed test
// into a Windows WPF app using WebView2.

using System;
using System.Text.Json;
using System.Threading.Tasks;
using System.Windows;
using Microsoft.Web.WebView2.Core;

namespace SpeedTest
{
    /// <summary>
    /// Speed test result data
    /// </summary>
    public class SpeedTestResult
    {
        public double Download { get; set; }
        public double Upload { get; set; }
        public double Latency { get; set; }
        public double Jitter { get; set; }
        public string? TestServer { get; set; }
        public string? IpAddress { get; set; }
        public string? Hostname { get; set; }
    }

    /// <summary>
    /// Speed test progress data
    /// </summary>
    public class SpeedTestProgress
    {
        public string TestType { get; set; } = "download";
        public double CurrentSpeed { get; set; }
        public double PercentComplete { get; set; }
        public int PassNumber { get; set; }
    }

    /// <summary>
    /// Speed test error data
    /// </summary>
    public class SpeedTestError
    {
        public int Code { get; set; }
        public string Message { get; set; } = "";
    }

    /// <summary>
    /// Message from WebView JavaScript
    /// </summary>
    public class WebViewMessage
    {
        public string Type { get; set; } = "";
        public JsonElement Data { get; set; }
    }

    /// <summary>
    /// Main window containing the speed test WebView
    /// </summary>
    public partial class SpeedTestWindow : Window
    {
        private bool _isRunning = false;

        public SpeedTestWindow()
        {
            InitializeComponent();
            InitializeWebView();
        }

        /// <summary>
        /// Initialize WebView2 and load the speed test HTML
        /// </summary>
        private async void InitializeWebView()
        {
            try
            {
                // Ensure WebView2 runtime is available
                await SpeedTestWebView.EnsureCoreWebView2Async();

                // Handle messages from JavaScript
                SpeedTestWebView.CoreWebView2.WebMessageReceived += OnWebMessageReceived;

                // Option 1: Load from embedded resource or file
                // Note: You'll need to set up the HTML as an embedded resource
                // or copy it to the output directory
                string htmlPath = System.IO.Path.Combine(
                    AppDomain.CurrentDomain.BaseDirectory,
                    "speedtest.html"
                );

                if (System.IO.File.Exists(htmlPath))
                {
                    SpeedTestWebView.CoreWebView2.Navigate($"file:///{htmlPath.Replace('\\', '/')}");
                }
                else
                {
                    // Option 2: Load from remote URL
                    SpeedTestWebView.CoreWebView2.Navigate("https://your-domain.com/speedtest.html");
                }

                UpdateStatus("Ready");
            }
            catch (Exception ex)
            {
                UpdateStatus($"Error initializing WebView: {ex.Message}");
            }
        }

        /// <summary>
        /// Handle messages received from the WebView JavaScript
        /// </summary>
        private void OnWebMessageReceived(object? sender, CoreWebView2WebMessageReceivedEventArgs e)
        {
            try
            {
                string json = e.WebMessageAsJson;
                var message = JsonSerializer.Deserialize<WebViewMessage>(json, new JsonSerializerOptions
                {
                    PropertyNameCaseInsensitive = true
                });

                if (message == null) return;

                // Must dispatch to UI thread
                Dispatcher.Invoke(() => HandleMessage(message));
            }
            catch (Exception ex)
            {
                Dispatcher.Invoke(() => UpdateStatus($"Error: {ex.Message}"));
            }
        }

        /// <summary>
        /// Process a message from the WebView
        /// </summary>
        private void HandleMessage(WebViewMessage message)
        {
            switch (message.Type)
            {
                case "ready":
                    _isRunning = false;
                    UpdateStatus("Ready to test");
                    break;

                case "started":
                    _isRunning = true;
                    ResultsPanel.Visibility = Visibility.Collapsed;
                    UpdateStatus("Starting test...");
                    break;

                case "progress":
                    var progress = JsonSerializer.Deserialize<SpeedTestProgress>(
                        message.Data.GetRawText(),
                        new JsonSerializerOptions { PropertyNameCaseInsensitive = true }
                    );
                    if (progress != null)
                    {
                        string testType = progress.TestType == "download" ? "Download" : "Upload";
                        UpdateStatus($"{testType}: {progress.CurrentSpeed:F1} Mbps ({progress.PercentComplete:F0}%)");
                    }
                    break;

                case "completed":
                    var result = JsonSerializer.Deserialize<SpeedTestResult>(
                        message.Data.GetRawText(),
                        new JsonSerializerOptions { PropertyNameCaseInsensitive = true }
                    );
                    if (result != null)
                    {
                        _isRunning = false;
                        UpdateStatus("Test complete");
                        DisplayResults(result);
                    }
                    break;

                case "error":
                    var error = JsonSerializer.Deserialize<SpeedTestError>(
                        message.Data.GetRawText(),
                        new JsonSerializerOptions { PropertyNameCaseInsensitive = true }
                    );
                    if (error != null)
                    {
                        _isRunning = false;
                        UpdateStatus($"Error {error.Code}: {error.Message}");
                    }
                    break;
            }
        }

        /// <summary>
        /// Update the status text
        /// </summary>
        private void UpdateStatus(string status)
        {
            StatusText.Text = status;
        }

        /// <summary>
        /// Display the test results in the native UI
        /// </summary>
        private void DisplayResults(SpeedTestResult result)
        {
            DownloadResult.Text = $"{result.Download} Mbps";
            UploadResult.Text = $"{result.Upload} Mbps";
            LatencyResult.Text = $"{result.Latency} ms";
            JitterResult.Text = $"{result.Jitter} ms";
            ResultsPanel.Visibility = Visibility.Visible;
        }
    }
}

/*
 * XAML for SpeedTestWindow.xaml:
 *
 * <Window x:Class="SpeedTest.SpeedTestWindow"
 *         xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
 *         xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
 *         xmlns:wv2="clr-namespace:Microsoft.Web.WebView2.Wpf;assembly=Microsoft.Web.WebView2.Wpf"
 *         Title="Speed Test" Height="600" Width="450"
 *         WindowStartupLocation="CenterScreen"
 *         Background="#F3F3F3">
 *     <Grid>
 *         <Grid.RowDefinitions>
 *             <RowDefinition Height="Auto"/>
 *             <RowDefinition Height="*"/>
 *             <RowDefinition Height="Auto"/>
 *         </Grid.RowDefinitions>
 *
 *         <!-- Status Bar -->
 *         <Border Grid.Row="0" Background="White" BorderBrush="#E1E1E1" BorderThickness="0,0,0,1" Padding="16">
 *             <TextBlock x:Name="StatusText" FontSize="14" Foreground="#1A1A1A"/>
 *         </Border>
 *
 *         <!-- WebView -->
 *         <wv2:WebView2 x:Name="SpeedTestWebView" Grid.Row="1"/>
 *
 *         <!-- Native Results Panel (optional) -->
 *         <Border x:Name="ResultsPanel" Grid.Row="2" Background="White"
 *                 BorderBrush="#E1E1E1" BorderThickness="0,1,0,0" Padding="16"
 *                 Visibility="Collapsed">
 *             <StackPanel>
 *                 <Grid Margin="0,8">
 *                     <TextBlock Text="Download" Foreground="#616161"/>
 *                     <TextBlock x:Name="DownloadResult" HorizontalAlignment="Right" FontWeight="SemiBold"/>
 *                 </Grid>
 *                 <Grid Margin="0,8">
 *                     <TextBlock Text="Upload" Foreground="#616161"/>
 *                     <TextBlock x:Name="UploadResult" HorizontalAlignment="Right" FontWeight="SemiBold"/>
 *                 </Grid>
 *                 <Grid Margin="0,8">
 *                     <TextBlock Text="Latency" Foreground="#616161"/>
 *                     <TextBlock x:Name="LatencyResult" HorizontalAlignment="Right" FontWeight="SemiBold"/>
 *                 </Grid>
 *                 <Grid Margin="0,8">
 *                     <TextBlock Text="Jitter" Foreground="#616161"/>
 *                     <TextBlock x:Name="JitterResult" HorizontalAlignment="Right" FontWeight="SemiBold"/>
 *                 </Grid>
 *             </StackPanel>
 *         </Border>
 *     </Grid>
 * </Window>
 */