Yes, the AX15 bell housing will boltup on 4.0's and 4.2's.
No, there is no way you will swap driveshafts without modification. Too much difference in length. I don't know how much difference in length there is between the two trannies but I believe the Ax15 is longer. You MIGHT have to change the length of 4 driveshafts if you do the swap. Most people would not swap a BA10 into a Wrangler. They are notoriously weak. (Also expensive to repair and hard to find in junk yards)