[Assign plugin] Add a compiler plugin for overloading assign ('=') operator

2022-09-01 07:51:38 +02:00
parent 6052962f50
commit 09d6dfc8bf
62 changed files with 2822 additions and 6 deletions
@@ -0,0 +1,55 @@
+annotation class ValueContainer
+
+@ValueContainer
+data class StringProperty(var v: String) {
+    fun assign(v: String) {
+        this.v = v
+    }
+    fun assign(v: StringProperty) {
+        this.v = v.get()
+    }
+    fun get(): String = v
+}
+
+data class Task(val input: StringProperty)
+
+fun `should not effect error reporting for other operators`() {
+    val task = Task(StringProperty("Fail"))
+    val nullTask: Task? = null
+
+    // a.b += c
+    task.input <!UNRESOLVED_REFERENCE_WRONG_RECEIVER!>+=<!> StringProperty("Fail")
+    nullTask?.input <!UNRESOLVED_REFERENCE_WRONG_RECEIVER!>+=<!> StringProperty("Fail")
+
+    // a.b <= c
+    task.input <!UNRESOLVED_REFERENCE!><=<!> StringProperty("Fail")
+    nullTask?.input <!UNRESOLVED_REFERENCE!><=<!> StringProperty("Fail")
+
+    // a.b >= c
+    task.input <!UNRESOLVED_REFERENCE!>>=<!> StringProperty("Fail")
+    nullTask?.input <!UNRESOLVED_REFERENCE!>>=<!> StringProperty("Fail")
+
+    // a.b[c] = d
+    task.input<!NO_SET_METHOD!><!UNRESOLVED_REFERENCE!>[<!>0<!UNRESOLVED_REFERENCE!>]<!><!> = StringProperty("Fail")
+    nullTask?.input<!NO_SET_METHOD!><!UNRESOLVED_REFERENCE!>[<!>0<!UNRESOLVED_REFERENCE!>]<!><!> = StringProperty("Fail")
+
+    // a.b[c, d] = e
+    task.input<!NO_SET_METHOD!><!UNRESOLVED_REFERENCE!>[<!>0, 0<!UNRESOLVED_REFERENCE!>]<!><!> = StringProperty("Fail")
+    nullTask?.input<!NO_SET_METHOD!><!UNRESOLVED_REFERENCE!>[<!>0, 0<!UNRESOLVED_REFERENCE!>]<!><!> = StringProperty("Fail")
+
+    // a.b[c,..,d] = e
+    task.input<!NO_SET_METHOD!><!UNRESOLVED_REFERENCE!>[<!>0, 0, 0<!UNRESOLVED_REFERENCE!>]<!><!> = StringProperty("Fail")
+    nullTask?.input<!NO_SET_METHOD!><!UNRESOLVED_REFERENCE!>[<!>0, 0, 0<!UNRESOLVED_REFERENCE!>]<!><!> = StringProperty("Fail")
+
+    // a?.b[c] += d
+    <!OPERATOR_MODIFIER_REQUIRED!>task.input<!NO_SET_METHOD!>[<!TOO_MANY_ARGUMENTS!>0<!>]<!><!> += StringProperty("Fail")
+    <!OPERATOR_MODIFIER_REQUIRED, UNSAFE_CALL!>nullTask?.input<!NO_SET_METHOD!>[<!TOO_MANY_ARGUMENTS!>0<!>]<!><!> += StringProperty("Fail")
+
+    // a[i] = b should not be translated to a.get(i).assign(b)
+    operator fun Task.get(i: Int) = this.input
+    task<!NO_SET_METHOD!><!UNRESOLVED_REFERENCE!>[<!>0<!UNRESOLVED_REFERENCE!>]<!><!> = StringProperty("Fail")
+
+    // a.get(i) = b should not be translated to a.get(i).assign(b)
+    task.<!VARIABLE_EXPECTED!>get(0)<!> = StringProperty("Fail")
+    nullTask?.<!VARIABLE_EXPECTED!>get(0)<!> = StringProperty("Fail")
+}