chore: import upstream snapshot with attribution

2026-07-13 12:06:18 +08:00
commit e84fb4a79e
474 changed files with 68321 additions and 0 deletions
@@ -0,0 +1,164 @@
+//===----------------------------------------------------------------------===//
+// Copyright © 2025-2026 Apple Inc. and the container 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 ContainerImagesServiceClient
+import Containerization
+import ContainerizationOCI
+import Foundation
+import Logging
+
+public actor ContentStoreService {
+    private let log: Logger
+    private let contentStore: LocalContentStore
+    private let root: URL
+
+    public init(root: URL, log: Logger) throws {
+        try FileManager.default.createDirectory(at: root, withIntermediateDirectories: true)
+        self.root = root.appendingPathComponent("content")
+        self.contentStore = try LocalContentStore(path: self.root)
+        self.log = log
+    }
+
+    public func get(digest: String) async throws -> URL? {
+        self.log.trace(
+            "ContentStoreService: enter",
+            metadata: [
+                "func": "\(#function)",
+                "digest": "\(digest)",
+            ]
+        )
+        defer {
+            self.log.trace(
+                "ContentStoreService: exit",
+                metadata: [
+                    "func": "\(#function)",
+                    "digest": "\(digest)",
+                ]
+            )
+        }
+
+        return try await self.contentStore.get(digest: digest)?.path
+    }
+
+    @discardableResult
+    public func delete(digests: [String]) async throws -> ([String], UInt64) {
+        self.log.trace(
+            "ContentStoreService: enter",
+            metadata: [
+                "func": "\(#function)",
+                "digests": "\(digests)",
+            ]
+        )
+        defer {
+            self.log.trace(
+                "ContentStoreService: exit",
+                metadata: [
+                    "func": "\(#function)",
+                    "digests": "\(digests)",
+                ]
+            )
+        }
+
+        return try await self.contentStore.delete(digests: digests)
+    }
+
+    @discardableResult
+    public func delete(keeping: [String]) async throws -> ([String], UInt64) {
+        self.log.debug(
+            "ContentStoreService: enter",
+            metadata: [
+                "func": "\(#function)",
+                "keeping": "\(keeping)",
+            ]
+        )
+        defer {
+            self.log.debug(
+                "ContentStoreService: exit",
+                metadata: [
+                    "func": "\(#function)",
+                    "keeping": "\(keeping)",
+                ]
+            )
+        }
+
+        return try await self.contentStore.delete(keeping: keeping)
+    }
+
+    public func newIngestSession() async throws -> (id: String, ingestDir: URL) {
+        self.log.debug(
+            "ContentStoreService: enter",
+            metadata: [
+                "func": "\(#function)"
+            ]
+        )
+        defer {
+            self.log.debug(
+                "ContentStoreService: exit",
+                metadata: [
+                    "func": "\(#function)"
+                ]
+            )
+        }
+        return try await self.contentStore.newIngestSession()
+    }
+
+    public func completeIngestSession(_ id: String) async throws -> [String] {
+        self.log.debug(
+            "ContentStoreService: enter",
+            metadata: [
+                "func": "\(#function)",
+                "id": "\(id)",
+            ]
+        )
+        defer {
+            self.log.debug(
+                "ContentStoreService: exit",
+                metadata: [
+                    "func": "\(#function)",
+                    "id": "\(id)",
+                ]
+            )
+        }
+
+        return try await self.contentStore.completeIngestSession(id)
+    }
+
+    public func cancelIngestSession(_ id: String) async throws {
+        self.log.debug(
+            "ContentStoreService: enter",
+            metadata: [
+                "func": "\(#function)",
+                "id": "\(id)",
+            ]
+        )
+        defer {
+            self.log.debug(
+                "ContentStoreService: exit",
+                metadata: [
+                    "func": "\(#function)",
+                    "id": "\(id)",
+                ]
+            )
+        }
+
+        return try await self.contentStore.cancelIngestSession(id)
+    }
+
+    /// Total bytes allocated on disk for the content store.
+    public func totalAllocatedSize() async throws -> UInt64 {
+        try await self.contentStore.totalAllocatedSize()
+    }
+}