/* foundry-test-manager.c
 *
 * Copyright 2025 Christian Hergert <chergert@redhat.com>
 *
 * This library is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of the
 * License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 * SPDX-License-Identifier: LGPL-2.1-or-later
 */

#include "config.h"

#include <libpeas.h>

#include "foundry-debug.h"
#include "foundry-model-manager.h"
#include "foundry-test.h"
#include "foundry-test-manager.h"
#include "foundry-test-provider-private.h"
#include "foundry-test-suite-private.h"
#include "foundry-service-private.h"
#include "foundry-util-private.h"

/**
 * FoundryTestManager:
 *
 * Manages plugins providing test providers and their tests.
 */

struct _FoundryTestManager
{
  FoundryService    parent_instance;
  PeasExtensionSet *addins;
};

struct _FoundryTestManagerClass
{
  FoundryServiceClass parent_class;
};

static void list_model_iface_init (GListModelInterface *iface);

G_DEFINE_FINAL_TYPE_WITH_CODE (FoundryTestManager, foundry_test_manager, FOUNDRY_TYPE_SERVICE,
                               G_IMPLEMENT_INTERFACE (G_TYPE_LIST_MODEL, list_model_iface_init))

static void
foundry_test_manager_provider_added (PeasExtensionSet *set,
                                     PeasPluginInfo   *plugin_info,
                                     GObject          *addin,
                                     gpointer          user_data)
{
  FoundryTestManager *self = user_data;

  g_assert (PEAS_IS_EXTENSION_SET (set));
  g_assert (PEAS_IS_PLUGIN_INFO (plugin_info));
  g_assert (FOUNDRY_IS_TEST_PROVIDER (addin));
  g_assert (FOUNDRY_IS_TEST_MANAGER (self));

  g_debug ("Adding FoundryTestProvider of type %s", G_OBJECT_TYPE_NAME (addin));

  dex_future_disown (foundry_test_provider_load (FOUNDRY_TEST_PROVIDER (addin)));
}

static void
foundry_test_manager_provider_removed (PeasExtensionSet *set,
                                       PeasPluginInfo   *plugin_info,
                                       GObject          *addin,
                                       gpointer          user_data)
{
  FoundryTestManager *self = user_data;

  g_assert (PEAS_IS_EXTENSION_SET (set));
  g_assert (PEAS_IS_PLUGIN_INFO (plugin_info));
  g_assert (FOUNDRY_IS_TEST_PROVIDER (addin));
  g_assert (FOUNDRY_IS_TEST_MANAGER (self));

  g_debug ("Removing FoundryTestProvider of type %s", G_OBJECT_TYPE_NAME (addin));

  dex_future_disown (foundry_test_provider_unload (FOUNDRY_TEST_PROVIDER (addin)));
}

static DexFuture *
foundry_test_manager_start_fiber (gpointer user_data)
{
  FoundryTestManager *self = user_data;
  g_autoptr(GPtrArray) futures = NULL;
  guint n_items;

  g_assert (FOUNDRY_IS_MAIN_THREAD ());
  g_assert (FOUNDRY_IS_TEST_MANAGER (self));
  g_assert (PEAS_IS_EXTENSION_SET (self->addins));

  g_signal_connect_object (self->addins,
                           "extension-added",
                           G_CALLBACK (foundry_test_manager_provider_added),
                           self,
                           0);
  g_signal_connect_object (self->addins,
                           "extension-removed",
                           G_CALLBACK (foundry_test_manager_provider_removed),
                           self,
                           0);
  g_signal_connect_object (self->addins,
                           "items-changed",
                           G_CALLBACK (g_list_model_items_changed),
                           self,
                           G_CONNECT_SWAPPED);

  n_items = g_list_model_get_n_items (G_LIST_MODEL (self->addins));
  futures = g_ptr_array_new_with_free_func (dex_unref);

  for (guint i = 0; i < n_items; i++)
    {
      g_autoptr(FoundryTestProvider) provider = g_list_model_get_item (G_LIST_MODEL (self->addins), i);
      g_ptr_array_add (futures, foundry_test_provider_load (provider));
    }

  g_list_model_items_changed (G_LIST_MODEL (self), 0, 0, n_items);

  if (futures->len > 0)
    dex_await (foundry_future_all (futures), NULL);

  return dex_future_new_true ();
}

static DexFuture *
foundry_test_manager_start (FoundryService *service)
{
  FoundryTestManager *self = (FoundryTestManager *)service;

  g_assert (FOUNDRY_IS_MAIN_THREAD ());
  g_assert (FOUNDRY_IS_TEST_MANAGER (self));
  g_assert (PEAS_IS_EXTENSION_SET (self->addins));

  return dex_scheduler_spawn (NULL, 0,
                              foundry_test_manager_start_fiber,
                              g_object_ref (self),
                              g_object_unref);
}

