feat: add character contact labels

Author: ErikKalkoken

internal/app/contact.go

@@ -3,6 +3,8 @@ package app
 import (
 	"fmt"
 
+	"github.com/ErikKalkoken/go-set"
+
 	"github.com/ErikKalkoken/evebuddy/internal/optional"
 )
 
@@ -53,5 +55,6 @@ type CharacterContact struct {
 	Contact     *EveEntity
 	IsBlocked   optional.Optional[bool]
 	IsWatched   optional.Optional[bool]
+	Labels      set.Set[string]
 	Standing    float64
 }

internal/app/storage/charactercontact.go

@@ -104,3 +104,62 @@ func (st *Storage) UpdateOrCreateCharacterContact(ctx context.Context, arg Updat
 	}
 	return nil
 }
+
+type CreateCharacterContactLabelParams struct {
+	CharacterID int64
+	LabelID     int64
+	Name        string
+}
+
+func (st *Storage) CreateCharacterContactLabel(ctx context.Context, arg CreateCharacterContactLabelParams) error {
+	err := st.qRW.CreateCharacterContactLabel(ctx, queries.CreateCharacterContactLabelParams{
+		CharacterID: arg.CharacterID,
+		LabelID:     arg.LabelID,
+		Name:        arg.Name,
+	})
+	if err != nil {
+		return fmt.Errorf("CreateCharacterContactLabel: %v: %w", arg, err)
+	}
+	return nil
+}
+
+func (st *Storage) GetCharacterContactLabel(ctx context.Context, characterID, labelID int64) (string, error) {
+	r, err := st.qRO.GetCharacterContactLabel(ctx, queries.GetCharacterContactLabelParams{
+		CharacterID: characterID,
+		LabelID:     labelID,
+	})
+	if err != nil {
+		return "", fmt.Errorf("GetCharacterContactLabel: %d %d: %w", characterID, labelID, err)
+	}
+	return r.Name, nil
+}
+
+func (st *Storage) ListCharacterContactLabels(ctx context.Context, characterID int64) (set.Set[string], error) {
+	rows, err := st.qRO.ListCharacterContactLabels(ctx, characterID)
+	if err != nil {
+		return set.Set[string]{}, fmt.Errorf("ListCharacterContactLabels for character %d: %w", characterID, err)
+	}
+	x := set.Collect(slices.Values(rows))
+	return x, nil
+}
+
+func (st *Storage) DeleteCharacterContactLabels(ctx context.Context, characterID int64, names set.Set[string]) error {
+	wrapErr := func(err error) error {
+		return fmt.Errorf("DeleteCharacterContactLabels for character %d and contact IDs: %v: %w", characterID, names, err)
+	}
+	if characterID == 0 {
+		return wrapErr(app.ErrInvalid)
+	}
+	if names.Size() == 0 {
+		return nil
+	}
+	err := st.qRW.DeleteCharacterContactLabels(ctx, queries.DeleteCharacterContactLabelsParams{
+		CharacterID: characterID,
+		Names:       slices.Collect(names.All()),
+	})
+	if err != nil {
+		return wrapErr(err)
+	}
+	slog.Info("Character labels deleted", "characterID", characterID, "labels", names)
+	return nil
+}

internal/app/storage/charactercontact_test.go

@@ -146,3 +146,51 @@ func TestCharacterContact(t *testing.T) {
 		xassert.Equal(t, want, got)
 	})
 }
+
+func TestCharacterContactLabel(t *testing.T) {
+	db, st, factory := testutil.NewDBInMemory()
+	defer db.Close()
+	ctx := context.Background()
+	t.Run("can create from scratch", func(t *testing.T) {
+		// given
+		testutil.MustTruncateTables(db)
+		c := factory.CreateCharacter()
+		// when
+		err := st.CreateCharacterContactLabel(ctx, storage.CreateCharacterContactLabelParams{
+			CharacterID: c.ID,
+			LabelID:     42,
+			Name:        "Alpha",
+		})
+		// then
+		require.NoError(t, err)
+		got, err := st.GetCharacterContactLabel(ctx, c.ID, 42)
+		require.NoError(t, err)
+		xassert.Equal(t, "Alpha", got)
+	})
+	t.Run("can list labels", func(t *testing.T) {
+		// given
+		testutil.MustTruncateTables(db)
+		c := factory.CreateCharacter()
+		l1 := factory.CreateCharacterContactLabel()
+		l2 := factory.CreateCharacterContactLabel()
+		// when
+		got, err := st.ListCharacterContactLabels(ctx, c.ID)
+		// then
+		require.NoError(t, err)
+		xassert.Equal(t, set.Of(l1, l2), got)
+	})
+	t.Run("can delete labels", func(t *testing.T) {
+		// given
+		testutil.MustTruncateTables(db)
+		c := factory.CreateCharacter()
+		l1 := factory.CreateCharacterContactLabel()
+		l2 := factory.CreateCharacterContactLabel()
+		// when
+		err := st.DeleteCharacterContactLabels(ctx, c.ID, set.Of(l1))
+		// then
+		require.NoError(t, err)
+		got, err := st.ListCharacterContactLabels(ctx, c.ID)
+		require.NoError(t, err)
+		xassert.Equal(t, set.Of(l2), got)
+	})
+}

