diff --git a/controller/files.go b/controller/files.go index ad3c3a0..d6989dd 100644 --- a/controller/files.go +++ b/controller/files.go @@ -6,6 +6,7 @@ import ( "git.lumeweb.com/LumeWeb/portal/controller/response" "git.lumeweb.com/LumeWeb/portal/logger" "git.lumeweb.com/LumeWeb/portal/middleware" + "git.lumeweb.com/LumeWeb/portal/service/auth" "git.lumeweb.com/LumeWeb/portal/service/files" "github.com/kataras/iris/v12" "go.uber.org/zap" @@ -33,7 +34,7 @@ func (f *FilesController) PostUpload() { return } - upload, err := files.Upload(file, meta.Size, nil) + upload, err := files.Upload(file, meta.Size, nil, auth.GetCurrentUserId(ctx)) if internalError(ctx, err) { logger.Get().Debug("failed uploading file", zap.Error(err)) diff --git a/service/files/files.go b/service/files/files.go index a30f717..fd48e37 100644 --- a/service/files/files.go +++ b/service/files/files.go @@ -54,7 +54,7 @@ func Init() { client.SetDisableWarn(true) } -func Upload(r io.ReadSeeker, size int64, hash []byte) (model.Upload, error) { +func Upload(r io.ReadSeeker, size int64, hash []byte, accountID uint) (model.Upload, error) { var upload model.Upload tree, hashBytes, err := bao.ComputeTree(r, size) @@ -146,7 +146,8 @@ func Upload(r io.ReadSeeker, size int64, hash []byte) (model.Upload, error) { } upload = model.Upload{ - Hash: hashHex, + Hash: hashHex, + AccountID: accountID, } if err = db.Get().Create(&upload).Error; err != nil {