Base d'API backend pour la manipulation des groupes de travail

Types:

type Workgroup {
  id: ID!
  name: String
  createdAt: Time!
  closedAt: Time
  members: [User]!
}

Mutations:

joinWorkgroup(workgroupId: ID!): Workgroup!
leaveWorkgroup(workgroupId: ID!): Workgroup!
createWorkgroup(changes: WorkgroupChanges!): Workgroup!
closeWorkgroup(workgroupId: ID!): Workgroup!
updateWorkgroup(workgroupId: ID!, changes: WorkgroupChanges!): Workgroup!

Queries:

workgroups: [Workgroup]!

internal/model/user.go

@@ -1,13 +1,17 @@
 package model
 
-import "time"
+import (
+	"time"
+
+	"github.com/jinzhu/gorm"
+)
 
 type User struct {
-	ID          *uint     `gorm:"primary_key"`
-	Name        *string   `json:"name"`
-	Email       string    `json:"email" gorm:"unique;not null"`
-	ConnectedAt time.Time `json:"connectedAt"`
-	CreatedAt   time.Time `json:"createdAt"`
+	gorm.Model
+	Name        *string      `json:"name"`
+	Email       string       `json:"email" gorm:"unique;not null"`
+	ConnectedAt time.Time    `json:"connectedAt"`
+	Workgroups  []*Workgroup `gorm:"many2many:users_workgroups;"`
 }
 
 type ProfileChanges struct {

internal/model/user_repository.go

@@ -15,8 +15,7 @@ type UserRepository struct {
 
 func (r *UserRepository) CreateOrConnectUser(ctx context.Context, email string) (*User, error) {
 	user := &User{
-		Email:     email,
-		CreatedAt: time.Now(),
+		Email: email,
 	}
 
 	err := orm.WithTx(ctx, r.db, func(ctx context.Context, tx *gorm.DB) error {
@@ -44,7 +43,7 @@ func (r *UserRepository) FindUserByEmail(ctx context.Context, email string) (*Us
 		Email: email,
 	}
 
-	err := r.db.First(user, "email = ?", email).Error
+	err := r.db.Model(user).Preload("Workgroups").First(user, "email = ?", email).Error
 	if err != nil {
 		return nil, errors.Wrap(err, "could not find user")
 	}

internal/model/workgroup.go

@@ -0,0 +1,18 @@
+package model
+
+import (
+	"time"
+
+	"github.com/jinzhu/gorm"
+)
+
+type Workgroup struct {
+	gorm.Model
+	Name     *string   `json:"name"`
+	ClosedAt time.Time `json:"closedAt"`
+	Members  []*User   `gorm:"many2many:users_workgroups;"`
+}
+
+type WorkgroupChanges struct {
+	Name *string `json:"name"`
+}

internal/model/workgroup_repository.go

@@ -0,0 +1,140 @@
+package model
+
+import (
+	"context"
+	"time"
+
+	"github.com/jinzhu/gorm"
+	"github.com/pkg/errors"
+)
+
+type WorkgroupRepository struct {
+	db *gorm.DB
+}
+
+func (r *WorkgroupRepository) FindWorkgroups(ctx context.Context, criteria ...interface{}) ([]*Workgroup, error) {
+	workgroups := make([]*Workgroup, 0)
+	if err := r.db.Model(&Workgroup{}).Preload("Members").Find(&workgroups, criteria...).Error; err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	return workgroups, nil
+}
+
+func (r *WorkgroupRepository) UpdateWorkgroup(ctx context.Context, workgroupID uint, changes WorkgroupChanges) (*Workgroup, error) {
+	workgroup := &Workgroup{
+		Name: changes.Name,
+	}
+	workgroup.ID = workgroupID
+
+	err := r.db.Model(workgroup).
+		Update(workgroup).
+		Error
+	if err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	err = r.db.Model(workgroup).Preload("Members").First(workgroup, "id = ?", workgroupID).Error
+	if err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	return workgroup, nil
+}
+
+func (r *WorkgroupRepository) CreateWorkgroup(ctx context.Context, changes WorkgroupChanges) (*Workgroup, error) {
+	workgroup := &Workgroup{
+		Name: changes.Name,
+	}
+
+	if err := r.db.Model(&Workgroup{}).Create(workgroup).Error; err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	return workgroup, nil
+}
+
+func (r *WorkgroupRepository) CloseWorkgroup(ctx context.Context, workgroupID uint) (*Workgroup, error) {
+	workgroup := &Workgroup{}
+
+	err := r.db.Model(workgroup).
+		Where("id = ?", workgroupID).
+		UpdateColumn("closedAt", time.Now()).
+		Error
+	if err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	err = r.db.Model(workgroup).Preload("Members").First(workgroup, "id = ?", workgroupID).Error
+	if err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	return workgroup, nil
+}
+
+func (r *WorkgroupRepository) AddUserToWorkgroup(ctx context.Context, userID, workgroupID uint) (*Workgroup, error) {
+	user := &User{}
+
+	err := r.db.Model(user).Preload("Workgroups").First(user, "id = ?", userID).Error
+	if err != nil {
+		return nil, errors.Wrap(err, "could not find user")
+	}
+
+	workgroup := &Workgroup{}
+	workgroup.ID = workgroupID
+
+	err = r.db.Model(user).
+		Association("Workgroups").
+		Append(workgroup).
+		Error
+
+	if err != nil {
+		return nil, errors.Wrap(err, "could not add user to workgroup")
+	}
+
+	err = r.db.Model(workgroup).
+		Preload("Members").
+		First(workgroup, "id = ?", workgroupID).
+		Error
+	if err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	return workgroup, nil
+}
+
+func (r *WorkgroupRepository) RemoveUserFromWorkgroup(ctx context.Context, userID, workgroupID uint) (*Workgroup, error) {
+	user := &User{}
+
+	err := r.db.First(user, "id = ?", userID).Error
+	if err != nil {
+		return nil, errors.Wrap(err, "could not find user")
+	}
+
+	workgroup := &Workgroup{}
+	workgroup.ID = workgroupID
+
+	err = r.db.Model(user).
+		Association("Workgroups").
+		Delete(workgroup).
+		Error
+
+	if err != nil {
+		return nil, errors.Wrap(err, "could not add user to workgroup")
+	}
+
+	err = r.db.Model(workgroup).
+		Preload("Members").
+		First(workgroup, "id = ?", workgroupID).
+		Error
+	if err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	return workgroup, nil
+}
+
+func NewWorkgroupRepository(db *gorm.DB) *WorkgroupRepository {
+	return &WorkgroupRepository{db}
+}