Note that you don’t have to implement SourceDirectorySet. You can define an interface with just the API you care about yourself, like I did for instance in the Xtext plugin.
I currently don’t know of a plan to make the DefaultSourceDirectorySet of the current Gradle model public.
The new software model already provides a public API for defining custom LanguageSourceSets.