com.netflix.java.refactor.fix.ChangeField.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-source-refactor Show documentation

Pluggable and distributed refactoring tool for Java source code

There is a newer version: 0.6.2

package com.netflix.java.refactor.fix

import com.netflix.java.refactor.RefactorFix
import com.netflix.java.refactor.RefactorTransaction
import com.netflix.java.refactor.ast.*
import com.sun.source.tree.VariableTree
import com.sun.tools.javac.tree.JCTree
import com.sun.tools.javac.util.Context
import java.util.*

data class ChangeField(val clazz: String, val tx: RefactorTransaction) : RefactoringAstScannerBuilder {
    var refactorTargetType: String? = null
    var refactorName: String? = null
    var refactorDelete: Boolean = false
    
    fun changeType(clazz: String): ChangeField {
        refactorTargetType = clazz
        return this
    }
    
    fun changeType(clazz: Class<*>) = changeType(clazz.name)

    fun changeName(name: String): ChangeField {
        refactorName = name
        return this
    }
    
    fun delete(): RefactorTransaction {
        refactorDelete = true
        return done()
    }
    
    fun done() = tx

    override fun scanner() =
            if (refactorTargetType != null) {
                IfThenScanner(
                        ifFixesResultFrom = ChangeFieldScanner(this),
                        then = arrayOf(
                                AddImport(refactorTargetType!!).scanner(),
                                RemoveImport(clazz).scanner()
                        )
                )
            } else ChangeFieldScanner(this)
}

class ChangeFieldScanner(val op: ChangeField) : FixingScanner() {
    override fun visitVariable(node: VariableTree, context: Context): List? {
        val decl = node as JCTree.JCVariableDecl
        
        if(decl.type?.toString() == op.clazz) {
            return refactorField(decl)
        }
        
        return super.visitVariable(node, context)
    }

    private fun refactorField(decl: JCTree.JCVariableDecl): ArrayList {
        val fixes = ArrayList()

        if (op.refactorDelete) {
            fixes.add(decl.delete())
            return fixes
        }
        
        if (op.refactorTargetType is String && !decl.vartype.matches(op.refactorTargetType!!)) {
            fixes.add(decl.vartype.replace(className(op.refactorTargetType!!)))
        }

        if (op.refactorName is String && decl.name.toString() != op.refactorName) {
            // unfortunately name is not represented with a JCTree, so we have to resort to extraordinary measures...
            val original = decl.name.toString()
            val start = decl.startPosition + sourceText.substring(decl.startPosition..decl.getEndPosition(cu.endPositions))
                .substringBefore(original).length
            
            fixes.add(replace(start..start+original.length, op.refactorName!!))
        }

        return fixes
    }
}