⚡ Add direct message resource to Twitter (#1118)

packages/nodes-base/nodes/Twitter/GenericFunctions.ts

@@ -3,6 +3,7 @@ import {
 } from 'request';
 
 import {
+	BINARY_ENCODING,
 	IExecuteFunctions,
 	IExecuteSingleFunctions,
 	IHookFunctions,
@@ -10,7 +11,8 @@ import {
 } from 'n8n-core';
 
 import {
-	IDataObject,
+	IBinaryKeyData,
+	IDataObject, INodeExecutionData,
 } from 'n8n-workflow';
 
 export async function twitterApiRequest(this: IExecuteFunctions | IExecuteSingleFunctions | ILoadOptionsFunctions | IHookFunctions, method: string, resource: string, body: any = {}, qs: IDataObject = {}, uri?: string, option: IDataObject = {}): Promise<any> { // tslint:disable-line:no-any
@@ -28,6 +30,9 @@ export async function twitterApiRequest(this: IExecuteFunctions | IExecuteSingle
 		if (Object.keys(body).length === 0) {
 			delete options.body;
 		}
+		if (Object.keys(qs).length === 0) {
+			delete options.qs;
+		}
 		//@ts-ignore
 		return await this.helpers.requestOAuth1.call(this, 'twitterOAuth1Api', options);
 	} catch (error) {
@@ -74,3 +79,105 @@ export function chunks (buffer: Buffer, chunkSize: number) {
 
 	return result;
 }
+
+export async function uploadAttachments(this: IExecuteFunctions, binaryProperties: string[], items: INodeExecutionData[], i: number) {
+	
+	const uploadUri = 'https://upload.twitter.com/1.1/media/upload.json';
+	
+	const media: IDataObject[] = [];
+
+	for (const binaryPropertyName of binaryProperties) {
+
+		const binaryData = items[i].binary as IBinaryKeyData;
+
+		if (binaryData === undefined) {
+			throw new Error('No binary data set. So file can not be written!');
+		}
+
+		if (!binaryData[binaryPropertyName]) {
+			continue;
+		}
+
+		let attachmentBody = {};
+		let response: IDataObject = {};
+
+		const isAnimatedWebp = (Buffer.from(binaryData[binaryPropertyName].data, 'base64').toString().indexOf('ANMF') !== -1);
+
+		const isImage = binaryData[binaryPropertyName].mimeType.includes('image');
+
+		if (isImage && isAnimatedWebp) {
+			throw new Error('Animated .webp images are not supported use .gif instead');
+		}
+
+		if (isImage) {
+
+			const attachmentBody = {
+				media_data: binaryData[binaryPropertyName].data,
+			};
+
+			response = await twitterApiRequest.call(this, 'POST', '', {}, {}, uploadUri, { form: attachmentBody });
+
+			media.push(response);
+
+		} else {
+
+			// https://developer.twitter.com/en/docs/media/upload-media/api-reference/post-media-upload-init
+
+			attachmentBody = {
+				command: 'INIT',
+				total_bytes: Buffer.from(binaryData[binaryPropertyName].data, BINARY_ENCODING).byteLength,
+				media_type: binaryData[binaryPropertyName].mimeType,
+			};
+
+			response = await twitterApiRequest.call(this, 'POST', '', {}, {},  uploadUri, { form: attachmentBody });
+
+			const mediaId = response.media_id_string;
+
+			// break the data on 5mb chunks (max size that can be uploaded at once)
+
+			const binaryParts = chunks(Buffer.from(binaryData[binaryPropertyName].data, BINARY_ENCODING), 5242880);
+
+			let index = 0;
+
+			for (const binaryPart of binaryParts) {
+
+				//https://developer.twitter.com/en/docs/media/upload-media/api-reference/post-media-upload-append
+
+				attachmentBody = {
+					name: binaryData[binaryPropertyName].fileName,
+					command: 'APPEND',
+					media_id: mediaId,
+					media_data: Buffer.from(binaryPart).toString('base64'),
+					segment_index: index,
+				};
+
+				response = await twitterApiRequest.call(this, 'POST', '', {}, {}, uploadUri, { form: attachmentBody });
+
+				index++;
+			}
+
+			//https://developer.twitter.com/en/docs/media/upload-media/api-reference/post-media-upload-finalize
+
+			attachmentBody = {
+				command: 'FINALIZE',
+				media_id: mediaId,
+			};
+
+			response = await twitterApiRequest.call(this, 'POST', '', {}, {}, uploadUri, { form: attachmentBody });
+
+			// data has not been uploaded yet, so wait for it to be ready
+			if (response.processing_info) {
+				const { check_after_secs } = (response.processing_info as IDataObject);
+				await new Promise((resolve, reject) => {
+					setTimeout(() => {
+						resolve();
+					}, (check_after_secs as number) * 1000);
+				});
+			}
+			
+			media.push(response);
+		}
+
+		return media;
+	}
+}