firebase-ios-sdk/FirebaseStorage/Sources/StorageListResult.swift

// Copyright 2022 Google LLC
//
// 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
//
//      http://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 Foundation

/** Contains the prefixes and items returned by a `StorageReference.list()` call. */
@objc(FIRStorageListResult) open class StorageListResult: NSObject {
  /**
   * The prefixes (folders) returned by a `list()` operation.
   */
  @objc public let prefixes: [StorageReference]

  /**
   * The objects (files) returned by a `list()` operation.
   */
  @objc public let items: [StorageReference]

  /**
   * A token that can be used to resume a previous `list()` operation. `nil`
   * indicates that there are no more results.
   */
  @objc public let pageToken: String?

  // MARK: - NSObject overrides

  @objc override open func copy() -> Any {
    return StorageListResult(withPrefixes: prefixes,
                             items: items,
                             pageToken: pageToken)
  }

  // MARK: - Internal APIs

  convenience init(with dictionary: [String: Any], reference: StorageReference) {
    var prefixes = [StorageReference]()
    var items = [StorageReference]()

    let rootReference = reference.root()
    if let prefixEntries = dictionary["prefixes"] as? [String] {
      for prefixEntry in prefixEntries {
        var pathWithoutTrailingSlash = prefixEntry
        if prefixEntry.hasSuffix("/") {
          pathWithoutTrailingSlash = String(prefixEntry.dropLast())
        }
        prefixes.append(rootReference.child(pathWithoutTrailingSlash))
      }
    }

    if let itemEntries = dictionary["items"] as? [[String: String]] {
      for itemEntry in itemEntries {
        if let item = itemEntry["name"] {
          items.append(rootReference.child(item))
        }
      }
    }
    let pageToken = dictionary["nextPageToken"] as? String
    self.init(withPrefixes: prefixes, items: items, pageToken: pageToken)
  }

  init(withPrefixes prefixes: [StorageReference],
       items: [StorageReference],
       pageToken: String?) {
    self.prefixes = prefixes
    self.items = items
    self.pageToken = pageToken
  }
}