Don't set Object as a superclass

1 files changed, 6 insertions, 7 deletions

diff --git a/src/main/java/org/zwobble/couscous/frontends/java/JavaTypes.java b/src/main/java/org/zwobble/couscous/frontends/java/JavaTypes.java
index 226e05d..c3c6b7f 100644
--- a/src/main/java/org/zwobble/couscous/frontends/java/JavaTypes.java
+++ b/src/main/java/org/zwobble/couscous/frontends/java/JavaTypes.java
@@ -5,6 +5,7 @@ import org.eclipse.jdt.core.dom.*;
 import org.zwobble.couscous.ast.TypeName;
 import org.zwobble.couscous.values.ObjectValues;
 
+import java.util.Optional;
 import java.util.Set;
 
 import static com.google.common.collect.Iterables.transform;
@@ -44,18 +45,16 @@ class JavaTypes {
 
     private static Set<TypeName> superTypes(ITypeBinding typeBinding) {
         ImmutableSet.Builder<TypeName> superTypes = ImmutableSet.builder();
-        superTypes.add(superClass(typeBinding));
+        superClass(typeBinding).ifPresent(superTypes::add);
         superTypes.addAll(transform(
             asList(typeBinding.getInterfaces()),
             JavaTypes::typeOf));
         return superTypes.build();
     }
 
-    private static TypeName superClass(ITypeBinding typeBinding) {
-        if (typeBinding.getSuperclass() == null) {
-            return ObjectValues.OBJECT;
-        } else {
-            return typeOf(typeBinding.getSuperclass());
-        }
+    private static Optional<TypeName> superClass(ITypeBinding typeBinding) {
+        return Optional.ofNullable(typeBinding.getSuperclass())
+            .map(JavaTypes::typeOf)
+            .filter(type -> !type.equals(ObjectValues.OBJECT));
     }
 }