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

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

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

import {
	pagerDutyApiRequest,
	pagerDutyApiRequestAllItems,
	keysToSnakeCase,
} from './GenericFunctions';

import {
	incidentFields,
	incidentOperations,
} from './IncidentDescription';

import {
	incidentNoteFields,
	incidentNoteOperations,
} from './IncidentNoteDescription';

import {
	logEntryFields,
	logEntryOperations,
} from './logEntryDescription';

import {
	IIncident,
} from './IncidentInterface';

import {
	snakeCase,
} from 'change-case';

import * as moment from 'moment-timezone';

export class PagerDuty implements INodeType {
	description: INodeTypeDescription = {
		displayName: 'PagerDuty',
		name: 'pagerDuty',
		icon: 'file:pagerDuty.png',
		group: ['output'],
		version: 1,
		subtitle: '={{$parameter["operation"] + ": " + $parameter["resource"]}}',
		description: 'Consume PagerDuty API',
		defaults: {
			name: 'PagerDuty',
			color: '#49a25f',
		},
		inputs: ['main'],
		outputs: ['main'],
		credentials: [
			{
				name: 'pagerDutyApi',
				required: true,
			},
		],
		properties: [
			{
				displayName: 'Resource',
				name: 'resource',
				type: 'options',
				options: [
					{
						name: 'Incident',
						value: 'incident',
					},
					{
						name: 'Incident Note',
						value: 'incidentNote',
					},
					{
						name: 'Log Entry',
						value: 'logEntry',
					},
				],
				default: 'incident',
				description: 'Resource to consume.',
			},
			// INCIDENT
			...incidentOperations,
			...incidentFields,
			// INCIDENT NOTE
			...incidentNoteOperations,
			...incidentNoteFields,
			// LOG ENTRY
			...logEntryOperations,
			...logEntryFields,
		],
	};

	methods = {
		loadOptions: {
			// Get all the available escalation policies to display them to user so that he can
			// select them easily
			async getEscalationPolicies(this: ILoadOptionsFunctions): Promise<INodePropertyOptions[]> {
				const returnData: INodePropertyOptions[] = [];
				const escalationPolicies = await pagerDutyApiRequestAllItems.call(this, 'escalation_policies', 'GET', '/escalation_policies');
				for (const escalationPolicy of escalationPolicies) {
					const escalationPolicyName = escalationPolicy.name;
					const escalationPolicyId = escalationPolicy.id;
					returnData.push({
						name: escalationPolicyName,
						value: escalationPolicyId,
					});
				}
				return returnData;
			},
			// Get all the available priorities to display them to user so that he can
			// select them easily
			async getPriorities(this: ILoadOptionsFunctions): Promise<INodePropertyOptions[]> {
				const returnData: INodePropertyOptions[] = [];
				const priorities = await pagerDutyApiRequestAllItems.call(this, 'priorities', 'GET', '/priorities');
				for (const priority of priorities) {
					const priorityName = priority.name;
					const priorityId = priority.id;
					const priorityDescription = priority.description;
					returnData.push({
						name: priorityName,
						value: priorityId,
						description: priorityDescription,
					});
				}
				return returnData;
			},
			// Get all the available services to display them to user so that he can
			// select them easily
			async getServices(this: ILoadOptionsFunctions): Promise<INodePropertyOptions[]> {
				const returnData: INodePropertyOptions[] = [];
				const services = await pagerDutyApiRequestAllItems.call(this, 'services', 'GET', '/services');
				for (const service of services) {
					const serviceName = service.name;
					const serviceId = service.id;
					returnData.push({
						name: serviceName,
						value: serviceId,
					});
				}
				return returnData;
			},
			// Get all the timezones to display them to user so that he can
			// select them easily
			async getTimezones(this: ILoadOptionsFunctions): Promise<INodePropertyOptions[]> {
				const returnData: INodePropertyOptions[] = [];
				for (const timezone of moment.tz.names()) {
					const timezoneName = timezone;
					const timezoneId = timezone;
					returnData.push({
						name: timezoneName,
						value: timezoneId,
					});
				}
				return returnData;
			},
		},
	};

