unpkg/server/CloudflareAPI.js

107 lines
2.5 KiB
JavaScript
Raw Normal View History

require('isomorphic-fetch')
2017-09-13 15:35:25 +00:00
const invariant = require('invariant')
const gunzip = require('gunzip-maybe')
const ndjson = require('ndjson')
const CloudflareAPIURL = 'https://api.cloudflare.com'
const CloudflareEmail = process.env.CLOUDFLARE_EMAIL
const CloudflareKey = process.env.CLOUDFLARE_KEY
2017-11-08 16:57:15 +00:00
invariant(CloudflareEmail, 'Missing the $CLOUDFLARE_EMAIL environment variable')
2017-11-08 16:57:15 +00:00
invariant(CloudflareKey, 'Missing the $CLOUDFLARE_KEY environment variable')
function get(path, headers) {
return fetch(`${CloudflareAPIURL}/client/v4${path}`, {
headers: Object.assign({}, headers, {
'X-Auth-Email': CloudflareEmail,
'X-Auth-Key': CloudflareKey
})
})
}
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 => {
2017-11-08 16:57:15 +00:00
return res.json()
})
2017-11-08 18:14:46 +00:00
.then(data => {
2017-11-08 16:57:15 +00:00
if (!data.success) {
console.error(`CloudflareAPI.getJSON failed at ${path}`)
console.error(data)
throw new Error('Failed to getJSON from Cloudflare')
}
2017-11-08 16:57:15 +00:00
return data.result
})
}
function getZones(domains) {
return Promise.all(
2017-11-08 18:14:46 +00:00
(Array.isArray(domains) ? domains : [domains]).map(domain => {
return getJSON(`/zones?name=${domain}`)
})
2017-11-08 18:14:46 +00:00
).then(results => {
return results.reduce((memo, zones) => {
return memo.concat(zones)
})
})
}
function reduceResults(target, values) {
Object.keys(values).forEach(key => {
const value = values[key]
if (typeof value === 'object' && value) {
target[key] = reduceResults(target[key] || {}, value)
} else if (typeof value === 'number') {
target[key] = (target[key] || 0) + values[key]
}
})
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()}`
)
})
2017-11-08 18:14:46 +00:00
).then(results => {
return results.reduce(reduceResults)
})
}
function getJSONStream(path, headers) {
2017-11-08 16:57:15 +00:00
const acceptGzipHeaders = Object.assign({}, headers, {
'Accept-Encoding': 'gzip'
})
2017-11-08 16:57:15 +00:00
return get(path, acceptGzipHeaders)
2017-11-08 18:14:46 +00:00
.then(res => {
2017-11-08 16:57:15 +00:00
return res.body.pipe(gunzip())
})
2017-11-08 18:14:46 +00:00
.then(stream => {
2017-11-08 16:57:15 +00:00
return stream.pipe(ndjson.parse())
})
}
function getLogs(zoneId, startTime, endTime) {
2017-11-08 16:57:15 +00:00
return getJSONStream(
`/zones/${zoneId}/logs/requests?start=${startTime}&end=${endTime}`
)
}
module.exports = {
get,
getJSON,
getZones,
getZoneAnalyticsDashboard,
getJSONStream,
getLogs
}