Add index.ts

Author: 1001mei

src/compiler/binary-writer.ts

@@ -39,6 +39,10 @@ export class BinaryWriter {
     this.constantPool = []
   }
 
+  generateBinary(classFile: ClassFile) {
+    return this.toBinary(classFile)
+  }
+
   writeBinary(classFile: ClassFile, filepath: string) {
     const filename = filepath + 'Main.class'
     const binary = this.toBinary(classFile)

src/compiler/compiler.ts

@@ -1,5 +1,3 @@
-import * as fs from 'fs'
-import * as peggy from 'peggy'
 import { ClassFile } from '../ClassFile/types'
 import { AST } from '../ast/types/packages-and-modules'
 import {
@@ -55,15 +53,6 @@ export class Compiler {
     return classFiles[0]
   }
 
-  compileFromSource(javaProgram: string) {
-    const javaPegGrammar = fs.readFileSync(__dirname + '/main.pegjs', 'utf8')
-    const parser = peggy.generate(javaPegGrammar, {
-      allowedStartRules: ['CompilationUnit']
-    })
-    const ast = parser.parse(javaProgram)
-    return this.compile(ast)
-  }
-
   private compileClass(classNode: ClassDeclaration): ClassFile {
     const parentClassName = 'java/lang/Object'
     this.className = classNode.typeIdentifier

src/compiler/index.ts

@@ -0,0 +1,32 @@
+import * as peggy from 'peggy'
+import { AST } from '../ast/types/packages-and-modules'
+import { Compiler } from './compiler'
+import { javaPegGrammar } from './grammar'
+import { BinaryWriter } from './binary-writer'
+
+export const compile = (ast: AST): string => {
+  const compiler = new Compiler()
+  const classFile = compiler.compile(ast)
+
+  const binaryWriter = new BinaryWriter()
+  const byteArray = binaryWriter.generateBinary(classFile)
+  const base64encoded = Buffer.from(byteArray).toString('base64')
+
+  return base64encoded
+}
+
+export const compileFromSource = (javaProgram: string): string => {
+  const parser = peggy.generate(javaPegGrammar, {
+    allowedStartRules: ['CompilationUnit'],
+    cache: true
+  })
+
+  let ast: AST
+  try {
+    ast = parser.parse(javaProgram)
+  } catch (e) {
+    throw new SyntaxError(e)
+  }
+
+  return compile(ast)
+}