1 files changed, 21 insertions, 0 deletions
diff --git a/dottydoc/src/dotty/tools/dottydoc/util/syntax.scala b/dottydoc/src/dotty/tools/dottydoc/util/syntax.scala
new file mode 100644
index 000000000..140b3e761
--- /dev/null
+++ b/dottydoc/src/dotty/tools/dottydoc/util/syntax.scala
@@ -0,0 +1,21 @@
+package dotty.tools
+package dottydoc
+package util
+
+import dotc.core.Contexts.{ Context, DocBase }
+import model.Package
+
+object syntax {
+  implicit class RichDocContext(val ctx: Context) extends AnyVal {
+    def docbase: DocBase = ctx.getDocbase getOrElse {
+      throw new IllegalStateException("DocBase must be set before running dottydoc phases")
+    }
+  }
+
+  implicit class RichDocBase(val db: DocBase) {
+    def packages: Map[String, Package] = db.packagesAs[Package].toMap
+
+    def packagesMutable: collection.mutable.Map[String, Package] =
+      db.packagesAs[Package]
+  }
+}