From cf83e79d0440cbe03cd0417af6676673d65ba0f7 Mon Sep 17 00:00:00 2001
From: Aaron Farnham <aaron@foreflight.com>
Date: Fri, 1 Sep 2023 10:14:36 -0500
Subject: [PATCH] Add support for .mlpackage bundles

---
 Sources/ProjectSpec/FileType.swift                | 1 +
 Tests/XcodeGenKitTests/SourceGeneratorTests.swift | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/Sources/ProjectSpec/FileType.swift b/Sources/ProjectSpec/FileType.swift
index c794b64e2..b6af3a9fa 100644
--- a/Sources/ProjectSpec/FileType.swift
+++ b/Sources/ProjectSpec/FileType.swift
@@ -87,6 +87,7 @@ extension FileType {
         "intentdefinition": FileType(buildPhase: .sources),
         "metal": FileType(buildPhase: .sources),
         "mlmodel": FileType(buildPhase: .sources),
+        "mlpackage" : FileType(buildPhase: .sources),
         "mlmodelc": FileType(buildPhase: .resources),
         "rcproject": FileType(buildPhase: .sources),
         "iig": FileType(buildPhase: .sources),
diff --git a/Tests/XcodeGenKitTests/SourceGeneratorTests.swift b/Tests/XcodeGenKitTests/SourceGeneratorTests.swift
index c48f6e050..2075abf3b 100644
--- a/Tests/XcodeGenKitTests/SourceGeneratorTests.swift
+++ b/Tests/XcodeGenKitTests/SourceGeneratorTests.swift
@@ -589,6 +589,7 @@ class SourceGeneratorTests: XCTestCase {
                     - file.xcassets
                     - file.metal
                     - file.mlmodel
+                    - file.mlpackage
                     - file.mlmodelc
                     - Info.plist
                     - Intent.intentdefinition
@@ -648,6 +649,7 @@ class SourceGeneratorTests: XCTestCase {
                 try pbxProj.expectFile(paths: ["C", "Info.plist"], buildPhase: BuildPhaseSpec.none)
                 try pbxProj.expectFile(paths: ["C", "file.metal"], buildPhase: .sources)
                 try pbxProj.expectFile(paths: ["C", "file.mlmodel"], buildPhase: .sources)
+                try pbxProj.expectFile(paths: ["C", "file.mlpackage"], buildPhase: .sources)
                 try pbxProj.expectFile(paths: ["C", "file.mlmodelc"], buildPhase: .resources)
                 try pbxProj.expectFile(paths: ["C", "Intent.intentdefinition"], buildPhase: .sources)
                 try pbxProj.expectFile(paths: ["C", "Configuration.storekit"], buildPhase: .resources)