// Copyright (c) Microsoft. All rights reserved.

using System.Collections.Generic;
using System.Text.Json.Serialization;
using System.Threading;
using System.Threading.Tasks;

namespace Microsoft.Agents.AI.UnitTests.AgentSkills;

/// <summary>
/// A simple in-memory <see cref="AgentSkill"/> implementation for unit tests.
/// </summary>
internal sealed class TestAgentSkill : AgentSkill
{
    private readonly AgentSkillFrontmatter _frontmatter;
    private readonly string _content;

    /// <summary>
    /// Initializes a new instance of the <see cref="TestAgentSkill"/> class.
    /// </summary>
    /// <param name="name">Kebab-case skill name.</param>
    /// <param name="description">Skill description.</param>
    /// <param name="content">Full skill content (body text).</param>
    public TestAgentSkill(string name, string description, string content)
    {
        this._frontmatter = new AgentSkillFrontmatter(name, description);
        this._content = content;
    }

    /// <inheritdoc/>
    public override AgentSkillFrontmatter Frontmatter => this._frontmatter;

    /// <inheritdoc/>
    public override ValueTask<string> GetContentAsync(CancellationToken cancellationToken = default) => new(this._content);
}

/// <summary>
/// A simple in-memory <see cref="AgentSkillsSource"/> implementation for unit tests.
/// </summary>
internal sealed class TestAgentSkillsSource : AgentSkillsSource
{
    private readonly IList<AgentSkill> _skills;

    /// <summary>
    /// Initializes a new instance of the <see cref="TestAgentSkillsSource"/> class.
    /// </summary>
    /// <param name="skills">The skills to return.</param>
    public TestAgentSkillsSource(IList<AgentSkill> skills)
    {
        this._skills = skills;
    }

    /// <summary>
    /// Initializes a new instance of the <see cref="TestAgentSkillsSource"/> class.
    /// </summary>
    /// <param name="skills">The skills to return.</param>
    public TestAgentSkillsSource(params AgentSkill[] skills)
    {
        this._skills = skills;
    }

    /// <inheritdoc/>
    public override Task<IList<AgentSkill>> GetSkillsAsync(AgentSkillsSourceContext context, CancellationToken cancellationToken = default)
    {
        return Task.FromResult(this._skills);
    }
}

/// <summary>
/// Custom input type accepted by skill script delegates in JSO tests.
/// </summary>
internal sealed class LookupRequest
{
    /// <summary>Gets or sets the search query.</summary>
    public string Query { get; set; } = string.Empty;

    /// <summary>Gets or sets the maximum number of results.</summary>
    public int MaxResults { get; set; }
}

/// <summary>
/// Custom output type returned by skill script delegates in JSO tests.
/// </summary>
internal sealed class LookupResponse
{
    /// <summary>Gets or sets the items found.</summary>
    public IList<string> Items { get; set; } = [];

    /// <summary>Gets or sets the total number of matches.</summary>
    public int TotalCount { get; set; }
}

/// <summary>
/// Custom output type returned by skill resource delegates in JSO tests.
/// </summary>
internal sealed class SkillConfig
{
    /// <summary>Gets or sets the theme name.</summary>
    public string Theme { get; set; } = string.Empty;

    /// <summary>Gets or sets whether verbose mode is enabled.</summary>
    public bool Verbose { get; set; }
}

/// <summary>
/// Source-generated JSON serializer context for skill test types.
/// Provides serialization support for <see cref="LookupRequest"/>, <see cref="LookupResponse"/>,
/// and <see cref="SkillConfig"/> without requiring runtime reflection.
/// </summary>
[JsonSourceGenerationOptions]
[JsonSerializable(typeof(LookupRequest))]
[JsonSerializable(typeof(LookupResponse))]
[JsonSerializable(typeof(SkillConfig))]
internal sealed partial class SkillTestJsonContext : JsonSerializerContext
{
}
