porter/internal/models/invite.go

package models

import (
	"time"

	"gorm.io/gorm"

	"github.com/porter-dev/porter/api/types"
)

// Invite type that extends gorm.Model
type Invite struct {
	gorm.Model

	Token  string `gorm:"unique"`
	Expiry *time.Time
	Email  string

	// Kind is the role kind that this refers to
	Kind string

	ProjectID      uint
	UserID         uint
	InvitingUserID uint
	Status         InviteStatus
}

type InviteStatus string

const (
	InvitePending  InviteStatus = "pending"
	InviteAccepted InviteStatus = "accepted"
	InviteDeclined InviteStatus = "declined"
)

// ToInviteType generates an external Invite to be shared over REST
func (i *Invite) ToInviteType() *types.Invite {
	return &types.Invite{
		ID:             i.Model.ID,
		Token:          i.Token,
		Email:          i.Email,
		Expired:        i.IsExpired(),
		Accepted:       i.IsAccepted(),
		Kind:           i.Kind,
		Status:         string(i.Status),
		InvitingUserID: i.InvitingUserID,
	}
}

func (i *Invite) IsExpired() bool {
	timeLeft := i.Expiry.Sub(time.Now())
	return timeLeft < 0
}

func (i *Invite) IsAccepted() bool {
	return i.UserID != 0
}