	async execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]> {
		const items = this.getInputData();
		const returnData: IDataObject[] = [];
		const length = items.length as unknown as number;
		let responseData;
		const qs: IDataObject = {};
		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 === 'incident') {
				//https://api-reference.pagerduty.com/#!/Incidents/post_incidents
				if (operation === 'create') {
					const title = this.getNodeParameter('title', i) as string;
					const serviceId = this.getNodeParameter('serviceId', i) as string;
					const email = this.getNodeParameter('email', i) as string;
					const additionalFields = this.getNodeParameter('additionalFields', i) as IDataObject;
					const conferenceBridge = (this.getNodeParameter('conferenceBridgeUi', i) as IDataObject).conferenceBridgeValues as IDataObject;
					const body: IIncident = {
						type: 'incident',
						title,
						service: {
							id: serviceId,
							type: 'service_reference',
						},
					};
					if (additionalFields.details) {
						body.body = {
							type: 'incident_body',
							details: additionalFields.details,
						};
					}
					if (additionalFields.priorityId) {
						body.priority = {
							id: additionalFields.priorityId,
							type: 'priority_reference',
						};
					}
					if (additionalFields.escalationPolicyId) {
						body.escalation_policy = {
							id: additionalFields.escalationPolicyId,
							type: 'escalation_policy_reference',
						};
					}
					if (additionalFields.urgency) {
						body.urgency = additionalFields.urgency as string;
					}
					if (additionalFields.incidentKey) {
						body.incident_key = additionalFields.incidentKey as string;
					}
					if (conferenceBridge) {
						body.conference_bridge = {
							conference_number: conferenceBridge.conferenceNumber,
							conference_url: conferenceBridge.conferenceUrl,
						};
					}
					responseData = await pagerDutyApiRequest.call(this, 'POST', '/incidents', { incident: body }, {}, undefined, { from: email });
					responseData = responseData.incident;
				}
				//https://api-reference.pagerduty.com/#!/Incidents/get_incidents_id
				if (operation === 'get') {
					const incidentId = this.getNodeParameter('incidentId', i) as string;
					responseData = await pagerDutyApiRequest.call(this, 'GET', `/incidents/${incidentId}`);
					responseData = responseData.incident;
				}
				//https://api-reference.pagerduty.com/#!/Incidents/get_incidents
				if (operation === 'getAll') {
					const returnAll = this.getNodeParameter('returnAll', 0) as boolean;
					const options = this.getNodeParameter('options', 0) as IDataObject;
					if (options.userIds) {
						options.userIds = (options.userIds as string).split(',') as string[];
					}
					if (options.teamIds) {
						options.teamIds = (options.teamIds as string).split(',') as string[];
					}
					if (options.include) {
						options.include = (options.include as string[]).map((e) => snakeCase(e));
					}
					if (options.sortBy) {
						options.sortBy = options.sortBy as string;
					}
					Object.assign(qs, keysToSnakeCase(options)[0]);
					if (returnAll) {
						responseData = await pagerDutyApiRequestAllItems.call(this, 'incidents', 'GET', '/incidents', {}, qs);
					} else {
						qs.limit = this.getNodeParameter('limit', 0) as number;
						responseData = await pagerDutyApiRequest.call(this, 'GET', '/incidents', {}, qs);
						responseData = responseData.incidents;
					}
				}
				//https://api-reference.pagerduty.com/#!/Incidents/put_incidents_id
				if (operation === 'update') {
					const incidentId = this.getNodeParameter('incidentId', i) as string;
					const email = this.getNodeParameter('email', i) as string;
					const conferenceBridge = (this.getNodeParameter('conferenceBridgeUi', i) as IDataObject).conferenceBridgeValues as IDataObject;
					const updateFields = this.getNodeParameter('updateFields', i) as IDataObject;
					const body: IIncident = {
						type: 'incident',
					};
					if (updateFields.title) {
						body.title = updateFields.title as string;
					}
					if (updateFields.escalationLevel) {
						body.escalation_level = updateFields.escalationLevel as number;
					}
					if (updateFields.details) {
						body.body = {
							type: 'incident_body',
							details: updateFields.details,
						};
					}
					if (updateFields.priorityId) {
						body.priority = {
							id: updateFields.priorityId,
							type: 'priority_reference',
						};
					}
					if (updateFields.escalationPolicyId) {
						body.escalation_policy = {
							id: updateFields.escalationPolicyId,
							type: 'escalation_policy_reference',
						};
					}
					if (updateFields.urgency) {
						body.urgency = updateFields.urgency as string;
					}
					if (updateFields.resolution) {
						body.resolution = updateFields.resolution as string;
					}
					if (updateFields.status) {
						body.status = updateFields.status as string;
					}
					if (conferenceBridge) {
						body.conference_bridge = {
							conference_number: conferenceBridge.conferenceNumber,
							conference_url: conferenceBridge.conferenceUrl,
						};
					}
					responseData = await pagerDutyApiRequest.call(this, 'PUT', `/incidents/${incidentId}`, { incident: body }, {}, undefined, { from: email });
					responseData = responseData.incident;
				}
			}
			if (resource === 'incidentNote') {
				//https://api-reference.pagerduty.com/#!/Incidents/post_incidents_id_notes
				if (operation === 'create') {
					const incidentId = this.getNodeParameter('incidentId', i) as string;
					const content = this.getNodeParameter('content', i) as string;
					const email = this.getNodeParameter('email', i) as string;
					const body: IDataObject = {
						content,
					};
					responseData = await pagerDutyApiRequest.call(this, 'POST', `/incidents/${incidentId}/notes`, { note: body }, {}, undefined, { from: email });
				}
				//https://api-reference.pagerduty.com/#!/Incidents/get_incidents_id_notes
				if (operation === 'getAll') {
					const incidentId = this.getNodeParameter('incidentId', i) as string;
					const returnAll = this.getNodeParameter('returnAll', 0) as boolean;
					if (returnAll) {
						responseData = await pagerDutyApiRequestAllItems.call(this, 'notes', 'GET', `/incidents/${incidentId}/notes`, {}, qs);
					} else {
						qs.limit = this.getNodeParameter('limit', 0) as number;
						responseData = await pagerDutyApiRequest.call(this, 'GET', `/incidents/${incidentId}/notes`, {}, qs);
						responseData = responseData.notes;
					}
				}
			}
			if (resource === 'logEntry') {
				//https://api-reference.pagerduty.com/#!/Log_Entries/get_log_entries_id
				if (operation === 'get') {
					const logEntryId = this.getNodeParameter('logEntryId', i) as string;
					responseData = await pagerDutyApiRequest.call(this, 'GET', `/log_entries/${logEntryId}`);
					responseData = responseData.log_entry;
				}
				//https://api-reference.pagerduty.com/#!/Log_Entries/get_log_entries
				if (operation === 'getAll') {
					const options = this.getNodeParameter('options', i) as IDataObject;
					Object.assign(qs, options);
					keysToSnakeCase(qs);
					const returnAll = this.getNodeParameter('returnAll', 0) as boolean;
					if (returnAll) {
						responseData = await pagerDutyApiRequestAllItems.call(this, 'log_entries', 'GET', '/log_entries', {}, qs);
					} else {
						qs.limit = this.getNodeParameter('limit', 0) as number;
						responseData = await pagerDutyApiRequest.call(this, 'GET', 'log_entries', {}, qs);
						responseData = responseData.log_entries;
					}
				}
			}
			if (Array.isArray(responseData)) {
				returnData.push.apply(returnData, responseData as IDataObject[]);
			} else {
				returnData.push(responseData as IDataObject);
			}
		}
		return [this.helpers.returnJsonArray(returnData)];
	}
}