CSRF regenerate with AJAX

I think I found a solution. Of course no page reloading. CSRF set to regenerate. A lot of forms on a single page. Please test, or point out if this may not work in some scenarios, if you have the time. Heres a simple, handy, dandy, vanilla javascript function that updates ALL present forms on the page with current csrf token:

Code:

function update_csrf_fields(value) {
    let all_forms = document.forms;
    for(e of all_forms) {
        e.querySelector('input[name=csrf_token]').value = value;
    }
}

Please note if your csrf_token is named something else - change the input name. Heres full example cycle:

Form submit using JQuery:

Code:

$(".form").submit(function(e) {
    e.preventDefault();
    let form = this;
    let choice = $(e.target).find("input[type=submit]:focus");
    if (choice) {
        if (choice[0].className === 'delete') {
            $.ajax({
                url: base_url + 'media/delete',
                type: 'POST',
                data: $(form).serializeArray(),
                dataType: 'json',
                headers: {'X-Requested-With': 'XMLHttpRequest'}
            }).done(function (data) {
                //show a message that I deleted the item, or do some fancy form.fadeOut() stuff
                update_csrf_fields(data.csrf_token); ////////TADA! This is where its AT!
            }).fail(function () {
                alert('Ajax Submit Failed ...');
            });
        }

        if (choice[0].className === 'save') {
            //a different submit button was pressed with save class
            //it will now do save actions with ajax
        }
    }
});

The controller ('media/delete'):

PHP Code:

public function delete()
{
        if ($this->request->isAJAX())
        {
            $formDataRaw = $this->request->getRawInput();
            $formDataRaw['csrf_token'] = csrf_hash();
            return $this->response->setJSON($formDataRaw);
        } else {
            return '{ \'error\': \'Invalid Request\'}';
        }
} 


You can see things I made with codeigniter here: itart.pro its not overly impressive as I have very little time to learn.