import Foundation

import SwabbleKit

import Testing

import XCTest

@Suite struct WakeWordGateTests {

@Test func matchRequiresGapAfterTrigger() {

final class WakeWordGateTests: XCTestCase {

func testMatchRequiresGapAfterTrigger() {

let transcript = "hey clawd do thing"

let segments = makeSegments(

transcript: transcript,

("thing", 0.5, 0.1),

])

let config = WakeWordGateConfig(triggers: ["clawd"], minPostTriggerGap: 0.3)

#expect(WakeWordGate.match(transcript: transcript, segments: segments, config: config) == nil)

XCTAssertNil(WakeWordGate.match(transcript: transcript, segments: segments, config: config))

}

@Test func matchAllowsGapAndExtractsCommand() {

func testMatchAllowsGapAndExtractsCommand() {

let transcript = "hey clawd do thing"

let segments = makeSegments(

transcript: transcript,

])

let config = WakeWordGateConfig(triggers: ["clawd"], minPostTriggerGap: 0.3)

let match = WakeWordGate.match(transcript: transcript, segments: segments, config: config)

#expect(match?.command == "do thing")

XCTAssertEqual(match?.command, "do thing")

}

@Test func matchHandlesMultiWordTriggers() {

func testMatchHandlesMultiWordTriggers() {

let transcript = "hey clawd do it"

let segments = makeSegments(

transcript: transcript,

])

let config = WakeWordGateConfig(triggers: ["hey clawd"], minPostTriggerGap: 0.3)

let match = WakeWordGate.match(transcript: transcript, segments: segments, config: config)

#expect(match?.command == "do it")

XCTAssertEqual(match?.command, "do it")

}

@Test func matchPrefersMostSpecificTriggerWhenOverlapping() {

func testMatchPrefersMostSpecificTriggerWhenOverlapping() {

let transcript = "hey clawd do it"

let segments = makeSegments(

transcript: transcript,

])

let config = WakeWordGateConfig(triggers: ["clawd", "hey clawd"], minPostTriggerGap: 0.3)

let match = WakeWordGate.match(transcript: transcript, segments: segments, config: config)

#expect(match?.trigger == "hey clawd")

XCTAssertEqual(match?.trigger, "hey clawd")

}

@Test func commandTextHandlesForeignRangeIndices() {

func testCommandTextHandlesForeignRangeIndices() {

let transcript = "hey clawd do thing"

let other = "do thing"

let foreignRange = other.range(of: "do")

segments: segments,

triggerEndTime: 0.3)

#expect(command == "do thing")

XCTAssertEqual(command, "do thing")

}

}