[FIR] Consider call site in argument mapping handling for set operator call

^KT-56542 Fixed
^KT-56714 Fixed

compiler/fir/analysis-tests/testData/resolve/arguments/setWithTrailingLambda.fir.txt

@@ -0,0 +1,38 @@
+FILE: setWithTrailingLambda.kt
+    public final fun test(m: R|MyMap<EditorData, kotlin/Any>|): R|kotlin/Unit| {
+        R|<local>/m|.R|SubstitutionOverride</MyMap.set: R|kotlin/Unit|>|<R|(kotlin/String, @R|kotlin/ParameterName|(name = String(bar))  kotlin/Any) -> kotlin/Unit|>(Q|SomeKey|, <L> = set@fun <anonymous>(<unused var>: R|kotlin/String|, <unused var>: R|@R|kotlin/ParameterName|(name = String(bar))  kotlin/Any|): R|kotlin/Unit| <inline=NoInline>  {
+            ^@set Unit
+        }
+        )
+        R|<local>/m|.R|SubstitutionOverride</MyMap.set: R|kotlin/Unit|>|<R|(kotlin/String, @R|kotlin/ParameterName|(name = String(bar))  kotlin/Any) -> kotlin/Unit|>(Q|SomeKey|, fun <anonymous>(<unused var>: R|kotlin/String|, <unused var>: R|@R|kotlin/ParameterName|(name = String(bar))  kotlin/Any|): R|kotlin/Unit| <inline=NoInline>  {
+            ^ Unit
+        }
+        )
+    }
+    public final data class EditorData : R|kotlin/Any| {
+        public constructor(meta: R|MyMap<EditorData, kotlin/Any>|): R|EditorData| {
+            super<R|kotlin/Any|>()
+        }
+
+        public final val meta: R|MyMap<EditorData, kotlin/Any>| = R|<local>/meta|
+            public get(): R|MyMap<EditorData, kotlin/Any>|
+
+        public final operator fun component1(): R|MyMap<EditorData, kotlin/Any>|
+
+        public final fun copy(meta: R|MyMap<EditorData, kotlin/Any>| = this@R|/EditorData|.R|/EditorData.meta|): R|EditorData|
+
+    }
+    public abstract interface MyMap<Domain, V : R|kotlin/Any|> : R|kotlin/Any| {
+        public abstract operator fun <T : R|V|> set(k: R|Key<T, Domain>|, v: R|T|): R|kotlin/Unit|
+
+    }
+    public abstract interface Key<V : R|kotlin/Any|, in Domain> : R|kotlin/Any| {
+    }
+    public abstract interface EditorDataKey<T : R|kotlin/Any|> : R|Key<T, EditorData>| {
+    }
+    public final object SomeKey : R|EditorDataKey<kotlin/Function2<kotlin/String, @R|kotlin/ParameterName|(name = String(bar))  kotlin/Any, kotlin/Unit>>| {
+        private constructor(): R|SomeKey| {
+            super<R|kotlin/Any|>()
+        }
+
+    }

compiler/fir/analysis-tests/testData/resolve/arguments/setWithTrailingLambda.kt

@@ -0,0 +1,18 @@
+// ISSUE: KT-56714
+
+fun test(m: MyMap<EditorData, Any>) {
+    m.set(SomeKey) { _, _ -> }
+    m[SomeKey] = { _, _ -> }
+}
+
+data class EditorData(val meta: MyMap<EditorData, Any>)
+
+interface MyMap<Domain, V : Any> {
+    operator fun <T : V> set(k: Key<T, Domain>, v: T)
+}
+
+interface Key<V : Any, in Domain>
+
+interface EditorDataKey<T : Any> : Key<T, EditorData>
+
+object SomeKey : EditorDataKey<(String, bar: Any) -> Unit>

compiler/fir/analysis-tests/testData/resolve/arguments/varargFromJava.fir.txt

@@ -0,0 +1,11 @@
+FILE: main.kt
+    public final class Some : R|kotlin/Any| {
+        public constructor(): R|Some| {
+            super<R|kotlin/Any|>()
+        }
+
+        public final fun test(handle: R|VarHandle|): R|kotlin/Unit| {
+            R|<local>/handle|.R|/VarHandle.set|(vararg(this@R|/Some|, Boolean(false)))
+        }
+
+    }

compiler/fir/analysis-tests/testData/resolve/arguments/varargFromJava.kt

@@ -0,0 +1,11 @@
+// FILE: VarHandle.java
+public class VarHandle {
+    public void set(Object... args) {}
+}
+
+// FILE: main.kt
+class Some {
+    fun test(handle: VarHandle) {
+        handle.set(this, false)
+    }
+}