//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

import BuildServerIntegration
@_spi(SourceKitLSP) import BuildServerProtocol
import Foundation
@_spi(SourceKitLSP) import LanguageServerProtocol
@_spi(SourceKitLSP) import LanguageServerProtocolExtensions
import SKLogging
import SKTestSupport
import TSCBasic
import XCTest

final class LegacyBuildServerTests: SourceKitLSPTestCase {
  func testBuildSettingsFromBuildServer() async throws {
    let project = try await ExternalBuildServerTestProject(
      files: [
        "Test.swift": """
        #if DEBUG
        #error("DEBUG SET")
        #else
        #error("DEBUG NOT SET")
        #endif
        """
      ],
      buildServer: """
        class BuildServer(LegacyBuildServer):
          def register_for_changes(self, notification: Dict[str, object]):
            if notification["action"] == "register":
              self.send_notification(
                "build/sourceKitOptionsChanged",
                {
                  "uri": notification["uri"],
                  "updatedOptions": {
                    "options": [r"$TEST_DIR/Test.swift", "-DDEBUG", $SDK_ARGS]
                  },
                },
              )
        """
    )

    let (uri, _) = try project.openDocument("Test.swift")
    try await repeatUntilExpectedResult {
      let diags = try await project.testClient.send(
        DocumentDiagnosticsRequest(textDocument: TextDocumentIdentifier(uri))
      )
      return diags.fullReport?.items.map(\.message) == ["DEBUG SET"]
    }
  }

  func testBuildSettingsFromBuildServerChanged() async throws {
    let project = try await ExternalBuildServerTestProject(
      files: [
        "Test.swift": """
        #if DEBUG
        #error("DEBUG SET")
        #else
        #error("DEBUG NOT SET")
        #endif
        """
      ],
      buildServer: """
        import threading

        class BuildServer(LegacyBuildServer):
          def send_delayed_options_changed(self, uri: str):
            self.send_sourcekit_options_changed(uri, [r"$TEST_DIR/Test.swift", "-DDEBUG", $SDK_ARGS])

          def register_for_changes(self, notification: Dict[str, object]):
            if notification["action"] != "register":
              return
            self.send_sourcekit_options_changed(
              notification["uri"],
              [r"$TEST_DIR/Test.swift", $SDK_ARGS]
            )
            threading.Timer(1, self.send_delayed_options_changed, [notification["uri"]]).start()
        """
    )

    let (uri, _) = try project.openDocument("Test.swift")
    try await repeatUntilExpectedResult {
      let diags = try await project.testClient.send(
        DocumentDiagnosticsRequest(textDocument: TextDocumentIdentifier(uri))
      )
      return diags.fullReport?.items.map(\.message) == ["DEBUG SET"]
    }
  }
}
