container/Sources/Services/ContainerImagesService/Server/ContentStoreService.swift

//===----------------------------------------------------------------------===//
// Copyright © 2025 Apple Inc. and the container project authors. All rights reserved.
//
// 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: \(#function) digest \(digest)")
        return try await self.contentStore.get(digest: digest)?.path
    }

    @discardableResult
    public func delete(digests: [String]) async throws -> ([String], UInt64) {
        self.log.debug("ContentStoreService: \(#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: \(#function) digests \(keeping)")
        return try await self.contentStore.delete(keeping: keeping)
    }

    public func newIngestSession() async throws -> (id: String, ingestDir: URL) {
        self.log.debug("ContentStoreService: \(#function)")
        return try await self.contentStore.newIngestSession()
    }

    public func completeIngestSession(_ id: String) async throws -> [String] {
        self.log.debug("ContentStoreService: \(#function) id \(id)")
        return try await self.contentStore.completeIngestSession(id)
    }

    public func cancelIngestSession(_ id: String) async throws {
        self.log.debug("ContentStoreService: \(#function) id \(id)")
        return try await self.contentStore.cancelIngestSession(id)
    }
}