path: root/src/CoreBindings.kt

package moe.nea.lisp

object CoreBindings {
	val def = LispData.externalRawCall("def") { context, callsite, stackFrame, args ->
		if (args.size != 2) {
			return@externalRawCall stackFrame.reportError("Function define expects exactly two arguments", callsite)
		}
		val (name, value) = args
		if (name !is LispAst.Reference) {
			return@externalRawCall stackFrame.reportError("Define expects a name as first argument", name)
		}
		if (name.label in stackFrame.variables) {
			return@externalRawCall stackFrame.reportError("Cannot redefine value in local context", name)
		}
		return@externalRawCall stackFrame.setValueLocal(name.label, context.resolveValue(stackFrame, value))
	}

	fun isTruthy(data: LispData): Boolean? {
		if (data == trueValue) return true
		if (data == falseValue) return false
		return null
	}

	val trueValue = LispData.Atom("true")
	val falseValue = LispData.Atom("false")

	val ifFun = LispData.externalRawCall("if") { context, callsite, stackFrame, args ->
		if (args.size != 3) {
			return@externalRawCall stackFrame.reportError("if requires 3 arguments", callsite)
		}
		val (cond, ifTrue, ifFalse) = args

		val c = isTruthy(context.resolveValue(stackFrame, cond))
		if (c == null) {
			return@externalRawCall stackFrame.reportError("Non boolean value $c used as condition for if", cond)
		}
		if (c) {
			return@externalRawCall context.resolveValue(stackFrame, ifTrue)
		} else {
			return@externalRawCall context.resolveValue(stackFrame, ifFalse)
		}
	}

	val pure = LispData.externalCall("pure") { args, reportError ->
		return@externalCall args.singleOrNull()?.let { value ->
			LispData.externalCall("pure.r") { args, reportError ->
				if (args.isNotEmpty())
					reportError("Pure function does not expect arguments")
				else
					value
			}
		} ?: reportError("Function pure expects exactly one argument")
	}

	val lambda = LispData.externalRawCall("lambda") { context, callsite, stackFrame, args ->
		if (args.size != 2) {
			return@externalRawCall stackFrame.reportError("Lambda needs exactly 2 arguments", callsite)
		}
		val (argumentNames, body) = args
		if (argumentNames !is LispAst.Parenthesis) {
			return@externalRawCall stackFrame.reportError("Lambda has invalid argument declaration", argumentNames)
		}
		val argumentNamesString = argumentNames.items.map {
			val ref = it as? LispAst.Reference
			if (ref == null) {
				return@externalRawCall stackFrame.reportError("Lambda has invalid argument declaration", it)
			}
			ref.label
		}
		if (body !is LispAst.Parenthesis) {
			return@externalRawCall stackFrame.reportError("Lambda has invalid body declaration", body)
		}
		LispData.createLambda(stackFrame, argumentNamesString, body)
	}

	val defun = LispData.externalRawCall("defun") { context, callSite, stackFrame, lispAsts ->
		if (lispAsts.size != 3) {
			return@externalRawCall stackFrame.reportError("Invalid function definition", callSite)
		}
		val (name, args, body) = lispAsts
		if (name !is LispAst.Reference) {
			return@externalRawCall stackFrame.reportError("Invalid function definition name", name)
		}
		if (name.label in stackFrame.variables) {
			return@externalRawCall stackFrame.reportError("Cannot redefine function in local context", name)
		}
		if (args !is LispAst.Parenthesis) {
			return@externalRawCall stackFrame.reportError("Invalid function definition arguments", args)
		}
		val argumentNames = args.items.map {
			val ref = it as? LispAst.Reference
				?: return@externalRawCall stackFrame.reportError("Invalid function definition argument name", it)
			ref.label
		}
		if (body !is LispAst.Parenthesis) {
			return@externalRawCall stackFrame.reportError("Invalid function definition body", body)
		}
		return@externalRawCall stackFrame.setValueLocal(
			name.label,
			LispData.createLambda(stackFrame, argumentNames, body, name.label)
		)
	}
	val seq = LispData.externalRawCall("seq") { context, callsite, stackFrame, args ->
		var lastResult: LispData? = null
		for (arg in args) {
			lastResult = context.executeLisp(stackFrame, arg)
		}
		lastResult ?: stackFrame.reportError("Seq cannot be invoked with 0 argumens", callsite)
	}

	internal fun stringify(thing: LispData): String {
		return when (thing) {
			is LispData.Atom -> ":${thing.label}"
			is LispData.JavaExecutable -> "<native function ${thing.name}>"
			LispData.LispNil -> "nil"
			is LispData.LispNode -> thing.node.toSource()
			is LispData.LispList -> thing.elements.joinToString(", ", "[", "]") { stringify(it) }
			is LispData.LispString -> thing.string
			is LispData.LispHash -> thing.map.asIterable().joinToString(", ", "{", "}") { it.key + ": " + it.value }
			is LispData.LispNumber -> thing.value.toString()
			is LispData.ForeignObject<*> -> "<foreign ${thing.obj}>"
			is LispData.LispInterpretedCallable -> "<function ${thing.name ?: "<anonymous>"} ${thing.argNames} ${thing.body.toSource()}>"
		}
	}

