wvlet.airframe.SourceCode.scala Maven / Gradle / Ivy

Go to download
/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package wvlet.airframe

/**
  * A hack to embed source code location where DI is used
  */
case class SourceCode(
    // Deprecated and hidden because the filePath can be too long and may contain private directory paths.
    // Removing this parameter causes binary incompatibility between AirSpec and Airframe, so preserving it here
    private val filePath: String,
    fileName: String,
    line: Int,
    col: Int
):
  override def toString = s"${fileName}:${line}"

object SourceCode:
  def apply()(implicit code: SourceCode) = code

  import scala.quoted.*

  inline implicit def generate: SourceCode = ${ generateImpl }

  private def generateImpl(using q: Quotes): Expr[SourceCode] =
    import q.reflect.*
    val pos    = Position.ofMacroExpansion
    val line   = Expr(pos.startLine)
    val column = Expr(pos.endColumn)
    val src    = pos.sourceFile
    // Hidden because embedding the absolute path is not good for privacy and code size
    val srcPath: java.nio.file.Path = java.nio.file.Paths.get(src.path)
    val fileName                    = Expr(srcPath.getFileName().toString)
    '{ SourceCode("", ${ fileName }, ${ line } + 1, ${ column }) }