static DexFuture *
foundry_test_manager_stop (FoundryService *service)
{
  FoundryTestManager *self = (FoundryTestManager *)service;
  g_autoptr(GPtrArray) futures = NULL;
  guint n_items;

  g_assert (FOUNDRY_IS_MAIN_THREAD ());
  g_assert (FOUNDRY_IS_SERVICE (service));

  g_signal_handlers_disconnect_by_func (self->addins,
                                        G_CALLBACK (foundry_test_manager_provider_added),
                                        self);
  g_signal_handlers_disconnect_by_func (self->addins,
                                        G_CALLBACK (foundry_test_manager_provider_removed),
                                        self);

  n_items = g_list_model_get_n_items (G_LIST_MODEL (self->addins));
  futures = g_ptr_array_new_with_free_func (dex_unref);

  for (guint i = 0; i < n_items; i++)
    {
      g_autoptr(FoundryTestProvider) provider = g_list_model_get_item (G_LIST_MODEL (self->addins), i);
      g_ptr_array_add (futures, foundry_test_provider_unload (provider));
    }

  g_list_model_items_changed (G_LIST_MODEL (self), 0, n_items, 0);

  g_clear_object (&self->addins);

  if (futures->len > 0)
    return foundry_future_all (futures);

  return dex_future_new_true ();
}

static void
foundry_test_manager_constructed (GObject *object)
{
  FoundryTestManager *self = (FoundryTestManager *)object;
  g_autoptr(FoundryContext) context = NULL;

  G_OBJECT_CLASS (foundry_test_manager_parent_class)->constructed (object);

  context = foundry_contextual_dup_context (FOUNDRY_CONTEXTUAL (self));

  self->addins = peas_extension_set_new (NULL,
                                         FOUNDRY_TYPE_TEST_PROVIDER,
                                         "context", context,
                                         NULL);
}

static void
foundry_test_manager_finalize (GObject *object)
{
  FoundryTestManager *self = (FoundryTestManager *)object;

  g_clear_object (&self->addins);

  G_OBJECT_CLASS (foundry_test_manager_parent_class)->finalize (object);
}

static void
foundry_test_manager_class_init (FoundryTestManagerClass *klass)
{
  GObjectClass *object_class = G_OBJECT_CLASS (klass);
  FoundryServiceClass *service_class = FOUNDRY_SERVICE_CLASS (klass);

  object_class->constructed = foundry_test_manager_constructed;
  object_class->finalize = foundry_test_manager_finalize;

  service_class->start = foundry_test_manager_start;
  service_class->stop = foundry_test_manager_stop;
}

static void
foundry_test_manager_init (FoundryTestManager *self)
{
}

static GType
foundry_test_manager_get_item_type (GListModel *model)
{
  return FOUNDRY_TYPE_TEST_PROVIDER;
}

static guint
foundry_test_manager_get_n_items (GListModel *model)
{
  FoundryTestManager *self = FOUNDRY_TEST_MANAGER (model);

  if (self->addins == NULL)
    return 0;

  return g_list_model_get_n_items (G_LIST_MODEL (self->addins));
}

static gpointer
foundry_test_manager_get_item (GListModel *model,
                               guint       position)
{
  FoundryTestManager *self = FOUNDRY_TEST_MANAGER (model);

  if (self->addins == NULL)
    return NULL;

  return g_list_model_get_item (G_LIST_MODEL (self->addins), position);
}

static void
list_model_iface_init (GListModelInterface *iface)
{
  iface->get_item_type = foundry_test_manager_get_item_type;
  iface->get_n_items = foundry_test_manager_get_n_items;
  iface->get_item = foundry_test_manager_get_item;
}

/**
 * foundry_test_manager_list_tests:
 * @self: a [class@Foundry.TestManager]
 *
 * Queries all [class@Foundry.TestProvider] for available unit tests.
 *
 * The resulting module may not be fully populated by all providers
 * by time it resolves. You may await the completion of all providers
 * by awaiting [func@Foundry.list_model_await] for the completion
 * of all providers.
 *
 * This allows the consumer to get a dynamically populating list model
 * for user interfaces without delay.
 *
 * Returns: (transfer full): a [class@Dex.Future] that resolves to a
 *   [iface@Gio.ListModel] of [class@Foundry.Test]
 */
DexFuture *
foundry_test_manager_list_tests (FoundryTestManager *self)
{
  g_autoptr(GPtrArray) futures = NULL;
  guint n_items;

  dex_return_error_if_fail (FOUNDRY_IS_TEST_MANAGER (self));

  if (self->addins == NULL)
    return foundry_future_new_not_supported ();

  futures = g_ptr_array_new_with_free_func (dex_unref);
  n_items = g_list_model_get_n_items (G_LIST_MODEL (self->addins));

  for (guint i = 0; i < n_items; i++)
    {
      g_autoptr(FoundryTestProvider) provider = g_list_model_get_item (G_LIST_MODEL (self->addins), i);
      g_ptr_array_add (futures, foundry_test_provider_list_tests (provider));
    }

  return _foundry_flatten_list_model_new_from_futures (futures);
}

