(function () {
    var content = document.querySelector('.content');
    if (!content) return;
    var headings = content.querySelectorAll('h1, h2, h3, h4, h5, h6');
    if (!headings.length) return;

    // Section 0 is pre-heading content, editable via full-page edit.
    // Sections 1..N each start at a heading; that is the index sent to the server.
    // Skip headings that already carry a server-rendered edit link.
    headings.forEach(function (h, i) {
        if (h.querySelector('a.btn')) return;
        var a = document.createElement('a');
        a.href = '?edit&section=' + (i + 1);
        a.className = 'btn btn-small';
        a.textContent = 'edit';
        h.appendChild(document.createTextNode(' '))
        h.appendChild(a);
    });
}());
