Horizontal and Angled Arrays in Dynamic Blocks

Good afternoon,

I was trying to make a block where a user could stretch a conveyor. I need to have both a 2D and a 3D version. I can't figure out a way to have one grip array both the 2D version (where it will be arraying horizontal) and the 3D (where the array will be angled). I've attached the block in question. If someone could help me figure out a way to do this, that would be great. 

Thanks!

Image may be NSFW.
Clik here to view.Image may be NSFW.
Clik here to view.Image may be NSFW.
Clik here to view.Image may be NSFW.
Clik here to view.Image may be NSFW.
Clik here to view.

Image may be NSFW.
Clik here to view.