mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-14 15:53:28 +01:00
22b4297fc5
- Fix: a few broken tests due to iana.org requiring a user agent in its requests rewrite: - introduced a new JSWorkerRewriter class in order to support rewriting via wombat workers in the context of all supported worker variants via - ensured rewriter app correctly sets the static prefix wombat: - add wombat as submodule!
31 lines
1.3 KiB
Python
31 lines
1.3 KiB
Python
from pywb.rewrite.content_rewriter import StreamingRewriter, WORKER_MODS
|
|
|
|
__all__ = ["JSWorkerRewriter"]
|
|
|
|
INJECT = "(function() { self.importScripts('%s'); new WBWombat(%s); })();"
|
|
INIT = "{'prefix': '%s', 'prefixMod': '%s/', 'originalURL': '%s'}"
|
|
|
|
|
|
class JSWorkerRewriter(StreamingRewriter):
|
|
"""A simple rewriter for rewriting web or service workers.
|
|
The only rewriting that occurs is the injection of the init code
|
|
for wombatWorkers.js.
|
|
This allows for all them to operate as expected on the live web.
|
|
"""
|
|
|
|
def __init__(self, url_rewriter, align_to_line=True, first_buff=''):
|
|
"""Initialize a new JSWorkerRewriter
|
|
|
|
:param UrlRewriter url_rewriter: The url rewriter for this rewrite
|
|
:param bool align_to_line: Should the response stream be aliened to line boundaries
|
|
:param str first_buff: The first string to be added to the rewrite
|
|
:rtype: None
|
|
"""
|
|
super(JSWorkerRewriter, self).__init__(url_rewriter, align_to_line, first_buff)
|
|
wb_url = self.url_rewriter.wburl
|
|
if wb_url.mod in WORKER_MODS:
|
|
rw_url = self.url_rewriter.pywb_static_prefix + "wombatWorkers.js"
|
|
prefix = self.url_rewriter.full_prefix
|
|
init = INIT % (prefix, prefix + 'wkrf_', wb_url.url)
|
|
self.first_buff = INJECT % (rw_url, init)
|