n8n/packages/nodes-base/nodes/PostHog/PostHog.node.ts
2022-11-08 15:28:21 +01:00

287 lines
7.3 KiB
TypeScript

import { IExecuteFunctions } from 'n8n-core';
import { IDataObject, INodeExecutionData, INodeType, INodeTypeDescription } from 'n8n-workflow';
import { IAlias, IEvent, IIdentity, ITrack, posthogApiRequest } from './GenericFunctions';
import { aliasFields, aliasOperations } from './AliasDescription';
import { eventFields, eventOperations } from './EventDescription';
import { trackFields, trackOperations } from './TrackDescription';
import { identityFields, identityOperations } from './IdentityDescription';
import moment from 'moment-timezone';
export class PostHog implements INodeType {
description: INodeTypeDescription = {
displayName: 'PostHog',
name: 'postHog',
icon: 'file:postHog.svg',
group: ['input'],
version: 1,
subtitle: '={{$parameter["operation"] + ": " + $parameter["resource"]}}',
description: 'Consume PostHog API',
defaults: {
name: 'PostHog',
},
inputs: ['main'],
outputs: ['main'],
credentials: [
{
name: 'postHogApi',
required: true,
},
],
properties: [
{
displayName: 'Resource',
name: 'resource',
type: 'options',
noDataExpression: true,
options: [
{
name: 'Alias',
value: 'alias',
},
{
name: 'Event',
value: 'event',
},
{
name: 'Identity',
value: 'identity',
},
{
name: 'Track',
value: 'track',
},
],
default: 'event',
},
...aliasOperations,
...aliasFields,
...eventOperations,
...eventFields,
...identityOperations,
...identityFields,
...trackOperations,
...trackFields,
],
};
async execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]> {
const items = this.getInputData();
const returnData: IDataObject[] = [];
const length = items.length;
let responseData;
const resource = this.getNodeParameter('resource', 0) as string;
const operation = this.getNodeParameter('operation', 0) as string;
if (resource === 'alias') {
if (operation === 'create') {
for (let i = 0; i < length; i++) {
try {
const distinctId = this.getNodeParameter('distinctId', i) as string;
const alias = this.getNodeParameter('alias', i) as string;
const additionalFields = this.getNodeParameter('additionalFields', i) as IDataObject;
const context =
(((additionalFields.contextUi as IDataObject) || {})
.contextValues as IDataObject[]) || [];
const event: IAlias = {
type: 'alias',
event: '$create_alias',
context: context.reduce(
(obj, value) => Object.assign(obj, { [`${value.key}`]: value.value }),
{},
),
properties: {
distinct_id: distinctId,
alias,
},
};
Object.assign(event, additionalFields);
if (additionalFields.timestamp) {
additionalFields.timestamp = moment(
additionalFields.timestamp as string,
).toISOString();
}
responseData = await posthogApiRequest.call(this, 'POST', '/batch', event);
returnData.push(responseData);
} catch (error) {
if (this.continueOnFail()) {
returnData.push({ error: error.message });
continue;
}
throw error;
}
}
}
}
if (resource === 'event') {
if (operation === 'create') {
try {
const events: IEvent[] = [];
for (let i = 0; i < length; i++) {
const eventName = this.getNodeParameter('eventName', i) as string;
const distinctId = this.getNodeParameter('distinctId', i) as string;
const additionalFields = this.getNodeParameter('additionalFields', i) as IDataObject;
const properties =
(((additionalFields.propertiesUi as IDataObject) || {})
.propertyValues as IDataObject[]) || [];
const event: IEvent = {
event: eventName,
properties: properties.reduce(
(obj, value) => Object.assign(obj, { [`${value.key}`]: value.value }),
{},
),
};
event.properties['distinct_id'] = distinctId;
Object.assign(event, additionalFields);
if (additionalFields.timestamp) {
additionalFields.timestamp = moment(
additionalFields.timestamp as string,
).toISOString();
}
//@ts-ignore
delete event.propertiesUi;
events.push(event);
}
responseData = await posthogApiRequest.call(this, 'POST', '/capture', { batch: events });
returnData.push(responseData);
} catch (error) {
if (this.continueOnFail()) {
returnData.push({ error: error.message });
} else {
throw error;
}
}
}
}
if (resource === 'identity') {
if (operation === 'create') {
for (let i = 0; i < length; i++) {
try {
const distinctId = this.getNodeParameter('distinctId', i) as string;
const additionalFields = this.getNodeParameter('additionalFields', i) as IDataObject;
const properties =
(((additionalFields.propertiesUi as IDataObject) || {})
.propertyValues as IDataObject[]) || [];
const event: IIdentity = {
event: '$identify',
properties: properties.reduce(
(obj, value) => Object.assign(obj, { [`${value.key}`]: value.value }),
{},
),
distinct_id: distinctId,
};
Object.assign(event, additionalFields);
if (additionalFields.timestamp) {
additionalFields.timestamp = moment(
additionalFields.timestamp as string,
).toISOString();
}
//@ts-ignore
delete event.propertiesUi;
responseData = await posthogApiRequest.call(this, 'POST', '/batch', event);
returnData.push(responseData);
} catch (error) {
if (this.continueOnFail()) {
returnData.push({ error: error.message });
continue;
}
throw error;
}
}
}
}
if (resource === 'track') {
if (operation === 'page' || operation === 'screen') {
for (let i = 0; i < length; i++) {
try {
const distinctId = this.getNodeParameter('distinctId', i) as string;
const name = this.getNodeParameter('name', i) as string;
const additionalFields = this.getNodeParameter('additionalFields', i) as IDataObject;
const context =
(((additionalFields.contextUi as IDataObject) || {})
.contextValues as IDataObject[]) || [];
const properties =
(((additionalFields.propertiesUi as IDataObject) || {})
.propertyValues as IDataObject[]) || [];
const event: ITrack = {
name,
type: operation,
event: `$${operation}`,
context: context.reduce(
(obj, value) => Object.assign(obj, { [`${value.key}`]: value.value }),
{},
),
distinct_id: distinctId,
properties: properties.reduce(
(obj, value) => Object.assign(obj, { [`${value.key}`]: value.value }),
{},
),
};
Object.assign(event, additionalFields);
if (additionalFields.timestamp) {
additionalFields.timestamp = moment(
additionalFields.timestamp as string,
).toISOString();
}
//@ts-ignore
delete event.propertiesUi;
responseData = await posthogApiRequest.call(this, 'POST', '/batch', event);
returnData.push(responseData);
} catch (error) {
if (this.continueOnFail()) {
returnData.push({ error: error.message });
continue;
}
throw error;
}
}
}
}
return [this.helpers.returnJsonArray(returnData)];
}
}