using System.Text.Json.Serialization;
using MoShiro.Protocol;

namespace MoShiro.Probe;

/// <summary>
/// <c>probe.json</c>：面板 <c>POST /api/admin/probe/identity</c> 生成的文件（只能下载一次）。
/// <c>limits</c> 是本地硬上限，面板无法修改；在路由器上建议调低，避免 Probe 自己的 CPU 成为瓶颈、反过来影响测量。
/// </summary>
public sealed class ProbeFile
{
    [JsonPropertyName("probe_id")]
    public string ProbeId { get; set; } = "";

    /// <summary>PKCS#8，base64。</summary>
    [JsonPropertyName("private_key")]
    public string PrivateKey { get; set; } = "";

    /// <summary>面板的地址（http 或 https）。Probe 直接连它：身份、Agent 列表、更新都从这里来。</summary>
    [JsonPropertyName("server")]
    public string Server { get; set; } = "";

    [JsonPropertyName("limits")]
    public ProbeLimits Limits { get; set; } = new();
}

public sealed class KnownAgent
{
    [JsonPropertyName("ip_version")]
    public string IpVersion { get; set; } = IpVersions.V4;

    [JsonIgnore]
    public string PathKey => AgentId + "~" + IpVersion;

    [JsonPropertyName("agent_id")]
    public string AgentId { get; set; } = "";

    [JsonPropertyName("host")]
    public string Host { get; set; } = "";

    [JsonPropertyName("tcp_port")]
    public int TcpPort { get; set; }

    /// <summary>Agent 的 TLS 证书指纹（SHA-256，base64）。</summary>
    [JsonPropertyName("spki")]
    public string Spki { get; set; } = "";
}

public sealed class ProbeLimits
{
    /// <summary>按需测试里单个阶段允许的最大码率（Mbit/s）。</summary>
    [JsonPropertyName("max_test_mbps")]
    public double MaxTestMbps { get; set; } = 100;

    /// <summary>按需测试里单个阶段允许的最大包速率。</summary>
    [JsonPropertyName("max_total_pps")]
    public long MaxTotalPps { get; set; } = 20_000;
}

[JsonSerializable(typeof(ProbeFile))]
[JsonSerializable(typeof(AgentsCache))]
public sealed partial class ProbeConfigContext : JsonSerializerContext;

/// <summary><c>agents.json</c>：最近一次从面板同步到的 Agent 列表（配置，不是测量数据）；面板连不上时 Probe 靠它继续测量。</summary>
public sealed class AgentsCache
{
    [JsonPropertyName("version")]
    public long Version { get; set; }

    [JsonPropertyName("agents")]
    public List<KnownAgent> Agents { get; set; } = [];
}