daddy/internal/graph/helper.go

package graph

import (
	"context"

	"forge.cadoles.com/Cadoles/daddy/internal/voter"

	"forge.cadoles.com/Cadoles/daddy/internal/model"
	"forge.cadoles.com/Cadoles/daddy/internal/orm"
	"forge.cadoles.com/Cadoles/daddy/internal/session"
	"github.com/jinzhu/gorm"

	"github.com/pkg/errors"
	"gitlab.com/wpetit/goweb/middleware/container"
)

func getDB(ctx context.Context) (*gorm.DB, error) {
	ctn, err := container.From(ctx)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	orm, err := orm.From(ctn)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return orm.DB(), nil
}

func getSessionUser(ctx context.Context) (*model.User, *gorm.DB, error) {
	db, err := getDB(ctx)
	if err != nil {
		return nil, nil, errors.WithStack(err)
	}

	userEmail, err := session.UserEmail(ctx)
	if err != nil {
		return nil, nil, errors.WithStack(err)
	}

	repo := model.NewUserRepository(db)

	user, err := repo.FindUserByEmail(ctx, userEmail)
	if err != nil {
		return nil, nil, errors.WithStack(err)
	}

	return user, db, nil
}

func isAuthorized(ctx context.Context, obj interface{}, action interface{}) (bool, error) {
	user, _, err := getSessionUser(ctx)
	if err != nil {
		return false, errors.WithStack(err)
	}

	ctn, err := container.From(ctx)
	if err != nil {
		return false, errors.WithStack(err)
	}

	voterSrv, err := voter.From(ctn)
	if err != nil {
		return false, errors.WithStack(err)
	}

	decision, err := voterSrv.Authorized(ctx, user, obj, action)
	if err != nil {
		return false, errors.WithStack(err)
	}

	if decision == voter.Allow {
		return true, nil
	}

	return false, nil
}