From f004673ea5ef041598f5c42c395833454903fe08 Mon Sep 17 00:00:00 2001
From: syuilo <syuilotan@yahoo.co.jp>
Date: Fri, 17 Feb 2017 05:46:14 +0900
Subject: NEW FEATURE: Add url upload

---
 src/api/endpoints.ts                             | 27 ++++++------
 src/api/endpoints/drive/files/upload_from_url.js | 55 ++++++++++++++++++++++++
 2 files changed, 69 insertions(+), 13 deletions(-)
 create mode 100644 src/api/endpoints/drive/files/upload_from_url.js

(limited to 'src/api')

diff --git a/src/api/endpoints.ts b/src/api/endpoints.ts
index 963d1df258..37e3348b86 100644
--- a/src/api/endpoints.ts
+++ b/src/api/endpoints.ts
@@ -54,19 +54,20 @@ export default [
 	{ name: 'notifications/mark_as_read',     shouldBeSignin: true, kind: 'notification-write' },
 	{ name: 'notifications/mark_as_read_all', shouldBeSignin: true, kind: 'notification-write' },
 
-	{ name: 'drive',                shouldBeSignin: true, kind: 'drive-read' },
-	{ name: 'drive/stream',         shouldBeSignin: true, kind: 'drive-read' },
-	{ name: 'drive/files',          shouldBeSignin: true, kind: 'drive-read' },
-	{ name: 'drive/files/create',   shouldBeSignin: true, limitDuration: hour, limitMax: 100, withFile: true, kind: 'drive-write' },
-	{ name: 'drive/files/show',     shouldBeSignin: true, kind: 'drive-read' },
-	{ name: 'drive/files/find',     shouldBeSignin: true, kind: 'drive-read' },
-	{ name: 'drive/files/delete',   shouldBeSignin: true, kind: 'drive-write' },
-	{ name: 'drive/files/update',   shouldBeSignin: true, kind: 'drive-write' },
-	{ name: 'drive/folders',        shouldBeSignin: true, kind: 'drive-read' },
-	{ name: 'drive/folders/create', shouldBeSignin: true, limitDuration: hour, limitMax: 50, kind: 'drive-write' },
-	{ name: 'drive/folders/show',   shouldBeSignin: true, kind: 'drive-read' },
-	{ name: 'drive/folders/find',   shouldBeSignin: true, kind: 'drive-read' },
-	{ name: 'drive/folders/update', shouldBeSignin: true, kind: 'drive-write' },
+	{ name: 'drive',                       shouldBeSignin: true, kind: 'drive-read' },
+	{ name: 'drive/stream',                shouldBeSignin: true, kind: 'drive-read' },
+	{ name: 'drive/files',                 shouldBeSignin: true, kind: 'drive-read' },
+	{ name: 'drive/files/create',          shouldBeSignin: true, limitDuration: hour, limitMax: 100, withFile: true, kind: 'drive-write' },
+	{ name: 'drive/files/upload_from_url', shouldBeSignin: true, limitDuration: hour, limitMax: 10, kind: 'drive-write' },
+	{ name: 'drive/files/show',            shouldBeSignin: true, kind: 'drive-read' },
+	{ name: 'drive/files/find',            shouldBeSignin: true, kind: 'drive-read' },
+	{ name: 'drive/files/delete',          shouldBeSignin: true, kind: 'drive-write' },
+	{ name: 'drive/files/update',          shouldBeSignin: true, kind: 'drive-write' },
+	{ name: 'drive/folders',               shouldBeSignin: true, kind: 'drive-read' },
+	{ name: 'drive/folders/create',        shouldBeSignin: true, limitDuration: hour, limitMax: 50, kind: 'drive-write' },
+	{ name: 'drive/folders/show',          shouldBeSignin: true, kind: 'drive-read' },
+	{ name: 'drive/folders/find',          shouldBeSignin: true, kind: 'drive-read' },
+	{ name: 'drive/folders/update',        shouldBeSignin: true, kind: 'drive-write' },
 
 	{ name: 'users',                    shouldBeSignin: false },
 	{ name: 'users/show',               shouldBeSignin: false },
diff --git a/src/api/endpoints/drive/files/upload_from_url.js b/src/api/endpoints/drive/files/upload_from_url.js
new file mode 100644
index 0000000000..3cc91b946b
--- /dev/null
+++ b/src/api/endpoints/drive/files/upload_from_url.js
@@ -0,0 +1,55 @@
+'use strict';
+
+/**
+ * Module dependencies
+ */
+import * as URL from 'url';
+const download = require('download');
+import * as mongo from 'mongodb';
+import File from '../../../models/drive-file';
+import { validateFileName } from '../../../models/drive-file';
+import User from '../../../models/user';
+import serialize from '../../../serializers/drive-file';
+import create from '../../../common/add-file-to-drive';
+
+/**
+ * Create a file from a URL
+ *
+ * @param {Object} params
+ * @param {Object} user
+ * @return {Promise<object>}
+ */
+module.exports = (params, user) =>
+	new Promise(async (res, rej) =>
+{
+	// Get 'url' parameter
+	const url = params.url;
+	if (url == null) {
+		return rej('url is required');
+	}
+
+	let name = URL.parse(url).pathname.split('/').pop();
+	if (!validateFileName(name)) {
+		name = null;
+	}
+
+	// Get 'folder_id' parameter
+	let folder = params.folder_id;
+	if (folder === undefined || folder === null) {
+		folder = null;
+	} else {
+		folder = new mongo.ObjectID(folder);
+	}
+
+	// Download file
+	const data = await download(url);
+
+	// Create file
+	const driveFile = await create(user, data, name, null, folder);
+
+	// Serialize
+	const fileObj = await serialize(driveFile);
+
+	// Response
+	res(fileObj);
+});
-- 
cgit v1.2.3-freya