blob: 8ea0591ab1680273104c25e271ed4b147960b2f6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
<?php
/**
* SPDX-FileCopyrightText: 2016-2024 Nextcloud GmbH and Nextcloud contributors
* SPDX-FileCopyrightText: 2016 ownCloud, Inc.
* SPDX-License-Identifier: AGPL-3.0-only
*/
namespace OCA\DAV\Upload;
use OC\Files\ObjectStore\ObjectStoreStorage;
use OCA\DAV\Connector\Sabre\Directory;
use OCP\Files\ObjectStore\IObjectStoreMultiPartUpload;
use OCP\Files\Storage\IStorage;
use Sabre\DAV\Exception\Forbidden;
use Sabre\DAV\ICollection;
class UploadFolder implements ICollection {
public function __construct(
private Directory $node,
private CleanupService $cleanupService,
private IStorage $storage,
) {
}
public function createFile($name, $data = null) {
// TODO: verify name - should be a simple number
try {
$this->node->createFile($name, $data);
} catch (\Exception $e) {
if ($this->node->childExists($name)) {
$child = $this->node->getChild($name);
$child->delete();
}
throw $e;
}
}
public function createDirectory($name) {
throw new Forbidden('Permission denied to create file (filename ' . $name . ')');
}
public function getChild($name) {
if ($name === '.file') {
return new FutureFile($this->node, '.file');
}
return new UploadFile($this->node->getChild($name));
}
public function getChildren() {
$tmpChildren = $this->node->getChildren();
$children = [];
$children[] = new FutureFile($this->node, '.file');
foreach ($tmpChildren as $child) {
$children[] = new UploadFile($child);
}
if ($this->storage->instanceOfStorage(ObjectStoreStorage::class)) {
/** @var ObjectStoreStorage $storage */
$objectStore = $this->storage->getObjectStore();
if ($objectStore instanceof IObjectStoreMultiPartUpload) {
$cache = \OC::$server->getMemCacheFactory()->createDistributed(ChunkingV2Plugin::CACHE_KEY);
$uploadSession = $cache->get($this->getName());
if ($uploadSession) {
$uploadId = $uploadSession[ChunkingV2Plugin::UPLOAD_ID];
$id = $uploadSession[ChunkingV2Plugin::UPLOAD_TARGET_ID];
$parts = $objectStore->getMultipartUploads($this->storage->getURN($id), $uploadId);
foreach ($parts as $part) {
$children[] = new PartFile($this->node, $part);
}
}
}
}
return $children;
}
public function childExists($name) {
if ($name === '.file') {
return true;
}
return $this->node->childExists($name);
}
public function delete() {
$this->node->delete();
// Background cleanup job is not needed anymore
$this->cleanupService->removeJob($this->getName());
}
public function getName() {
return $this->node->getName();
}
public function setName($name) {
throw new Forbidden('Permission denied to rename this folder');
}
public function getLastModified() {
return $this->node->getLastModified();
}
public function getStorage() {
return $this->storage;
}
}
|
