Add RichQPromise

jokade · Feb 23, 2016 · 1e43123 · 1e43123
1 parent 34b04ce
commit 1e43123
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/src/main/scala/biz/enef/angulate/core/QPromise.scala b/src/main/scala/biz/enef/angulate/core/QPromise.scala
@@ -9,18 +9,24 @@ import scala.scalajs.js
 
 trait QPromise extends js.Object {
 
-  def `then`(successCallback: js.Function1[js.Any,js.Any]) : QPromise = js.native
+  def `then`(successCallback: js.Function1[js.Any,_]) : QPromise = js.native
 
-  def `then`(successCallback: js.Function1[js.Any,js.Any],
-             errorCallback: js.Function1[js.Any,Unit]) : QPromise = js.native
+  def `then`(successCallback: js.Function1[js.Any,_],
+             errorCallback: js.Function1[js.Any,_]) : QPromise = js.native
 
-  def `then`(successCallback: js.Function1[js.Any,js.Any],
-             errorCallback: js.Function1[js.Any,Unit],
-             notifyCallback: js.Function1[js.Any,Unit]) : QPromise = js.native
+  def `then`(successCallback: js.Function1[js.Any,_],
+             errorCallback: js.Function1[js.Any,_],
+             notifyCallback: js.Function1[js.Any,_]) : QPromise = js.native
 
-  def `catch`(errorCallback: js.Function1[js.Any,Unit]) : QPromise = js.native
+  def `catch`(errorCallback: js.Function1[js.Any,_]) : QPromise = js.native
 
-  def `finally`(callback: js.Function1[js.Any,Unit]) : QPromise = js.native
+  def `finally`(callback: js.Function1[js.Any,_]) : QPromise = js.native
+}
+
+object QPromise {
+  implicit class RichQPromise(val p: QPromise) extends AnyVal {
+    def onSuccess(f: (js.Any)=>Unit) = p.`then`(f:js.Function1[js.Any,Unit])
+  }
 }
 
 trait QService extends ProvidedService {