package models import ( "time" ) type Gist struct { ID uint `gorm:"primaryKey"` Uuid string Title string `validate:"max=50" form:"title"` Preview string PreviewFilename string Description string `validate:"max=150" form:"description"` Private bool `form:"private"` UserID uint User User `validate:"-"` NbFiles int NbLikes int CreatedAt int64 UpdatedAt int64 Likes []User `gorm:"many2many:likes;constraint:OnUpdate:CASCADE,OnDelete:CASCADE"` Files []File `gorm:"-" validate:"min=1,dive"` } type File struct { Filename string `validate:"excludes=\x2f,excludes=\x5c,max=50"` OldFilename string `validate:"excludes=\x2f,excludes=\x5c,max=50"` Content string `validate:"required"` } type Commit struct { Hash string Author string Timestamp string Changed string Files []File } func GetGist(user string, gistUuid string) (*Gist, error) { gist := new(Gist) err := db.Preload("User"). Where("gists.uuid = ? AND users.username like ?", gistUuid, user). Joins("join users on gists.user_id = users.id"). First(&gist).Error return gist, err } func GetGistByID(gistId string) (*Gist, error) { gist := new(Gist) err := db.Preload("User"). Where("gists.id = ?", gistId). First(&gist).Error return gist, err } func GetAllGistsForCurrentUser(currentUserId uint, offset int, sort string, order string) ([]*Gist, error) { var gists []*Gist err := db.Preload("User"). Where("gists.private = 0 or gists.user_id = ?", currentUserId). Limit(11). Offset(offset * 10). Order(sort + "_at " + order). Find(&gists).Error return gists, err } func GetAllGists(offset int) ([]*Gist, error) { var all []*Gist err := db.Preload("User"). Limit(11). Offset(offset * 10). Order("id asc"). Find(&all).Error return all, err } func GetAllGistsFromUser(fromUser string, currentUserId uint, offset int, sort string, order string) ([]*Gist, error) { var gists []*Gist err := db.Preload("User"). Where("users.username = ? and ((gists.private = 0) or (gists.private = 1 and gists.user_id = ?))", fromUser, currentUserId). Joins("join users on gists.user_id = users.id"). Limit(11). Offset(offset * 10). Order("gists." + sort + "_at " + order). Find(&gists).Error return gists, err } func CreateGist(gist *Gist) error { return db.Create(&gist).Error } func UpdateGist(gist *Gist) error { return db.Save(&gist).Error } func DeleteGist(gist *Gist) error { return db.Delete(&gist).Error } func GistLastActiveNow(gistID uint) error { return db.Model(&Gist{}). Where("id = ?", gistID). Update("updated_at", time.Now().Unix()).Error } func AppendUserLike(gist *Gist, user *User) error { db.Model(&gist).Omit("updated_at").Update("nb_likes", gist.NbLikes+1) return db.Model(&gist).Omit("updated_at").Association("Likes").Append(user) } func RemoveUserLike(gist *Gist, user *User) error { db.Model(&gist).Omit("updated_at").Update("nb_likes", gist.NbLikes-1) return db.Model(&gist).Omit("updated_at").Association("Likes").Delete(user) } func GetUsersLikesForGists(gist *Gist, offset int) ([]*User, error) { var users []*User err := db.Model(&gist). Where("gist_id = ?", gist.ID). Limit(31). Offset(offset * 30). Association("Likes").Find(&users) return users, err } func UserCanWrite(user *User, gist *Gist) bool { return !(user == nil) && (gist.UserID == user.ID) }