Merge pull request #87 from JoelGerboreLaser/master

XCGroup: modification to read and write correctly PBXVariantGroupType

Author: jasperblues

Source/XCGroup.h

@@ -13,6 +13,7 @@
 #import <Foundation/Foundation.h>
 #import "XcodeGroupMember.h"
 #import "XcodeSourceFileType.h"
+#import "XCBuildFile.h"
 
 @class XCProject;
 @class XCClassDefinition;
@@ -29,7 +30,7 @@
 * Represents a _group container in an Xcode project. A group can contain members of type `XCSourceFile` or other
 * groups.
 */
-@interface XCGroup : NSObject <XcodeGroupMember>
+@interface XCGroup : NSObject <XcodeGroupMember, XCBuildFile>
 {
 
     NSString* _pathRelativeToParent;
@@ -42,8 +43,12 @@
     NSMutableArray* _children;
     NSMutableArray* _members;
 
+    NSNumber *_isBuildFile;
+    NSString *_buildFileKey;
+
     XCFileOperationQueue* _fileOperationQueue;
     XCProject* _project;
+    XcodeMemberType _memberType;
 
 }
 
@@ -77,8 +82,12 @@
 
 + (XCGroup*)groupWithProject:(XCProject*)project key:(NSString*)key alias:(NSString*)alias path:(NSString*)path children:(NSArray<id<XcodeGroupMember>>*)children;
 
++ (XCGroup*)groupWithProject:(XCProject*)project key:(NSString*)key alias:(NSString*)alias path:(NSString*)path children:(NSArray<id<XcodeGroupMember>>*)children memberType:(XcodeMemberType)groupType;
+
 - (id)initWithProject:(XCProject*)project key:(NSString*)key alias:(NSString*)alias path:(NSString*)path children:(NSArray<id<XcodeGroupMember>>*)children;
 
+- (id)initWithProject:(XCProject*)project key:(NSString*)key alias:(NSString*)alias path:(NSString*)path children:(NSArray<id<XcodeGroupMember>>*)children memberType:(XcodeMemberType)groupType;
+
 #pragma mark Parent group
 
 - (void)removeFromParentGroup;
@@ -134,6 +143,11 @@
  */
 - (XCGroup*)addGroupWithAlias:(NSString *)alias;
 
+/**
+ * Adds a _group with an alias to this _group and a particular type (only PBXGroupType and PBXVariantGroupType are valid)
+ */
+- (XCGroup*)addGroupWithAlias:(NSString *)alias groupType:(XcodeMemberType)type;
+
 /**
  * Adds a version group with a path relative to this group.
  */

Source/XCGroup.m

@@ -37,17 +37,24 @@ + (XCGroup *)groupWithProject:(XCProject *)project key:(NSString *)key alias:(NS
                      children:(NSArray *)children
 {
 
-    return [[XCGroup alloc] initWithProject:project key:key alias:alias path:path children:children];
+    return [[XCGroup alloc] initWithProject:project key:key alias:alias path:path children:children memberType:PBXGroupType];
+}
+
++ (XCGroup *)groupWithProject:(XCProject *)project key:(NSString *)key alias:(NSString *)alias path:(NSString *)path children:(NSArray<id<XcodeGroupMember>> *)children memberType:(XcodeMemberType)groupType
+{
+    return [[XCGroup alloc]initWithProject:project key:key alias:alias path:path children:children memberType:groupType];
 }
 
 //-------------------------------------------------------------------------------------------
 #pragma mark - Initialization & Destruction
 //-------------------------------------------------------------------------------------------
 
-- (id)initWithProject:(XCProject *)project key:(NSString *)key alias:(NSString *)alias path:(NSString *)path
-             children:(NSArray *)children
+- (id)initWithProject:(XCProject *)project key:(NSString *)key alias:(NSString *)alias path:(NSString *)path children:(NSArray<id<XcodeGroupMember>> *)children memberType:(XcodeMemberType)groupType
 {
     self = [super init];
+
+    assert(groupType == PBXGroupType || groupType == PBXVariantGroupType);
+
     if (self) {
         _project = project;
         _fileOperationQueue = [_project fileOperationQueue];
@@ -59,10 +66,18 @@ - (id)initWithProject:(XCProject *)project key:(NSString *)key alias:(NSString *
         if (!_children) {
             _children = [[NSMutableArray alloc] init];
         }
+
+        _memberType = groupType;
     }
     return self;
 }
 
+- (id)initWithProject:(XCProject *)project key:(NSString *)key alias:(NSString *)alias path:(NSString *)path
+             children:(NSArray *)children
+{
+    return [self initWithProject:project key:key alias:alias path:path children:children memberType:PBXGroupType];
+}
+
 //-------------------------------------------------------------------------------------------
 #pragma mark - Interface Methods
 //-------------------------------------------------------------------------------------------
@@ -276,7 +291,12 @@ - (XCGroup *)addGroupWithPath:(NSString *)path
     return group;
 }
 
-- (XCGroup*)addGroupWithAlias:(NSString *)alias
+- (XCGroup *)addGroupWithAlias:(NSString *)alias
+{
+    return [self addGroupWithAlias:alias groupType:PBXGroupType];
+}
+
+- (XCGroup*)addGroupWithAlias:(NSString *)alias groupType:(XcodeMemberType)type
 {
     NSString *groupKey = [[XCKeyBuilder forItemNamed:alias] build];
 
@@ -290,7 +310,7 @@ - (XCGroup*)addGroupWithAlias:(NSString *)alias
         }
     }
 
