You can set the beforeunload listener as follows:
var preventUser = function() {
return "Your work will be lost";
}
window.addEventListener('beforeunload',preventUser);
, , :
function onSubmitForm(){
window.removeEventListener('beforeunload',preventUser);
}
:
<button onclick="onSubmitForm()">Submit Form</button>