const ddg_result_selector = "a[data-testid=\"result-title-a\""
const ddg_result_list_selector = "ol.react-results--main"

let result_list_node
let result_list_observer
const page_links = new Map()

class SearchLink {
    constructor(link_node) {
        this.node = link_node
        this.target = link_node.getAttribute("href")
        this.checked = false
        this.result = undefined
    }
}

function check_links(links) {
    // send a message to background script with a list of URLs to check
    browser.runtime.sendMessage({type: "check", urls: links})
    links.forEach((link) => {page_links.get(link).checked = true})
}

async function message_listener(message) {
    if(message.type === "check_result") {
        console.log(message.url, message.result)
        page_links.get(message.url).result = message.result
    }
}

function get_initial_links() {
    //get links
    const links = document.querySelectorAll(ddg_result_selector)
    links.forEach((node) => {
        const link = new SearchLink(node)
        page_links.set(link.target, link)
    })
    const link_targets = page_links.keys().toArray()
    check_links(link_targets)
}

function update_links() {
    // the result list has updated, add new links and check them
    const links = document.querySelectorAll(ddg_result_selector)
    links.forEach((node) => {
        const link = new SearchLink(node)
        if (page_links.has(link.target)) return
        page_links.set(link.target, link)
    })
    const link_arr = page_links.keys().filter((key) => {
        return !(page_links.get(key).checked)
    }).toArray()

    check_links(link_arr)
}

function setup_result_observer() {
    const config = { childList: true }
    result_list_observer = new MutationObserver(update_links)
    result_list_observer.observe(result_list_node, config)
}

function onload_handler() {
    // get results ol node
    result_list_node = document.querySelector(ddg_result_list_selector)

    get_initial_links()
    setup_result_observer()
}

browser.runtime.onMessage.addListener(message_listener)

if (document.readyState === "loading") {
    document.addEventListener("DOMContentLoaded", onload_handler)
} else {
    onload_handler()
}