doc-tool/src/dotty/tools/dottydoc/core/PackageObjectsPhase.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

package dotty.tools
package dottydoc
package core

import dotty.tools.dotc.core.Symbols.Symbol
import dotc.core.Contexts.Context
import dotc.ast.tpd

import model._
import model.internal._
import util.syntax._
import transform.DocMiniPhase

class PackageObjectsPhase extends DocMiniPhase {

  override def transformPackage(implicit ctx: Context) = { case pkg: PackageImpl =>
    pkg
      .members
      .collect { case o: Object if o.symbol.isPackageObject => o }
      .headOption
      .map { obj =>
        pkg.copy(
          members = obj.members ++ pkg.members,
          superTypes = obj.superTypes,
          comment = obj.comment
        )
      }
      .getOrElse(pkg)
  }

  override def transformObject(implicit ctx: Context) = { case obj: Object =>
    if (obj.symbol.isPackageObject) NonEntity
    else obj
  }
}