Rick and Morty VR Game Style Puzzle

I want to implement a puzzle element into my game and would like to have it like the puzzle system seen in the Rick and Morty VR game. Specifically the engine puzzle. How would I go about having items snap into place and creating a logic that recognizes the correct formations?