go/pkg/repos/registry_test.go

package repos

import (
	"testing"

	"forge.lthn.ai/core/go/pkg/io"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

// ── LoadRegistry ───────────────────────────────────────────────────

func TestLoadRegistry_Good(t *testing.T) {
	m := io.NewMockMedium()
	yaml := `
version: 1
org: host-uk
base_path: /tmp/repos
repos:
  core:
    type: foundation
    description: Core package
`
	_ = m.Write("/tmp/repos.yaml", yaml)

	reg, err := LoadRegistry(m, "/tmp/repos.yaml")
	assert.NoError(t, err)
	assert.NotNil(t, reg)
	assert.Equal(t, "host-uk", reg.Org)
	assert.Equal(t, "/tmp/repos", reg.BasePath)
	assert.Equal(t, m, reg.medium)

	repo, ok := reg.Get("core")
	assert.True(t, ok)
	assert.Equal(t, "core", repo.Name)
	assert.Equal(t, "/tmp/repos/core", repo.Path)
	assert.Equal(t, reg, repo.registry)
}

func TestLoadRegistry_Good_WithDefaults(t *testing.T) {
	m := io.NewMockMedium()
	yaml := `
version: 1
org: host-uk
base_path: /tmp/repos
defaults:
  ci: github-actions
  license: EUPL-1.2
  branch: main
repos:
  core-php:
    type: foundation
    description: Foundation
  core-admin:
    type: module
    description: Admin panel
`
	_ = m.Write("/tmp/repos.yaml", yaml)

	reg, err := LoadRegistry(m, "/tmp/repos.yaml")
	require.NoError(t, err)

	php, ok := reg.Get("core-php")
	require.True(t, ok)
	assert.Equal(t, "github-actions", php.CI)

	admin, ok := reg.Get("core-admin")
	require.True(t, ok)
	assert.Equal(t, "github-actions", admin.CI)
}

func TestLoadRegistry_Good_CustomRepoPath(t *testing.T) {
	m := io.NewMockMedium()
	yaml := `
version: 1
org: host-uk
base_path: /tmp/repos
repos:
  special:
    type: module
    path: /opt/special-repo
`
	_ = m.Write("/tmp/repos.yaml", yaml)

	reg, err := LoadRegistry(m, "/tmp/repos.yaml")
	require.NoError(t, err)

	repo, ok := reg.Get("special")
	require.True(t, ok)
	assert.Equal(t, "/opt/special-repo", repo.Path)
}

func TestLoadRegistry_Good_CIOverride(t *testing.T) {
	m := io.NewMockMedium()
	yaml := `
version: 1
org: test
base_path: /tmp
defaults:
  ci: default-ci
repos:
  a:
    type: module
  b:
    type: module
    ci: custom-ci
`
	_ = m.Write("/tmp/repos.yaml", yaml)

	reg, err := LoadRegistry(m, "/tmp/repos.yaml")
	require.NoError(t, err)

	a, _ := reg.Get("a")
	assert.Equal(t, "default-ci", a.CI)

	b, _ := reg.Get("b")
	assert.Equal(t, "custom-ci", b.CI)
}

func TestLoadRegistry_Bad_FileNotFound(t *testing.T) {
	m := io.NewMockMedium()
	_, err := LoadRegistry(m, "/nonexistent/repos.yaml")
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "failed to read")
}

func TestLoadRegistry_Bad_InvalidYAML(t *testing.T) {
	m := io.NewMockMedium()
	_ = m.Write("/tmp/bad.yaml", "{{{{not yaml at all")

	_, err := LoadRegistry(m, "/tmp/bad.yaml")
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "failed to parse")
}

// ── List / Get / ByType ────────────────────────────────────────────

func newTestRegistry(t *testing.T) *Registry {
	t.Helper()
	m := io.NewMockMedium()
	yaml := `
version: 1
org: host-uk
base_path: /tmp/repos
repos:
  core-php:
    type: foundation
    description: Foundation
  core-admin:
    type: module
    depends_on: [core-php]
    description: Admin
  core-tenant:
    type: module
    depends_on: [core-php]
    description: Tenancy
  core-bio:
    type: product
    depends_on: [core-php, core-tenant]
    description: Bio product
`
	_ = m.Write("/tmp/repos.yaml", yaml)
	reg, err := LoadRegistry(m, "/tmp/repos.yaml")
	require.NoError(t, err)
	return reg
}

func TestRegistry_List_Good(t *testing.T) {
	reg := newTestRegistry(t)
	repos := reg.List()
	assert.Len(t, repos, 4)
}

func TestRegistry_Get_Good(t *testing.T) {
	reg := newTestRegistry(t)
	repo, ok := reg.Get("core-php")
	assert.True(t, ok)
	assert.Equal(t, "core-php", repo.Name)
}

