You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
276 lines
7.0 KiB
276 lines
7.0 KiB
package model
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/HFO4/cloudreve/pkg/cache"
|
|
"github.com/HFO4/cloudreve/pkg/hashid"
|
|
"github.com/HFO4/cloudreve/pkg/util"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/jinzhu/gorm"
|
|
"math"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// Share 分享模型
|
|
type Share struct {
|
|
gorm.Model
|
|
Password string // 分享密码，空值为非加密分享
|
|
IsDir bool // 原始资源是否为目录
|
|
UserID uint // 创建用户ID
|
|
SourceID uint // 原始资源ID
|
|
Views int // 浏览数
|
|
Downloads int // 下载数
|
|
RemainDownloads int // 剩余下载配额，负值标识无限制
|
|
Expires *time.Time // 过期时间，空值表示无过期时间
|
|
Score int // 每人次下载扣除积分
|
|
PreviewEnabled bool // 是否允许直接预览
|
|
SourceName string `gorm:"index:source"` // 用于搜索的字段
|
|
|
|
// 数据库忽略字段
|
|
User User `gorm:"PRELOAD:false,association_autoupdate:false"`
|
|
File File `gorm:"PRELOAD:false,association_autoupdate:false"`
|
|
Folder Folder `gorm:"PRELOAD:false,association_autoupdate:false"`
|
|
}
|
|
|
|
// Create 创建分享
|
|
func (share *Share) Create() (uint, error) {
|
|
if err := DB.Create(share).Error; err != nil {
|
|
util.Log().Warning("无法插入数据库记录, %s", err)
|
|
return 0, err
|
|
}
|
|
return share.ID, nil
|
|
}
|
|
|
|
// GetShareByHashID 根据HashID查找分享
|
|
func GetShareByHashID(hashID string) *Share {
|
|
id, err := hashid.DecodeHashID(hashID, hashid.ShareID)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
var share Share
|
|
result := DB.First(&share, id)
|
|
if result.Error != nil {
|
|
return nil
|
|
}
|
|
|
|
return &share
|
|
}
|
|
|
|
// IsAvailable 返回此分享是否可用（是否过期）
|
|
func (share *Share) IsAvailable() bool {
|
|
if share.RemainDownloads == 0 {
|
|
return false
|
|
}
|
|
if share.Expires != nil && time.Now().After(*share.Expires) {
|
|
return false
|
|
}
|
|
|
|
// 检查创建者状态
|
|
if share.Creator().Status != Active {
|
|
return false
|
|
}
|
|
|
|
// 检查源对象是否存在
|
|
var sourceID uint
|
|
if share.IsDir {
|
|
folder := share.SourceFolder()
|
|
sourceID = folder.ID
|
|
} else {
|
|
file := share.SourceFile()
|
|
sourceID = file.ID
|
|
}
|
|
if sourceID == 0 {
|
|
// TODO 是否要在这里删除这个无效分享？
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// Creator 获取分享的创建者
|
|
func (share *Share) Creator() *User {
|
|
if share.User.ID == 0 {
|
|
share.User, _ = GetUserByID(share.UserID)
|
|
}
|
|
return &share.User
|
|
}
|
|
|
|
// Source 返回源对象
|
|
func (share *Share) Source() interface{} {
|
|
if share.IsDir {
|
|
return share.SourceFolder()
|
|
}
|
|
return share.SourceFile()
|
|
}
|
|
|
|
// SourceFolder 获取源目录
|
|
func (share *Share) SourceFolder() *Folder {
|
|
if share.Folder.ID == 0 {
|
|
folders, _ := GetFoldersByIDs([]uint{share.SourceID}, share.UserID)
|
|
if len(folders) > 0 {
|
|
share.Folder = folders[0]
|
|
}
|
|
}
|
|
return &share.Folder
|
|
}
|
|
|
|
// SourceFile 获取源文件
|
|
func (share *Share) SourceFile() *File {
|
|
if share.File.ID == 0 {
|
|
files, _ := GetFilesByIDs([]uint{share.SourceID}, share.UserID)
|
|
if len(files) > 0 {
|
|
share.File = files[0]
|
|
}
|
|
}
|
|
return &share.File
|
|
}
|
|
|
|
// CanBeDownloadBy 返回此分享是否可以被给定用户下载
|
|
func (share *Share) CanBeDownloadBy(user *User) error {
|
|
// 用户组权限
|
|
if !user.Group.OptionsSerialized.ShareDownload {
|
|
if user.IsAnonymous() {
|
|
return errors.New("未登录用户无法下载")
|
|
}
|
|
return errors.New("您当前的用户组无权下载")
|
|
}
|
|
|
|
// 需要积分但未登录
|
|
if share.Score > 0 && user.IsAnonymous() {
|
|
return errors.New("未登录用户无法下载")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// WasDownloadedBy 返回分享是否已被用户下载过
|
|
func (share *Share) WasDownloadedBy(user *User, c *gin.Context) (exist bool) {
|
|
if user.IsAnonymous() {
|
|
exist = util.GetSession(c, fmt.Sprintf("share_%d_%d", share.ID, user.ID)) != nil
|
|
} else {
|
|
_, exist = cache.Get(fmt.Sprintf("share_%d_%d", share.ID, user.ID))
|
|
}
|
|
|
|
return exist
|
|
}
|
|
|
|
// DownloadBy 增加下载次数、检查积分等，匿名用户不会缓存
|
|
func (share *Share) DownloadBy(user *User, c *gin.Context) error {
|
|
if !share.WasDownloadedBy(user, c) {
|
|
if err := share.Purchase(user); err != nil {
|
|
return err
|
|
}
|
|
share.Downloaded()
|
|
if !user.IsAnonymous() {
|
|
cache.Set(fmt.Sprintf("share_%d_%d", share.ID, user.ID), true,
|
|
GetIntSetting("share_download_session_timeout", 2073600))
|
|
} else {
|
|
util.SetSession(c, map[string]interface{}{fmt.Sprintf("share_%d_%d", share.ID, user.ID): true})
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Purchase 使用积分购买分享
|
|
func (share *Share) Purchase(user *User) error {
|
|
// 不需要付积分
|
|
if share.Score == 0 || user.Group.OptionsSerialized.ShareFree || user.ID == share.UserID {
|
|
return nil
|
|
}
|
|
|
|
ok := user.PayScore(share.Score)
|
|
if !ok {
|
|
return errors.New("积分不足")
|
|
}
|
|
|
|
scoreRate := GetIntSetting("share_score_rate", 100)
|
|
gainedScore := int(math.Ceil(float64(share.Score*scoreRate) / 100))
|
|
share.Creator().AddScore(gainedScore)
|
|
|
|
return nil
|
|
}
|
|
|
|
// Viewed 增加访问次数
|
|
func (share *Share) Viewed() {
|
|
share.Views++
|
|
DB.Model(share).UpdateColumn("views", gorm.Expr("views + ?", 1))
|
|
}
|
|
|
|
// Downloaded 增加下载次数
|
|
func (share *Share) Downloaded() {
|
|
share.Downloads++
|
|
if share.RemainDownloads > 0 {
|
|
share.RemainDownloads--
|
|
}
|
|
DB.Model(share).Updates(map[string]interface{}{
|
|
"downloads": share.Downloads,
|
|
"remain_downloads": share.RemainDownloads,
|
|
})
|
|
}
|
|
|
|
// Update 更新分享属性
|
|
func (share *Share) Update(props map[string]interface{}) error {
|
|
return DB.Model(share).Updates(props).Error
|
|
}
|
|
|
|
// Delete 删除分享
|
|
func (share *Share) Delete() error {
|
|
return DB.Model(share).Delete(share).Error
|
|
}
|
|
|
|
// DeleteShareBySourceIDs 根据原始资源类型和ID删除文件
|
|
func DeleteShareBySourceIDs(sources []uint, isDir bool) error {
|
|
return DB.Where("source_id in (?) and is_dir = ?", sources, isDir).Delete(&Share{}).Error
|
|
}
|
|
|
|
// ListShares 列出UID下的分享
|
|
func ListShares(uid uint, page, pageSize int, order string, publicOnly bool) ([]Share, int) {
|
|
var (
|
|
shares []Share
|
|
total int
|
|
)
|
|
dbChain := DB
|
|
dbChain = dbChain.Where("user_id = ?", uid)
|
|
if publicOnly {
|
|
dbChain = dbChain.Where("password = ?", "")
|
|
}
|
|
|
|
// 计算总数用于分页
|
|
dbChain.Model(&Share{}).Count(&total)
|
|
|
|
// 查询记录
|
|
dbChain.Limit(pageSize).Offset((page - 1) * pageSize).Order(order).Find(&shares)
|
|
return shares, total
|
|
}
|
|
|
|
// SearchShares 根据关键字搜索分享
|
|
func SearchShares(page, pageSize int, order, keywords string) ([]Share, int) {
|
|
var (
|
|
shares []Share
|
|
total int
|
|
)
|
|
|
|
keywordList := strings.Split(keywords, " ")
|
|
availableList := make([]string, 0, len(keywordList))
|
|
for i := 0; i < len(keywordList); i++ {
|
|
if len(keywordList[i]) > 0 {
|
|
availableList = append(availableList, keywordList[i])
|
|
}
|
|
}
|
|
if len(availableList) == 0 {
|
|
return shares, 0
|
|
}
|
|
|
|
dbChain := DB
|
|
dbChain = dbChain.Where("password = ? and remain_downloads <> 0 and (expires is NULL or expires > ?) and source_name like ?", "", time.Now(), "%"+strings.Join(availableList, "%")+"%")
|
|
|
|
// 计算总数用于分页
|
|
dbChain.Model(&Share{}).Count(&total)
|
|
|
|
// 查询记录
|
|
dbChain.Limit(pageSize).Offset((page - 1) * pageSize).Order(order).Find(&shares)
|
|
return shares, total
|
|
}
|