Dont know what your states rules are but in vic the brakes have to be the equivalent of the motor your putting in meaning get good brakes and you should be able to get it engineered providing everything else is good.
Dont get me wrong a head swap like you mentioned sounds cool and all but wouldn't you rather keep the 20v and all etc