nir: Refactor algebraic's block walk