Fix issue

nekocode · Sep 24, 2016 · b50b6b3 · b50b6b3
1 parent 1702e2f
commit b50b6b3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/META-INF/plugin.xml b/META-INF/plugin.xml
@@ -1,7 +1,7 @@
 <idea-plugin version="2">
     <id>cn.nekocode.plugin.parcelablegenerator</id>
     <name>Parcelable Code Generator(for kotlin)</name>
-    <version>0.5.0</version>
+    <version>0.5.1</version>
     <vendor email="[email protected]" url="https://github.com/nekocode/android-parcelable-intellij-plugin-kotlin">nekocode</vendor>
 
     <description><![CDATA[

diff --git a/src/cn/nekocode/plugin/parcelablegenerator/CodeGenerator.java b/src/cn/nekocode/plugin/parcelablegenerator/CodeGenerator.java
@@ -41,11 +41,13 @@ private String generateStaticCreator(KtClass ktClass, KtClassBody oldBodyOfCompa
         String className = ktClass.getName();
 
         StringBuilder oldBodyText = new StringBuilder();
-        List<KtDeclaration> declarations = oldBodyOfCompanion.getDeclarations();
-        for (KtDeclaration declaration : declarations) {
-            String declarationName = declaration.getName();
-            if (declarationName != null && !declarationName.equals("CREATOR")) {
-                oldBodyText.append(declaration.getText()).append("\n\n");
+        if (oldBodyOfCompanion != null) {
+            List<KtDeclaration> declarations = oldBodyOfCompanion.getDeclarations();
+            for (KtDeclaration declaration : declarations) {
+                String declarationName = declaration.getName();
+                if (declarationName != null && !declarationName.equals("CREATOR")) {
+                    oldBodyText.append(declaration.getText()).append("\n\n");
+                }
             }
         }