summaryrefslogtreecommitdiff
path: root/plugins/kotlin/jps/jps-plugin/src/org/jetbrains/kotlin/jps/build/KotlinResourcesRootProvider.kt
blob: aff24b468872c4151f461ae8268769a399a64f9b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
// Copyright 2000-2021 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.

package org.jetbrains.kotlin.jps.build

import org.jetbrains.jps.builders.AdditionalRootsProviderService
import org.jetbrains.jps.builders.BuildTarget
import org.jetbrains.jps.builders.java.ResourceRootDescriptor
import org.jetbrains.jps.builders.java.ResourcesTargetType
import org.jetbrains.jps.builders.storage.BuildDataPaths
import org.jetbrains.jps.incremental.ResourcesTarget
import org.jetbrains.jps.model.java.JavaResourceRootProperties
import org.jetbrains.kotlin.config.ResourceKotlinRootType
import org.jetbrains.kotlin.config.TestResourceKotlinRootType

class KotlinResourcesRootProvider : AdditionalRootsProviderService<ResourceRootDescriptor>(ResourcesTargetType.ALL_TYPES) {
    override fun getAdditionalRoots(
        target: BuildTarget<ResourceRootDescriptor>,
        dataPaths: BuildDataPaths?
    ): List<ResourceRootDescriptor> {
        val moduleBuildTarget = target as? ResourcesTarget ?: return listOf()
        val module = moduleBuildTarget.module

        val result = mutableListOf<ResourceRootDescriptor>()

        // Add source roots with type KotlinResourceRootType.
        // See the note in KotlinSourceRootProvider
        val kotlinResourceRootType = if (target.isTests) TestResourceKotlinRootType else ResourceKotlinRootType
        module.getSourceRoots(kotlinResourceRootType).forEach {
            result.add(
                ResourceRootDescriptor(
                    it.file,
                    target,
                    it.properties.packagePrefix,
                    setOf()
                )
            )
        }

        return result
    }
}

/**
 * Copied from implementation of org.jetbrains.jps.incremental.ResourcesTarget.computeRootDescriptors
 */
private val JavaResourceRootProperties.packagePrefix: String
    get() = relativeOutputPath.replace('/', '.')