Initial commit

internal/models/gist.go

@@ -0,0 +1,136 @@
+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)
+}