Imperial-MEng/WACC_37
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: fix formatting

Browse Source
This commit is contained in:
Jonny
2025-03-02 03:14:58 +00:00
parent 85a82aabb4
commit 9a5ccea1f6
1 changed files with 32 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
src/main/wacc/Main.scala
View File

@@ -19,15 +19,17 @@ import org.typelevel.log4cats.Logger
import assemblyIR as asm
// TODO: IO correctness, --greedy, parallelisable, and probably splitting this file up
given Argument[Path] = Argument.from("path") { str =>
val path = Path.of(str)
(
Either.cond(Files.exists(path), path, s"File '${path.toAbsolutePath}' does not exist"),
Either.cond(Files.isRegularFile(path), path, s"File '${path.toAbsolutePath}' must be a regular file"),
Either.cond(
Files.isRegularFile(path),
path,
s"File '${path.toAbsolutePath}' must be a regular file"
),
Either.cond(path.toString.endsWith(".wacc"), path, "File must have .wacc extension")
).mapN((_, _, _) => path).toValidatedNel
}
@@ -42,7 +44,6 @@ val outputOpt: Opts[Option[Path]] =
val filesOpt: Opts[NonEmptyList[Path]] = Opts.arguments[Path]("files")
def frontend(
contents: String
): IO[Either[Int, microWacc.Program]] = {
@@ -64,9 +65,11 @@ def frontend(
else {
// TODO: multiple traversal of error content, should be a foldleft or co
given errorContent: String = contents
val exitCode = errResult.collectFirst {
case _: Error.InternalError => 201
}.getOrElse(200)
val exitCode = errResult
.collectFirst { case _: Error.InternalError =>
201
}
.getOrElse(200)
val formattedErrors = errResult.map(formatError).mkString("\n")
@@ -85,35 +88,39 @@ def compile(filename: String, outputDir: Option[Path], log: Boolean): IO[Int] =
if (log) logger.info(_)
else (_ => IO.unit)
for {
contents <- IO.delay(os.read(os.Path(filename))) // TODO: Is IO as a wrapper ok or do we require .delay - also, should it be .blocking?
contents <- IO.delay(
os.read(os.Path(filename))
) // TODO: Is IO as a wrapper ok or do we require .delay - also, should it be .blocking?
_ <- logAction(s"Compiling file: $filename")
result <- frontend(contents)
exitCode <- result.fold(
code => logger.error(s"Compilation failed for $filename\nExit code: $code").as(code),
typedProg =>
val outDir = outputDir.getOrElse(Paths.get(filename).getParent)
IO.delay(Files.createDirectories(outDir)) // TODO: Is IO as a wrapper ok or do we require .delay - also, should it be .blocking?
IO.delay(
Files.createDirectories(outDir)
) // TODO: Is IO as a wrapper ok or do we require .delay - also, should it be .blocking?
val outputFile = outDir.resolve(filename.stripSuffix(".wacc") + ".s")
writer.writeTo(backend(typedProg), outputFile) *> // TODO: I dont think we need IO here if we look at the implementation of writer
logAction(s"Compilation succeeded: $filename").as(0)
writer.writeTo(
backend(typedProg),
outputFile
) *> // TODO: I dont think we need IO here if we look at the implementation of writer
logAction(s"Compilation succeeded: $filename").as(0)
)
} yield exitCode
// TODO: this is sequential, thus should be what occurs when --greedy is passed in
val compileCommand: Opts[IO[ExitCode]] =
(filesOpt, logOpt, outputOpt).mapN{
(files, log, outDir) =>
files
.traverse{ file =>
compile(
file.toAbsolutePath.toString,
outDir,
log)
}.map {
exitCodes =>
if (exitCodes.exists(_ != 0)) ExitCode.Error // TODO- it should be the first one to exit when parallelised :)
else ExitCode.Success
}
(filesOpt, logOpt, outputOpt).mapN { (files, log, outDir) =>
files
.traverse { file =>
compile(file.toAbsolutePath.toString, outDir, log)
}
.map { exitCodes =>
if (exitCodes.exists(_ != 0))
ExitCode.Error // TODO- it should be the first one to exit when parallelised :)
else ExitCode.Success
}
}
@@ -127,4 +134,4 @@ object Main
def main: Opts[IO[ExitCode]] =
compileCommand
}
}