Passed
Push — master ( 8113df...5a27e5 )
by Morris
27:51 queued 12:24
created

Azure::objectExists()   A

Complexity

Conditions 3
Paths 3

Size

Total Lines 9
Code Lines 8

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 3
eloc 8
nc 3
nop 1
dl 0
loc 9
rs 10
c 0
b 0
f 0
1
<?php
2
/**
3
 * @copyright Copyright (c) 2018 Robin Appelman <[email protected]>
4
 *
5
 * @license GNU AGPL version 3 or any later version
6
 *
7
 * This program is free software: you can redistribute it and/or modify
8
 * it under the terms of the GNU Affero General Public License as
9
 * published by the Free Software Foundation, either version 3 of the
10
 * License, or (at your option) any later version.
11
 *
12
 * This program is distributed in the hope that it will be useful,
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15
 * GNU Affero General Public License for more details.
16
 *
17
 * You should have received a copy of the GNU Affero General Public License
18
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
19
 *
20
 */
21
22
namespace OC\Files\ObjectStore;
23
24
use MicrosoftAzure\Storage\Blob\BlobRestProxy;
25
use MicrosoftAzure\Storage\Common\Exceptions\ServiceException;
26
use OCP\Files\ObjectStore\IObjectStore;
27
28
class Azure implements IObjectStore {
29
	/** @var string */
30
	private $containerName;
31
	/** @var string */
32
	private $accountName;
33
	/** @var string */
34
	private $accountKey;
35
	/** @var BlobRestProxy|null */
36
	private $blobClient = null;
37
	/** @var string|null */
38
	private $endpoint = null;
39
	/** @var bool  */
40
	private $autoCreate = false;
41
42
	/**
43
	 * @param array $parameters
44
	 */
45
	public function __construct($parameters) {
46
		$this->containerName = $parameters['container'];
47
		$this->accountName = $parameters['account_name'];
48
		$this->accountKey = $parameters['account_key'];
49
		if (isset($parameters['endpoint'])) {
50
			$this->endpoint = $parameters['endpoint'];
51
		}
52
		if (isset($parameters['autocreate'])) {
53
			$this->autoCreate = $parameters['autocreate'];
54
		}
55
	}
56
57
	/**
58
	 * @return BlobRestProxy
59
	 */
60
	private function getBlobClient() {
61
		if (!$this->blobClient) {
62
			$protocol = $this->endpoint ? substr($this->endpoint, 0, strpos($this->endpoint, ':')) : 'https';
63
			$connectionString = "DefaultEndpointsProtocol=" . $protocol . ";AccountName=" . $this->accountName . ";AccountKey=" . $this->accountKey;
64
			if ($this->endpoint) {
65
				$connectionString .= ';BlobEndpoint=' . $this->endpoint;
66
			}
67
			$this->blobClient = BlobRestProxy::createBlobService($connectionString);
68
69
			if ($this->autoCreate) {
70
				try {
71
					$this->blobClient->createContainer($this->containerName);
72
				} catch (ServiceException $e) {
73
					if ($e->getCode() === 409) {
74
						// already exists
75
					} else {
76
						throw $e;
77
					}
78
				}
79
			}
80
		}
81
		return $this->blobClient;
82
	}
83
84
	/**
85
	 * @return string the container or bucket name where objects are stored
86
	 */
87
	public function getStorageId() {
88
		return 'azure::blob::' . $this->containerName;
89
	}
90
91
	/**
92
	 * @param string $urn the unified resource name used to identify the object
93
	 * @return resource stream with the read data
94
	 * @throws \Exception when something goes wrong, message will be logged
95
	 */
96
	public function readObject($urn) {
97
		$blob = $this->getBlobClient()->getBlob($this->containerName, $urn);
98
		return $blob->getContentStream();
99
	}
100
101
	/**
102
	 * @param string $urn the unified resource name used to identify the object
103
	 * @param resource $stream stream with the data to write
104
	 * @throws \Exception when something goes wrong, message will be logged
105
	 */
106
	public function writeObject($urn, $stream) {
107
		$this->getBlobClient()->createBlockBlob($this->containerName, $urn, $stream);
108
	}
109
110
	/**
111
	 * @param string $urn the unified resource name used to identify the object
112
	 * @return void
113
	 * @throws \Exception when something goes wrong, message will be logged
114
	 */
115
	public function deleteObject($urn) {
116
		$this->getBlobClient()->deleteBlob($this->containerName, $urn);
117
	}
118
119
	public function objectExists($urn) {
120
		try {
121
			$this->getBlobClient()->getBlobMetadata($this->containerName, $urn);
122
			return true;
123
		} catch (ServiceException $e) {
124
			if ($e->getCode() === 404) {
125
				return false;
126
			} else {
127
				throw $e;
128
			}
129
		}
130
	}
131
}
132