feat: add registry cleaning (#201)

Author: CodeReaper

internal/command/updater.go

@@ -52,6 +52,16 @@ func (cmd Updater) Run() error {
 		return err
 	}
 
+	if cmd.registry != nil {
+		validKeys := validSourceKeys(configurations)
+		for key := range cmd.registry.Sources {
+			if !validKeys[key] {
+				delete(cmd.registry.Sources, key)
+				fmt.Fprintf(cmd.Logger, "Removing stale registry entry: %s\n", key)
+			}
+		}
+	}
+
 	tmpDir, err := os.MkdirTemp("", "output")
 	if err != nil {
 		return fmt.Errorf("failed to create temp dir: %w", err)
@@ -154,6 +164,14 @@ func splitConfigurations(configurations []configuration.Configuration) []configu
 	return updated
 }
 
+func validSourceKeys(configs []configuration.Configuration) map[string]bool {
+	valid := make(map[string]bool)
+	for _, cfg := range splitConfigurations(configs) {
+		valid[cfg.Name] = true
+	}
+	return valid
+}
+
 // merge will move created files in generated into the schema repository,
 // and will attempt to write a file to schema repository if moving it fails
 func merge(generatedRepository, schemaRepository string) error {

internal/command/updater_test.go

@@ -100,6 +100,33 @@ func TestSplittingConfiguration(t *testing.T) {
 	}
 }
 
+func TestValidSourceKeys(t *testing.T) {
+	assert.Empty(t, validSourceKeys(nil))
+	assert.Empty(t, validSourceKeys([]configuration.Configuration{}))
+
+	configs := []configuration.Configuration{{Name: "foo"}}
+	keys := validSourceKeys(configs)
+	assert.True(t, keys["foo"])
+	assert.Len(t, keys, 1)
+
+	configs = []configuration.Configuration{{Name: "bar", Entries: []string{"a", "b"}}}
+	keys = validSourceKeys(configs)
+	assert.True(t, keys["bar.a"])
+	assert.True(t, keys["bar.b"])
+	assert.Len(t, keys, 2)
+
+	configs = []configuration.Configuration{
+		{Name: "single"},
+		{Name: "multi", Entries: []string{"x", "y", "z"}},
+	}
+	keys = validSourceKeys(configs)
+	assert.True(t, keys["single"])
+	assert.True(t, keys["multi.x"])
+	assert.True(t, keys["multi.y"])
+	assert.True(t, keys["multi.z"])
+	assert.Len(t, keys, 4)
+}
+
 func TestReadConfiguration(t *testing.T) {
 	_, err := readConfiguration("testdata/does-not.exist")
 	assert.NotNil(t, err)
@@ -221,6 +248,41 @@ func TestRunWithRegistrySavesUpdates(t *testing.T) {
 	assert.Contains(t, string(content), "1.0.0")
 }
 
+func TestRunWithRegistryRemovesStaleEntries(t *testing.T) {
+	server, config, tmpDir := setup(t)
+	defer server.Close()
+
+	registryPath := path.Join(tmpDir, "registry.yaml")
+	initialContent := `sources:
+  http:
+    kind: http
+    version: 1.0.0
+    lastUpdated: "2026-01-01T00:00:00Z"
+  stale1:
+    kind: helm
+    version: v1.0.0
+    lastUpdated: "2026-01-01T00:00:00Z"
+  stale2:
+    kind: git
+    version: v1.0.0
+    lastUpdated: "2026-01-01T00:00:00Z"
+`
+	os.WriteFile(registryPath, []byte(initialContent), 0664)
+
+	updater := NewUpdater(config, tmpDir, tmpDir, registryPath, bytes.NewBuffer([]byte{}), nil)
+
+	err := updater.Run()
+	assert.Nil(t, err)
+
+	content, err := os.ReadFile(registryPath)
+	assert.Nil(t, err)
+
+	assert.Contains(t, string(content), "http")
+	assert.Contains(t, string(content), "1.0.0")
+	assert.NotContains(t, string(content), "stale1")
+	assert.NotContains(t, string(content), "stale2")
+}
+
 // Test is skipped during unit testing and meant for step-debugging a local check
 func TestCheckLocal(t *testing.T) {
 	output := "../../build/ephemeral/schema"