"use strict"

const { pathToFileURL } = require("node:url")
const fs = require("node:fs")

const {
  filterTokens,
  addError,
  convertHeadingToHTMLFragment,
  getMarkdownHeadings,
} = require("./utils.js")

const customRule = {
  names: ["relative-links"],
  description: "Relative links should be valid",
  tags: ["links"],
  function: (params, onError) => {
    filterTokens(params, "inline", (token) => {
      for (const child of token.children) {
        const { lineNumber, type, attrs } = child

        /** @type {string | null} */
        let hrefSrc = null

        if (type === "link_open") {
          for (const attr of attrs) {
            if (attr[0] === "href") {
              hrefSrc = attr[1]
              break
            }
          }
        }

        if (type === "image") {
          for (const attr of attrs) {
            if (attr[0] === "src") {
              hrefSrc = attr[1]
              break
            }
          }
        }

        if (hrefSrc != null) {
          const url = new URL(hrefSrc, pathToFileURL(params.name))
          const isRelative =
            url.protocol === "file:" && !hrefSrc.startsWith("/")
          if (isRelative) {
            const detail = `Link "${hrefSrc}"`

            if (!fs.existsSync(url)) {
              addError(
                onError,
                lineNumber,
                `${detail} should exist in the file system`,
              )
              continue
            }

            if (type === "link_open" && url.hash !== "") {
              const fileContent = fs.readFileSync(url, { encoding: "utf8" })
              const headings = getMarkdownHeadings(fileContent)

              /** @type {Map<string, number>} */
              const fragments = new Map()

              const headingsHTMLFragments = headings.map((heading) => {
                const fragment = convertHeadingToHTMLFragment(heading)
                const count = fragments.get(fragment) ?? 0
                fragments.set(fragment, count + 1)
                if (count !== 0) {
                  return `${fragment}-${count}`
                }
                return fragment
              })

              if (!headingsHTMLFragments.includes(url.hash)) {
                addError(
                  onError,
                  lineNumber,
                  `${detail} should have a valid fragment`,
                )
              }
            }
          }
        }
      }
    })
  },
}

module.exports = customRule