package xyz.pixelatedw.recipe.utils

import android.content.Context
import android.net.Uri
import androidx.core.text.trimmedLength
import androidx.documentfile.provider.DocumentFile
import io.github.wasabithumb.jtoml.JToml
import xyz.pixelatedw.recipe.data.Recipe
import xyz.pixelatedw.recipe.data.RecipesView
import java.io.BufferedReader
import java.io.InputStreamReader

fun getRecipes(ctx: Context, view: RecipesView, uri: Uri?) {
	if (uri == null) {
		return
	}

	val dir = DocumentFile.fromTreeUri(ctx, uri)
	if (dir != null) {
		val fileList: Array<DocumentFile> = dir.listFiles()
		for (file in fileList) {
			if (file.isFile && file.name?.endsWith(".md") == true) {
				val recipe = parseRecipe(ctx, file)
				if (recipe != null) {
					view.addRecipe(recipe)
				}
			}
		}
	}
}

private fun parseRecipe(ctx: Context, file: DocumentFile): Recipe? {
	val lastModified = file.lastModified()

	val inputStream = ctx.contentResolver.openInputStream(file.uri)
	val reader = BufferedReader(InputStreamReader(inputStream))
	val text = reader.readText()
	val lines = text.lines()

	val sb = StringBuilder()
	var hasToml = false
	var frontMatterSize = 0
	// TODO This could use some improvements as it always assumes frontmatter is the very first thing in the file
	for (i in 0..lines.size) {
		val line = lines[i]
		frontMatterSize += line.trimmedLength() + 1

		if (line == "+++") {
			if (hasToml) {
				break
			}
			hasToml = true
			continue
		}

		sb.appendLine(line)
	}

	val toml = JToml.jToml()
	val doc = toml.readFromString(sb.toString())

	val tags = arrayListOf<String>()
	for (tomlElem in doc["tags"]!!.asArray()) {
		tags.add(tomlElem!!.asPrimitive().asString())
	}

	val content = text.substring(frontMatterSize..<text.length)

	val recipe = Recipe(
		title = doc["title"]!!.asPrimitive().asString(),
		tags = tags,
		lastModified = lastModified,
		content = content
	)

	return recipe
}