service以下にinterface

internal/model/account_manager.go

@@ -1,18 +1,51 @@
-//go:generate go tool mockgen -source=$GOFILE -destination=mock_$GOPACKAGE/mock_$GOFILE -package=mock_$GOPACKAGE
 package model
 
 import (
 	"context"
 
 	"github.com/google/uuid"
+	"github.com/samber/lo"
+	"github.com/traPtitech/Jomon/internal/ent"
+	"github.com/traPtitech/Jomon/internal/ent/user"
+	"github.com/traPtitech/Jomon/internal/service"
 )
 
-type AccountManager struct {
-	ID uuid.UUID `json:"id"`
+func (repo *EntRepository) GetAccountManagers(
+	ctx context.Context,
+) ([]*service.AccountManager, error) {
+	users, err := repo.client.User.
+		Query().
+		Where(user.AccountManager(true)).
+		All(ctx)
+	if err != nil {
+		return nil, defaultEntErrorConverter.convert(err)
+	}
+
+	accountManagers := lo.Map(users, func(u *ent.User, _ int) *service.AccountManager {
+		return &service.AccountManager{
+			ID: u.ID,
+		}
+	})
+
+	return accountManagers, nil
+}
+
+func (repo *EntRepository) AddAccountManagers(ctx context.Context, userIDs []uuid.UUID) error {
+	_, err := repo.client.User.
+		Update().
+		Where(user.IDIn(userIDs...)).
+		SetAccountManager(true).
+		Save(ctx)
+
+	return defaultEntErrorConverter.convert(err)
 }
 
-type AccountManagerRepository interface {
-	GetAccountManagers(ctx context.Context) ([]*AccountManager, error)
-	AddAccountManagers(ctx context.Context, userIDs []uuid.UUID) error
-	DeleteAccountManagers(ctx context.Context, userIDs []uuid.UUID) error
+func (repo *EntRepository) DeleteAccountManagers(ctx context.Context, userIDs []uuid.UUID) error {
+	_, err := repo.client.User.
+		Update().
+		Where(user.IDIn(userIDs...)).
+		SetAccountManager(false).
+		Save(ctx)
+
+	return defaultEntErrorConverter.convert(err)
 }

internal/model/account_manager_impl_test.go → internal/model/account_manager_test.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/google/uuid"
 	"github.com/stretchr/testify/require"
+	"github.com/traPtitech/Jomon/internal/service"
 	"github.com/traPtitech/Jomon/internal/testutil"
 	"github.com/traPtitech/Jomon/internal/testutil/random"
 )
@@ -35,7 +36,7 @@ func TestEntRepository_GetAccountManagers(t *testing.T) {
 
 		got, err := repo.GetAccountManagers(ctx)
 		require.NoError(t, err)
-		exp := []*AccountManager{
+		exp := []*service.AccountManager{
 			{ID: user1.ID},
 			{ID: user2.ID},
 		}