-    XCGroup *group = [[XCGroup alloc] initWithProject:_project key:groupKey alias:alias path:nil children:nil];
+    XCGroup *group = [[XCGroup alloc] initWithProject:_project key:groupKey alias:alias path:nil children:nil memberType:type];
     NSDictionary *groupDict = [group asDictionary];
 
     [_project objects][groupKey] = groupDict;
@@ -591,6 +611,68 @@ - (NSString *)pathRelativeToProjectRoot
     return _pathRelativeToProjectRoot;
 }
 
+//-------------------------------------------------------------------------------------------
+#pragma mark - XCBuildFile Methods
+
+- (BOOL) canBecomeBuildFile
+{
+    return _memberType == PBXVariantGroupType;
+}
+
+- (XcodeMemberType)buildPhase
+{
+    if (_memberType == PBXVariantGroupType)
+        return PBXResourcesBuildPhaseType;
+
+    return PBXNilType;
+}
+
+- (NSString *)buildFileKey
+{
+    if (_buildFileKey == nil) {
+        [[_project objects] enumerateKeysAndObjectsUsingBlock:^(NSString *key, NSDictionary *obj, BOOL *stop) {
+            if ([[obj valueForKey:@"isa"] xce_hasBuildFileType]) {
+                if ([[obj valueForKey:@"fileRef"] isEqualToString:_key]) {
+                    _buildFileKey = [key copy];
+                }
+            }
+        }];
+    }
+    return [_buildFileKey copy];
+
+}
+
+
+- (void)becomeBuildFile
+{
+    if (![self isBuildFile]) {
+        if ([self canBecomeBuildFile]) {
+            NSMutableDictionary *sourceBuildFile = [NSMutableDictionary dictionary];
+            sourceBuildFile[@"isa"] = [NSString xce_stringFromMemberType:PBXBuildFileType];
+            sourceBuildFile[@"fileRef"] = _key;
+            NSString *buildFileKey = [[XCKeyBuilder forItemNamed:[self.displayName stringByAppendingString:@".buildFile"]] build];
+            [_project objects][buildFileKey] = sourceBuildFile;
+        }
+    }
+}
+
+- (BOOL)isBuildFile
+{
+    if ([self canBecomeBuildFile] && _isBuildFile == nil) {
+        _isBuildFile = @NO;
+        [[_project objects] enumerateKeysAndObjectsUsingBlock:^(NSString *key, NSDictionary *obj, BOOL *stop) {
+            if ([[obj valueForKey:@"isa"] xce_hasBuildFileType]) {
+                if ([[obj valueForKey:@"fileRef"] isEqualToString:_key]) {
+                    _isBuildFile = nil;
+
+                    _isBuildFile = @YES;
+                }
+            }
+        }];
+    }
+    return [_isBuildFile boolValue];
+}
+
 //-------------------------------------------------------------------------------------------
 #pragma mark - Utility Methods
 
@@ -828,7 +910,7 @@ - (NSDictionary *)makeFileReferenceWithPath:(NSString *)path name:(NSString *)na
 - (NSDictionary *)asDictionary
 {
     NSMutableDictionary *groupData = [NSMutableDictionary dictionary];
-    groupData[@"isa"] = [NSString xce_stringFromMemberType:PBXGroupType];
+    groupData[@"isa"] = [NSString xce_stringFromMemberType:_memberType];
     groupData[@"sourceTree"] = @"<group>";
 
     if (_alias != nil) {
@@ -859,4 +941,4 @@ - (void)addSourceFile:(XCSourceFile *)sourceFile toTargets:(NSArray *)targets
     }
 }
 
-@end
+@end

Source/XCProject.m

@@ -300,8 +300,12 @@ - (XCGroup *)groupWithPathFromRoot:(NSString *)path
 
 - (XCGroup *)createGroupWithDictionary:(NSDictionary *)dictionary forKey:(NSString *)key
 {
-    return [XCGroup groupWithProject:self key:key alias:[dictionary valueForKey:@"name"]
-                                path:[dictionary valueForKey:@"path"] children:[dictionary valueForKey:@"children"]];
+    return [XCGroup groupWithProject:self
+                                 key:key
+                               alias:[dictionary valueForKey:@"name"]
+                                path:[dictionary valueForKey:@"path"]
+                            children:[dictionary valueForKey:@"children"]
+            memberType:[[dictionary valueForKey:@"isa"] xce_asMemberType]];
 }
 
 //-------------------------------------------------------------------------------------------
@@ -535,4 +539,4 @@ - (BOOL)doPruneEmptyGroups
     return hadEmptyGroups;
 }
 
-@end
+@end

XcodeEditorTests/XCGroupTests.m

@@ -27,7 +27,7 @@ @interface XCFrameworkPath : NSObject
 @implementation XCFrameworkPath
 
 static const NSString *SDK_PATH =
-        @"/Applications/Xcode.app/Contents/Developer/Platforms/iPhoneOS.platform/Developer/SDKs/iPhoneOS9.1.sdk";
+        @"/Applications/Xcode.app/Contents/Developer/Platforms/iPhoneOS.platform/Developer/SDKs/iPhoneOS.sdk";
 
 + (NSString *)eventKitUIPath
 {
@@ -488,4 +488,4 @@ - (void)test_allows_deleting_a_group_after_adding_contents
 }
 
 
-@end
+@end