From 1ece3b42051d26050cd612a3ed9a20122d501746 Mon Sep 17 00:00:00 2001
From: mo khan <mo@mokhan.ca>
Date: Mon, 21 Apr 2025 13:06:56 -0600
Subject: feat: attach current user if they are in the db

---
 pkg/db/entity.go                    |  7 ++++++
 pkg/db/in_memory_repository.go      | 39 +++++++++++++++++++++++++++++
 pkg/db/in_memory_repository_test.go | 50 +++++++++++++++++++++++++++++++++++++
 pkg/db/repository.go                | 36 +-------------------------
 pkg/db/repository_test.go           | 33 ------------------------
 5 files changed, 97 insertions(+), 68 deletions(-)
 create mode 100644 pkg/db/entity.go
 create mode 100644 pkg/db/in_memory_repository.go
 create mode 100644 pkg/db/in_memory_repository_test.go
 delete mode 100644 pkg/db/repository_test.go

(limited to 'pkg/db')

diff --git a/pkg/db/entity.go b/pkg/db/entity.go
new file mode 100644
index 0000000..1dcf4c3
--- /dev/null
+++ b/pkg/db/entity.go
@@ -0,0 +1,7 @@
+package db
+
+type Entity interface {
+	GetID() string
+	SetID(id string) error
+	Validate() error
+}
diff --git a/pkg/db/in_memory_repository.go b/pkg/db/in_memory_repository.go
new file mode 100644
index 0000000..5859c0b
--- /dev/null
+++ b/pkg/db/in_memory_repository.go
@@ -0,0 +1,39 @@
+package db
+
+import (
+	"github.com/xlgmokha/x/pkg/x"
+	"gitlab.com/gitlab-org/software-supply-chain-security/authorization/sparkled/pkg/pls"
+)
+
+type inMemoryRepository[T Entity] struct {
+	items []T
+}
+
+func NewRepository[T Entity]() Repository[T] {
+	return &inMemoryRepository[T]{
+		items: []T{},
+	}
+}
+
+func (r *inMemoryRepository[T]) All() []T {
+	return r.items
+}
+
+func (r *inMemoryRepository[T]) Find(id string) T {
+	return x.Find(r.All(), func(item T) bool {
+		return item.GetID() == id
+	})
+}
+
+func (r *inMemoryRepository[T]) Save(item T) error {
+	if err := item.Validate(); err != nil {
+		return err
+	}
+
+	if item.GetID() == "" {
+		item.SetID(pls.GenerateULID())
+	}
+
+	r.items = append(r.items, item)
+	return nil
+}
diff --git a/pkg/db/in_memory_repository_test.go b/pkg/db/in_memory_repository_test.go
new file mode 100644
index 0000000..382a656
--- /dev/null
+++ b/pkg/db/in_memory_repository_test.go
@@ -0,0 +1,50 @@
+package db
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"gitlab.com/gitlab-org/software-supply-chain-security/authorization/sparkled/pkg/domain"
+)
+
+func TestInMemoryRepository(t *testing.T) {
+	storage := NewRepository[*domain.Sparkle]()
+
+	t.Run("Save", func(t *testing.T) {
+		t.Run("an invalid Sparkle", func(t *testing.T) {
+			err := storage.Save(&domain.Sparkle{Reason: "because"})
+
+			assert.Error(t, err)
+			assert.Equal(t, 0, len(storage.All()))
+		})
+
+		t.Run("a valid Sparkle", func(t *testing.T) {
+			sparkle := &domain.Sparkle{Sparklee: "@tanuki", Reason: "because"}
+			require.NoError(t, storage.Save(sparkle))
+
+			sparkles := storage.All()
+			assert.Equal(t, 1, len(sparkles))
+			assert.NotEmpty(t, sparkles[0].ID)
+			assert.Equal(t, "@tanuki", sparkles[0].Sparklee)
+			assert.Equal(t, "because", sparkles[0].Reason)
+		})
+	})
+
+	t.Run("Find", func(t *testing.T) {
+		t.Run("when the entity exists", func(t *testing.T) {
+			sparkle, err := domain.NewSparkle("@tanuki for testing this func")
+			require.NoError(t, err)
+			require.NoError(t, storage.Save(sparkle))
+
+			result := storage.Find(sparkle.ID)
+			require.NotNil(t, result)
+			require.Equal(t, sparkle, result)
+		})
+
+		t.Run("when the entity does not exist", func(t *testing.T) {
+			result := storage.Find("unknown")
+			require.Nil(t, result)
+		})
+	})
+}
diff --git a/pkg/db/repository.go b/pkg/db/repository.go
index 79c7ae3..397eee7 100644
--- a/pkg/db/repository.go
+++ b/pkg/db/repository.go
@@ -1,41 +1,7 @@
 package db
 
-import "gitlab.com/gitlab-org/software-supply-chain-security/authorization/sparkled/pkg/pls"
-
-type Entity interface {
-	GetID() string
-	SetID(id string) error
-	Validate() error
-}
-
 type Repository[T Entity] interface {
 	All() []T
+	Find(string) T
 	Save(T) error
 }
-
-type inMemoryRepository[T Entity] struct {
-	items []T
-}
-
-func NewRepository[T Entity]() Repository[T] {
-	return &inMemoryRepository[T]{
-		items: []T{},
-	}
-}
-
-func (r *inMemoryRepository[T]) All() []T {
-	return r.items
-}
-
-func (r *inMemoryRepository[T]) Save(item T) error {
-	if err := item.Validate(); err != nil {
-		return err
-	}
-
-	if item.GetID() == "" {
-		item.SetID(pls.GenerateULID())
-	}
-
-	r.items = append(r.items, item)
-	return nil
-}
diff --git a/pkg/db/repository_test.go b/pkg/db/repository_test.go
deleted file mode 100644
index bb788d2..0000000
--- a/pkg/db/repository_test.go
+++ /dev/null
@@ -1,33 +0,0 @@
-package db
-
-import (
-	"testing"
-
-	"github.com/stretchr/testify/assert"
-	"github.com/stretchr/testify/require"
-	"gitlab.com/gitlab-org/software-supply-chain-security/authorization/sparkled/pkg/domain"
-)
-
-func TestRepository(t *testing.T) {
-	storage := NewRepository[*domain.Sparkle]()
-
-	t.Run("Save", func(t *testing.T) {
-		t.Run("an invalid Sparkle", func(t *testing.T) {
-			err := storage.Save(&domain.Sparkle{Reason: "because"})
-
-			assert.Error(t, err)
-			assert.Equal(t, 0, len(storage.All()))
-		})
-
-		t.Run("a valid Sparkle", func(t *testing.T) {
-			sparkle := &domain.Sparkle{Sparklee: "@tanuki", Reason: "because"}
-			require.NoError(t, storage.Save(sparkle))
-
-			sparkles := storage.All()
-			assert.Equal(t, 1, len(sparkles))
-			assert.NotEmpty(t, sparkles[0].ID)
-			assert.Equal(t, "@tanuki", sparkles[0].Sparklee)
-			assert.Equal(t, "because", sparkles[0].Reason)
-		})
-	})
-}
-- 
cgit v1.2.3