• Home
• dev.inmo
• micro_utils.ksp.generator
• 0.22.6
• source code

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

o_utils.ksp.generator.0.22.6.source-code.WalkOnKSFiles.kt Maven / Gradle / Ivy

package dev.inmo.micro_ksp.generator

import com.google.devtools.ksp.getAllSuperTypes
import com.google.devtools.ksp.symbol.KSAnnotated
import com.google.devtools.ksp.symbol.KSClassDeclaration
import com.google.devtools.ksp.symbol.KSDeclarationContainer
import com.google.devtools.ksp.symbol.KSFile

fun KSClassDeclaration.findSubClasses(subSymbol: KSAnnotated): Sequence {
    return when (subSymbol) {
        is KSClassDeclaration -> if (subSymbol.getAllSuperTypes().map { it.declaration }.contains(this)) {
            sequenceOf(subSymbol)
        } else {
            sequenceOf()
        }
        else -> sequenceOf()
    } + if (subSymbol is KSDeclarationContainer) {
        subSymbol.declarations.flatMap {
            findSubClasses(it)
        }
    } else {
        sequenceOf()
    }
}

fun KSClassDeclaration.findSubClasses(files: Sequence): Sequence {
    return files.flatMap {
        findSubClasses(it)
    }
}