Refactor

1 files changed, 20 insertions, 0 deletions

diff --git a/src/misc/get-drive-file-url.ts b/src/misc/get-drive-file-url.ts
new file mode 100644
index 0000000000..0fe467261e
--- /dev/null
+++ b/src/misc/get-drive-file-url.ts
@@ -0,0 +1,20 @@
+import { IDriveFile } from '../models/drive-file';
+import config from '../config';
+
+export default function(file: IDriveFile, thumbnail = false): string {
+	if (file == null) return null;
+
+	if (file.metadata.withoutChunks) {
+		if (thumbnail) {
+			return file.metadata.thumbnailUrl || file.metadata.url;
+		} else {
+			return file.metadata.url;
+		}
+	} else {
+		if (thumbnail) {
+			return `${config.drive_url}/${file._id}?thumbnail`;
+		} else {
+			return `${config.drive_url}/${file._id}`;
+		}
+	}
+}