import { OptionsWithUri, } from 'request'; import { IExecuteFunctions, IExecuteSingleFunctions, ILoadOptionsFunctions, } from 'n8n-core'; import { IDataObject, } from 'n8n-workflow'; export async function googleApiRequest(this: IExecuteFunctions | IExecuteSingleFunctions | ILoadOptionsFunctions, method: string, resource: string, body: any = {}, qs: IDataObject = {}, uri?: string, headers: IDataObject = {}): Promise { // tslint:disable-line:no-any const options: OptionsWithUri = { headers: { 'Content-Type': 'application/json', }, method, body, qs, uri: uri || `https://bigquery.googleapis.com/bigquery${resource}`, json: true, }; try { if (Object.keys(headers).length !== 0) { options.headers = Object.assign({}, options.headers, headers); } if (Object.keys(body).length === 0) { delete options.body; } //@ts-ignore return await this.helpers.requestOAuth2.call(this, 'googleBigQueryOAuth2Api', options); } catch (error) { if (error.response && error.response.body && error.response.body.error) { let errors = error.response.body.error.errors; errors = errors.map((e: IDataObject) => e.message); // Try to return the error prettier throw new Error( `Google BigQuery error response [${error.statusCode}]: ${errors.join('|')}`, ); } throw error; } } export async function googleApiRequestAllItems(this: IExecuteFunctions | ILoadOptionsFunctions, propertyName: string, method: string, endpoint: string, body: any = {}, query: IDataObject = {}): Promise { // tslint:disable-line:no-any const returnData: IDataObject[] = []; let responseData; query.maxResults = 100; do { responseData = await googleApiRequest.call(this, method, endpoint, body, query); query.pageToken = responseData['nextPageToken']; returnData.push.apply(returnData, responseData[propertyName]); } while ( responseData['nextPageToken'] !== undefined && responseData['nextPageToken'] !== '' ); return returnData; } export function simplify(rows: IDataObject[], fields: string[]) { const results = []; for (const row of rows) { const record: IDataObject = {}; for (const [index, field] of fields.entries()) { record[field] = (row.f as IDataObject[])[index].v; } results.push(record); } return results; }