// ==UserScript==
// @name         Show only pretest result when participating in virtual contest in Codeforces
// @version      0.0
// @description  Show only pretest result when participating in virtual contest in Codeforces
// @match        *://codeforces.com/*
// @grant        none
// @run-at       document-start
// ==/UserScript==


(function(){
	let searchParams=new URL(location).searchParams
	// always_show, reset_button

	let cache={} // problemId -> result
	let cacheSubmissions={} // submissionId -> item
	let participantId // assume participantId is fixed
	let csrf_token

	unsafeWindow.cache=cache
	unsafeWindow.cacheSubmissions=cacheSubmissions

	function getData(problemId,callback){
		if(cache[problemId]!==undefined)
			callback(cache[problemId])
		else{
			if(csrf_token===undefined){
				let elements=document.getElementsByName('X-Csrf-Token')
				if(elements.length===0){
					console.log('Something unexpected happened. The displayed result might be wrong.')
					return
				}
				if(elements.length>1)
					console.log('Something unexpected happened, but it probably does not affect the result.')
				csrf_token=elements[0].content
			}

			$.post('/data/standings',{
					participantId:participantId,
					problemId:problemId,
					csrf_token:csrf_token
				},function(result){
					cache[problemId]=result
					result.forEach(function(x){
						cacheSubmissions[x.submissionId]=x
					})
					callback(result)
				}
			)
		}
	}


	if(location.href.match('://codeforces.com/contest/\\d*/my')){
		let restoreFunctions=[]

		function restoreAll(){
			observer.disconnect()
			for(let t of restoreFunctions){
				t()
			}
			restoreFunctions=[]
		}

		let button
		function clickResetButton(){
			restoreAll()
			if(searchParams.has('reset_button')){
				document.body.removeChild(button)
				button=undefined
			}
		}
		function createResetButton(){
			if(button===undefined){
				if(searchParams.has('reset_button')){
					button=document.createElement('button')
					button.innerHTML='Reset'
					button.onclick=clickResetButton
					document.body.appendChild(button)
				}
			}
		}


		let observer=new MutationObserver(function(mutations, observer){
			for (let r of mutations){
				let t=r.target // must be in local scope

				if(t.tagName==='SPAN'){
					if(t.classList.contains('contest-state-phase')){
						if(t.textContent==='Finished'&&!searchParams.has('always_show')){
							if(restoreFunctions.length!==0){ // unlikely
								clickResetButton()
							}

							observer.disconnect()
							return
						}
					}else if(t.classList.contains('verdict-accepted')){
						if(t.textContent==='Accepted'){
							restoreFunctions.push(function(){
								t.textContent='Accepted'
							})
							createResetButton()
							t.textContent='Pretest passed'
						}
					}else if(t.classList.contains('verdict-rejected')&&t.textContent.includes('on test')){
						// for some reason it's not possible to use unsafeWindow.getProblemId now
						let verdictWrapper=t.parentNode
						let tableRow=verdictWrapper.parentNode.parentNode
						let problemId=tableRow.children[3].getAttribute('data-problemId') // int-parseable string
						let submissionId=parseInt(tableRow.children[0].textContent)

						if(participantId===undefined)
							participantId=parseInt(tableRow.children[2].getAttribute('data-participantId'))

						// temporarily set accepted, otherwise for slow network the user may briefly see WA for main test failed
						{
							let oldTextContent=t.textContent // must be in local scope
							t.classList.replace('verdict-rejected','verdict-accepted')
							t.textContent='Pretest passed'
							restoreFunctions.push(function(){
								t.classList.replace('verdict-accepted','verdict-rejected')
								t.textContent=oldTextContent
							})
							createResetButton()

							getData(problemId,function(result /*unused */){
								if(!cacheSubmissions[submissionId].verdict.includes('main tests')){
									t.classList.replace('verdict-accepted','verdict-rejected')
									t.textContent=t.oldTextContent.replace('on test','on pretest')
								}
							})
						}
					}
				}
			}
		})

		observer.observe(document,{
			childList:true,
			subtree:true,
			attributes:false
		});

		window.addEventListener('load',function(){
		})

	}
})()


// TODO incomplete (rewrite standings table)


//if(location.href.match('://codeforces.com/contest/\\d*/standings')){
//	let observer=new MutationObserver(function(mutations, observer){
//		/*
//		let logout_link=document.querySelector('[href$="logout"]')
//		if(logout_link!==null)
//		let handle=logout_link.previousElementSibling.textContent)
//		*/
//
//		GM_addStyle(String.raw`
//			.highlighted-row td[problemid] {
//				visibility: hidden;
//			}
//		`)
//		observer.disconnect()
//	})
//	observer.observe(document,{
//		childList:true,
//		subtree:true,
//		attributes:true
//	});
//}


// vim: set ts=4 sw=4 fdm=indent: