fixed Xcode crash bug #14

kimsungwhee · Jul 15, 2015 · ca5b9f9 · ca5b9f9
1 parent 2fb4797
commit ca5b9f9
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/KSHObjcUML/KSHObjcUML.m b/KSHObjcUML/KSHObjcUML.m
@@ -46,22 +46,38 @@ + (NSBundle *)pluginBundle
 
 - (NSString *)scriptPath
 {
-    return [[self.directoryPath stringByAppendingPathComponent:FOLDAR_NAME] stringByAppendingPathComponent:@"script.rb"];
+    NSString *tempScriptPath = [[self.directoryPath stringByAppendingPathComponent:FOLDAR_NAME] stringByAppendingPathComponent:@"script.rb"];
+    if (tempScriptPath) {
+        _scriptPath = tempScriptPath;
+    }
+    return _scriptPath;
 }
 
 - (NSString *)projectName
 {
-    return [VWKProject projectForKeyWindow].projectOriginalName;
+    NSString *tempProjectName = [VWKProject projectForKeyWindow].projectOriginalName;
+    if (tempProjectName) {
+        _projectName = tempProjectName;
+    }
+    return _projectName;
 }
 
 - (NSString *)zipFilePath
 {
-    return [[KSHObjcUML pluginBundle] pathForResource:FOLDAR_NAME ofType:@"zip"];
+    NSString *tempZipFilePath = [[KSHObjcUML pluginBundle] pathForResource:FOLDAR_NAME ofType:@"zip"];
+    if (tempZipFilePath) {
+        _zipFilePath = tempZipFilePath;
+    }
+    return _zipFilePath;
 }
 
 - (NSString *)directoryPath
 {
-    return [VWKProject projectForKeyWindow].directoryPath;
+    NSString *tempDirectoryPath = [VWKProject projectForKeyWindow].directoryPath;
+    if (tempDirectoryPath) {
+        _directoryPath = tempDirectoryPath;
+    }
+    return _directoryPath;
 }
 
 + (void)pluginDidLoad:(NSBundle *)plugin