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

using System.Collections.Generic;
using Microsoft.Extensions.AI;

namespace Microsoft.Agents.AI.Workflows;

/// <summary>
/// Provides extension methods for treating workflows as <see cref="AIAgent"/>
/// </summary>
public static class WorkflowHostingExtensions
{
    /// <summary>
    /// Convert a workflow with the appropriate primary input type to an <see cref="AIAgent"/>.
    /// </summary>
    /// <param name="workflow">The workflow to be hosted by the resulting <see cref="AIAgent"/></param>
    /// <param name="id">A unique id for the hosting <see cref="AIAgent"/>.</param>
    /// <param name="name">A name for the hosting <see cref="AIAgent"/>.</param>
    /// <param name="description">A description for the hosting <see cref="AIAgent"/>.</param>
    /// <param name="checkpointManager">A <see cref="CheckpointManager"/> to enable persistence of run state.</param>
    /// <param name="executionEnvironment">Specify the execution environment to use when running the workflows. See
    /// <see cref="InProcessExecution.OffThread"/>, <see cref="InProcessExecution.Concurrent"/> and
    /// <see cref="InProcessExecution.Lockstep"/> for the in-process environments.</param>
    /// <param name="includeExceptionDetails">If <see langword="true"/>, will include <see cref="System.Exception.Message"/>
    /// in the <see cref="ErrorContent"/> representing the workflow error.</param>
    /// <param name="includeWorkflowOutputsInResponse">If <see langword="true"/>, will transform outgoing workflow outputs
    /// into into content in <see cref="AgentResponseUpdate"/>s or the <see cref="AgentResponse"/> as appropriate.</param>
    /// <returns></returns>
    public static AIAgent AsAgent(
        this Workflow workflow,
        string? id = null,
        string? name = null,
        string? description = null,
        CheckpointManager? checkpointManager = null,
        IWorkflowExecutionEnvironment? executionEnvironment = null,
        bool includeExceptionDetails = false,
        bool includeWorkflowOutputsInResponse = false)
    {
        return new WorkflowHostAgent(workflow, id, name, description, checkpointManager, executionEnvironment, includeExceptionDetails, includeWorkflowOutputsInResponse);
    }

    internal static FunctionCallContent ToFunctionCall(this ExternalRequest request)
    {
        Dictionary<string, object?> parameters = new()
        {
            { "data", request.Data}
        };

        return new FunctionCallContent(request.RequestId, request.PortInfo.PortId, parameters);
    }
}