static DexFuture *
foundry_test_manager_filter_test (DexFuture *completed,
                                  gpointer   data)
{
  const char *id = data;
  g_autoptr(GListModel) tests = NULL;
  guint n_items;

  g_assert (DEX_IS_FUTURE (completed));
  g_assert (id != NULL);

  if (!(tests = dex_await_object (dex_ref (completed), NULL)))
    return NULL;

  n_items = g_list_model_get_n_items (tests);

  for (guint i = 0; i < n_items; i++)
    {
      g_autoptr(FoundryTest) test = g_list_model_get_item (tests, i);
      g_autofree char *test_id = foundry_test_dup_id (test);

      if (foundry_str_equal0 (test_id, id))
        return dex_future_new_take_object (g_steal_pointer (&test));
    }

  return dex_future_new_reject (G_IO_ERROR,
                                G_IO_ERROR_NOT_FOUND,
                                "No such test `%s`", id);
}

static DexFuture *
foundry_test_manager_finish_cb (DexFuture *completed,
                                gpointer   data)
{
  g_autoptr(GListModel) tests = dex_await_object (dex_ref (completed), NULL);

  return dex_future_then (foundry_list_model_await (tests),
                          foundry_future_return_object,
                          g_object_ref (tests),
                          g_object_unref);
}

/**
 * foundry_test_manager_find_test:
 * @self: a [class@Foundry.TestManager]
 * @test_id: the identifier of the test
 *
 * Returns: (transfer full): a [class@Dex.Future] that resolves to a
 *   [class@Foundry.Test] or rejects with error.
 */
DexFuture *
foundry_test_manager_find_test (FoundryTestManager *self,
                                const char         *test_id)
{
  dex_return_error_if_fail (FOUNDRY_IS_TEST_MANAGER (self));
  dex_return_error_if_fail (test_id != NULL);

  return dex_future_then (dex_future_then (foundry_test_manager_list_tests (self),
                                           foundry_test_manager_finish_cb,
                                           NULL, NULL),
                          foundry_test_manager_filter_test,
                          g_strdup (test_id),
                          g_free);
}

static void
add_to_suite (GListStore  *suites,
              GHashTable  *name_to_suite,
              const char  *name,
              FoundryTest *test)
{
  FoundryTestSuite *suite;

  g_assert (G_IS_LIST_STORE (suites));
  g_assert (name_to_suite != NULL);
  g_assert (name != NULL);
  g_assert (FOUNDRY_IS_TEST (test));

  if (!(suite = g_hash_table_lookup (name_to_suite, name)))
    {
      suite = _foundry_test_suite_new (name);
      g_hash_table_replace (name_to_suite, g_strdup (name), suite);
      g_list_store_append (suites, suite);
    }

  _foundry_test_suite_add (suite, test);
}

static DexFuture *
foundry_test_manager_list_suites_fiber (gpointer user_data)
{
  FoundryTestManager *self = user_data;
  g_autoptr(GListModel) tests = NULL;
  g_autoptr(GListStore) suites = NULL;
  g_autoptr(GHashTable) name_to_suite = NULL;
  g_autoptr(GError) error = NULL;
  guint n_tests;

  g_assert (FOUNDRY_IS_TEST_MANAGER (self));

  if (!(tests = dex_await_object (foundry_test_manager_list_tests (self), &error)))
    return dex_future_new_for_error (g_steal_pointer (&error));

  dex_await (foundry_list_model_await (tests), NULL);

  name_to_suite = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, g_object_unref);
  suites = g_list_store_new (FOUNDRY_TYPE_TEST_SUITE);
  n_tests = g_list_model_get_n_items (tests);

  for (guint i = 0; i < n_tests; i++)
    {
      g_autoptr(FoundryTest) test = g_list_model_get_item (tests, i);
      g_auto(GStrv) suite_names = foundry_test_dup_suites (test);

      if (suite_names == NULL || suite_names[0] == NULL)
        {
          add_to_suite (suites, name_to_suite, "", test);
          continue;
        }

      for (guint j = 0; suite_names[j]; j++)
        add_to_suite (suites, name_to_suite, suite_names[j], test);
    }

  return dex_future_new_take_object (g_steal_pointer (&suites));
}

/**
 * foundry_test_manager_list_suites:
 * @self: a [class@Foundry.TestManager]
 *
 * Loads available tests and groups them within a [class@Foundry.TestSuite].
 *
 * Returns: (transfer full): a [class@Dex.Future] that resolves to a
 *   [iface@Gio.ListModel] of [class@Foundry.TestSuite].
 *
 * Since: 1.1
 */
DexFuture *
foundry_test_manager_list_suites (FoundryTestManager *self)
{
  dex_return_error_if_fail (FOUNDRY_IS_TEST_MANAGER (self));

  return dex_scheduler_spawn (NULL, 0,
                              foundry_test_manager_list_suites_fiber,
                              g_object_ref (self),
                              g_object_unref);
}