bagas/filekeeper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add file filtering functionality

Browse Source
This commit is contained in:
Bagas Aulia Rezki
2024-09-17 22:46:32 +07:00
parent ab73c5cdd3
commit 1a444bb6eb
22 changed files with 1669 additions and 1278 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
db/database.go
View File

@ -271,13 +271,28 @@ func (db *mySQLdb) GetUserFile(name string, ownerID string) (*models.File, error
return &file, nil
}
func (db *mySQLdb) GetFiles(ownerID string) ([]*models.File, error) {
func (db *mySQLdb) GetFiles(ownerID string, query string, status types.FileStatus) ([]*models.File, error) {
var files []*models.File
err := db.DB.Table("files").Where("owner_id = ?", ownerID).Find(&files).Error
tx := db.DB.Table("files").Where("owner_id = ?", ownerID)
if query != "" {
tx = tx.Where("name LIKE ?", "%"+query+"%")
}
switch status {
case types.Private:
tx = tx.Where("is_private = ?::boolean", true)
case types.Public:
tx = tx.Where("is_private = ?::boolean", false)
default:
}
err := tx.Find(&files).Error
if err != nil {
return nil, err
}
return files, err
return files, nil
}
func (db *mySQLdb) IncrementDownloadCount(fileID string) error {
@ -450,13 +465,38 @@ func (db *postgresDB) GetUserFile(name string, ownerID string) (*models.File, er
return &file, nil
}
func (db *postgresDB) GetFiles(ownerID string) ([]*models.File, error) {
func (db *postgresDB) GetFiles(ownerID string, query string, status types.FileStatus) ([]*models.File, error) {
var files []*models.File
err := db.DB.Table("files").Where("owner_id = $1", ownerID).Find(&files).Error
tx := db.DB.Table("files").Where("owner_id = $1", ownerID)
if query != "" {
tx = tx.Where("name LIKE $2", "%"+query+"%")
}
if query == "" {
switch status {
case types.Private:
tx = tx.Where("is_private = $2::boolean", true)
case types.Public:
tx = tx.Where("is_private = $2::boolean", false)
default:
}
} else {
switch status {
case types.Private:
tx = tx.Where("is_private = $3::boolean", true)
case types.Public:
tx = tx.Where("is_private = $3::boolean", false)
default:
}
}
err := tx.Find(&files).Error
if err != nil {
return nil, err
}
return files, err
return files, nil
}
func (db *postgresDB) IncrementDownloadCount(fileID string) error {