tgdrive/teldrive
1
1
Fork 0
mirror of https://github.com/tgdrive/teldrive.git synced 2024-09-20 08:15:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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
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])
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)
messages, err := getTGMessages(c, client, file.Parts, file.ChannelID, user)
if err != nil {
return err
}
channel, err := GetChannelById(c, client, file.ChannelID, user)
channel, err := GetChannelById(ctx, client, channelId, user)
if err != nil {
return err
}
for _, message := range messages.Messages {
for i, message := range messages.Messages {
item := message.(*tg.Message)
media := item.Media.(*tg.MessageMediaDocument)
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
@ -380,8 +389,9 @@ func (fs *FileService) CopyFile(c *gin.Context) (*schemas.FileOut, *types.AppErr
dbFile.Starred = false
dbFile.Status = "active"
dbFile.ParentID = dest.ID
dbFile.ChannelID = &file.ChannelID
dbFile.ChannelID = &channelId
dbFile.Encrypted = file.Encrypted
dbFile.Category = file.Category
if err := fs.db.Create(&dbFile).Error; err != nil {
return nil, &types.AppError{Error: err}