aboutsummaryrefslogtreecommitdiff
path: root/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/ErrorTypeProcessor.kt
blob: 956aa16d11c48b2d97a7a06cdf6ca6b51e4c6fe6 (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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
/*
 * Copyright 2020 Google LLC
 * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.google.devtools.ksp.processor

import com.google.devtools.ksp.getClassDeclarationByName
import com.google.devtools.ksp.processing.Resolver
import com.google.devtools.ksp.symbol.KSAnnotated
import com.google.devtools.ksp.symbol.KSPropertyDeclaration
import com.google.devtools.ksp.symbol.KSType

class ErrorTypeProcessor : AbstractTestProcessor() {
    val result = mutableListOf<String>()

    override fun toResult(): List<String> {
        return result
    }

    override fun process(resolver: Resolver): List<KSAnnotated> {
        val classC = resolver.getClassDeclarationByName(resolver.getKSNameFromString("C"))!!
        val errorAtTop = classC.declarations.single { it.simpleName.asString() == "errorAtTop" }
            as KSPropertyDeclaration
        val errorInComponent = classC.declarations.single { it.simpleName.asString() == "errorInComponent" }
            as KSPropertyDeclaration
        result.add(errorAtTop.type.resolve().print() ?: "")
        result.add(errorInComponent.type.resolve().print() ?: "")
        errorInComponent.type.resolve().arguments.forEach { result.add(it.type!!.resolve().print()) }
        result.add(
            "errorInComponent is assignable from errorAtTop: ${
            errorAtTop.type.resolve().isAssignableFrom(errorAtTop.type.resolve())
            }"
        )
        result.add(
            "errorInComponent is assignable from class C: ${
            errorAtTop.type.resolve().isAssignableFrom(classC.asStarProjectedType())
            }"
        )
        result.add(
            "Any is assignable from errorInComponent: ${
            resolver.builtIns.anyType.isAssignableFrom(errorAtTop.type.resolve())
            }"
        )
        result.add(
            "class C is assignable from errorInComponent: ${
            classC.asStarProjectedType().isAssignableFrom(errorAtTop.type.resolve())
            }"
        )
        result.add(
            "Any is assignable from class C: ${
            resolver.builtIns.anyType.isAssignableFrom(classC.asStarProjectedType())
            }"
        )
        val Cls = resolver.getClassDeclarationByName("Cls")!!
        val type = Cls.superTypes.toList()[0].resolve()
        result.add("Cls's super type is Error type: ${type.isError}")
        Cls.annotations.forEach {
            val annotation = it.annotationType.resolve()
            result.add("Cls's annotation is Error type: ${annotation.isError}")
        }
        return emptyList()
    }

    private fun KSType.print(): String {
        return if (this.isError) {
            if (this.declaration.qualifiedName == null) "ERROR TYPE" else
                throw IllegalStateException("Error type should resolve to KSErrorTypeClassDeclaration")
        } else this.declaration.qualifiedName!!.asString()
    }
}