[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
// available methods, so should be safe
$mimetype = $sfile->getMimeType();
Event::handle('AttachmentValidation', [&$sfile, &$mimetype]);
Event::handle('AttachmentValidation', [&$sfile, &$mimetype, &$title]);
$attachment = Attachment::create([
'file_hash' => $hash,
'gsactor_id' => $actor_id,
@ -55,6 +55,7 @@ class GSFile
'title' => $title ?: _m('Untitled attachment'),
'filename' => $hash,
'is_local' => $is_local,
'size' => $sfile->getSize(),
]);
$sfile->move($dest_dir, $hash);
DB::persist($attachment);

View File

@ -54,6 +54,7 @@ class Attachment extends Entity
private ?bool $is_local;
private ?int $source;
private ?int $scope;
private ?int $size;
private \DateTimeInterface $modified;
public function setId(int $id): self
@ -177,6 +178,17 @@ class Attachment extends Entity
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
{
$this->modified = $modified;
@ -259,6 +271,7 @@ class Attachment extends Entity
'is_local' => ['type' => 'bool', 'description' => 'whether the file is stored locally'],
'source' => ['type' => 'int', 'default' => null, 'description' => 'Source of the Attachment (upload, TFN, embed)'],
'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'],
],
'primary key' => ['id'],