package initdb

import (
	"fmt"
	"testing"

	"github.com/glebarez/sqlite"
	"github.com/stretchr/testify/require"
	"gorm.io/gorm"
	"senlinai-agent/backend/internal/models"
)

func TestInitDatasetCreatesDefaultSourcesWhenTableIsEmpty(t *testing.T) {
	database := newDatasetInitTestDatabase(t)
	require.NoError(t, InitUser(database))

	require.NoError(t, InitDataset(database))
	require.NoError(t, InitDataset(database))

	var root models.SaUser
	require.NoError(t, database.Where("email = ?", rootUsername).First(&root).Error)
	var sources []models.SaDatasetSource
	require.NoError(t, database.Order("id asc").Find(&sources).Error)
	require.Len(t, sources, len(defaultDatasetSources))
	for index, expected := range defaultDatasetSources {
		require.Equal(t, expected.Name, sources[index].Name)
		require.Equal(t, expected.URL, sources[index].URL)
		require.Equal(t, expected.IconURL, sources[index].IconURL)
		require.Equal(t, "rss", sources[index].Kind)
		require.True(t, sources[index].Enabled)
		require.Equal(t, root.ID, sources[index].CreatedBy)
		require.Equal(t, root.Identity, sources[index].CreatedByIdentity)
	}
}

func TestInitDatasetDoesNothingWhenSourceTableIsNotEmpty(t *testing.T) {
	database := newDatasetInitTestDatabase(t)
	require.NoError(t, InitUser(database))
	var root models.SaUser
	require.NoError(t, database.Where("email = ?", rootUsername).First(&root).Error)
	existing := models.SaDatasetSource{
		CreatedBy: root.ID,
		Name:      "Existing source",
		Kind:      "manual",
		Enabled:   true,
	}
	require.NoError(t, database.Create(&existing).Error)

	require.NoError(t, InitDataset(database))

	var sources []models.SaDatasetSource
	require.NoError(t, database.Find(&sources).Error)
	require.Len(t, sources, 1)
	require.Equal(t, existing.Identity, sources[0].Identity)
}

func TestInitDatasetUsesFirstExistingUserWhenRootIsAbsent(t *testing.T) {
	database := newDatasetInitTestDatabase(t)
	user := models.SaUser{
		Email:        "existing@example.com",
		DisplayName:  "Existing",
		PasswordHash: "hash",
		Role:         "user",
	}
	require.NoError(t, database.Create(&user).Error)

	require.NoError(t, InitDataset(database))

	var sources []models.SaDatasetSource
	require.NoError(t, database.Find(&sources).Error)
	require.Len(t, sources, len(defaultDatasetSources))
	for _, source := range sources {
		require.Equal(t, user.ID, source.CreatedBy)
		require.Equal(t, user.Identity, source.CreatedByIdentity)
	}
}

func newDatasetInitTestDatabase(t *testing.T) *gorm.DB {
	t.Helper()
	database, err := gorm.Open(sqlite.Open(fmt.Sprintf("file:%s?mode=memory&cache=shared", t.Name())), &gorm.Config{})
	require.NoError(t, err)
	require.NoError(t, database.AutoMigrate(&models.SaUser{}, &models.SaDatasetSource{}))
	return database
}