darwin-apk/Sources/dpk-cli/Subcommands/DpkGraphCommand.swift

/*
 * darwin-apk © 2024, 2025 Gay Pizza Specifications
 * SPDX-License-Identifier: Apache-2.0
 */

import Foundation
import ArgumentParser
import darwin_apk

struct DpkGraphCommand: AsyncParsableCommand {
  static let configuration = CommandConfiguration(commandName: "graph")

  func run() async throws(ExitCode) {
    let graph: ApkPackageGraph
    do {
      let localRepositories = try await ApkRepositoriesConfig()

      var timerStart = DispatchTime.now()
      graph = ApkPackageGraph(index:
        try await ApkIndexReader.resolve(localRepositories.repositories, fetch: .lazy))
      print("Index build took \(timerStart.distance(to: .now()).seconds) seconds")
      try graph.pkgIndex.description.write(to: URL(filePath: "packages.txt"), atomically: false, encoding: .utf8)

      timerStart = DispatchTime.now()
      graph.buildGraphNode()
      print("Graph build took \(timerStart.distance(to: .now()).seconds) seconds")

      try graph.shallowIsolates.map { $0.package.nameDescription }.joined(separator: "\n")
        .write(to: URL(filePath: "shallowIsolates.txt"), atomically: false, encoding: .utf8)
      try graph.deepIsolates.map { $0.package.nameDescription }.joined(separator: "\n")
        .write(to: URL(filePath: "deepIsolates.txt"), atomically: false, encoding: .utf8)

      timerStart = DispatchTime.now()
#if false
      let sorted = try graph.parallelOrderSort()
      print("Parallel sort took \(timerStart.distance(to: .now()).seconds) seconds")

      if var out = TextFileWriter(URL(filePath: "sorted.txt")) {
        for (i, set) in sorted.enumerated() {
          print("\(i):\n", to: &out)
          for item in set {
            print("\(item.description)", to: &out)
          }
        }
      }
#else
      let sorted = try graph.orderSort()
      print("Order sort took \(timerStart.distance(to: .now()).seconds) seconds")

      try sorted.map(String.init).joined(separator: "\n")
        .write(to: URL(filePath: "sorted.txt"), atomically: false, encoding: .utf8)
#endif
    } catch {
      fatalError(error.localizedDescription)
    }
  }
}

fileprivate extension DispatchTimeInterval {
  var seconds: Double {
    switch self {
    case .seconds(let value):      Double(value)
    case .milliseconds(let value): Double(value) / 1_000
    case .microseconds(let value): Double(value) / 1_000_000
    case .nanoseconds(let value):  Double(value) / 1_000_000_000
    case .never: .infinity
    @unknown default:
    fatalError("Unsupported")
    }
  }
}