n8n/packages/nodes-base/nodes/HomeAssistant/GenericFunctions.ts

92 lines
2.6 KiB
TypeScript
Raw Normal View History

import { OptionsWithUri } from 'request';
import { IExecuteFunctions, ILoadOptionsFunctions } from 'n8n-core';
import { IDataObject, INodePropertyOptions, NodeApiError, NodeOperationError } from 'n8n-workflow';
export async function homeAssistantApiRequest(
this: IExecuteFunctions | ILoadOptionsFunctions,
method: string,
resource: string,
body: IDataObject = {},
qs: IDataObject = {},
uri?: string,
option: IDataObject = {},
) {
const credentials = await this.getCredentials('homeAssistantApi');
let options: OptionsWithUri = {
headers: {
Authorization: `Bearer ${credentials.accessToken}`,
},
method,
qs,
body,
uri:
uri ??
`${credentials.ssl === true ? 'https' : 'http'}://${credentials.host}:${
credentials.port
}/api${resource}`,
json: true,
};
options = Object.assign({}, options, option);
if (Object.keys(options.body).length === 0) {
delete options.body;
}
try {
2021-10-17 09:48:24 -07:00
if (this.helpers.request) {
return await this.helpers.request(options);
}
} catch (error) {
throw new NodeApiError(this.getNode(), error);
}
}
2021-10-17 09:48:24 -07:00
export async function getHomeAssistantEntities(
this: IExecuteFunctions | ILoadOptionsFunctions,
domain = '',
) {
2021-10-17 09:48:24 -07:00
const returnData: INodePropertyOptions[] = [];
const entities = await homeAssistantApiRequest.call(this, 'GET', '/states');
for (const entity of entities) {
const entityId = entity.entity_id as string;
if (domain === '' || (domain && entityId.startsWith(domain))) {
const entityName = (entity.attributes.friendly_name as string) || entityId;
2021-10-17 09:48:24 -07:00
returnData.push({
name: entityName,
value: entityId,
});
}
}
return returnData;
}
export async function getHomeAssistantServices(
this: IExecuteFunctions | ILoadOptionsFunctions,
domain = '',
) {
2021-10-17 09:48:24 -07:00
const returnData: INodePropertyOptions[] = [];
const services = await homeAssistantApiRequest.call(this, 'GET', '/services');
if (domain === '') {
// If no domain specified return domains
const domains = services.map(({ domain }: IDataObject) => domain as string).sort();
returnData.push(...domains.map((service: string) => ({ name: service, value: service })));
return returnData;
} else {
// If we have a domain, return all relevant services
const domainServices = services.filter((service: IDataObject) => service.domain === domain);
for (const domainService of domainServices) {
for (const [serviceID, value] of Object.entries(domainService.services)) {
const serviceProperties = value as IDataObject;
const serviceName = serviceProperties.description || serviceID;
returnData.push({
name: serviceName as string,
value: serviceID,
});
}
}
}
return returnData;
}