cloud.orbit.dsl.OrbitFileParser.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of orbit-codegen-dsl Show documentation
The Orbit project is a collection of JVM libraries that makes it easier to build distributed and scalable online services.
The newest version!
/*
 Copyright (C) 2015 - 2019 Electronic Arts Inc.  All rights reserved.
 This file is part of the Orbit Project .
 See license in LICENSE.
 */

package cloud.orbit.dsl

import cloud.orbit.dsl.ast.ActorDeclaration
import cloud.orbit.dsl.ast.ActorMethod
import cloud.orbit.dsl.ast.CompilationUnit
import cloud.orbit.dsl.ast.DataDeclaration
import cloud.orbit.dsl.ast.DataField
import cloud.orbit.dsl.ast.EnumDeclaration
import cloud.orbit.dsl.ast.EnumMember
import cloud.orbit.dsl.ast.MethodParameter
import cloud.orbit.dsl.ast.Type
import org.antlr.v4.runtime.CharStreams
import org.antlr.v4.runtime.CommonTokenStream
import org.antlr.v4.runtime.ConsoleErrorListener

class OrbitFileParser : OrbitBaseVisitor() {
    private val enums = mutableListOf()
    private val data = mutableListOf()
    private val actors = mutableListOf()

    // This code is not thread safe. Need to find a way to pass the context into the visitor
    fun parse(input: String, packageName: String): CompilationUnit {
        val lexer = OrbitLexer(CharStreams.fromString(input))
        val parser = OrbitParser(CommonTokenStream(lexer)).also {
            it.addErrorListener(ThrowingErrorListener())
            it.removeErrorListener(ConsoleErrorListener.INSTANCE)
        }

        actors.clear()
        data.clear()

        visitFile(parser.file())

        return CompilationUnit(
            packageName,
            enums,
            data,
            actors
        )
    }

    override fun visitEnumDeclaration(ctx: OrbitParser.EnumDeclarationContext?) = enums.add(
        EnumDeclaration(
            ctx!!.name.text,
            ctx.children
                .asSequence()
                .filterIsInstance(OrbitParser.EnumMemberContext::class.java)
                .map { EnumMember(it.name.text, it.index.text.toInt()) }
                .toList()))

    override fun visitDataDeclaration(ctx: OrbitParser.DataDeclarationContext?) = data.add(
        DataDeclaration(
            ctx!!.name.text,
            ctx.children
                .asSequence()
                .filterIsInstance(OrbitParser.DataFieldContext::class.java)
                .map { DataField(it.name.text, makeType(it.type()), it.index.text.toInt()) }
                .toList()))

    override fun visitActorDeclaration(ctx: OrbitParser.ActorDeclarationContext?) = actors.add(
        ActorDeclaration(
            ctx!!.name.text,
            ctx.children
                .asSequence()
                .filterIsInstance(OrbitParser.ActorMethodContext::class.java)
                .map { m ->
                    ActorMethod(
                        name = m.name.text,
                        returnType = makeType(m.returnType),
                        params = m.children
                            .asSequence()
                            .filterIsInstance(OrbitParser.MethodParamContext::class.java)
                            .map { p -> MethodParameter(p.name.text, makeType(p.type())) }
                            .toList())
                }
                .toList()))

    private fun makeType(ctx: OrbitParser.TypeContext): Type =
        Type(
            ctx.name.text, ctx.children
                .asSequence()
                .filterIsInstance(OrbitParser.TypeContext::class.java)
                .map(::makeType)
                .toList()
        )
}