Improve PDF / AI (Adobe Illustrator) recognition

velocityzen · Feb 18, 2022 · 0211a7e · 0211a7e
1 parent 6839f1c
commit 0211a7e
Show file tree

Hide file tree

Showing 8 changed files with 12,778 additions and 342 deletions.
diff --git a/Sources/FileType/FileTypeMatch.swift b/Sources/FileType/FileTypeMatch.swift
@@ -679,7 +679,7 @@ struct FileTypeMatch {
     FileTypeMatch(
       type: .ai,
       matchString: ["%PDF"],
-      match: { findString($0, "Adobe Illustrator", ignore: 1350) }
+      match: { findString($0, "AIPrivateData", ignore: 1350) }
     ),
 
     FileTypeMatch(

diff --git a/Tests/FileTypeTests/FileTypeTests.swift b/Tests/FileTypeTests/FileTypeTests.swift
@@ -176,6 +176,14 @@ final class FileTypeTests: XCTestCase {
     testFileType("fixture-2.doc.cfb", type: .cfb)
   }
 
+  func testPDF() {
+    testFileType("fixture-without-pdf-compatibility.ai", type: .ai)
+    testFileType("fixture-adobe-illustrator.pdf", type: .pdf)
+    testFileType("fixture-smallest.pdf", type: .pdf)
+    testFileType("fixture-fast-web.pdf", type: .pdf)
+    testFileType("fixture-printed.pdf", type: .pdf)
+  }
+
   func testBytesCountForType() {
     XCTAssertEqual(FileType.getBytesCountFor(type: .ac3), 2)
     XCTAssertEqual(FileType.getBytesCountFor(type: .zip), 4)
@@ -210,6 +218,7 @@ final class FileTypeTests: XCTestCase {
     ("testASF", testASF),
     ("testJXL", testJXL),
     ("testCFB", testCFB),
+    ("testPDF", testPDF),
 
     ("testBytesCountForType", testBytesCountForType),
     ("testBytesCountForTypes", testBytesCountForTypes),

diff --git a/Tests/Fixtures/fixture-adobe-illustrator.pdf b/Tests/Fixtures/fixture-adobe-illustrator.pdf
diff --git a/Tests/Fixtures/fixture-fast-web.pdf b/Tests/Fixtures/fixture-fast-web.pdf
diff --git a/Tests/Fixtures/fixture-printed.pdf b/Tests/Fixtures/fixture-printed.pdf
diff --git a/Tests/Fixtures/fixture-smallest.pdf b/Tests/Fixtures/fixture-smallest.pdf
diff --git a/Tests/Fixtures/fixture-without-pdf-compatibility.ai b/Tests/Fixtures/fixture-without-pdf-compatibility.ai
diff --git a/Tests/Fixtures/fixture.ai b/Tests/Fixtures/fixture.ai