tgdrive/teldrive
1
1
Fork 0
mirror of https://github.com/tgdrive/teldrive.git synced 2025-10-09 23:06:04 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: server side copy

Browse source
This commit is contained in:
divyam234 2024-05-03 23:48:39 +05:30
parent c682930496
commit fe3f802a6e
1 changed files with 16 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
pkg/services/file.go
View file

@ -304,24 +304,33 @@ func (fs *FileService) CopyFile(c *gin.Context) (*schemas.FileOut, *types.AppErr
var res []models.File var res []models.File
fs.db.Model(&models.File{}).Where("id = ?", payload.ID).Find(&res) if err := fs.db.Model(&models.File{}).Where("id = ?", payload.ID).Find(&res).Error; err != nil {
return nil, &types.AppError{Error: err}
}
file := mapper.ToFileOutFull(res[0]) file := mapper.ToFileOutFull(res[0])
newIds := models.Parts{} newIds := models.Parts{}
err := tgc.RunWithAuth(c, client, "", func(ctx context.Context) error { channelId, err := GetDefaultChannel(c, fs.db, userId)
if err != nil {
return nil, &types.AppError{Error: err}
}
err = tgc.RunWithAuth(c, client, "", func(ctx context.Context) error {
user := strconv.FormatInt(userId, 10) user := strconv.FormatInt(userId, 10)
messages, err := getTGMessages(c, client, file.Parts, file.ChannelID, user) messages, err := getTGMessages(c, client, file.Parts, file.ChannelID, user)
if err != nil { if err != nil {
return err return err
} }
channel, err := GetChannelById(c, client, file.ChannelID, user) channel, err := GetChannelById(ctx, client, channelId, user)
if err != nil { if err != nil {
return err return err
} }
for _, message := range messages.Messages { for i, message := range messages.Messages {
item := message.(*tg.Message) item := message.(*tg.Message)
media := item.Media.(*tg.MessageMediaDocument) media := item.Media.(*tg.MessageMediaDocument)
document := media.Document.(*tg.Document) document := media.Document.(*tg.Document)
@ -351,7 +360,7 @@ func (fs *FileService) CopyFile(c *gin.Context) (*schemas.FileOut, *types.AppErr
} }
} }
newIds = append(newIds, models.Part{ID: int64(msg.ID)}) newIds = append(newIds, models.Part{ID: int64(msg.ID), Salt: file.Parts[i].Salt})
} }
return nil return nil
@ -380,8 +389,9 @@ func (fs *FileService) CopyFile(c *gin.Context) (*schemas.FileOut, *types.AppErr
dbFile.Starred = false dbFile.Starred = false
dbFile.Status = "active" dbFile.Status = "active"
dbFile.ParentID = dest.ID dbFile.ParentID = dest.ID
dbFile.ChannelID = &file.ChannelID dbFile.ChannelID = &channelId
dbFile.Encrypted = file.Encrypted dbFile.Encrypted = file.Encrypted
dbFile.Category = file.Category
if err := fs.db.Create(&dbFile).Error; err != nil { if err := fs.db.Create(&dbFile).Error; err != nil {
return nil, &types.AppError{Error: err} return nil, &types.AppError{Error: err}