import Cocoa
import Doubt
import Either
import Prelude
import Madness

func benchmark<T>(label: String? = nil, _ f: () -> T) -> T {
	let start = NSDate.timeIntervalSinceReferenceDate()
	let result = f()
	let end = NSDate.timeIntervalSinceReferenceDate()
	print((label.map { "\($0): " } ?? "") + "\(end - start)s")
	return result
}

let arguments = BoundsCheckedArray(array: Process.arguments)

let empty = "{}\n"
print(parse(json, input: empty))
let dict = "{\"hello\":\"world\"}"
print(parse(json, input: dict))
let dictWithSpaces = "{ \"hello\" : \"world\" }"
print(parse(json, input: dictWithSpaces))

let dictWithMembers = "{\"hello\":\"world\",\"sup\":\"cat\"}"
print(parse(json, input: dictWithMembers))

let dictWithArray = "{\"hello\": [\"world\"],\"sup\": [\"cat\", \"dog\", \"keith\"] }"
print(parse(json, input: dictWithArray))

func diffAndSerialize(a aString: String, b bString: String, to: String) throws {
	let aParsed = benchmark("parsing a") { curry(parse)(json)(aString) }
	guard let a = aParsed.right else {
		_ = aParsed.left.map { print("error parsing a:", $0) }
		return
	}
	let bParsed = benchmark("parsing b") { curry(parse)(json)(bString) }
	guard let b = bParsed.right else {
		_ = bParsed.left.map { print("error parsing b:", $0) }
		return
	}

	let diff = benchmark("diffing a & b") {
		Interpreter<CofreeJSON>(equal: CofreeJSON.equals(annotation: const(true), leaf: ==), comparable: const(true), cost: Free.sum(Patch.difference)).run(a, b)
	}

	let range: Range<String.Index> -> Doubt.JSON = {
		let start = Int(String($0.startIndex))!
		let end = Int(String($0.endIndex))!
		return [
			.Number(Double(start)),
			.Number(Double(end - start)),
		]
	}
	let JSON: Doubt.JSON = benchmark("converting diff to JSON") {
		[
			"before": .String(aString),
			"after": .String(bString),
			"diff": diff.JSON(pure: { $0.JSON { $0.JSON(annotation: range, leaf: { $0.JSON }) } }, leaf: { $0.JSON }, annotation: {
				[
					"before": range($0),
					"after": range($1),
				]
			}),
		]
	}

	let data = benchmark("serializing JSON to NSData") {
		JSON.serialize()
	}

	try data.writeToFile(to, options: .DataWritingAtomic)
}

let readFile = { (path: String) -> String? in
	guard let data = try? NSString(contentsOfFile: path, encoding: NSUTF8StringEncoding) else { return nil }
	return data as String?
}

if let a = arguments[1].flatMap(readFile), b = arguments[2].flatMap(readFile), c = arguments[3] {
	try diffAndSerialize(a: a, b: b, to: c)
}