#include "mcp/tools/common/context.h"

#include <stdlib.h>

#include "base/allocator.h"

/* mcpc's callback API has no user-data parameter. The stdio server is single-instance. */
static tiny_mcp_tools_context_t *active_context;

tiny_search_index_t *
tiny_mcp_tools_build_search_index(const tiny_store_t *store)
{
  tiny_search_index_t *index = tiny_search_index_create();
  size_t position;
  if (!index)
    {
      return NULL;
    }
  for (position = 0; position < tiny_store_entry_count(store); ++position)
    {
      if (!tiny_search_index_add(index, &tiny_store_entry(store, position)->message, position))
        {
          tiny_search_index_free(index);
          return NULL;
        }
    }
  return index;
}

tiny_mcp_tools_context_t *
tiny_mcp_tools_context_new(tiny_store_t *store)
{
  tiny_mcp_tools_context_t *context;
  if (!store)
    {
      return NULL;
    }
  context = malloc(sizeof(*context));
  if (!context)
    {
      return NULL;
    }
  context->store = store;
  context->search_index = tiny_mcp_tools_build_search_index(store);
  if (!context->search_index)
    {
      free(context);
      return NULL;
    }
  return context;
}

void
tiny_mcp_tools_context_free(tiny_mcp_tools_context_t *context)
{
  if (!context)
    {
      return;
    }
  tiny_search_index_free(context->search_index);
  free(context);
}

bool
tiny_mcp_tools_context_activate(tiny_mcp_tools_context_t *context)
{
  if (!context || active_context)
    {
      return false;
    }
  active_context = context;
  return true;
}

void
tiny_mcp_tools_context_deactivate(tiny_mcp_tools_context_t *context)
{
  if (active_context == context)
    {
      active_context = NULL;
    }
}

tiny_mcp_tools_context_t *
tiny_mcp_tools_context_current(void)
{
  return active_context;
}