	val tostring = LispData.externalCall("tostring") { args, reportError ->
		LispData.LispString(args.joinToString(" ") { stringify(it) })
	}

	val debuglog = LispData.externalRawCall("debuglog") { context, callsite, stackFrame, args ->
		OutputCapture.print(
			stackFrame,
			args.joinToString(" ", postfix = "\n") { stringify(context.resolveValue(stackFrame, it)) })
		LispData.LispNil
	}
	val add = LispData.externalCall("add") { args, reportError ->
		if (args.size == 0) {
			return@externalCall reportError("Cannot call add without at least 1 argument")
		}
		LispData.LispNumber(args.fold(0.0) { a, b ->
			a + (b as? LispData.LispNumber
				?: return@externalCall reportError("Unexpected argument $b, expected number")).value
		})
	}
	val sub = LispData.externalCall("sub") { args, reportError ->
		if (args.size == 0) {
			return@externalCall reportError("Cannot call sub without at least 1 argument")
		}
		val c = args.map {
			(it as? LispData.LispNumber
				?: return@externalCall reportError("Unexpected argument $it, expected number")
					).value
		}
		LispData.LispNumber(c.drop(1).fold(c.first()) { a, b ->
			a - b
		})
	}
	val mul = LispData.externalCall("mul") { args, reportError ->
		if (args.size == 0) {
			return@externalCall reportError("Cannot call mul without at least 1 argument")
		}
		LispData.LispNumber(args.fold(1.0) { a, b ->
			a * (b as? LispData.LispNumber
				?: return@externalCall reportError("Unexpected argument $b, expected number")).value
		})
	}
	val eq = LispData.externalCall("eq") { args, reportError ->
		if (args.size == 0) {
			return@externalCall reportError("Cannot call eq without at least 1 argument")
		}
		LispData.boolean(!args.zipWithNext().any { it.first != it.second })
	}
	val div = LispData.externalCall("div") { args, reportError ->
		if (args.size == 0) {
			return@externalCall reportError("Cannot call div without at least 1 argument")
		}
		val c = args.map {
			(it as? LispData.LispNumber
				?: return@externalCall reportError("Unexpected argument $it, expected number")
					).value
		}
		LispData.LispNumber(c.drop(1).fold(c.first()) { a, b ->
			a / b
		})
	}
	val less = LispData.externalCall("less") { args, reportError ->
		if (args.size != 2) {
			return@externalCall reportError("Cannot call less without exactly 2 arguments")
		}
		val (left, right) = args.map {
			(it as? LispData.LispNumber
				?: return@externalCall reportError("Unexpected argument $it, expected number")
					).value
		}
		LispData.boolean(left < right)
	}

	private fun atomOrStringToString(thing: LispData): String? {
		return when (thing) {
			is LispData.Atom -> thing.label
			is LispData.LispString -> thing.string
			else -> null
		}
	}

	val import = LispData.externalRawCall("import") { context, callsite, stackFrame, args ->
		if (args.size != 1) {
			return@externalRawCall stackFrame.reportError("import needs at least one argument", callsite)
		}
		// TODO: aliased / namespaced imports
		val moduleName = atomOrStringToString(context.resolveValue(stackFrame, args[0]))
			?: return@externalRawCall stackFrame.reportError("import needs a string or atom as argument", callsite)
		context.importModule(moduleName, stackFrame, callsite)
		return@externalRawCall LispData.LispNil
	}

	val reflect = LispData.externalCall("reflect.type") { args, reportError ->
		if (args.size != 1) {
			return@externalCall reportError("reflect.type can only return the type for one argument")
		}

		return@externalCall when (args[0]) {
			is LispData.Atom -> LispData.Atom("atom")
			is LispData.LispExecutable -> LispData.Atom("callable")
			is LispData.LispList -> LispData.Atom("list")
			LispData.LispNil -> LispData.Atom("nil")
			is LispData.LispHash -> LispData.Atom("hash")
			is LispData.LispNode -> LispData.Atom("ast")
			is LispData.ForeignObject<*> -> LispData.Atom("foreign")
			is LispData.LispNumber -> LispData.Atom("number")
			is LispData.LispString -> LispData.Atom("string")
		}
	}

	val makeList = LispData.externalCall("list.make") { args, reportError ->
		return@externalCall LispData.LispList(args)
	}

