Add manifest support

sotasan · Oct 18, 2023 · 9df70f7 · 9df70f7
1 parent 54b7273
commit 9df70f7
Show file tree

Hide file tree

Showing 7 changed files with 92 additions and 9 deletions.
diff --git a/src/main/java/com/sotasan/decompiler/controllers/TabController.java b/src/main/java/com/sotasan/decompiler/controllers/TabController.java
@@ -1,6 +1,7 @@
 package com.sotasan.decompiler.controllers;
 
 import com.sotasan.decompiler.models.FileModel;
+import com.sotasan.decompiler.types.ClassType;
 import com.sotasan.decompiler.views.TabView;
 import lombok.Getter;
 import org.fife.ui.rsyntaxtextarea.SyntaxConstants;
@@ -21,8 +22,8 @@ public TabController(@NotNull FileModel fileModel) {
     public void update() {
         try {
             getView().getTextArea().setText(getText(fileModel));
-            if (fileModel.isClass())
-                getView().getTextArea().setSyntaxEditingStyle(SyntaxConstants.SYNTAX_STYLE_JAVA);
+            if (fileModel.getType() != null)
+                getView().getTextArea().setSyntaxEditingStyle(fileModel.getType().getSyntax());
         } catch (Exception e) {
             getView().getTextArea().setText(e.getMessage());
             getView().getTextArea().setSyntaxEditingStyle(SyntaxConstants.SYNTAX_STYLE_NONE);
@@ -31,7 +32,7 @@ public void update() {
     }
 
     private String getText(@NotNull FileModel fileModel) throws Exception {
-        return fileModel.isClass()
+        return fileModel.getType() instanceof ClassType
                 ? TabsController.getINSTANCE().getTransformer().getInstance().transform(fileModel)
                 : new String(fileModel.getBytes());
     }

diff --git a/src/main/java/com/sotasan/decompiler/controllers/TabsController.java b/src/main/java/com/sotasan/decompiler/controllers/TabsController.java
@@ -2,6 +2,7 @@
 
 import com.sotasan.decompiler.models.FileModel;
 import com.sotasan.decompiler.transformers.Transformer;
+import com.sotasan.decompiler.types.ClassType;
 import com.sotasan.decompiler.views.TabsView;
 import com.sotasan.decompiler.views.TabView;
 import lombok.Getter;
@@ -23,7 +24,7 @@ private TabsController() {
     public void actionPerformed(ActionEvent event) {
         for (int i = 0; i < getView().getTabCount(); i++) {
             TabController controller = ((TabView) getView().getComponentAt(i)).getController();
-            if (controller.getFileModel().isClass())
+            if (controller.getFileModel().getType() instanceof ClassType)
                 controller.update();
         }
     }

diff --git a/src/main/java/com/sotasan/decompiler/models/FileModel.java b/src/main/java/com/sotasan/decompiler/models/FileModel.java
@@ -1,5 +1,7 @@
 package com.sotasan.decompiler.models;
 
+import com.sotasan.decompiler.services.TypeService;
+import com.sotasan.decompiler.types.Type;
 import lombok.Getter;
 import lombok.SneakyThrows;
 import org.jetbrains.annotations.NotNull;
@@ -11,16 +13,14 @@ public class FileModel extends BaseModel {
 
     private final JarFile jarFile;
     private final JarEntry jarEntry;
+    private final Type type;
 
     public FileModel(JarFile jarFile, @NotNull JarEntry jarEntry) {
         super(jarEntry.getName(), false);
         this.jarFile = jarFile;
         this.jarEntry = jarEntry;
-        setIcon(isClass() ? "icons/class.png" : "icons/file.png");
-    }
-
-    public boolean isClass() {
-        return getName().toLowerCase().endsWith(".class");
+        type = TypeService.getType(this);
+        setIcon(type != null ? type.getIcon() : "icons/file.png");
     }
 
     @SneakyThrows

diff --git a/src/main/java/com/sotasan/decompiler/services/TypeService.java b/src/main/java/com/sotasan/decompiler/services/TypeService.java
@@ -0,0 +1,26 @@
+package com.sotasan.decompiler.services;
+
+import com.sotasan.decompiler.models.FileModel;
+import com.sotasan.decompiler.types.ClassType;
+import com.sotasan.decompiler.types.ManifestType;
+import com.sotasan.decompiler.types.Type;
+import lombok.experimental.UtilityClass;
+import org.jetbrains.annotations.Nullable;
+import java.util.List;
+
+@UtilityClass
+public class TypeService {
+
+    private static final List<Type> TYPES = List.of(
+            new ClassType(),
+            new ManifestType()
+    );
+
+    public static @Nullable Type getType(FileModel fileModel) {
+        for (Type format : TYPES)
+            if (format.isFormat(fileModel))
+                return format;
+        return null;
+    }
+
+}
diff --git a/src/main/java/com/sotasan/decompiler/types/ClassType.java b/src/main/java/com/sotasan/decompiler/types/ClassType.java
@@ -0,0 +1,18 @@
+package com.sotasan.decompiler.types;
+
+import com.sotasan.decompiler.models.FileModel;
+import org.fife.ui.rsyntaxtextarea.SyntaxConstants;
+import org.jetbrains.annotations.NotNull;
+
+public class ClassType extends Type {
+
+    public ClassType() {
+        super("icons/class.png", SyntaxConstants.SYNTAX_STYLE_JAVA);
+    }
+
+    @Override
+    public boolean isFormat(@NotNull FileModel fileModel) {
+        return fileModel.getName().toLowerCase().endsWith(".class");
+    }
+
+}
diff --git a/src/main/java/com/sotasan/decompiler/types/ManifestType.java b/src/main/java/com/sotasan/decompiler/types/ManifestType.java
@@ -0,0 +1,18 @@
+package com.sotasan.decompiler.types;
+
+import com.sotasan.decompiler.models.FileModel;
+import org.fife.ui.rsyntaxtextarea.SyntaxConstants;
+import org.jetbrains.annotations.NotNull;
+
+public class ManifestType extends Type {
+
+    public ManifestType() {
+        super("icons/manifest.png", SyntaxConstants.SYNTAX_STYLE_PROPERTIES_FILE);
+    }
+
+    @Override
+    public boolean isFormat(@NotNull FileModel fileModel) {
+        return fileModel.getName().toLowerCase().endsWith(".mf");
+    }
+
+}
diff --git a/src/main/java/com/sotasan/decompiler/types/Type.java b/src/main/java/com/sotasan/decompiler/types/Type.java
@@ -0,0 +1,19 @@
+package com.sotasan.decompiler.types;
+
+import com.sotasan.decompiler.models.FileModel;
+import lombok.Getter;
+
+@Getter
+public abstract class Type {
+
+    private final String icon;
+    private final String syntax;
+
+    public Type(String icon, String syntax) {
+        this.icon = icon;
+        this.syntax = syntax;
+    }
+
+    public abstract boolean isFormat(FileModel fileModel);
+
+}