Our current position on this is that extensions will not be supported in the iOS version of the companion as it likely violates App Store Review Guideline 2.5.2, which prohibits apps from downloading and executing code to change the behavior of the app.
However, once we have the system (companion plus build servers) stabilized we may decide to support extensions in compiled apps, in which case you will need to write a Swift version in addition to the Java version.