54
54
import org .gradle .api .tasks .bundling .Jar ;
55
55
import org .gradle .api .tasks .scala .ScalaDoc ;
56
56
import org .gradle .api .tasks .testing .Test ;
57
+ import org .gradle .api .tasks .ScalaSourceDirectorySet ;
57
58
58
59
import static org .gradle .api .plugins .JavaBasePlugin .DOCUMENTATION_GROUP ;
59
60
import static org .gradle .api .plugins .JavaBasePlugin .VERIFICATION_GROUP ;
@@ -291,19 +292,18 @@ public void execute(ComponentMetadataContext componentMetadataContext) {
291
292
292
293
@ Override
293
294
public void apply (final Project project ) {
295
+
294
296
SparkVariantPluginExtension extension = project .getExtensions ().create ("sparkVariants" , SparkVariantPluginExtension .class , project );
295
- final JavaPluginConvention javaPluginConvention = project .getConvention ().getPlugin (JavaPluginConvention .class );
296
297
final JavaPluginExtension javaPluginExtension = project .getExtensions ().getByType (JavaPluginExtension .class );
297
298
298
299
// Add a rule that annotates scala-library dependencies with the scala-library capability
299
300
project .getDependencies ().getComponents ().all (ScalaRuntimeCapability .class );
300
301
301
- extension .defaultVariant (sparkVariant -> configureDefaultVariant (project , sparkVariant , javaPluginExtension , javaPluginConvention ));
302
- extension .featureVariants (sparkVariant -> configureVariant (project , sparkVariant , javaPluginExtension , javaPluginConvention ));
302
+ extension .defaultVariant (sparkVariant -> configureDefaultVariant (project , sparkVariant , javaPluginExtension ));
303
+ extension .featureVariants (sparkVariant -> configureVariant (project , sparkVariant , javaPluginExtension ));
303
304
}
304
305
305
- private static void configureDefaultVariant (Project project , SparkVariant sparkVariant , JavaPluginExtension javaPluginExtension ,
306
- JavaPluginConvention javaPluginConvention ) {
306
+ private static void configureDefaultVariant (Project project , SparkVariant sparkVariant , JavaPluginExtension javaPluginExtension ) {
307
307
ConfigurationContainer configurations = project .getConfigurations ();
308
308
String capability = sparkVariant .getCapabilityName (project .getVersion ());
309
309
@@ -316,9 +316,8 @@ private static void configureDefaultVariant(Project project, SparkVariant sparkV
316
316
configureScalaJarClassifiers (project , sparkVariant );
317
317
}
318
318
319
- private static void configureVariant (Project project , SparkVariant sparkVariant , JavaPluginExtension javaPluginExtension ,
320
- JavaPluginConvention javaPluginConvention ) {
321
- SourceSetContainer sourceSets = javaPluginConvention .getSourceSets ();
319
+ private static void configureVariant (Project project , SparkVariant sparkVariant , JavaPluginExtension javaPluginExtension ) {
320
+ SourceSetContainer sourceSets = javaPluginExtension .getSourceSets ();
322
321
ConfigurationContainer configurations = project .getConfigurations ();
323
322
TaskContainer tasks = project .getTasks ();
324
323
Object version = project .getVersion ();
@@ -341,9 +340,8 @@ private static void configureVariant(Project project, SparkVariant sparkVariant,
341
340
}
342
341
343
342
public static SourceSet configureAdditionalVariantSourceSet (Project project , SparkVariant sparkVariant , String sourceSetName ) {
344
- final JavaPluginConvention javaPluginConvention = project .getConvention ().getPlugin (JavaPluginConvention .class );
345
343
final JavaPluginExtension javaPluginExtension = project .getExtensions ().getByType (JavaPluginExtension .class );
346
- SourceSetContainer sourceSets = javaPluginConvention .getSourceSets ();
344
+ SourceSetContainer sourceSets = javaPluginExtension .getSourceSets ();
347
345
ConfigurationContainer configurations = project .getConfigurations ();
348
346
String version = project .getVersion ().toString ();
349
347
@@ -381,7 +379,7 @@ private static SourceSet createVariantSourceSet(SparkVariant sparkVariant, Sourc
381
379
SourceDirectorySet resourcesSourceSet = sourceSet .getResources ();
382
380
resourcesSourceSet .setSrcDirs (Collections .singletonList ("src/" + sourceSetName + "/resources" ));
383
381
384
- SourceDirectorySet scalaSourceSet = getScalaSourceSet (sourceSet ). getScala ( );
382
+ ScalaSourceDirectorySet scalaSourceSet = getScalaSourceSet (sourceSet );
385
383
scalaSourceSet .setSrcDirs (Arrays .asList (
386
384
"src/" + sourceSetName + "/scala" ,
387
385
"src/" + sourceSetName + "/" + sparkVariant .getName ()
@@ -401,9 +399,8 @@ private static void configureAdditionalSourceSetClasspaths(Project project, Conf
401
399
additionalSourceSet .setRuntimeClasspath (project .files (additionalSourceSet .getOutput (), mainSourceSet .getOutput (), additionalRuntimeClasspath ));
402
400
}
403
401
404
- private static DefaultScalaSourceSet getScalaSourceSet (SourceSet sourceSet ) {
405
- Convention sourceSetConvention = (Convention ) InvokerHelper .getProperty (sourceSet , "convention" );
406
- return (DefaultScalaSourceSet ) sourceSetConvention .getPlugins ().get ("scala" );
402
+ private static ScalaSourceDirectorySet getScalaSourceSet (SourceSet sourceSet ) {
403
+ return sourceSet .getExtensions ().getByType (ScalaSourceDirectorySet .class );
407
404
}
408
405
409
406
private static void registerMainVariant (JavaPluginExtension java , SparkVariant sparkVariant , SourceSet main , Object version ) {
@@ -456,7 +453,7 @@ private static void registerVariantScaladoc(Project project, TaskContainer tasks
456
453
scaladocClasspath .from (main .getCompileClasspath ());
457
454
458
455
scalaDoc .setClasspath (scaladocClasspath );
459
- scalaDoc .setSource (getScalaSourceSet (main ). getScala () );
456
+ scalaDoc .setSource (getScalaSourceSet (main ));
460
457
});
461
458
}
462
459
0 commit comments