	val indexList = LispData.externalCall("list.at") { args, reportError ->
		if (args.size != 2) {
			return@externalCall reportError("list.at takes exactly two arguments")
		}
		val list =
			args[0] as? LispData.LispList ?: return@externalCall reportError("the first argument to list.at is a list")
		val index = (args[1] as? LispData.LispNumber)?.value?.toInt()
			?: return@externalCall reportError("the second argument to list.at is a number")
		if (index !in list.elements.indices)
			return@externalCall reportError("index out of bounds: $list[$index]")
		return@externalCall list.elements[index]
	}

	val joinList = LispData.externalCall("list.join") { args, reportError ->
		if (args.size != 2) {
			return@externalCall reportError("list.join takes exactly two arguments")
		}
		val listA =
			args[0] as? LispData.LispList
				?: return@externalCall reportError("the first argument to list.join is a list")
		val listB =
			args[1] as? LispData.LispList
				?: return@externalCall reportError("the second argument to list.join is a list")
		return@externalCall LispData.LispList(listA.elements + listB.elements)
	}

	val sliceList = LispData.externalCall("list.slice") { args, reportError ->
		if (args.size != 3) {
			return@externalCall reportError("list.slice takes exactly three arguments")
		}
		val list =
			args[0] as? LispData.LispList
				?: return@externalCall reportError("the first argument to list.slice is a list")
		val start =
			(args[1] as? LispData.LispNumber)?.value?.toInt()
				?: return@externalCall reportError("the second argument to list.slice is the start index")
		val end =
			(args[2] as? LispData.LispNumber)?.value?.toInt()
				?: return@externalCall reportError("the third argument to list.slice is the end index")
		return@externalCall LispData.LispList(list.elements.slice(start until end))
	}

	val lengthList = LispData.externalCall("list.length") { args, reportError ->
		if (args.size != 1) {
			return@externalCall reportError("list.length takes exactly three arguments")
		}
		val list =
			args[0] as? LispData.LispList
				?: return@externalCall reportError("the first argument to list.length is a list")
		return@externalCall LispData.LispNumber(list.elements.size.toDouble())
	}

	fun offerListTo(bindings: StackFrame) {
		bindings.setValueLocal("core.list.length", lengthList)
		bindings.setValueLocal("core.list.slice", sliceList)
		bindings.setValueLocal("core.list.join", joinList)
		bindings.setValueLocal("core.list.at", indexList)
		bindings.setValueLocal("core.list.new", makeList)
	}


	fun offerArithmeticTo(bindings: StackFrame) {
		bindings.setValueLocal("core.arith.add", add)
		bindings.setValueLocal("core.arith.div", div)
		bindings.setValueLocal("core.arith.mul", mul)
		bindings.setValueLocal("core.arith.sub", sub)
		bindings.setValueLocal("core.arith.less", less)
		bindings.setValueLocal("core.arith.eq", eq)
	}

	fun offerHashesTo(bindings: StackFrame) {
		bindings.setValueLocal("core.newhash", LispData.externalCall("newhash") { args, reportError ->
			if (args.size % 2 != 0) {
				return@externalCall reportError("Hash creation needs to have an even number of arguments")
			}

			LispData.LispHash(
				args.chunked(2).associate { (a, b) ->
					(atomOrStringToString(a)
						?: return@externalCall reportError("Hash key needs to be string or atom")) to b
				})
		})
		bindings.setValueLocal("core.gethash", LispData.externalCall("gethash") { args, reportError ->
			if (args.size != 2) {
				return@externalCall reportError("Hash access needs 2 arguments")
			}
			val (hash, name) = args
			if (hash !is LispData.LispHash) {
				return@externalCall reportError("$hash is not a hash")
			}
			val nameS =
				atomOrStringToString(name) ?: return@externalCall reportError("$name is not an atom or a string")
			hash.map[nameS] ?: LispData.LispNil
		})
		bindings.setValueLocal("core.mergehash", LispData.externalCall("mergehash") { args, reportError ->
			val m = mutableMapOf<String, LispData>()
			for (arg in args) {
				if (arg !is LispData.LispHash) {
					return@externalCall reportError("$arg is not a hash")
				}
				m.putAll(arg.map)
			}
			LispData.LispHash(m)
		})
	}

	fun offerAllTo(bindings: StackFrame) {
		bindings.setValueLocal("core.if", ifFun)
		bindings.setValueLocal("core.nil", LispData.LispNil)
		bindings.setValueLocal("core.def", def)
		bindings.setValueLocal("core.tostring", tostring)
		bindings.setValueLocal("core.pure", pure)
		bindings.setValueLocal("core.lambda", lambda)
		bindings.setValueLocal("core.defun", defun)
		bindings.setValueLocal("core.seq", seq)
		bindings.setValueLocal("core.import", import)
		bindings.setValueLocal("core.reflect.type", reflect)
		bindings.setValueLocal("core.debuglog", debuglog)
		offerArithmeticTo(bindings)
		offerListTo(bindings)
		offerHashesTo(bindings)
	}
}