Fix type resolution for parent elements in nested for directives

Author: xterao

src/main/kotlin/org/domaframework/doma/intellij/common/util/ForDirectiveUtil.kt

@@ -186,7 +186,9 @@ class ForDirectiveUtil {
             forDirectiveBlocks.drop(1).forEach { directive ->
                 // Get the definition type of the target directive
                 val formItem = ForItem(directive.item)
-                if (targetForItem != null && formItem.element.textOffset > targetForItem.textOffset) {
+                if (targetForItem != null &&
+                    formItem.element.textOffset > targetForItem.textOffset
+                ) {
                     return parentClassType
                 }
                 val forDirectiveExpr = formItem.getParentForDirectiveExpr()

src/main/kotlin/org/domaframework/doma/intellij/contributor/sql/provider/SqlParameterCompletionProvider.kt

@@ -421,9 +421,9 @@ class SqlParameterCompletionProvider : CompletionProvider<CompletionParameters>(
     ): Boolean {
         val project = top.project
         val forDirectiveBlocks = ForDirectiveUtil.getForDirectiveBlocks(top)
-        ForDirectiveUtil.findForItem(top, forDirectives = forDirectiveBlocks) ?: return false
+        val forItem = ForDirectiveUtil.findForItem(top, forDirectives = forDirectiveBlocks) ?: return false
 
-        val forItemClassType = ForDirectiveUtil.getForDirectiveItemClassType(project, forDirectiveBlocks) ?: return false
+        val forItemClassType = ForDirectiveUtil.getForDirectiveItemClassType(project, forDirectiveBlocks, forItem) ?: return false
         val specifiedClassType = ForDirectiveUtil.resolveForDirectiveItemClassTypeBySuffixElement(top.text)
         val topClassType =
             if (specifiedClassType != null) {

src/main/kotlin/org/domaframework/doma/intellij/document/generator/DocumentDaoParameterGenerator.kt

@@ -45,8 +45,9 @@ class DocumentDaoParameterGenerator(
         val searchElement = fieldAccessBlocks?.firstOrNull() ?: originalElement
 
         var isBatchAnnotation = false
-        if (ForDirectiveUtil.findForItem(searchElement, forDirectives = forDirectives) != null) {
-            val forItemClassType = ForDirectiveUtil.getForDirectiveItemClassType(project, forDirectives)
+        val forItem = ForDirectiveUtil.findForItem(searchElement, forDirectives = forDirectives)
+        if (forItem != null) {
+            val forItemClassType = ForDirectiveUtil.getForDirectiveItemClassType(project, forDirectives, forItem)
             val specifiedClassType =
                 ForDirectiveUtil.resolveForDirectiveItemClassTypeBySuffixElement(
                     searchElement.text,