override fun applyFix()

in src/main/kotlin/com/netflix/dgs/plugin/hints/DgsInputArgumentValidationInspector.kt [166:180]

15 lines of code
3 McCabe index (conditional complexity)


        override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
            val method = methodPointer.element ?: return
            val file = methodPointer.element?.sourcePsi?.parentOfType<PsiFile>()
            if(file is PsiJavaFile) {
                val factory: PsiElementFactory = JavaPsiFacade.getInstance(project).elementFactory
                    val param = factory.createParameterFromText(newInputArgument, method)
                    descriptor.psiElement.replace(param)
                project.getService(DgsService::class.java).clearCache()
            } else if(file is KtFile) {
                val psiFactory = KtPsiFactory(project)
                val param = psiFactory.createParameter(newInputArgument)
                descriptor.psiElement.replace(param)
                project.getService(DgsService::class.java).clearCache()
            }
        }