[wgsl-in] Let lowering contexts point to a Function directly.

Change `StatementContext` and `RuntimeExpressionContext` in
`front::wgsl::lower` to hold a `&mut crate::Function` reference,
rather than separate pointers to individual fields of the `Function`.
This replaces three fields with one, and clarifies their
relationships.