summaryrefslogtreecommitdiffstats
path: root/plugins/urls-rewriting.lua
blob: 192781e609a36b5506e2a785145d5b6ea6809048 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
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)