Skip to content

Instantly share code, notes, and snippets.

@Nooshu
Created March 2, 2021 00:57
Show Gist options
  • Save Nooshu/2de919f6eb8b69c13a3741700bf8742c to your computer and use it in GitHub Desktop.
Save Nooshu/2de919f6eb8b69c13a3741700bf8742c to your computer and use it in GitHub Desktop.
Easily adding multiple scripts to the head and body of a HTML page.
// set the site we are modifying
const site = 'www.example.com';
// do this on a fetch
addEventListener('fetch', event => {
const request = event.request
const url = new URL(request.url)
event.respondWith(handleRequest(request))
});
async function handleRequest(request) {
// store the URL
const url = new URL(request.url);
// disallow crawlers (write a robots.txt file)
if(url.pathname === "/robots.txt") {
return new Response('User-agent: *\nDisallow: /', {status: 200});
}
// when overrideHost is used in a WPT script, WPT sets x-host to original host i.e. site we want to proxy
// store the value of x-host
const xhost = request.headers.get('x-host');
// If this header is missing, abort
if(!xhost) {
return new Response('x-host header missing', {status: 403});
}
// set our hostname to that listed in the xhost header
url.hostname = xhost;
// look for header that allows us to bypass the transform entirely
const bypassTransform = request.headers.get('x-bypass-transform');
// get the accept header to allow us to examine the type of request it is
const acceptHeader = request.headers.get('accept');
if(xhost === site && (!bypassTransform || (bypassTransform && bypassTransform.indexOf('true') === -1))){
if(acceptHeader && acceptHeader.indexOf('text/html') >= 0){
// store this particular request for modification
let oldResponse = await fetch(url.toString(), request)
// create a new response
let newResponse = new HTMLRewriter()
// remove all external scripts from the page
.on("body > script[src]", new removeElement())
// insert scripts back before the closing body tag
.on("body", new reinsertBodyScripts())
// insert scripts back before the closing head tag
.on("head", new reinsertHeadScripts())
.transform(oldResponse)
// return the modified page along with custom headers
return newResponse
}
}
// otherwise just proxy the request unmodified
return fetch(url.toString(), request)
}
class removeElement {
element(element) {
element.remove();
}
}
class reinsertBodyScripts {
element(element) {
var srcArray = [
'/assets/js/body-script-1.js',
'/assets/js/body-script-2.js'
]
srcArray.forEach(function(val){
element.append(`<script src="${val}"></script>`, {html: true});
})
}
}
class reinsertHeadScripts {
element(element) {
var srcArray = [
'/assets/js/deferred-head-script-1.js',
'/assets/js/deferred-head-script-2.js'
]
srcArray.forEach(function(val){
element.append(`<script src="${val}" defer></script>`, {html: true});
})
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment