Try a shift kit before you go tearing that transmission out. Mine had similar behavior and the shift kit fixed it.
The rough idle in drive could be other issues such as a dirty throttle body or IAC, etc.
Can't shift kits increase the chance of a hard failure? As far as TB and IAC go I've been going through this since before I ported and polished my TB, while doing so I cleaned the whole thing out.