Exclude files or code from obfuscation

.gitignore

@@ -1,2 +1,5 @@
 
 *.profraw
+.DS_Store
+Index/DataStore/*
+Build/Intermediates.noindex/*

SwiftShieldTests/AutomaticSwiftShieldTests/AutomaticSwiftShieldTests.swift

@@ -27,16 +27,17 @@ class AutomaticSwiftShieldTests: XCTestCase {
                           ReferenceData(name: "MyType", line: 6, column: 28),
                           ReferenceData(name: "ViewController", line: 10, column: 15),
                           ReferenceData(name: "fakeMethod", line: 10, column: 30)]
+
         let originalFileData = loadFile("MockOriginalFile", ofType: "txt")
         let originalFile = String(data: originalFileData, encoding: .utf8)!
-        let obfuscatedFile = AutomaticSwiftShield(basePath: "abc", projectToBuild: "abc", schemeToBuild: "abc", modulesToIgnore: [], protectedClassNameSize: 0).generateObfuscatedFile(fromString: originalFile, references: references, obfuscationData: obfuscationData)
+        let obfuscatedFile = AutomaticSwiftShield(basePath: "abc", projectToBuild: "abc", schemeToBuild: "abc", modulesToIgnore: [], classesToIgnore: [], protectedClassNameSize: 0, excludedPrefixTag: "", excludedSuffixTag: "").generateObfuscatedFile(fromString: originalFile, references: references, obfuscationData: obfuscationData)
         let expectedFileData = loadFile("MockObfuscatedFile", ofType: "txt")
         let expectedFile = String(data: expectedFileData, encoding: .utf8)!
         XCTAssertEqual(obfuscatedFile, expectedFile)
     }
 
     func testPlistExtractor() {
-        let protector = AutomaticSwiftShield(basePath: "abc", projectToBuild: "abc", schemeToBuild: "abc", modulesToIgnore: [], protectedClassNameSize: 0)
+        let protector = AutomaticSwiftShield(basePath: "abc", projectToBuild: "abc", schemeToBuild: "abc", modulesToIgnore: [], classesToIgnore: [], protectedClassNameSize: 0, excludedPrefixTag: "", excludedSuffixTag: "")
         let plist = path(for: "MockPlist", ofType: "plist")
         let file = File(filePath: plist)
         let data = protector.getPlistVersionAndNumber(file)!
@@ -45,7 +46,7 @@ class AutomaticSwiftShieldTests: XCTestCase {
     }
 
     func testPlistPrincipalClassObfuscation() {
-        let protector = AutomaticSwiftShield(basePath: "abc", projectToBuild: "abc", schemeToBuild: "abc", modulesToIgnore: [], protectedClassNameSize: 0)
+        let protector = AutomaticSwiftShield(basePath: "abc", projectToBuild: "abc", schemeToBuild: "abc", modulesToIgnore: [], classesToIgnore: [], protectedClassNameSize: 0, excludedPrefixTag: "", excludedSuffixTag: "")
         let plist = path(for: "MockPlist", ofType: "plist")
         let file = MockFile(path: plist)
         let obfuscationData = AutomaticObfuscationData(modules: [Module(name: "mock", plists: [file])])

swiftshield-Sources/AutomaticSwiftShield.swift

@@ -5,7 +5,10 @@ class AutomaticSwiftShield: Protector {
     let projectToBuild: String
     let schemeToBuild: String
     let modulesToIgnore: Set<String>
-
+    let filesToIgnore: Set<String>
+    let excludedPrefixTag: String
+    let excludedSuffixTag: String
+
     var isWorkspace: Bool {
         return projectToBuild.hasSuffix(".xcworkspace")
     }
@@ -14,10 +17,16 @@ class AutomaticSwiftShield: Protector {
          projectToBuild: String,
          schemeToBuild: String,
          modulesToIgnore: Set<String>,
-         protectedClassNameSize: Int) {
+         classesToIgnore: Set<String>,
+         protectedClassNameSize: Int,
+         excludedPrefixTag: String,
+         excludedSuffixTag: String) {
         self.projectToBuild = projectToBuild
         self.schemeToBuild = schemeToBuild
         self.modulesToIgnore = modulesToIgnore
+        self.filesToIgnore = classesToIgnore
+        self.excludedPrefixTag = excludedPrefixTag
+        self.excludedSuffixTag = excludedSuffixTag
         super.init(basePath: basePath, protectedClassNameSize: protectedClassNameSize)
         if self.schemeToBuild.isEmpty || self.projectToBuild.isEmpty {
             Logger.log(.helpText)
@@ -108,6 +117,19 @@ extension AutomaticSwiftShield {
         guard let name = dict.getString(key: sourceKit.nameID)?.trueName, let usr = dict.getString(key: sourceKit.usrID) else {
             return nil
         }
+
+        if self.filesToIgnore.contains(name) {
+            return nil
+        }
+
+        if self.excludedPrefixTag != "" && name.hasPrefix(self.excludedPrefixTag) {
+            return nil
+        }
+
+        if self.excludedSuffixTag != "" && name.hasSuffix(excludedSuffixTag) {
+            return nil
+        }
+
         guard let protected = obfuscationData.obfuscationDict[name] else {
             let newName = String.random(length: self.protectedClassNameSize, excluding: obfuscationData.allObfuscatedNames)
             obfuscationData.obfuscationDict[name] = newName

swiftshield-Sources/SourceKit.swift

@@ -27,9 +27,9 @@ final class SourceKit {
              "struct",
              "protocol":
             return .object
-//        case "var.instance",
-//             "var.class":
-//            return .property
+        case "var.instance",
+             "var.class":
+            return .property
         case "function.free",
              "function.method.instance",
              "function.method.static",

swiftshield-Sources/main.swift

@@ -37,7 +37,11 @@ if automatic {
     let schemeToBuild = UserDefaults.standard.string(forKey: "automatic-project-scheme") ?? ""
     let projectToBuild = UserDefaults.standard.string(forKey: "automatic-project-file") ?? ""
     let modulesToIgnore = UserDefaults.standard.string(forKey: "ignore-modules")?.components(separatedBy: ",") ?? []
-    protector = AutomaticSwiftShield(basePath: basePath, projectToBuild: projectToBuild, schemeToBuild: schemeToBuild, modulesToIgnore: Set(modulesToIgnore), protectedClassNameSize: protectedClassNameSize)
+    let excludedSuffixTag = UserDefaults.standard.string(forKey: "excluded-suffix-tag") ?? ""
+    let excludedPrefixTag = UserDefaults.standard.string(forKey: "excluded-prefix-tag") ?? ""
+    let filesToIgnore = UserDefaults.standard.string(forKey: "ignore-files")?.components(separatedBy: ",") ?? []
+
+    protector = AutomaticSwiftShield(basePath: basePath, projectToBuild: projectToBuild, schemeToBuild: schemeToBuild, modulesToIgnore: Set(modulesToIgnore), classesToIgnore: Set(filesToIgnore), protectedClassNameSize: protectedClassNameSize, excludedPrefixTag: excludedPrefixTag, excludedSuffixTag: excludedSuffixTag)
 } else {
     let tag = UserDefaults.standard.string(forKey: "tag") ?? "__s"
     protector = ManualSwiftShield(basePath: basePath, tag: tag, protectedClassNameSize: protectedClassNameSize)