unpkg/modules/CloudflareAPI.js

106 lines
2.7 KiB
JavaScript
Raw Normal View History

2018-12-17 17:38:05 +00:00
require('isomorphic-fetch');
const invariant = require('invariant');
const gunzip = require('gunzip-maybe');
const ndjson = require('ndjson');
2018-12-17 17:38:05 +00:00
const cloudflareURL = 'https://api.cloudflare.com/client/v4';
2018-04-04 05:32:32 +00:00
const cloudflareEmail = process.env.CLOUDFLARE_EMAIL;
const cloudflareKey = process.env.CLOUDFLARE_KEY;
2018-02-18 02:00:56 +00:00
invariant(
2018-04-04 05:32:32 +00:00
cloudflareEmail,
2018-12-17 17:38:05 +00:00
'Missing the $CLOUDFLARE_EMAIL environment variable'
2018-02-18 02:00:56 +00:00
);
2018-12-17 17:38:05 +00:00
invariant(cloudflareKey, 'Missing the $CLOUDFLARE_KEY environment variable');
function get(path, headers) {
return fetch(`${cloudflareURL}${path}`, {
headers: Object.assign({}, headers, {
2018-12-17 17:38:05 +00:00
'X-Auth-Email': cloudflareEmail,
'X-Auth-Key': cloudflareKey
})
2018-02-18 02:00:56 +00:00
});
}
function getJSON(path, headers) {
2017-11-08 16:57:15 +00:00
return get(path, headers)
2017-11-08 18:14:46 +00:00
.then(res => {
2018-02-18 02:00:56 +00:00
return res.json();
2017-11-08 16:57:15 +00:00
})
2017-11-08 18:14:46 +00:00
.then(data => {
2017-11-08 16:57:15 +00:00
if (!data.success) {
2018-02-18 02:00:56 +00:00
console.error(`CloudflareAPI.getJSON failed at ${path}`);
console.error(data);
2018-12-17 17:38:05 +00:00
throw new Error('Failed to getJSON from Cloudflare');
2017-11-08 16:57:15 +00:00
}
2018-02-18 02:00:56 +00:00
return data.result;
});
}
function getZones(domains) {
return Promise.all(
2018-05-21 20:35:14 +00:00
(Array.isArray(domains) ? domains : [domains]).map(domain =>
getJSON(`/zones?name=${domain}`)
)
).then(results => results.reduce((memo, zones) => memo.concat(zones)));
}
function reduceResults(target, values) {
Object.keys(values).forEach(key => {
2018-02-18 02:00:56 +00:00
const value = values[key];
2018-12-17 17:38:05 +00:00
if (typeof value === 'object' && value) {
2018-02-18 02:00:56 +00:00
target[key] = reduceResults(target[key] || {}, value);
2018-12-17 17:38:05 +00:00
} else if (typeof value === 'number') {
2018-02-18 02:00:56 +00:00
target[key] = (target[key] || 0) + values[key];
}
2018-02-18 02:00:56 +00:00
});
2018-02-18 02:00:56 +00:00
return target;
}
function getZoneAnalyticsDashboard(zones, since, until) {
return Promise.all(
2017-11-08 18:14:46 +00:00
(Array.isArray(zones) ? zones : [zones]).map(zone => {
2017-11-08 16:57:15 +00:00
return getJSON(
`/zones/${
zone.id
}/analytics/dashboard?since=${since.toISOString()}&until=${until.toISOString()}`
2018-02-18 02:00:56 +00:00
);
})
2018-05-21 20:35:14 +00:00
).then(results => results.reduce(reduceResults));
}
function getJSONStream(path, headers) {
const gzipHeaders = Object.assign({}, headers, {
2018-12-17 17:38:05 +00:00
'Accept-Encoding': 'gzip'
2018-02-18 02:00:56 +00:00
});
2017-11-08 16:57:15 +00:00
return get(path, gzipHeaders)
2018-05-21 20:35:14 +00:00
.then(res => res.body.pipe(gunzip()))
.then(stream => stream.pipe(ndjson.parse()));
}
function getLogs(zoneId, startTime, endTime, fieldsArray) {
2018-12-17 17:38:05 +00:00
const fields = fieldsArray.join(',');
// console.log(
// `https://api.cloudflare.com/client/v4/zones/${zoneId}/logs/received?start=${startTime}&end=${endTime}&fields=${fields}`
// );
2018-02-18 02:00:56 +00:00
return getJSONStream(
`/zones/${zoneId}/logs/received?start=${startTime}&end=${endTime}&fields=${fields}`
2018-02-18 02:00:56 +00:00
);
}
module.exports = {
get,
getJSON,
getZones,
getZoneAnalyticsDashboard,
getJSONStream,
getLogs
2018-02-18 02:00:56 +00:00
};