all possible, wie ease even as the differs between shallow and deep water anyways and flying units do need to need anyways putting it behind a toggle or switch is not that hard.

what restrictions youre talking about, depending on what you could try a script, but be aware that there is no easy solution. e.g. you could try make ai better with offensive with some formula. (the league of legends dev block contains some strong formula for that purpose)