feat: add file visibility toggle for files

This commit is contained in:
2024-09-15 23:59:19 +07:00
parent dc330cc72f
commit b246d85fac
22 changed files with 738 additions and 144 deletions

View File

@ -275,6 +275,15 @@ func (db *mySQLdb) IncrementDownloadCount(fileID string) error {
return nil
}
func (db *mySQLdb) ChangeFileVisibility(fileID string) error {
err := db.DB.Model(&models.File{}).Where("id = ?", fileID).Select("is_private").
Updates(map[string]interface{}{"is_private": gorm.Expr("NOT is_private")}).Error
if err != nil {
return err
}
return nil
}
func (db *mySQLdb) InitializeTotp(email string, secret string) error {
var user models.User
err := db.DB.Table("users").Where("email = ?", email).First(&user).Error
@ -426,6 +435,16 @@ func (db *postgresDB) IncrementDownloadCount(fileID string) error {
return nil
}
func (db *postgresDB) ChangeFileVisibility(fileID string) error {
err := db.DB.Model(&models.File{}).Where("id = $1", fileID).Select("is_private").
Updates(map[string]interface{}{"is_private": gorm.Expr("NOT is_private")}).Error
if err != nil {
return err
}
return nil
}
func (db *postgresDB) InitializeTotp(email string, secret string) error {
var user models.User
err := db.DB.Table("users").Where("email = $1", email).First(&user).Error