chore: import upstream snapshot with attribution

2026-07-13 12:25:30 +08:00
commit 680845cb1c
445 changed files with 103779 additions and 0 deletions
@@ -0,0 +1,90 @@
+//===----------------------------------------------------------------------===//
+// Copyright © 2025-2026 Apple Inc. and the Containerization project authors.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//   https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//===----------------------------------------------------------------------===//
+
+import ContainerizationError
+import ContainerizationOCI
+import Foundation
+
+extension ImageStore {
+    /// A ReferenceManager handles the mappings between an image's
+    /// reference and the underlying descriptor inside of a content store.
+    internal actor ReferenceManager {
+        private let path: URL
+
+        private typealias State = [String: Descriptor]
+        private var images: State
+
+        public init(path: URL) throws {
+            try FileManager.default.createDirectory(at: path, withIntermediateDirectories: true)
+
+            self.path = path
+            self.images = [:]
+        }
+
+        private func load() throws -> State {
+            let statePath = self.path.appendingPathComponent("state.json")
+            guard FileManager.default.fileExists(atPath: statePath.absolutePath()) else {
+                return [:]
+            }
+            do {
+                let data = try Data(contentsOf: statePath)
+                return try JSONDecoder().decode(State.self, from: data)
+            } catch {
+                throw ContainerizationError(.internalError, message: "failed to load image state \(error.localizedDescription)")
+            }
+        }
+
+        private func save(_ state: State) throws {
+            let statePath = self.path.appendingPathComponent("state.json")
+            try JSONEncoder().encode(state).write(to: statePath)
+        }
+
+        public func delete(reference: String) throws {
+            var state = try self.load()
+            state.removeValue(forKey: reference)
+            try self.save(state)
+        }
+
+        public func delete(image: Image.Description) throws {
+            try self.delete(reference: image.reference)
+        }
+
+        public func create(description: Image.Description) throws {
+            var state = try self.load()
+            state[description.reference] = description.descriptor
+            try self.save(state)
+        }
+
+        public func list() throws -> [Image.Description] {
+            let state = try self.load()
+            return state.map { key, val in
+                let description = Image.Description(reference: key, descriptor: val)
+                return description
+            }
+        }
+
+        public func get(reference: String) throws -> Image.Description {
+            let images = try self.list()
+            let hit = images.first(where: { image in
+                image.reference == reference
+            })
+            guard let hit else {
+                throw ContainerizationError(.notFound, message: "image \(reference) not found")
+            }
+            return hit
+        }
+    }
+}