internal/app/storage/migrations/0019a_add_contact_labels.sql

@@ -0,0 +1,25 @@
+CREATE TABLE character_contact_labels (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    character_id INTEGER NOT NULL,
+    label_id INTEGER NOT NULL,
+    name TEXT NOT NULL,
+    FOREIGN KEY (character_id) REFERENCES characters (id) ON DELETE CASCADE,
+    UNIQUE (character_id, label_id)
+);
+
+CREATE INDEX character_contact_labels_idx1 ON character_contact_labels (character_id);
+
+CREATE INDEX character_contact_labels_idx2 ON character_contact_labels (label_id);
+
+CREATE TABLE character_contacts_labels (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    contact_id INTEGER NOT NULL,
+    label_id INTEGER NOT NULL,
+    FOREIGN KEY (contact_id) REFERENCES character_contacts (id) ON DELETE CASCADE,
+    FOREIGN KEY (label_id) REFERENCES character_contact_labels (id) ON DELETE CASCADE,
+    UNIQUE (contact_id, label_id)
+);
+
+CREATE INDEX character_contacts_labels_idx1 ON character_contacts_labels (contact_id);
+
+CREATE INDEX character_contacts_labels_idx2 ON character_contacts_labels (label_id);

internal/app/storage/queries/character_contacts.sql

@@ -49,3 +49,32 @@ SET
     is_blocked = ?3,
     is_watched = ?4,
     standing = ?5;
+
+-- name: CreateCharacterContactLabel :exec
+INSERT INTO
+    character_contact_labels (character_id, label_id, name)
+VALUES
+    (?, ?, ?);
+
+-- name: DeleteCharacterContactLabels :exec
+DELETE FROM character_contact_labels
+WHERE
+    character_id = ?
+    AND name IN (sqlc.slice('names'));
+
+-- name: GetCharacterContactLabel :one
+SELECT
+    *
+FROM
+    character_contact_labels
+WHERE
+    character_id = ?
+    AND label_id = ?;
+
+-- name: ListCharacterContactLabels :many
+SELECT
+    name
+FROM
+    character_contact_labels
+WHERE
+    character_id = ?;

internal/app/storage/queries/character_contacts.sql.go

@@ -245,6 +245,38 @@ func (f Factory) CreateCharacterContact(args ...storage.UpdateOrCreateCharacterC
 	return o
 }
 
+func (f Factory) CreateCharacterContactLabel(args ...storage.CreateCharacterContactLabelParams) string {
+	ctx := context.Background()
+	var arg storage.CreateCharacterContactLabelParams
+	if len(args) > 0 {
+		arg = args[0]
+	}
+	if arg.CharacterID == 0 {
+		x := f.CreateCharacter()
+		arg.CharacterID = x.ID
+	}
+	if arg.LabelID == 0 {
+		arg.LabelID = f.calcNewIDWithParam(
+			"character_contact_labels",
+			"label_id",
+			"character_id",
+			arg.CharacterID,
+		)
+	}
+	if arg.Name == "" {
+		arg.Name = fake.Color()
+	}
+	err := f.st.CreateCharacterContactLabel(ctx, arg)
+	if err != nil {
+		panic(fmt.Sprintf("%s|%+v", err, arg))
+	}
+	v, err := f.st.GetCharacterContactLabel(ctx, arg.CharacterID, arg.LabelID)
+	if err != nil {
+		panic(err)
+	}
+	return v
+}
+
 func (f Factory) CreateCharacterContract(args ...storage.CreateCharacterContractParams) *app.CharacterContract {
 	ctx := context.Background()
 	var arg storage.CreateCharacterContractParams