diff options
Diffstat (limited to 'plugins/urls-rewriting.lua')
| -rw-r--r-- | plugins/urls-rewriting.lua | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/plugins/urls-rewriting.lua b/plugins/urls-rewriting.lua new file mode 100644 index 0000000..192781e --- /dev/null +++ b/plugins/urls-rewriting.lua @@ -0,0 +1,46 @@ +prefix_url = config["prefix_url"] + +if not prefix_url then + Plugin.fail("Missing mandatory field: `prefix_url'") +end + +if not Regex.match(prefix_url, "^/(.*)") then + prefix_url = "/" .. prefix_url +end + +if not Regex.match(prefix_url, "(.*)/$") then + prefix_url = prefix_url .. "/" +end + +function prefix_urls (links, attr, prefix_url) + index, link = next(links) + + while index do + href = HTML.get_attribute(link, attr) + + if href then + todo = not Regex.match(href, "^" .. prefix_url .. "*") + + if todo then + if Regex.match(href, "^/") then + href = Regex.replace(href, "^/*", "") + href = prefix_url .. href + end + + if Regex.match(href, "index.html$") then + href = Regex.replace(href, "index.html$", "") + end + + HTML.set_attribute(link, attr, href) + end + end + + index, link = next(links, index) + end +end + +prefix_urls(HTML.select(page, "a"), "href", prefix_url) +prefix_urls(HTML.select(page, "link"), "href", prefix_url) +prefix_urls(HTML.select(page, "img"), "src", prefix_url) +prefix_urls(HTML.select(page, "script"), "src", prefix_url) +prefix_urls(HTML.select(page, "use"), "href", prefix_url) |