mirror of
https://github.com/thomiceli/opengist.git
synced 2024-12-23 13:02:39 +00:00
136 lines
3.4 KiB
Go
136 lines
3.4 KiB
Go
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)
|
|
}
|