lib/installer.rb in cocoapods-keys-0.9.0 vs lib/installer.rb in cocoapods-keys-0.9.5

@@ -15,27 +15,37 @@
       Pod::UI.section 'Adding keys' do
         key_master = KeyMaster.new(keyring)
 
         keys_folder = File.join(@sandbox_root, 'Keys')
         keys_headers_folder = File.join(@sandbox_root, 'Headers', 'Public', 'CocoaPods-Keys')
-        interface_file = File.join(keys_headers_folder, key_master.name + '.h')
+        interface_name = key_master.name + '.h'
+        interface_file = File.join(keys_headers_folder, interface_name)
         implementation_file = File.join(keys_folder, key_master.name + '.m')
+     
         Dir.mkdir keys_folder unless Dir.exists? keys_folder
         Dir.mkdir keys_headers_folder unless Dir.exists? keys_headers_folder
         File.open(interface_file, 'w') { |f| f.write(key_master.interface) }
         File.open(implementation_file, 'w') { |f| f.write(key_master.implementation) }
 
         project = Xcodeproj::Project.open File.join(@sandbox_root, 'Pods.xcodeproj')
 
         group = project.new_group('Keys')
-        group.new_file(interface_file)
+        interface = group.new_file(interface_file)
         implementation = group.new_file(implementation_file)
 
         pods_target = project.targets.detect { |t| t.name == 'Pods' }
         unless pods_target
           pods_target = project.targets.detect { |t| t.name == 'Pods-' + keyring.name }
         end
+
         pods_target.add_file_references [implementation]
+
+        # Swift Pod support
+        if pods_target.product_type == "com.apple.product-type.framework"
+          pods_target.add_file_references [interface]
+          header_ref = pods_target.headers_build_phase.files[-1]
+          header_ref.settings = { "ATTRIBUTES" => ["Public"] }
+        end
 
         project.save
       end
     end
   end