extends SceneTree

const EntryShell = preload("res://entry/entry_shell.gd")
const SettingsStore = preload("res://entry/settings_store.gd")
const DraftService = preload("res://entry/initial_company_draft_service.gd")
const ContractService = preload("res://entry/guild_contract_service.gd")

const DRAFT_PATH := "user://dev4_capture_draft.json"
const COMPANY_PATH := "user://dev4_capture_company.json"
const SEED := "ffeeddccbbaa00998877665544332211"
const CONTRACT_OP := "ababababcdcdcdcdefefefef01010101"
const HIRE_OP_1 := "11111111333333335555555577777777"
const HIRE_OP_2 := "22222222444444446666666688888888"
const VIEWPORT := Vector2i(844, 390)

var failures: Array[String] = []
var output_root := ""


func _initialize() -> void:
	call_deferred("run_capture")


func run_capture() -> void:
	output_root = OS.get_environment("ABG_EVIDENCE_DIR")
	if output_root.is_empty():
		output_root = "/tmp/abg-dev4-party-capture"
	output_root = output_root.path_join("party_line")
	DirAccess.make_dir_recursive_absolute(output_root)
	_cleanup()
	var company := _create_company()
	if company.is_empty():
		_finish(1)
		return
	var viewport := SubViewport.new()
	viewport.size = VIEWPORT
	viewport.render_target_update_mode = SubViewport.UPDATE_ALWAYS
	root.add_child(viewport)
	var shell := EntryShell.new()
	shell.auto_start = false
	shell.initial_company_service.path = DRAFT_PATH
	shell.initial_company_service.company_path = COMPANY_PATH
	viewport.add_child(shell)
	await _frames(3)
	var states := [
		{"name": "01-party-zh", "locale": "zh_CN", "scale": 1.0, "selected": "", "swap": false},
		{"name": "02-selected-zh", "locale": "zh_CN", "scale": 1.0, "select_position_zero": true, "swap": false},
		{"name": "03-swapped-zh", "locale": "zh_CN", "scale": 1.0, "select_position_zero": true, "swap": true},
		{"name": "04-party-en130", "locale": "en", "scale": 1.3, "selected": "", "swap": false},
	]
	for state in states:
		var settings := SettingsStore.new().defaults()
		settings.setup_completed = true
		settings.locale = String(state.locale)
		settings.text_scale = float(state.scale)
		shell.settings = settings
		shell.setup_draft = settings.duplicate(true)
		shell.party_draft_order.clear()
		shell.party_selected_recruit_id = ""
		shell.show_party_line(company)
		await _frames(5)
		var selected_id := String(shell.party_draft_order[0]) if bool(state.get("select_position_zero", false)) else String(state.get("selected", ""))
		shell.party_selected_recruit_id = selected_id
		if bool(state.swap):
			var swapped: Array = shell.party_draft_order.duplicate()
			var temporary = swapped[0]
			swapped[0] = swapped[3]
			swapped[3] = temporary
			shell.party_draft_order.assign(swapped)
		if not selected_id.is_empty() or bool(state.swap):
			shell.show_party_line(company)
			await _frames(5)
		await RenderingServer.frame_post_draw
		var image := viewport.get_texture().get_image()
		var path := output_root.path_join(String(state.name) + ".png")
		if image.save_png(path) != OK:
			failures.append("capture failed: %s" % state.name)
	shell.queue_free()
	viewport.queue_free()
	await _frames(100)
	_cleanup()
	if failures.is_empty():
		print("ABG_DEV4_PARTY_CAPTURE_OK frames=4 viewport=844x390 locales=zh_CN,en text_130=true root=%s" % output_root)
		quit(0)
	else:
		_finish(1)


func _create_company() -> Dictionary:
	var draft: Dictionary = _draft_service().begin_or_load(SEED)
	if not draft.ok:
		return {}
	_draft_service().set_selection([String(draft.draft.offers[0].offer_id), String(draft.draft.offers[2].offer_id)])
	var committed: Dictionary = _draft_service().commit_company()
	if not committed.ok:
		return {}
	var created: Dictionary = _contract_service().create_contract(CONTRACT_OP, "CTR-R1-01", "standard")
	if not created.ok:
		return {}
	var offers: Array = created.company.active_contract.market_board.recruit_offers
	_contract_service().hire_recruit(HIRE_OP_1, String(offers[0].source_offer_id), 14)
	var second: Dictionary = _contract_service().hire_recruit(HIRE_OP_2, String(offers[1].source_offer_id), 14)
	return second.company if second.ok else {}


func _draft_service() -> RefCounted:
	var service := DraftService.new()
	service.path = DRAFT_PATH
	service.company_path = COMPANY_PATH
	return service


func _contract_service() -> RefCounted:
	var service := ContractService.new()
	service.company_path = COMPANY_PATH
	return service


func _frames(count: int) -> void:
	for _index in range(count):
		await process_frame


func _cleanup() -> void:
	for path in [DRAFT_PATH, "%s.bak" % DRAFT_PATH, "%s.candidate" % DRAFT_PATH, "%s.bak.candidate" % DRAFT_PATH, COMPANY_PATH, "%s.bak" % COMPANY_PATH, "%s.candidate" % COMPANY_PATH, "%s.bak.candidate" % COMPANY_PATH, "%s.corrupt" % COMPANY_PATH]:
		if FileAccess.file_exists(path):
			DirAccess.remove_absolute(path)


func _finish(code: int) -> void:
	for failure in failures:
		push_error(failure)
	quit(code)