Codegen now generates "output.jar"

src/main/java/abstractSyntaxTree/Program.java

@@ -10,6 +10,8 @@ import abstractSyntaxTree.Parameter.ParameterList;
 import org.objectweb.asm.ClassWriter;
 import org.objectweb.asm.Opcodes;
 
+import java.io.File;
+import java.io.FileInputStream;
 import java.util.*;
 import java.io.FileOutputStream;
 import java.io.IOException;
@@ -71,28 +73,60 @@ public class Program implements Node {
         return result;
     }
 
-    public void codeGen() throws Exception{
+    public void codeGen() throws Exception {
+        // Store the names of the generated class files
+        List<String> classFileNames = new ArrayList<>();
+
         try {
             for (RefType oneClass : classes) {
                 ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);
                 cw.visit(Opcodes.V1_8, Opcodes.ACC_PUBLIC, oneClass.name, null, "java/lang/Object", null);
 
-               oneClass.codeGen(cw, methodContext, typeContext);
+                oneClass.codeGen(cw, methodContext, typeContext);
 
                 cw.visitEnd();
                 byte[] bytecode = cw.toByteArray();
 
-                try (FileOutputStream fos = new FileOutputStream(oneClass.name + ".class")) {
+                String classFileName = oneClass.name + ".class";
+                classFileNames.add(classFileName);
+
+                try (FileOutputStream fos = new FileOutputStream(classFileName)) {
                     fos.write(bytecode);
                 } catch (IOException e) {
                     e.printStackTrace();
                 }
             }
+
+            // Now create the JAR file
+            try (FileOutputStream fos = new FileOutputStream("output.jar");
+                 JarOutputStream jos = new JarOutputStream(fos)) {
+
+                for (String classFileName : classFileNames) {
+                    addFileToJar(jos, classFileName);
+                }
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
         } catch (Exception e) {
             throw new RuntimeException(e);
         }
     }
 
+    private void addFileToJar(JarOutputStream jos, String fileName) throws IOException {
+        File file = new File(fileName);
+        try (FileInputStream fis = new FileInputStream(file)) {
+            JarEntry entry = new JarEntry(fileName);
+            jos.putNextEntry(entry);
+
+            byte[] buffer = new byte[1024];
+            int bytesRead;
+            while ((bytesRead = fis.read(buffer)) != -1) {
+                jos.write(buffer, 0, bytesRead);
+            }
+            jos.closeEntry();
+        }
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;