raw.compiler.base.source.SourcePrettyPrinter.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of raw-compiler_2.12 Show documentation

raw-compiler

The newest version!

/*
 * Copyright 2023 RAW Labs S.A.
 *
 * Use of this software is governed by the Business Source License
 * included in the file licenses/BSL.txt.
 *
 * As of the Change Date specified in that file, in accordance with
 * the Business Source License, use of this software will be governed
 * by the Apache License, Version 2.0, included in the file
 * licenses/APL.txt.
 */

package raw.compiler.base.source

import com.typesafe.scalalogging.StrictLogging
import raw.compiler.base.{Keywords, PrettyPrinter, SyntaxAnalyzer}
import raw.utils._

import scala.language.implicitConversions

trait SourcePrettyPrinter extends PrettyPrinter with Keywords with StrictLogging {

  implicit protected def baseNodeToDoc(n: BaseNode): Doc = toDoc(n)

  protected def sepArgs(sep: Doc, elems: Doc*): Doc = parens(group(nest(lsep(elems.to, sep))))

  def args(n: Doc*): Doc = sepArgs(comma, n: _*)

  def method(name: Doc, n: Doc*): Doc = {
    if (n.nonEmpty) group(name <> args(n: _*))
    else name
  }

  protected def dquoted(s: String): Doc = dquotes(descape(s))

  def ident(idn: String): String = {
    if (SyntaxAnalyzer.identRegex.unapplySeq(idn).isDefined && !isReserved(idn)) {
      // If simple word (no spaces, no escaping) and not reserved, print word as-is.
      idn
    } else {
      s"`$idn`"
    }
  }

  override def toDoc(n: BaseNode): Doc = n match {
    case _: AnythingType => "anything"
    case _: NotValueType =>
      // TODO (msb): This message could be improved but kept for backward compatibility.
      "non-executable query"
    case _ => super.toDoc(n)
  }

}