simple support for one-line function

Surely it's not smart, and probably won't work well for mixed one-liner
and line-break. But I suppose we generally don't write code like that.