My advise - go for Shimano roller brakes, not disc brakes. Though discs have a bit more stopping power, roller brakes really have enough, they are even used on heavy cargo bikes, so any non-cargo application including an electric bike they can handle.
Here is my electric bike building thread...