compilers/compiler/src/main/scala/Main.scala

package compiler

import scala.util.{Try, Failure, Success}

import java.io.{BufferedReader, InputStreamReader, PrintWriter}
import java.net.{ServerSocket, Socket};
import java.util.stream.Stream

import compiler.Tokenizer._
import compiler.Parser.{Unit => _, _} // Scala also has Unit
import java.util.Base64
import java.nio.charset.StandardCharsets


@main def main(): Unit =
  println("Hello world!")
  println(msg)
  run_server()

def msg = "I was compiled by Scala 3. :)"

def base64Encode(source: String): String =
  return Base64.getEncoder.encodeToString(source.getBytes(StandardCharsets.UTF_8))

def run_server(): Unit =
  println("Starting server...")
  val serverSocket: ServerSocket = new ServerSocket(3000);
  println("Waiting for connections...")

  while true do
    // Accept incoming client connection
    val clientSocket: Socket = serverSocket.accept();
    println("Client connected!");

    // Setup input and output streams for communication with the client
    val in: BufferedReader = new BufferedReader(new InputStreamReader(clientSocket.getInputStream()));
    val out: PrintWriter = new PrintWriter(clientSocket.getOutputStream(), true);

    // Read message from client
    val message: String = in.readLine();
    println("Client says: " + message);

    val json: ujson.Value = ujson.read(message)
    val command = json("command").str

    val response = command match {
      case "ping" => "{}"
      case "compile" => s"{\"program\": \"${base64Encode(compile(json("code").str))}\"}"
      case _ => s"{\"error\": \"Incorrect json request body\"}"
    }


    println(s"Responding with: ${response}")
    // Send response to the client
    out.println(response)
    out.close()

def compile(source: String): String =
  val tokens = Tokenizer.tokenize(source)
  val ast = tokens match {
    case Success(tokens) => Parser.parse(tokens)
    case Failure(err) => Failure(err)
  }
  return ast.toString()