[ENTITY] Add field size to attachments, used for quota calculations

This commit is contained in:
Hugo Sales 2021-04-29 20:29:21 +00:00
parent 29457ef50d
commit 7509913fcf
Signed by untrusted user: someonewithpc
GPG Key ID: 7D0C7EAFC9D835A0
2 changed files with 15 additions and 1 deletions

View File

@ -47,7 +47,7 @@ class GSFile
// The following properly gets the mimetype with `file` or other // The following properly gets the mimetype with `file` or other
// available methods, so should be safe // available methods, so should be safe
$mimetype = $sfile->getMimeType(); $mimetype = $sfile->getMimeType();
Event::handle('AttachmentValidation', [&$sfile, &$mimetype]); Event::handle('AttachmentValidation', [&$sfile, &$mimetype, &$title]);
$attachment = Attachment::create([ $attachment = Attachment::create([
'file_hash' => $hash, 'file_hash' => $hash,
'gsactor_id' => $actor_id, 'gsactor_id' => $actor_id,
@ -55,6 +55,7 @@ class GSFile
'title' => $title ?: _m('Untitled attachment'), 'title' => $title ?: _m('Untitled attachment'),
'filename' => $hash, 'filename' => $hash,
'is_local' => $is_local, 'is_local' => $is_local,
'size' => $sfile->getSize(),
]); ]);
$sfile->move($dest_dir, $hash); $sfile->move($dest_dir, $hash);
DB::persist($attachment); DB::persist($attachment);

View File

@ -54,6 +54,7 @@ class Attachment extends Entity
private ?bool $is_local; private ?bool $is_local;
private ?int $source; private ?int $source;
private ?int $scope; private ?int $scope;
private ?int $size;
private \DateTimeInterface $modified; private \DateTimeInterface $modified;
public function setId(int $id): self public function setId(int $id): self
@ -177,6 +178,17 @@ class Attachment extends Entity
return $this->scope; return $this->scope;
} }
public function setSize(?int $size): self
{
$this->size = $size;
return $this;
}
public function getSize(): ?int
{
return $this->size;
}
public function setModified(DateTimeInterface $modified): self public function setModified(DateTimeInterface $modified): self
{ {
$this->modified = $modified; $this->modified = $modified;
@ -259,6 +271,7 @@ class Attachment extends Entity
'is_local' => ['type' => 'bool', 'description' => 'whether the file is stored locally'], 'is_local' => ['type' => 'bool', 'description' => 'whether the file is stored locally'],
'source' => ['type' => 'int', 'default' => null, 'description' => 'Source of the Attachment (upload, TFN, embed)'], 'source' => ['type' => 'int', 'default' => null, 'description' => 'Source of the Attachment (upload, TFN, embed)'],
'scope' => ['type' => 'int', 'default' => null, 'description' => 'visibility scope for this attachment'], 'scope' => ['type' => 'int', 'default' => null, 'description' => 'visibility scope for this attachment'],
'size' => ['type' => 'int', 'description' => 'size of resource when available'],
'modified' => ['type' => 'timestamp', 'not null' => true, 'default' => 'CURRENT_TIMESTAMP', 'description' => 'date this record was modified'], 'modified' => ['type' => 'timestamp', 'not null' => true, 'default' => 'CURRENT_TIMESTAMP', 'description' => 'date this record was modified'],
], ],
'primary key' => ['id'], 'primary key' => ['id'],