Created
March 1, 2019 16:48
-
-
Save mhartington/e7232331998e7b5a5bdfa3ab7569be40 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env node | |
const puppeteer = require('puppeteer'); | |
const fs = require('fs'); | |
const args = process.argv.slice(2); | |
const host = args[0] || 'http://127.0.0.1:8080'; | |
const indexMatches = [ | |
'.gz', | |
'.map', | |
'3rdpartylicenses.txt', | |
'ngsw-manifest.json', | |
'ngsw.json', | |
'worker-basic.min.js', | |
'ngsw-worker.js', | |
'favicon', | |
'index.html', | |
'manifest.webmanifest', | |
'.svg' | |
]; | |
let results = ''; | |
async function main() { | |
const browser = await puppeteer.launch(); | |
const page = await browser.newPage(); | |
await page._client.send('ServiceWorker.disable') | |
page.on('console', msg => console.log(msg.text())); | |
console.log('Page: loaded'); | |
await page.goto(host); | |
const allRequests = new Map(); | |
page.on('request', req => { | |
allRequests.set(req.url(), req); | |
}); | |
await page.reload({ waitUntil: 'networkidle2' }); | |
Array.from(allRequests.values()).forEach(req => { | |
const url = req.url(); | |
// filter out urls that match these extensions | |
for (const exlude of indexMatches) { | |
if (url.indexOf(exlude) != -1) return false; | |
} | |
// if external, dont worry about it for now | |
// | |
const origin = new URL(host); | |
if (url.indexOf(origin.origin) === -1) return false; | |
// Format the url to remove the host | |
const formatted = url.replace(`${origin.origin}/`, ''); | |
if (origin.pathname.includes(formatted)) return false; | |
// if it's an empty string, just ignore it | |
if (!formatted) return false; | |
let type = url.substr(-3) == 'css' ? 'style' : 'script'; | |
results += `</${formatted}>;rel=preload;as=${type},`; | |
}); | |
await browser.close(); | |
updateWith(results); | |
} | |
function updateWith(result) { | |
fs.readFile('firebase.json', 'utf8', function(err, data) { | |
if (err) { | |
return console.log(err); | |
} | |
let re = /("headers":\s*\[\s*{\s*"key":\s*"Link",\s*"value":\s*")(.*)("\s*}\s*\])/gm; | |
if (re.exec(data)) { | |
let newConfig = data.replace(re, `$1${result}$3`); | |
fs.writeFile('firebase.json', newConfig, 'utf8', function(err) { | |
if (err) return console.log(err); | |
console.log('firebase.json updated successfully.'); | |
}); | |
} else { | |
console.log("Couldn't find a valid firebase config to update."); | |
return; | |
} | |
}); | |
} | |
main(); |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment