mirror of
https://github.com/wgh136/nysoure.git
synced 2025-09-27 12:17:24 +00:00
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package model
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type File struct {
|
|
gorm.Model
|
|
UUID string `gorm:"uniqueIndex;not null"`
|
|
Filename string
|
|
Description string
|
|
StorageKey string
|
|
StorageID *uint `gorm:"default:null"`
|
|
Storage Storage
|
|
ResourceID uint
|
|
RedirectUrl string
|
|
Resource Resource `gorm:"foreignKey:ResourceID"`
|
|
UserID uint
|
|
User User `gorm:"foreignKey:UserID"`
|
|
Size int64
|
|
Hash string `gorm:"default:null"`
|
|
}
|
|
|
|
type FileView struct {
|
|
ID string `json:"id"`
|
|
Filename string `json:"filename"`
|
|
Description string `json:"description"`
|
|
Size int64 `json:"size"`
|
|
IsRedirect bool `json:"is_redirect"`
|
|
User UserView `json:"user"`
|
|
Resource *ResourceView `json:"resource,omitempty"`
|
|
Hash string `json:"hash,omitempty"`
|
|
}
|
|
|
|
func (f *File) ToView() *FileView {
|
|
return &FileView{
|
|
ID: f.UUID,
|
|
Filename: f.Filename,
|
|
Description: f.Description,
|
|
Size: f.Size,
|
|
IsRedirect: f.RedirectUrl != "",
|
|
User: f.User.ToView(),
|
|
Hash: f.Hash,
|
|
}
|
|
}
|
|
|
|
func (f *File) ToViewWithResource() *FileView {
|
|
var resource *ResourceView
|
|
if f.Resource.ID != 0 {
|
|
rv := f.Resource.ToView()
|
|
resource = &rv
|
|
}
|
|
|
|
return &FileView{
|
|
ID: f.UUID,
|
|
Filename: f.Filename,
|
|
Description: f.Description,
|
|
Size: f.Size,
|
|
IsRedirect: f.RedirectUrl != "",
|
|
User: f.User.ToView(),
|
|
Resource: resource,
|
|
Hash: f.Hash,
|
|
}
|
|
}
|