I believe you're looking for dependent types: https://en.wikipedia.org/wiki/Dependent_type