Workletize top-level functions in a file with 'worklet' directive

[tjzel]

Summary

With this pull request we are enable workletizing whole files to reduce boilerplate and allow for more elaborate worklet constructs.

Test plan

• [ ] Add relevant test cases.