//
//  FuzzyRangeTests.swift
//
//  CotEditor
//  https://coteditor.com
//
//  Created by 1024jp on 2020-01-14.
//
//  ---------------------------------------------------------------------------
//
//  © 2020-2023 1024jp
//
//  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
//
//  https://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 XCTest
@testable import CotEditor

final class FuzzyRangeTests: XCTestCase {
    
    func testFuzzyCharacterRange() {
        
        let string = "0123456789"
        
        XCTAssertEqual(string.range(in: FuzzyRange(location: 2, length: 2)), NSRange(location: 2, length: 2))
        XCTAssertEqual(string.range(in: FuzzyRange(location: -1, length: 0)), NSRange(location: 10, length: 0))
        XCTAssertEqual(string.range(in: FuzzyRange(location: -2, length: 1)), NSRange(location: 9, length: 1))
        XCTAssertEqual(string.range(in: FuzzyRange(location: 3, length: -1)), NSRange(3..<9))
        XCTAssertEqual(string.range(in: FuzzyRange(location: 3, length: -2)), NSRange(location: 3, length: "45678".utf16.count))
    }
    
    
    func testFuzzyLineRange() throws {
        
        let string = "1\r\n2\r\n3\r\n4"  // 1 based
        var range: NSRange
        
        range = try XCTUnwrap(string.rangeForLine(in: FuzzyRange(location: 1, length: 2)))
        XCTAssertEqual((string as NSString).substring(with: range), "1\r\n2\r\n")
        
        range = try XCTUnwrap(string.rangeForLine(in: FuzzyRange(location: 4, length: 1)))
        XCTAssertEqual((string as NSString).substring(with: range), "4")
        
        range = try XCTUnwrap(string.rangeForLine(in: FuzzyRange(location: 3, length: 0)))
        XCTAssertEqual((string as NSString).substring(with: range), "3\r\n")
        
        range = try XCTUnwrap(string.rangeForLine(in: FuzzyRange(location: -1, length: 1)))
        XCTAssertEqual((string as NSString).substring(with: range), "4")
        
        range = try XCTUnwrap(string.rangeForLine(in: FuzzyRange(location: -2, length: 1)))
        XCTAssertEqual((string as NSString).substring(with: range), "3\r\n")
        
        range = try XCTUnwrap(string.rangeForLine(in: FuzzyRange(location: 2, length: -2)))
        XCTAssertEqual((string as NSString).substring(with: range), "2\r\n")
        
        range = try XCTUnwrap("1\n".rangeForLine(in: FuzzyRange(location: -1, length: 0)))
        XCTAssertEqual(range, NSRange(location: 2, length: 0))
        
        range = try XCTUnwrap(string.rangeForLine(in: FuzzyRange(location: 1, length: 2), includingLineEnding: false))
        XCTAssertEqual((string as NSString).substring(with: range), "1\r\n2")
    }
    
    
    func testFormattingFuzzyRange() {
        
        XCTAssertEqual(FuzzyRange(location: 0, length: 0).formatted(), "0")
        XCTAssertEqual(FuzzyRange(location: 1, length: 0).formatted(), "1")
        XCTAssertEqual(FuzzyRange(location: 1, length: 1).formatted(), "1")
        XCTAssertEqual(FuzzyRange(location: 1, length: 2).formatted(), "1:2")
        XCTAssertEqual(FuzzyRange(location: -1, length: 0).formatted(), "-1")
        XCTAssertEqual(FuzzyRange(location: -1, length: -1).formatted(), "-1:-1")
    }
    
    
    func testParsingFuzzyRange() throws {
        
        let parser = FuzzyRangeParseStrategy()
        
        XCTAssertEqual(try parser.parse("0"), FuzzyRange(location: 0, length: 0))
        XCTAssertEqual(try parser.parse("1"), FuzzyRange(location: 1, length: 0))
        XCTAssertEqual(try parser.parse("1:2"), FuzzyRange(location: 1, length: 2))
        XCTAssertEqual(try parser.parse("-1"), FuzzyRange(location: -1, length: 0))
        XCTAssertEqual(try parser.parse("-1:-1"), FuzzyRange(location: -1, length: -1))
        XCTAssertThrowsError(try parser.parse(""))
        XCTAssertThrowsError(try parser.parse("abc"))
        XCTAssertThrowsError(try parser.parse("1:a"))
        XCTAssertThrowsError(try parser.parse("1:1:1"))
    }
    
    
    func testFuzzyLocation() throws {
        
        let string = "1\r\n2\r\n3\r\n456\n567"  // 1 based
        
        XCTAssertEqual(try string.fuzzyLocation(line: 0), 0)
        XCTAssertEqual(try string.fuzzyLocation(line: 0, column: 1), 1)
        
        XCTAssertEqual(try string.fuzzyLocation(line: 1), 0)
        XCTAssertEqual(try string.fuzzyLocation(line: 2), 3)
        XCTAssertEqual(try string.fuzzyLocation(line: 4), 9)
        XCTAssertEqual(try string.fuzzyLocation(line: 5), 13)
        XCTAssertEqual(try string.fuzzyLocation(line: -1), 13)
        XCTAssertEqual(try string.fuzzyLocation(line: -2), 9)
        XCTAssertEqual(try string.fuzzyLocation(line: -5), 0)
        XCTAssertThrowsError(try string.fuzzyLocation(line: -6))
        
        // line with a line ending
        XCTAssertEqual(try string.fuzzyLocation(line: 4, column: 0), 9)
        XCTAssertEqual(try string.fuzzyLocation(line: 4, column: 1), 10)
        XCTAssertEqual(try string.fuzzyLocation(line: 4, column: 3), 12)
        XCTAssertThrowsError(try string.fuzzyLocation(line: 4, column: 4))
        XCTAssertEqual(try string.fuzzyLocation(line: 4, column: -1), 12)
        XCTAssertEqual(try string.fuzzyLocation(line: 4, column: -2), 11)
        
        // line without any line endings (the last line)
        XCTAssertEqual(try string.fuzzyLocation(line: 5, column: 0), 13)
        XCTAssertEqual(try string.fuzzyLocation(line: 5, column: 1), 14)
        XCTAssertEqual(try string.fuzzyLocation(line: 5, column: 3), 16)
        XCTAssertThrowsError(try string.fuzzyLocation(line: 5, column: 4))
        XCTAssertEqual(try string.fuzzyLocation(line: 5, column: -1), 16)
        XCTAssertEqual(try string.fuzzyLocation(line: 5, column: -2), 15)
    }
}