n8n/packages/nodes-base/nodes/TravisCi/TravisCi.node.ts

import {
	IExecuteFunctions,
} from 'n8n-core';

import {
	IDataObject,
	INodeExecutionData,
	INodeType,
	INodeTypeDescription,
} from 'n8n-workflow';

import {
	buildFields,
	buildOperations,
} from './BuildDescription';

import {
	travisciApiRequest,
	travisciApiRequestAllItems,
} from './GenericFunctions';

export class TravisCi implements INodeType {
	description: INodeTypeDescription = {
		displayName: 'TravisCI',
		name: 'travisCi',
		icon: 'file:travisci.png',
		group: ['output'],
		version: 1,
		subtitle: '={{$parameter["operation"] + ": " + $parameter["resource"]}}',
		description: 'Consume TravisCI API',
		defaults: {
			name: 'TravisCI',
			color: '#666666',
		},
		inputs: ['main'],
		outputs: ['main'],
		credentials: [
			{
				name: 'travisCiApi',
				required: true,
			},
		],
		properties: [
			{
				displayName: 'Resource',
				name: 'resource',
				type: 'options',
				options: [
					{
						name: ' Build',
						value: 'build',
					},
				],
				default: 'build',
				description: 'Resource to consume.',
			},
			...buildOperations,
			...buildFields,
		],
	};

	async execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]> {
		const items = this.getInputData();
		const returnData: IDataObject[] = [];
		const length = items.length as unknown as number;
		const qs: IDataObject = {};
		let responseData;
		const resource = this.getNodeParameter('resource', 0) as string;
		const operation = this.getNodeParameter('operation', 0) as string;

		for (let i = 0; i < length; i++) {
			if (resource === 'build') {
				//https://developer.travis-ci.com/resource/build#find
				if (operation === 'get') {
					const buildId = this.getNodeParameter('buildId', i) as string;
					const additionalFields = this.getNodeParameter('additionalFields', i) as IDataObject;

					if (additionalFields.include) {
						qs.include = additionalFields.include as string;
					}

					responseData = await travisciApiRequest.call(this, 'GET', `/build/${buildId}`, {}, qs);
				}
				//https://developer.travis-ci.com/resource/builds#for_current_user
				if (operation === 'getAll') {
					const additionalFields = this.getNodeParameter('additionalFields', i) as IDataObject;
					const returnAll = this.getNodeParameter('returnAll', i) as boolean;

					if (additionalFields.sortBy) {
						qs.sort_by = additionalFields.sortBy;
					}

					if (additionalFields.sortBy && additionalFields.order) {
						qs.sort_by = `${additionalFields.sortBy}:${additionalFields.order}`;
					}

					if (additionalFields.include) {
						qs.include = additionalFields.include;
					}

					if (returnAll === true) {
						responseData = await travisciApiRequestAllItems.call(this, 'builds', 'GET', '/builds', {}, qs);

					} else {
						qs.limit = this.getNodeParameter('limit', i) as number;
						responseData = await travisciApiRequest.call(this, 'GET', '/builds', {}, qs);
						responseData = responseData.builds;
					}
				}
				//https://developer.travis-ci.com/resource/build#cancel
				if (operation === 'cancel') {
					const buildId = this.getNodeParameter('buildId', i) as string;
					responseData = await travisciApiRequest.call(this, 'POST', `/build/${buildId}/cancel`, {}, qs);
				}
				//https://developer.travis-ci.com/resource/build#restart
				if (operation === 'restart') {
					const buildId = this.getNodeParameter('buildId', i) as string;
					responseData = await travisciApiRequest.call(this, 'POST', `/build/${buildId}/restart`, {}, qs);
				}
				//https://developer.travis-ci.com/resource/requests#create
				if (operation === 'trigger') {
					let slug = this.getNodeParameter('slug', i) as string;
					const branch = this.getNodeParameter('branch', i) as string;
					const additionalFields = this.getNodeParameter('additionalFields', i) as IDataObject;

					slug = slug.replace(new RegExp(/\//g), '%2F');

					const request: IDataObject = {
						branch,
					};

					if (additionalFields.message) {
						request.message = additionalFields.message as string;
					}

					if (additionalFields.mergeMode) {
						request.merge_mode = additionalFields.mergeMode as string;
					}

					responseData = await travisciApiRequest.call(this, 'POST', `/repo/${slug}/requests`, JSON.stringify({ request }));
				}
			}
			if (Array.isArray(responseData)) {
				returnData.push.apply(returnData, responseData as IDataObject[]);
			} else {
				returnData.push(responseData as IDataObject);
			}
		}
		return [this.helpers.returnJsonArray(returnData)];
	}
}