refactoring

Resolve #7779

packages/backend/src/services/drive/upload-from-url.ts

@@ -0,0 +1,62 @@
+import { URL } from 'url';
+import create from './add-file';
+import { User } from '@/models/entities/user';
+import { driveLogger } from './logger';
+import { createTemp } from '@/misc/create-temp';
+import { downloadUrl } from '@/misc/download-url';
+import { DriveFolder } from '@/models/entities/drive-folder';
+import { DriveFile } from '@/models/entities/drive-file';
+import { DriveFiles } from '@/models/index';
+
+const logger = driveLogger.createSubLogger('downloader');
+
+export default async (
+	url: string,
+	user: { id: User['id']; host: User['host'] } | null,
+	folderId: DriveFolder['id'] | null = null,
+	uri: string | null = null,
+	sensitive = false,
+	force = false,
+	link = false,
+	comment = null
+): Promise<DriveFile> => {
+	let name = new URL(url).pathname.split('/').pop() || null;
+	if (name == null || !DriveFiles.validateFileName(name)) {
+		name = null;
+	}
+
+	// If the comment is same as the name, skip comment
+	// (image.name is passed in when receiving attachment)
+	if (comment !== null && name == comment) {
+		comment = null;
+	}
+
+	// Create temp file
+	const [path, cleanup] = await createTemp();
+
+	// write content at URL to temp file
+	await downloadUrl(url, path);
+
+	let driveFile: DriveFile;
+	let error;
+
+	try {
+		driveFile = await create(user, path, name, comment, folderId, force, link, url, uri, sensitive);
+		logger.succ(`Got: ${driveFile.id}`);
+	} catch (e) {
+		error = e;
+		logger.error(`Failed to create drive file: ${e}`, {
+			url: url,
+			e: e
+		});
+	}
+
+	// clean-up
+	cleanup();
+
+	if (error) {
+		throw error;
+	} else {
+		return driveFile!;
+	}
+};