func TestRegistry_Get_Bad_NotFound(t *testing.T) {
	reg := newTestRegistry(t)
	_, ok := reg.Get("nonexistent")
	assert.False(t, ok)
}

func TestRegistry_ByType_Good(t *testing.T) {
	reg := newTestRegistry(t)

	foundations := reg.ByType("foundation")
	assert.Len(t, foundations, 1)
	assert.Equal(t, "core-php", foundations[0].Name)

	modules := reg.ByType("module")
	assert.Len(t, modules, 2)

	products := reg.ByType("product")
	assert.Len(t, products, 1)
}

func TestRegistry_ByType_Good_NoMatch(t *testing.T) {
	reg := newTestRegistry(t)
	templates := reg.ByType("template")
	assert.Empty(t, templates)
}

// ── TopologicalOrder ───────────────────────────────────────────────

func TestTopologicalOrder_Good(t *testing.T) {
	reg := newTestRegistry(t)
	order, err := TopologicalOrder(reg)
	require.NoError(t, err)
	assert.Len(t, order, 4)

	// core-php must come before everything that depends on it.
	phpIdx := -1
	for i, r := range order {
		if r.Name == "core-php" {
			phpIdx = i
			break
		}
	}
	require.GreaterOrEqual(t, phpIdx, 0, "core-php not found")

	for i, r := range order {
		for _, dep := range r.DependsOn {
			depIdx := -1
			for j, d := range order {
				if d.Name == dep {
					depIdx = j
					break
				}
			}
			assert.Less(t, depIdx, i, "%s should come before %s", dep, r.Name)
		}
	}
}

func TopologicalOrder(reg *Registry) ([]*Repo, error) {
	return reg.TopologicalOrder()
}

func TestTopologicalOrder_Bad_CircularDep(t *testing.T) {
	m := io.NewMockMedium()
	yaml := `
version: 1
org: test
base_path: /tmp
repos:
  a:
    type: module
    depends_on: [b]
  b:
    type: module
    depends_on: [a]
`
	_ = m.Write("/tmp/repos.yaml", yaml)
	reg, err := LoadRegistry(m, "/tmp/repos.yaml")
	require.NoError(t, err)

	_, err = reg.TopologicalOrder()
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "circular dependency")
}

func TestTopologicalOrder_Bad_UnknownDep(t *testing.T) {
	m := io.NewMockMedium()
	yaml := `
version: 1
org: test
base_path: /tmp
repos:
  a:
    type: module
    depends_on: [nonexistent]
`
	_ = m.Write("/tmp/repos.yaml", yaml)
	reg, err := LoadRegistry(m, "/tmp/repos.yaml")
	require.NoError(t, err)

	_, err = reg.TopologicalOrder()
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "unknown repo")
}

func TestTopologicalOrder_Good_NoDeps(t *testing.T) {
	m := io.NewMockMedium()
	yaml := `
version: 1
org: test
base_path: /tmp
repos:
  a:
    type: module
  b:
    type: module
`
	_ = m.Write("/tmp/repos.yaml", yaml)
	reg, err := LoadRegistry(m, "/tmp/repos.yaml")
	require.NoError(t, err)

	order, err := reg.TopologicalOrder()
	require.NoError(t, err)
	assert.Len(t, order, 2)
}

// ── ScanDirectory ──────────────────────────────────────────────────

func TestScanDirectory_Good(t *testing.T) {
	m := io.NewMockMedium()

	// Create mock repos with .git dirs.
	_ = m.EnsureDir("/workspace/repo-a/.git")
	_ = m.EnsureDir("/workspace/repo-b/.git")
	_ = m.EnsureDir("/workspace/not-a-repo") // No .git

	// Write a file (not a dir) at top level.
	_ = m.Write("/workspace/README.md", "hello")

	reg, err := ScanDirectory(m, "/workspace")
	require.NoError(t, err)

	assert.Len(t, reg.Repos, 2)

	a, ok := reg.Repos["repo-a"]
	assert.True(t, ok)
	assert.Equal(t, "/workspace/repo-a", a.Path)
	assert.Equal(t, "module", a.Type) // Default type.

	_, ok = reg.Repos["not-a-repo"]
	assert.False(t, ok)
}

func TestScanDirectory_Good_DetectsGitHubOrg(t *testing.T) {
	m := io.NewMockMedium()

	_ = m.EnsureDir("/workspace/my-repo/.git")
	_ = m.Write("/workspace/my-repo/.git/config", `[core]
	repositoryformatversion = 0
[remote "origin"]
	url = git@github.com:host-uk/my-repo.git
	fetch = +refs/heads/*:refs/remotes/origin/*
`)

	reg, err := ScanDirectory(m, "/workspace")
	require.NoError(t, err)
	assert.Equal(t, "host-uk", reg.Org)
}

