website/pkg/photo/quota.go

23 lines
484 B
Go
Raw Normal View History

package photo
import (
"git.kirsle.net/apps/gosocial/pkg/config"
"git.kirsle.net/apps/gosocial/pkg/models"
)
// QuoteForUser returns the current photo usage quota for a given user.
func QuotaForUser(u *models.User) (current, allowed int) {
// Count their photos.
count := models.CountPhotos(u.ID)
// What is their quota at?
var quota int
if !u.Certified {
quota = config.PhotoQuotaUncertified
} else {
quota = config.PhotoQuotaCertified
}
return int(count), quota
}