Skip to content

Commit 5089291

Browse files
committed
Enhance SQL injection processing by adding DAO class check for Java/Kotlin files
1 parent 9214e89 commit 5089291

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/main/kotlin/org/domaframework/doma/intellij/formatter/processor/SqlInjectionPostProcessor.kt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ import com.intellij.psi.PsiElement
2121
import com.intellij.psi.PsiFile
2222
import com.intellij.psi.codeStyle.CodeStyleSettings
2323
import com.intellij.psi.impl.source.codeStyle.PostFormatProcessor
24+
import org.domaframework.doma.intellij.common.dao.getDaoClass
2425
import org.domaframework.doma.intellij.common.isJavaOrKotlinFileType
2526
import org.domaframework.doma.intellij.formatter.visitor.DaoInjectionSqlVisitor
2627

@@ -35,7 +36,7 @@ class SqlInjectionPostProcessor : PostFormatProcessor {
3536
rangeToReformat: TextRange,
3637
settings: CodeStyleSettings,
3738
): TextRange {
38-
if (!isJavaOrKotlinFileType(source)) return rangeToReformat
39+
if (!isJavaOrKotlinFileType(source) || getDaoClass(source) == null) return rangeToReformat
3940

4041
processInjected(source)
4142
return rangeToReformat

0 commit comments

Comments
 (0)