// ==UserScript==
// @name			Bonk.io Auto Colour Switching
// @author			Figurative Lag
// @description		Automatically rotate through bonk icon colours/teams
// @match			https://bonk.io/*
// @version			1.8
// @namespace			https://github.com/michaelskyba
// @run-at			document-idle
// @grant			none
// @license			Apache 2.0
// @downloadURL none
// ==/UserScript==

// Index 0 is actually supposed to be Spectate, which we're not going to use
// So, we have to add 1 to our index when submitting data to the web socket
const coloursList = [
	"FFA",
	"Red",
	"Blue",
	"Green",
	"Yellow"
]
const enabled = [
	true,
	true,
	true,
	true,
	true
]
let currentColour = 0

const RNG = (min, max) => {
	return Math.round(Math.random() * (max - min)) + min
}

function getFrame() {
	return document.getElementById("maingameframe").contentWindow
}

function getDoc() {
	return getFrame().document
}

function getId(id) {
	let frame = getDoc()
	return frame.getElementById(id)
}

let websocket = null
const originalSend = getFrame().WebSocket.prototype.send

getFrame().WebSocket.prototype.send = function(args) {
	const invalidSocket = websocket == null || websocket.readState != websocket.OPEN
	const validURL = this.url.includes(".bonk.io/socket.io/?EIO=3&transport=websocket&sid=")

	if (validURL && invalidSocket)
		websocket = this

	originalSend.call(this, args)
}

function setColour(colour) {
	currentColour = colour

	// We have to add one because we removed "spectate" from index 0
	if (websocket)
		websocket.send('42[6,{"targetTeam":' + (colour+1) + '}]')
}

let menu = document.getElementById("descriptioninner")
menu.style.cssText = "background-color: black !important;"

// Clear old screen
while (menu.children.length > 0) {
	menu.children[0].remove()
}

function createCheckbox(colour, i) {
	let id = `${colour}Checkbox`
	let label = document.createElement("label")
	label.innerHTML = colour
	label.htmlFor = id
	menu.appendChild(label)

	let checkbox = document.createElement("input")
	checkbox.type = "checkbox"
	checkbox.id = id
	checkbox.checked = true

	checkbox.onchange = function() {
		enabled[i] = this.checked
	}

	menu.appendChild(checkbox)
	menu.appendChild(document.createElement("br"))
}

let h3 = document.createElement("h3")
h3.innerHTML = "Automatic Colour (Team) Switching"
h3.style.margin = 0
menu.appendChild(h3)

let p = document.createElement("p")
p.innerHTML = "Made by Figurative Lag"
p.style.margin = 0
menu.appendChild(p)

menu.appendChild(document.createElement("hr"))

coloursList.forEach((colour, i) => {
	createCheckbox(colour, i)
})

menu.appendChild(document.createElement("hr"))
p = document.createElement("p")
p.innerHTML = "<strong>Switch Interval Modification</strong><br>Warning: bonk.io will quickly rate limit you, making it switch even slower than before you reduced the delay. So, I recommend leaving it at 1000 ms for consistency."
menu.appendChild(p)

let intervalInput = document.createElement("input")
intervalInput.type = "number"
intervalInput.value = 1000
intervalInput.placeholder = "Interval (ms)"
menu.appendChild(intervalInput)

let intervalSubmit = document.createElement("input")
intervalSubmit.type = "button"
intervalSubmit.value = "Submit"
intervalSubmit.onclick = () => {
	clearInterval(interval)
	interval = setInterval(timeout, parseInt(intervalInput.value))
}
menu.appendChild(intervalSubmit)

function timeout() {
	listChecked = []
	for (let i = 0; i < enabled.length; i++) {
		if (enabled[i])
			listChecked.push(i)
	}

	if (listChecked.length == 0)
		return
	if (listChecked.length == 1 && currentColour == listChecked[0])
		return

	// Don't set the colour that is already set
	let colour = currentColour
	while (colour == currentColour) {
		colour = listChecked[RNG(0, listChecked.length-1)]
	}

	setColour(colour)
}
let interval = setInterval(timeout, 1000)