Planner/models/plans.go

package models

import (
	"errors"
	"gorm.io/gorm"
	. "planner/errors"
)

func GetPlan(orm *gorm.DB, user User, id uint) (*Plan, error) {
	var plan Plan = Plan{
		ID: id,
	}
	result := orm.Take(&plan)

	if errors.Is(result.Error, gorm.ErrRecordNotFound) {
		return nil, ErrNotFound
	} else if result.Error != nil {
		return nil, result.Error
	}

	if plan.Owner == user.Username {
		return &plan, nil
	}

	isMember, err := plan.IsMember(orm, &user)

	if !isMember || err != nil {
		return nil, err
	}

	return &plan, nil
}

func (p *Plan) GetAllUsers(orm *gorm.DB) ([]Member, error) {
	var members []Member
	err := orm.Model(p).Association("Members").Find(&members)
	return members, err
}

func (p *Plan) IsMember(orm *gorm.DB, u *User) (bool, error) {
	var member_count int64
	result := orm.
		Table("members").
		Where("user_id=? AND plan_id=?", u.Username, p.ID).
		Count(&member_count).Error
	if result != nil {
		return false, ErrNotMember
	}
	return member_count == 1, nil
}