func TestScanDirectory_Good_DetectsHTTPSOrg(t *testing.T) {
	m := io.NewMockMedium()

	_ = m.EnsureDir("/workspace/my-repo/.git")
	_ = m.Write("/workspace/my-repo/.git/config", `[remote "origin"]
	url = https://github.com/lethean-io/my-repo.git
`)

	reg, err := ScanDirectory(m, "/workspace")
	require.NoError(t, err)
	assert.Equal(t, "lethean-io", reg.Org)
}

func TestScanDirectory_Good_EmptyDir(t *testing.T) {
	m := io.NewMockMedium()
	_ = m.EnsureDir("/empty")

	reg, err := ScanDirectory(m, "/empty")
	require.NoError(t, err)
	assert.Empty(t, reg.Repos)
	assert.Equal(t, "", reg.Org)
}

func TestScanDirectory_Bad_InvalidDir(t *testing.T) {
	m := io.NewMockMedium()
	_, err := ScanDirectory(m, "/nonexistent")
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "failed to read directory")
}

// ── detectOrg ──────────────────────────────────────────────────────

func TestDetectOrg_Good_SSHRemote(t *testing.T) {
	m := io.NewMockMedium()
	_ = m.Write("/repo/.git/config", `[remote "origin"]
	url = git@github.com:host-uk/core.git
`)
	assert.Equal(t, "host-uk", detectOrg(m, "/repo"))
}

func TestDetectOrg_Good_HTTPSRemote(t *testing.T) {
	m := io.NewMockMedium()
	_ = m.Write("/repo/.git/config", `[remote "origin"]
	url = https://github.com/snider/project.git
`)
	assert.Equal(t, "snider", detectOrg(m, "/repo"))
}

func TestDetectOrg_Bad_NoConfig(t *testing.T) {
	m := io.NewMockMedium()
	assert.Equal(t, "", detectOrg(m, "/nonexistent"))
}

func TestDetectOrg_Bad_NoRemote(t *testing.T) {
	m := io.NewMockMedium()
	_ = m.Write("/repo/.git/config", `[core]
	repositoryformatversion = 0
`)
	assert.Equal(t, "", detectOrg(m, "/repo"))
}

func TestDetectOrg_Bad_NonGitHubRemote(t *testing.T) {
	m := io.NewMockMedium()
	_ = m.Write("/repo/.git/config", `[remote "origin"]
	url = ssh://git@forge.lthn.ai:2223/core/go.git
`)
	assert.Equal(t, "", detectOrg(m, "/repo"))
}

// ── expandPath ─────────────────────────────────────────────────────

func TestExpandPath_Good_Tilde(t *testing.T) {
	got := expandPath("~/Code/repos")
	assert.NotContains(t, got, "~")
	assert.Contains(t, got, "Code/repos")
}

func TestExpandPath_Good_NoTilde(t *testing.T) {
	assert.Equal(t, "/absolute/path", expandPath("/absolute/path"))
	assert.Equal(t, "relative/path", expandPath("relative/path"))
}

// ── Repo.Exists / IsGitRepo ───────────────────────────────────────

func TestRepo_Exists_Good(t *testing.T) {
	m := io.NewMockMedium()
	reg := &Registry{
		medium:   m,
		BasePath: "/tmp/repos",
		Repos:    make(map[string]*Repo),
	}
	repo := &Repo{
		Name:     "core",
		Path:     "/tmp/repos/core",
		registry: reg,
	}

	assert.False(t, repo.Exists())

	_ = m.EnsureDir("/tmp/repos/core")
	assert.True(t, repo.Exists())
}

func TestRepo_IsGitRepo_Good(t *testing.T) {
	m := io.NewMockMedium()
	reg := &Registry{
		medium:   m,
		BasePath: "/tmp/repos",
		Repos:    make(map[string]*Repo),
	}
	repo := &Repo{
		Name:     "core",
		Path:     "/tmp/repos/core",
		registry: reg,
	}

	assert.False(t, repo.IsGitRepo())

	_ = m.EnsureDir("/tmp/repos/core/.git")
	assert.True(t, repo.IsGitRepo())
}

// ── getMedium fallback ─────────────────────────────────────────────

func TestGetMedium_Good_FallbackToLocal(t *testing.T) {
	repo := &Repo{Name: "orphan", Path: "/tmp/orphan"}
	// No registry set — should fall back to io.Local.
	m := repo.getMedium()
	assert.Equal(t, io.Local, m)
}

func TestGetMedium_Good_NilMediumFallback(t *testing.T) {
	reg := &Registry{} // medium is nil.
	repo := &Repo{Name: "test", registry: reg}
	m := repo.getMedium()
	assert.Equal(t, io.Local, m)
}