com.nawforce.runtime.workspace.ApexClassLoader.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2022 FinancialForce.com, inc. All rights reserved.
 */
package com.nawforce.runtime.workspace

import com.financialforce.oparser._
import com.nawforce.pkgforce.diagnostics._
import com.nawforce.pkgforce.documents.MetadataDocument
import com.nawforce.pkgforce.path.Location
import com.nawforce.runtime.parsers.SourceData

import java.util.concurrent.ExecutorService
import scala.concurrent.duration.Duration
import scala.concurrent.{Await, ExecutionContext, ExecutionContextExecutor, Future}

final class ApexClassLoader(
  loadingPool: ExecutorService,
  module: IPM.Module,
  factory: TypeDeclFactory[IMutableModuleTypeDeclaration, IPM.Module]
) {
  private implicit val ec: ExecutionContextExecutor = ExecutionContext.fromExecutor(loadingPool)

  def loadClasses(
    documents: Iterator[MetadataDocument],
    logger: IssueLogger
  ): Array[(MetadataDocument, IMutableModuleTypeDeclaration)] = {

    val loadFutures = Future.traverse(documents)(document => {
      Future {
        val source = document.path.readSourceData()
        source match {
          case Left(err) =>
            Left(Some(new Issue(document.path, Diagnostic(ERROR_CATEGORY, Location.empty, err))))
          case Right(source) =>
            parseSource(document, source)
        }
      }
    })

    val results = Await.result(loadFutures, Duration.Inf)

    results.toArray.flatMap {
      case Right(td) =>
        Some(td)
      case Left(optIssue) =>
        optIssue.foreach(logger.log)
        None
    }
  }

  private def parseSource(
    document: MetadataDocument,
    source: SourceData
  ): Either[Option[Issue], (MetadataDocument, IMutableModuleTypeDeclaration)] = {
    val (_, reason, td) =
      OutlineParser.parse(document.path.toString, source.asString, factory, module)
    td
      .map(td => Right(document, td))
      .getOrElse(
        Left(Some(new Issue(document.path, Diagnostic(ERROR_CATEGORY, Location.empty, reason.get))))
      )
  }

}