// swift-tools-version:5.9

import PackageDescription
import Foundation

var unsupportedTests: Set<String> = []
#if !os(macOS) && !os(iOS) && !os(watchOS) && !os(tvOS)
unsupportedTests.insert("ObjectiveCNoBridgingStubs")
unsupportedTests.insert("ObjectiveCBridging")
unsupportedTests.insert("ObjectiveCBridgingStubs")
#endif

unsupportedTests.insert("SimpleArraySpecialization")

//===---
// Single Source Libraries
//

/// Return the source files in subDirectory that we will translate into
/// libraries. Each source library will be compiled as its own module.
func getSingleSourceLibraries(subDirectory: String) -> [String] {
  let f = FileManager.`default`
  let dirURL = URL(fileURLWithPath: subDirectory)
  let fileURLs = try! f.contentsOfDirectory(at: dirURL,
                                            includingPropertiesForKeys: nil)
  return fileURLs.compactMap { (path: URL) -> String? in
    guard let lastDot = path.lastPathComponent.lastIndex(of: ".") else {
      return nil
    }
    let ext = String(path.lastPathComponent.suffix(from: lastDot))
    guard ext == ".swift" else { return nil }

    let name = String(path.lastPathComponent.prefix(upTo: lastDot))

    // Test names must have a single component.
    if name.contains(".") { return nil }

    if unsupportedTests.contains(name) {
      // We do not support this test.
      return nil
    }

    return name
  }
}

var singleSourceLibraryDirs: [String] = []
singleSourceLibraryDirs.append("single-source")

var singleSourceLibraries: [String] = singleSourceLibraryDirs.flatMap {
  getSingleSourceLibraries(subDirectory: $0)
}

var cxxSingleSourceLibraryDirs: [String] = ["cxx-source"]
var cxxSingleSourceLibraries: [String] = cxxSingleSourceLibraryDirs.flatMap {
  getSingleSourceLibraries(subDirectory: $0)
}

//===---
// Multi Source Libraries
//

func getMultiSourceLibraries(subDirectory: String) -> [(String, String)] {
  let f = FileManager.`default`
  let dirURL = URL(string: subDirectory)!
  let subDirs = try! f.contentsOfDirectory(at: dirURL, includingPropertiesForKeys: nil)
  return subDirs.map { (subDirectory, $0.lastPathComponent) }
}

var multiSourceLibraryDirs: [String] = []
multiSourceLibraryDirs.append("multi-source")

var multiSourceLibraries: [(parentSubDir: String, name: String)] = multiSourceLibraryDirs.flatMap {
  getMultiSourceLibraries(subDirectory: $0)
}

//===---
// Products
//

var products: [Product] = []
products.append(.library(name: "TestsUtils", type: .static, targets: ["TestsUtils"]))
//products.append(.library(name: "SimpleArray", type: .static, targets: ["SimpleArray"]))
products.append(.library(name: "DriverUtils", type: .static, targets: ["DriverUtils"]))
#if os(macOS) || os(iOS) || os(watchOS) || os(tvOS)
products.append(.library(name: "ObjectiveCTests", type: .static, targets: ["ObjectiveCTests"]))
#endif
products.append(.executable(name: "SwiftBench", targets: ["SwiftBench"]))

products += singleSourceLibraries.map { .library(name: $0, type: .static, targets: [$0]) }
products += cxxSingleSourceLibraries.map { .library(name: $0, type: .static, targets: [$0]) }
products += multiSourceLibraries.map {
  return .library(name: $0.name, type: .static, targets: [$0.name])
}

//===---
// Targets
//

var targets: [Target] = []
targets.append(.target(name: "TestsUtils", path: "utils", sources: ["TestsUtils.swift"]))
// targets.append(.target(
//   name: "SimpleArray",
//   path: "utils",
//   sources: ["SimpleArray.swift"],
//   swiftSettings: [.unsafeFlags(["-Xfrontend",
//                                 "-enable-experimental-feature",
//                                 "LayoutPrespecialization"])]))
targets.append(.systemLibrary(name: "LibProc", path: "utils/LibProc"))
targets.append(
  .target(name: "DriverUtils",
    dependencies: [.target(name: "TestsUtils"), "LibProc"],
    path: "utils",
    sources: ["DriverUtils.swift", "ArgParse.swift"]))

var swiftBenchDeps: [Target.Dependency] = [.target(name: "TestsUtils")]
#if os(macOS) || os(iOS) || os(watchOS) || os(tvOS)
swiftBenchDeps.append(.target(name: "ObjectiveCTests"))
#endif
swiftBenchDeps.append(.target(name: "DriverUtils"))
swiftBenchDeps += singleSourceLibraries.map { .target(name: $0) }
swiftBenchDeps += cxxSingleSourceLibraries.map { .target(name: $0) }
swiftBenchDeps += multiSourceLibraries.map { .target(name: $0.name) }

targets.append(
    .target(name: "SwiftBench",
    dependencies: swiftBenchDeps,
    path: "utils",
    sources: ["main.swift"],
    swiftSettings: [.interoperabilityMode(.Cxx),
                    .unsafeFlags(["-I",
                                  "utils/CxxTests"])]))

#if os(macOS) || os(iOS) || os(watchOS) || os(tvOS)
targets.append(
  .target(name: "ObjectiveCTests",
    path: "utils/ObjectiveCTests",
    publicHeadersPath: "."))
#endif

var singleSourceDeps: [Target.Dependency] = [.target(name: "TestsUtils"), /* .target(name: "SimpleArray") */]
#if os(macOS) || os(iOS) || os(watchOS) || os(tvOS)
singleSourceDeps.append(.target(name: "ObjectiveCTests"))
#endif

targets += singleSourceLibraries.map { name in
  if name == "ObjectiveCNoBridgingStubs" {
    return .target(
      name: name,
      dependencies: singleSourceDeps,
      path: "single-source",
      sources: ["\(name).swift"],
      swiftSettings: [.unsafeFlags(["-Xfrontend",
                                    "-disable-swift-bridge-attr"])])
  }
  return .target(name: name,
      dependencies: singleSourceDeps,
      path: "single-source",
      sources: ["\(name).swift"])
}

targets += cxxSingleSourceLibraries.map { name in
  return .target(
    name: name,
    dependencies: singleSourceDeps,
    path: "cxx-source",
    sources: ["\(name).swift"],
    swiftSettings: [.interoperabilityMode(.Cxx),
                    .unsafeFlags(["-I",
                                  "utils/CxxTests",
                                  // FIXME: https://github.com/apple/swift/issues/61453
                                  "-Xfrontend", "-validate-tbd-against-ir=none"])])
}

targets += multiSourceLibraries.map { lib in
  return .target(
    name: lib.name,
    dependencies: [
      .target(name: "TestsUtils")
    ],
    path: "\(lib.parentSubDir)/\(lib.name)")
}

//===---
// Top Level Definition
//

var p = Package(
  name: "swiftbench",
  products: products,
  targets: targets,
  swiftLanguageVersions: [.v4],
  cxxLanguageStandard: .cxx20
)

// Let's build for Swift 5.5-aligned runtimes.
#if os(macOS) || os(iOS) || os(watchOS) || os(tvOS)
p.platforms = [.macOS(.v12), .iOS(.v15), .watchOS(.v8), .tvOS(.v15)]
#endif