The Kelvin timeline did have its interpretation of this plot in the comics. It stuck pretty close to the TOS episode, but with better visuals and pacing. Basically what the episode would have been like if it were filmed today with a modern budget.