class LeFormatter {

	static manualButton(text, id, { onClick }) {
		const button = gradioApp().getElementById(id).cloneNode()

		button.id = 'manual-format'
		button.classList.remove('gr-button-lg', 'gr-button-primary', 'lg', 'primary')
		button.classList.add('secondary')
		button.textContent = text
		button.addEventListener('click', onClick)

		return button
	}

	static injectButton(id, { onClick }) {
		const button = gradioApp().getElementById(id)
		button.addEventListener('click', onClick)
	}

	static checkbox(text, def, { onChange }) {
		const label = document.createElement('label')
		label.style.display = 'flex'
		label.style.alignItems = 'center'
		label.style.margin = '2px 8px'

		const checkbox = gradioApp().querySelector('input[type=checkbox]').cloneNode()
		checkbox.checked = def
		checkbox.addEventListener('change', (event) => {
			onChange(event.target.checked)
		})

		const span = document.createElement('span')
		span.style.marginLeft = 'var(--size-2, 4px)'
		span.textContent = text

		label.appendChild(checkbox)
		label.appendChild(span)

		return label
	}

	static formatString(input, dedupe, removeUnderscore) {
		// Remove Duplicate
		if (dedupe) {
			const temp = input.split(',')

			const cleanArray = []
			const finalArray = []

			temp.forEach((tag) => {
				const cleanedTag = tag.replace(/\[|\]|\(|\)|\s+/g, '').trim()

				if (!cleanArray.includes(cleanedTag)) {
					cleanArray.push(cleanedTag)
					finalArray.push(tag)
					return
				}

				if (/^(AND|BREAK)$/.test(cleanedTag)) {
					finalArray.push(tag)
					return
				}

				finalArray.push(tag.replace(cleanedTag, ''))
			})

			input = finalArray.join(', ')
		}

		// Fix Bracket & Comma
		input = input.replace(/,\s*\)/g, '),').replace(/,\s*\]/g, '],').replace(/\(\s*,/g, ',(').replace(/\[\s*,/g, ',[')

		// Remove Commas
		let tags = input.split(',').map(word => (removeUnderscore ? word.replace(/_/g, ' ') : word).trim()).filter(word => word !== '')

		// Remove Stray Brackets
		const patterns = [/^\(+$/, /^\)+$/, /^\[+$/, /^\]+$/]
		tags = tags.filter(word => !patterns[0].test(word)).filter(word => !patterns[1].test(word)).filter(word => !patterns[2].test(word)).filter(word => !patterns[3].test(word))

		// Remove Spaces
		input = tags.join(', ').replace(/\s+/g, ' ')

		// Fix Bracket & Space
		input = input.replace(/\s+\)/g, ')').replace(/\s+\]/g, ']').replace(/\(\s+/g, '(').replace(/\[\s+/g, '[')

		// Fix Empty Bracket
		input = input.replace(/\(\s+\)/g, '').replace(/\[\s+\]/g, '')

		while (input.match(/\(\s*\)|\[\s*\]/g))
			input = input.replace(/\(\s*\)|\[\s*\]/g, '')

		return input.split(',').map(word => word.trim()).filter(word => word !== '').join(', ')
	}

	static grabBrackets(str, index) {
		let openBracket = -1
		let closeBracket = -1

		for (let i = index; i >= 0; i--) {
			if (str[i] === '(') {
				openBracket = i
				break;
			}
			if (str[i] === ')' && i !== index) {
				break;
			}
		}

		for (let i = index; i < str.length; i++) {
			if (str[i] === ')') {
				closeBracket = i
				break;
			}
			if (str[i] === '(' && i !== index) {
				break;
			}
		}

		if (openBracket !== -1 && closeBracket !== -1 && openBracket !== closeBracket)
			return [openBracket, closeBracket]
		else
			return null
	}

	static injectTagShift(id) {
		const textarea = gradioApp().getElementById(id).querySelector('textarea')

		textarea.addEventListener('wheel', (event) => {
			if (event.shiftKey) {
				event.preventDefault()

				if (textarea.selectionStart !== textarea.selectionEnd)
					return;

				if (event.deltaY === 0)
					return;

				const shift = event.deltaY < 0 ? 1 : -1
				const tags = textarea.value.split(',').map(t => t.trim())

				var cursor = textarea.selectionStart

				var index = 0

				for (let i = 0; i < textarea.selectionStart; i++) {
					if (textarea.value[i] === ',')
						index++
				}

				if (index === 0 && shift === -1)
					return;
				if (index === tags.length - 1 && shift === 1)
					return;

				const shifted = []

				if (shift < 0) {
					for (let i = 0; i < index - 1; i++)
						shifted.push(tags[i])

					shifted.push(tags[index])
					shifted.push(tags[index - 1])

					cursor -= tags[index - 1].length + 2

					for (let i = index + 1; i < tags.length; i++)
						shifted.push(tags[i])
				} else {
					for (let i = 0; i < index; i++)
						shifted.push(tags[i])

					shifted.push(tags[index + 1])
					shifted.push(tags[index])

					cursor -= tags[index + 1].length * -1 - 2

					for (let i = index + 2; i < tags.length; i++)
						shifted.push(tags[i])
				}

				textarea.value = shifted.join(', ')

				textarea.selectionStart = cursor
				textarea.selectionEnd = cursor

				updateInput(textarea)
			}
		})
	}

	static injectBracketEscape(id) {
		const textarea = gradioApp().getElementById(id).querySelector('textarea')

		textarea.addEventListener('keydown', (event) => {
			if (event.ctrlKey && event.key === '\\') {
				event.preventDefault()

				let cursorPosition = textarea.selectionStart

				if (textarea.selectionStart !== textarea.selectionEnd)
					cursorPosition++

				let result = LeFormatter.grabBrackets(textarea.value, cursorPosition)

				if (result) {
					const original = textarea.value

					if (result[0] !== 0 && textarea.value[result[0] - 1] === '\\' && textarea.value[result[1] - 1] === '\\') {
						textarea.value = original.slice(0, result[0] - 1) + original.slice(result[0] - 1, result[1]).replace(/\\/g, '') + original.slice(result[1])
						textarea.selectionStart = result[0] - 1
						textarea.selectionEnd = result[1] - 1
					}
					else {
						textarea.value = original.slice(0, result[0]) + '\\' + original.slice(result[0], result[1]) + '\\' + original.slice(result[1])
						textarea.selectionStart = result[0]
						textarea.selectionEnd = result[1] + 3
					}

					updateInput(textarea)
				}
			}
		})
	}

}

