Configuring source jars for gradle dependencies

Is there a way to configure source jars for dependencies from the gradle build files when using IntelliJs import?

For Eclipse/Buildship I do the following in build.gradle because Buildship uses the eclipse dsl in the gradle files:

apply plugin: 'eclipse'
apply plugin: 'java'

dependencies {
  compile group: "somegroup", name: 'stuff'
  testCompile group: "somegroup", name: 'stuff', classifier: "test"
}

repositories {
 flatDir name: "flatdir-repo", dir: "stuffimports/"
}

eclipse {
 classpath {
   file {
     whenMerged {
       def libs = entries.findAll { it.path ==~ /.+stuff(|-test)\.jar$/ }
       libs.each {
         it.sourcePath = fileReference(file(it.path.replaceAll(/\.jar/, /-src.jar/)))
       }
     }
   }
 }

Basically: Is there any documentation on how/if the IntelliJs "Import existing module from Gradle" interacts with Gradle's 'idea' plugin? Or any good examples?

https://www.jetbrains.com/help/idea/2017.1/gradle.html and https://docs.gradle.org/current/userguide/idea_plugin.html doesn't say anything about this.

/Thanks