//
//  JSONParser.swift
//  Doubt
//
//  Created by Josh Vera on 10/16/15.
//  Copyright © 2015 GitHub. All rights reserved.
//

import Foundation
import Madness
import Prelude

public typealias CofreeJSON = Cofree<JSONLeaf, (Range<Line>, Range<Column>, Range<String.Index>)>
public typealias JSONParser = Parser<String.CharacterView, CofreeJSON>.Function

// Inlined for performance reasons
let whitespace = many(oneOf(" \n\r\t"))

// TODO: Parse unicode escape sequence
let escapeChar: StringParser = curry(+) <^> %"\\" <*> ({ String($0) } <^> oneOf("\\\"bfnrt"))
let otherChar: StringParser = { String($0) } <^> noneOf("\"\\")

// Quoted strings parser
// TODO: Improve string parsing
let stringBody: StringParser = { $0.joinWithSeparator("") } <^> many(escapeChar <|> otherChar)
let quoted = %"\"" *> stringBody <* %"\""

typealias MembersParser = Parser<String.CharacterView, [(String, CofreeJSON)]>.Function;

// Parses an array of (String, CofreeJSON) object members
func members(json: JSONParser) -> MembersParser {
	let keyAndKeyTerm: Parser<String.CharacterView, (String, CofreeJSON)>.Function = quoted --> { (lines, columns, range, key) in
		(key, Cofree((lines, columns, range), .Leaf(.String(key))))
	}
	let pairs: Parser<String.CharacterView, (String, CofreeJSON)>.Function = (curry(pair) <^>
		keyAndKeyTerm
		<* whitespace
		<* %":"
		<* whitespace
		<*> json) --> { lines, columns, range, values in
			(values.0.0, Cofree((lines, columns, range), .Fixed([values.0.1, values.1])))
		}

	return sepBy(pairs, whitespace <* %"," <* whitespace)
}

public let json: JSONParser = fix { json in
	let string: JSONParser = quoted --> {
		Cofree(($0, $1, $2), .Leaf(.String($3)))
	} <?> "string"

	let array: JSONParser =  %"["
		<* whitespace
		*> sepBy(whitespace *> json, whitespace <* %"," <* whitespace)
		<* whitespace
		<* %"]"
		--> {
			Cofree(($0, $1, $2), .Indexed($3))
		} <?> "array"

	let object: JSONParser = %"{"
		*> whitespace
		*> members(json)
		<* whitespace
		<* %"}"
		--> { (lines, columns, range, values: [(String, CofreeJSON)]) in
			Cofree((lines, columns, range), .Keyed(Dictionary(elements: values)))
		} <?> "object"

	let numberParser: JSONParser = (number --> { lines, columns, range, value in
		Cofree((lines, columns, range), .Leaf(JSONLeaf.Number(value)))
	}) <?> "number"
	
	let null: JSONParser = %"null" --> { lines, columns, range, value in
		return Cofree((lines, columns, range), .Leaf(.Null))
	} <?> "null"
	
	let boolean: JSONParser = %"false" <|> %"true" --> { lines, columns, range, value in
		let boolean = value == "true"
		return Cofree((lines, columns, range), .Leaf(.Boolean(boolean)))
	} <?> "boolean"

	// TODO: This should be JSON = dict <|> array and
	// Value = dict | array | string | number | null | bool
	return (object <|> array <|> string <|> numberParser <|> boolean <|> null) <* whitespace
}