onUiLoaded(async () => {
	const Modes = ['txt', 'img']

	let autoRun = true
	let dedupe = false
	let removeUnderscore = false
	let refreshInput = null

	try {
		const temp = gradioApp().getElementById('pf-config-true')
		temp.remove()
		refreshInput = false
	} catch {
		refreshInput = true
	}

	const manualBtn = LeFormatter.manualButton('Format', 'txt2img_generate', {
		onClick: () => {
			const ids = ['txt2img_prompt', 'txt2img_neg_prompt', 'img2img_prompt', 'img2img_neg_prompt']

			ids.forEach((id) => {
				const textArea = gradioApp().getElementById(id).querySelector('textarea')

				let lines = textArea.value.split('\n')

				for (let i = 0; i < lines.length; i++)
					lines[i] = LeFormatter.formatString(lines[i], dedupe, removeUnderscore)

				textArea.value = lines.join('\n')
				updateInput(textArea)
			})
		}
	})

	manualBtn.style.display = 'none'

	const autoCB = LeFormatter.checkbox('Auto Format', autoRun, {
		onChange: (checked) => {
			autoRun = checked
			manualBtn.style.display = autoRun ? 'none' : 'block'
		}
	})

	const dedupeCB = LeFormatter.checkbox('Remove Duplicates', dedupe, {
		onChange: (checked) => { dedupe = checked }
	})

	const underlineCB = LeFormatter.checkbox('Remove Underscores', removeUnderscore, {
		onChange: (checked) => { removeUnderscore = checked }
	})

	const formatter = document.createElement('div')
	formatter.id = 'le-formatter'
	formatter.style.display = 'flex'
	formatter.style.flex.direction = 'row'

	formatter.appendChild(autoCB)
	formatter.appendChild(manualBtn)
	formatter.appendChild(dedupeCB)
	formatter.appendChild(underlineCB)

	const tools = document.getElementById('quicksettings')
	tools.after(formatter)

	Modes.forEach((mode) => {

		LeFormatter.injectButton(mode + '2img_generate', {
			onClick: () => {
				if (!autoRun)
					return;

				const ids = [mode + '2img_prompt', mode + '2img_neg_prompt']
				const textAreas = [gradioApp().getElementById(ids[0]).querySelector('textarea'), gradioApp().getElementById(ids[1]).querySelector('textarea')]

				let lines = [textAreas[0].value.split('\n'), textAreas[1].value.split('\n')]

				for (let m = 0; m < 2; m++) {

					for (let i = 0; i < lines[m].length; i++)
						lines[m][i] = LeFormatter.formatString(lines[m][i], dedupe, removeUnderscore)

					textAreas[m].value = lines[m].join('\n')

					if (refreshInput)
						updateInput(textAreas[m])
				}
			}
		})

		LeFormatter.injectBracketEscape(mode + '2img_prompt')
		LeFormatter.injectBracketEscape(mode + '2img_neg_prompt')
		LeFormatter.injectTagShift(mode + '2img_prompt')
		LeFormatter.injectTagShift(mode + '2img_neg_prompt')

	})
})