[terraform] IJPL-185785 Added post condition block to self context of resource and data source

GitOrigin-RevId: 8339b597037f833a4af790b358e5cf9f7ec75b46

Author: Nurullokhon Gulomkodirov

terraform/src/org/intellij/terraform/hcl/psi/HCLPsiUtil.kt

@@ -10,6 +10,8 @@ import com.intellij.psi.util.PsiTreeUtil
 import com.intellij.psi.util.parentOfType
 import com.intellij.psi.util.siblings
 import org.intellij.terraform.config.Constants.HCL_CONNECTION_IDENTIFIER
+import org.intellij.terraform.config.Constants.HCL_LIFECYCLE_IDENTIFIER
+import org.intellij.terraform.config.Constants.HCL_POSTCONDITION_BLOCK_IDENTIFIER
 import org.intellij.terraform.config.Constants.HCL_PROVISIONER_IDENTIFIER
 import org.intellij.terraform.config.model.Module
 import org.intellij.terraform.config.model.Variable
@@ -232,7 +234,7 @@ internal fun getDefinedLocalsInModule(element: BaseExpression): List<String> {
 
 internal fun getHclBlockForSelfContext(position: BaseExpression): HCLBlock? {
   val host = position.getHCLHost() ?: return null
-  return getProvisionerOfResource(host) ?: getConnectionOfResource(host)
+  return getProvisionerOfResource(host) ?: getConnectionOfResource(host) ?: getPostConditionOfBlock(host)
 }
 
 internal fun getProvisionerOfResource(host: HCLElement): HCLBlock? {
@@ -254,6 +256,22 @@ internal fun getConnectionOfResource(host: HCLElement): HCLBlock? {
   }
 }
 
+internal fun getPostConditionOfBlock(host: HCLElement): HCLBlock? {
+  val postCondition = host.parentOfType<HCLBlock>() ?: return null
+  if (postCondition.getNameElementUnquoted(0) != HCL_POSTCONDITION_BLOCK_IDENTIFIER)
+    return null
+
+  val lifecycle = postCondition.parentOfType<HCLBlock>()
+  if (lifecycle?.getNameElementUnquoted(0) != HCL_LIFECYCLE_IDENTIFIER) {
+    return null
+  }
+
+  val resourceOrData = lifecycle.parentOfType<HCLBlock>()
+  return if (TfPsiPatterns.ResourceRootBlock.accepts(resourceOrData) || TfPsiPatterns.DataSourceRootBlock.accepts(resourceOrData))
+    resourceOrData
+  else null
+}
+
 internal fun getParentResourceBlock(element: HCLElement): HCLBlock? {
   val resource = element.parentOfType<HCLBlock>()
   return if (TfPsiPatterns.ResourceRootBlock.accepts(resource)) resource else null