using System;
using System.Threading;
using Azure.AI.OpenAI;
using Azure.AI.OpenAI.Chat;
using OpenAI;
using OpenAI.Assistants;
using OpenAI.Chat;

#pragma warning disable OPENAI001 // Type is for evaluation purposes only and is subject to change or removal in future updates. Suppress this diagnostic to proceed.

public class OpenAIAssistantsClass
{
    public AssistantThread? Thread;

    public OpenAIAssistantsClass(OpenAIClient client, string assistantId)
    {
        _assistantClient = client.GetAssistantClient();
        _assistantId = assistantId;
    }

    public async Task CreateThreadAsync()
    {
        var result = await _assistantClient.CreateThreadAsync();
        Thread = result.Value;
    }

    public async Task RetrieveThreadAsync(string threadId)
    {
        var result = await _assistantClient.GetThreadAsync(threadId);
        Thread = result.Value;
    }

    public async Task GetThreadMessagesAsync(Action<string, string> callback)
    {
        var options = new MessageCollectionOptions() { Order = ListOrder.OldestFirst };
        await foreach (var message in _assistantClient.GetMessagesAsync(Thread, options).GetAllValuesAsync())
        {
            var content = string.Join("", message.Content.Select(c => c.Text));
            var role = message.Role == MessageRole.User ? "user" : "assistant";
            callback(role, content);
        }
    }

    public async Task<string> GetResponseAsync(string userInput)
    {
        await _assistantClient.CreateMessageAsync(Thread, MessageRole.User, [ userInput ]);
        var assistant = await _assistantClient.GetAssistantAsync(_assistantId);

        var result = await _assistantClient.CreateRunAsync(Thread, assistant);
        var run = result.Value;

        while (!run.Status.IsTerminal)
        {
            System.Threading.Thread.Sleep(TimeSpan.FromMilliseconds(100));
            result = _assistantClient.GetRun(run.ThreadId, run.Id);
            run = result.Value;
        }

        var options = new MessageCollectionOptions() { Order = ListOrder.OldestFirst };
        await foreach (var message in _assistantClient.GetMessagesAsync(run.ThreadId, options).GetAllValuesAsync())
        {
            if (message.Role == MessageRole.Assistant)
            {
                var content = string.Join("", message.Content.Select(c => c.Text));
                return content;
            }
        }

        return string.Empty;
    }

    private readonly string _assistantId;
    private readonly AssistantClient _assistantClient;
}