context in PageQueryImplDsl + withContext

intracer · Aug 7, 2016 · b84deeb · b84deeb
1 parent b065187
commit b84deeb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/scalawiki-core/src/main/scala/org/scalawiki/query/PageQueryImplDsl.scala b/scalawiki-core/src/main/scala/org/scalawiki/query/PageQueryImplDsl.scala
@@ -16,8 +16,10 @@ import scala.concurrent.Future
 
 class PageQueryImplDsl(query: Either[Set[Long], Set[String]],
                        bot: MwBot,
-                       context: Map[String, String] = Map.empty
-                      ) extends PageQuery with SinglePageQuery {
+                       context: Map[String, String] = Map.empty) extends PageQuery with SinglePageQuery {
+
+  override def withContext(context: Map[String, String]) =
+    new PageQueryImplDsl(query, bot, context)
 
   override def revisions(namespaces: Set[Int], props: Set[String], continueParam: Option[(String, String)]): Future[Seq[Page]] = {
 

diff --git a/scalawiki-core/src/main/scala/org/scalawiki/query/SinglePageQuery.scala b/scalawiki-core/src/main/scala/org/scalawiki/query/SinglePageQuery.scala
@@ -38,4 +38,5 @@ trait SinglePageQuery {
              comment: Option[String] = None,
              ignoreWarnings: Boolean = false): Future[String]
 
+  def withContext(context: Map[String, String]): SinglePageQuery
 }