using BotSharp.Abstraction.Conversations.Enums;

namespace BotSharp.Abstraction.Conversations.Models;

public class Conversation
{
    public string Id { get; set; } = string.Empty;
    public string AgentId { get; set; } = string.Empty;
    public string UserId { get; set; } = string.Empty;

    /// <summary>
    /// Agent task id
    /// </summary>
    public string? TaskId { get; set; }
    public string Title { get; set; } = string.Empty;

    [JsonIgnore]
    public List<DialogElement> Dialogs { get; set; } = new List<DialogElement>();

    [JsonIgnore]
    public Dictionary<string, string> States { get; set; } = new Dictionary<string, string>();

    public string Status { get; set; } = ConversationStatus.Open;

    public string Channel { get; set; } = ConversationChannel.OpenAPI;

    public int DialogCount { get; set; }

    public DateTime UpdatedTime { get; set; } = DateTime.UtcNow;
    public DateTime CreatedTime { get; set; } = DateTime.UtcNow;
}

public class DialogElement
{
    [JsonPropertyName("meta_data")]
    public DialogMetaData MetaData { get; set; }

    [JsonPropertyName("content")]
    public string Content { get; set; }

    [JsonPropertyName("secondary_content")]
    public string? SecondaryContent { get; set; }

    [JsonPropertyName("rich_content")]
    public string? RichContent { get; set; }

    [JsonPropertyName("secondary_rich_content")]
    public string? SecondaryRichContent { get; set; }

    [JsonPropertyName("payload")]
    public string? Payload { get; set; }

    public DialogElement()
    {

    }

    public DialogElement(DialogMetaData meta, string content, string? richContent = null,
        string? secondaryContent = null, string? secondaryRichContent = null, string? payload = null)
    {
        MetaData = meta;
        Content = content;
        RichContent = richContent;
        SecondaryContent = secondaryContent;
        SecondaryRichContent = secondaryRichContent;
        Payload = payload;
    }

    public override string ToString()
    {
        return $"{MetaData.Role}: {Content} [{MetaData?.CreateTime}]";
    }
}

public class DialogMetaData
{
    [JsonPropertyName("role")]
    public string Role { get; set; }

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

    [JsonPropertyName("message_id")]
    public string MessageId { get; set; }

    [JsonPropertyName("function_name")]
    public string? FunctionName { get; set; }

    [JsonPropertyName("sender_id")]
    public string? SenderId { get; set; }

    [JsonPropertyName("create_at")]
    public DateTime CreateTime { get; set; }
}