Skip to content

Commit 03fe79d

Browse files
authored
add @doc annotation for all steps; also document delegators, ... (#114)
* document delegator steps on nodes * also fix release * add @doc annotation for all steps
1 parent 80d24ce commit 03fe79d

File tree

2 files changed

+14
-3
lines changed

2 files changed

+14
-3
lines changed

Diff for: codegen/src/main/scala/overflowdb/codegen/CodeGen.scala

+7-3
Original file line numberDiff line numberDiff line change
@@ -470,7 +470,7 @@ class CodeGen(schema: Schema) {
470470

471471
s"""/** ${adjacentNode.customStepDoc.getOrElse("")}
472472
| * Traverse to ${neighbor.name} via ${adjacentNode.viaEdge.name}.
473-
| */
473+
| */ ${docAnnotationMaybe(adjacentNode.customStepDoc)}
474474
|def $accessorName: ${fullScalaType(neighbor, cardinality)} =
475475
| $edgeAccessorName.collectAll[${neighbor.className}]$appendix""".stripMargin
476476
}
@@ -896,11 +896,15 @@ class CodeGen(schema: Schema) {
896896
val nodeDelegators = neighborInfo.nodeInfos.collect {
897897
case neighborNodeInfo if !neighborNodeInfo.isInherited =>
898898
val accessorNameForNode = accessorName(neighborNodeInfo)
899-
s"def $accessorNameForNode: ${neighborNodeInfo.returnType} = get().$accessorNameForNode"
899+
s"""/** ${neighborNodeInfo.customStepDoc.getOrElse("")}
900+
| * Traverse to ${neighborNodeInfo.neighborNode.name} via ${neighborNodeInfo.edge.name}.
901+
| */ ${docAnnotationMaybe(neighborNodeInfo.customStepDoc)}
902+
|def $accessorNameForNode: ${neighborNodeInfo.returnType} = get().$accessorNameForNode""".stripMargin
900903
}.mkString("\n")
901904

902905
s"""def $edgeAccessorName = get().$edgeAccessorName
903906
|override def _$edgeAccessorName = get()._$edgeAccessorName
907+
|
904908
|$nodeDelegators
905909
|""".stripMargin
906910
}.mkString("\n")
@@ -1162,7 +1166,7 @@ class CodeGen(schema: Schema) {
11621166
}
11631167
s"""/** ${customStepDoc.getOrElse("")}
11641168
| * Traverse to ${neighbor.name} via ${viaEdge.name} - this relationship was given a customStepName in the schema.
1165-
| */
1169+
| */ ${docAnnotationMaybe(customStepDoc)}
11661170
|def $customStepName: Traversal[${neighbor.className}] =
11671171
| traversal.$mapOrFlatMap(_.$customStepName)
11681172
|""".stripMargin

Diff for: codegen/src/main/scala/overflowdb/codegen/Helpers.scala

+7
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,13 @@ object Helpers {
6464
}
6565
}
6666

67+
def docAnnotationMaybe(customStepDoc: Option[String]): String = {
68+
customStepDoc match {
69+
case Some(doc) => s"""\[email protected]("$doc")"""
70+
case None => ""
71+
}
72+
}
73+
6774
def isNodeBaseTrait(baseTraits: Seq[NodeBaseType], nodeName: String): Boolean =
6875
nodeName == DefaultNodeTypes.AbstractNodeName || baseTraits.map(_.name).contains(nodeName)
6976

0 commit comments

Comments
 (0)