[code-completion] Add type context for single-expression functions
Extend the support for single-expression closures to handle
single-expression functions of all kinds. This allows, e.g.
    func foo() -> MyEnum { .<here> }
to complete members of `MyEnum`.