const WC_VARIABLES = {
	mainWidth : 100,
	selectWidth : 300,
	textColor: 'var(--fg-color)',
	bgMain : 'var(--bg-color)',
	itemColor: 'var(--item-color)',
	borderColor : 'var(--fg-color)',
	selectedColor : 'var(--sel-color)',
}

const OPE_VARIABLES = {
	mainWidth : 100,
	mainHeight: 100,
	topIdent : 20,
	borderColor : 'var(--fg-color)',
	bgMain : 'var(--bg-color)',
	editorURL : 'https://zhuyu1997.github.io/open-pose-editor/'
}

log = console.log;

class WC{
	constructor(TARGET){
		this.mainFrame = $Add('div', TARGET, {className : 'topMenu'});
		
		this.comfyMenuHidden = false;
		
		this.loader = $Add("input", $d.body, {
			type: "file",
			accept: ".json",
			style: { display: "none" },
			onchange: () => {
				this.restore(this.loader.files[0]);
			},
		});
		
		//this.slideFrame = $Add('div', this.mainFrame, {innerHTML : '↔', style : {background : WC_VARIABLES.bgMain, position: 'absolute', left: `${WC_VARIABLES.mainWidth}px`, border : `solid 1px ${WC_VARIABLES.borderColor}`, borderLeft : `solid 0 white`, cursor: 'pointer'}, onclick : ()=>{this.slide()} } );
		this.Title = $Add('span', this.mainFrame, {innerHTML: 'ComfyUI', className : 'title'});
		
		this.queuePanel = $Add('span', this.mainFrame, {style : {position: 'absolute', right: 0} } );
		this.queueInfoButton = $Add('span', this.queuePanel, {innerHTML : '≡', className : 'button', onclick : ()=>{this.showHideComfyMenu()} } );
		this.generateButton = $Add('span', this.queuePanel, {innerHTML : 'GENERATE', className : 'button' , onclick : function(){app.queuePrompt(0, this.batchCount)} } );

		this.controlPanel = $Add('span', this.mainFrame, {className : 'workflowControlPanel'});
		
		$Add('span', this.controlPanel, {innerHTML : '	|	'});
		
		this.itemSelect = $Add('select', this.controlPanel, {className: 'comboBox', onchange : () => {this.select() } } );
		
		this.addWorkflow = $Add('span', this.controlPanel, {type : 'button',  innerHTML : 'Add', className : 'button', onclick : () => {this.add(prompt(), jsonEncode(app.graph.serialize() ) ) } } );
		this.removeWorkflow = $Add('span', this.controlPanel, {type : 'button', innerHTML : 'Remove', className : 'button', onclick : () => {this.remove() } } );
		this.renameWorkflow = $Add('span', this.controlPanel, {type : 'button', innerHTML : 'Rename', className : 'button', onclick : () => {this.rename() } } );
		this.backupWorkflow = $Add('span', this.controlPanel, {type : 'button', innerHTML : 'Backup', className : 'button', onclick : () => {this.backup() } } );
		this.restoreWorkflow = $Add('span', this.controlPanel, {type : 'button', innerHTML : 'Restore', className : 'button', onclick : () => {this.loader.click() } } );
		
		$Add('span', this.controlPanel, {innerHTML : '	|	'});
		
		this.unselectWorkflow = $Add('option', this.itemSelect, {value : -1, style : {}, innerHTML : 'Not selected' } );
		
		this.selected = undefined;
		this.items = [];
		
		this.load();
		
		this.saveInterval = setInterval(() => {
			if (this.selected != undefined){
				this.items[this.selected].save(jsonEncode(app.graph.serialize()));
			}
			this.save();
		}, 1000);
		
		this.opened = true;
		//this.slide();
	}
	showHideComfyMenu(){
		if (this.comfyMenuHidden){
			qs('.comfy-menu').style.display = '';
		}else{
			qs('.comfy-menu').style.display = 'none';
		}
		this.comfyMenuHidden = !this.comfyMenuHidden;
	}
	slide(){
		if (this.opened){
			this.opened = !this.opened;
			this.mainFrame.style.left = `${0 - WC_VARIABLES.mainWidth}px`;
		}else{
			this.opened = !this.opened;
			this.mainFrame.style.left = 0;
		}
	}
	add(NAME, GRAPH){
		if (NAME != null && NAME != '')this.items.push(new WCItem(this, NAME, GRAPH));
	}
	remove(){
		if (this.selected == undefined) return 0;
		this.items[this.selected].remove();
		this.items.splice(this.selected, 1);
		this.unselect();
		for (let i = 0; i < this.items.length; i++){
			this.items[i].id = i;
		}
	}
	rename(){
		if (this.selected == undefined) return 0;
		this.items[this.selected].rename(prompt());
	}
	save(){
		let data = [];
		for(let i = 0; i < this.items.length; i++){
			data.push({name : this.items[i].name, graph : this.items[i].graph});
		}
		lsSet('WorkflowCollection', jsonEncode(data));
	}
	backup(){
		const json = lsGet('WorkflowCollection');
			const blob = new Blob([json], { type: "application/json" });
			const url = URL.createObjectURL(blob);
			const a = $Add("a", $d.body, {
				href: url,
				download: "WorkflowCollection.json",
				style: { display: "none" }
			});
			a.click();
			setTimeout(function () {
				a.remove();
				window.URL.revokeObjectURL(url);
			}, 0);
	}
	restore(FILE){
		if (FILE.type === "application/json" || FILE.name.endsWith(".json")) {
			const reader = new FileReader();
			reader.onload = () => {
				this.load(jsonDecode(reader.result));
			};
			reader.readAsText(FILE);
		}
	}
	load(DATA){
		if (DATA != undefined){
			for (let i = 0; i < this.items.length; i++){
				this.items[i].remove();
			}
			this.items = [];
			lsSet('WorkflowCollection', jsonEncode(DATA));
		}
		if (lsGet('WorkflowCollection') == null) {
			//lsSet('WorkflowCollection', jsonEncode(DEFAULT_COLLECTIONS));
			this.loadDefault();
			return
		}
		let data = jsonDecode(lsGet('WorkflowCollection'));
		if (data == null) {data = []};
		for (let i = 0; i < data.length; i++){
			this.add(data[i].name, data[i].graph);
		}
		this.unselect();
	}
	async loadDefault(){
		let listResp = await fetch('DefaultWorkflows/index.json');
		log(listResp);
		let list = await listResp.json();
		log(list);
		let data = [];
		log(data);
		for (let i = 0; i < list.length; i++){
			log(i);
			let temp = await fetch(`DefaultWorkflows/${list[i]}`);
			log(temp);
			data.push( {name : list[i].replace('.json', ''), graph : jsonEncode(await temp.json()) } );
			log(data);
		}
		this.load(data);
	}
	select(){
		this.selected = (this.itemSelect.value == -1) ? undefined : this.itemSelect.value;
		for (let i = 0; i < this.items.length; i++){
			if (i == this.selected){
				this.items[i].frame.style.background = WC_VARIABLES.selectedColor;
			}else{
				this.items[i].frame.style.background = WC_VARIABLES.itemColor;
			}
		}
		if (this.selected == undefined){
			this.unselectWorkflow.style.background = WC_VARIABLES.selectedColor;
		}else{
			this.unselectWorkflow.style.background = WC_VARIABLES.itemColor;
		}
		if (this.selected != undefined) this.items[Number(this.selected)].select()
	}
	unselect(){
		this.selected = undefined;
		this.select();
	}
}

class WCItem{
	constructor(PARENT, NAME, GRAPH){
		this.parent = PARENT;
		this.name = NAME;
		this.frame = $Add('option', this.parent.itemSelect, {innerHTML: this.name, value: this.parent.items.length, style:{} });
		this.id = this.parent.items.length;
		this.graph = GRAPH;
	}
	remove(){
		this.frame.remove();
		delete this;
	}
	select(){
		//this.parent.selected = this.id;
		//this.parent.select();
		app.graph.clear();
		this.load();
	}
	rename(NAME){
		this.name = NAME;
		this.frame.innerHTML = this.name;
	}
	save(GRAPH){
		this.graph = GRAPH;
	}
	load(){
		app.loadGraphData(jsonDecode(this.graph));
	}
}

class OPE{
	constructor(TARGET){
		this.mainFrame = $Add('div', TARGET, {className : 'FullWindow'});
		this.slideFrame = $Add('span', wc.mainFrame, {innerHTML : 'OPE', className : 'button', onclick : ()=>{this.slide()} } );
		
		this.iFrame = $Add('iframe', this.mainFrame, {width: '100%', height: '100%', src : OPE_VARIABLES.editorURL, style : {border : 'none'} } );
		
		this.opened = false;
	}
	slide(){
		if (this.opened){
			this.opened = !this.opened;
			this.mainFrame.style.left = `-${OPE_VARIABLES.mainWidth}%`;
			this.slideFrame.style.background = OPE_VARIABLES.bgMain;
		}else{
			this.opened = !this.opened;
			this.mainFrame.style.left = 0;
			this.slideFrame.style.background = WC_VARIABLES.selectedColor;
		}
	}
}

class NCP{
	constructor(TARGET){
		this.mainFrame = $Add('div', TARGET, {className : 'FullWindow'});
		this.slideFrame = $Add('span', wc.mainFrame, {innerHTML : 'NCP', className : 'button', onclick : ()=>{this.slide()} } );
		
		this.opened = false;
		this.editMode = false;
		
		$Add('div', this.mainFrame, {style : {height : '25px'} } );
		this.panel = $Add('div', this.mainFrame, {});
		
		this.editModeButton = $Add('div', this.mainFrame, {innerHTML : 'Edit', className : 'button', style : {position : 'absolute', left : 0}, onclick : ()=>{this.editSwitch();} } );
	}
	slide(){
		if (this.opened){
			this.opened = !this.opened;
			this.mainFrame.style.left = `-${OPE_VARIABLES.mainWidth}%`;
			this.slideFrame.style.background = OPE_VARIABLES.bgMain;
		}else{
			this.opened = !this.opened;
			this.mainFrame.style.left = 0;
			this.slideFrame.style.background = WC_VARIABLES.selectedColor;
		}
	}
	editSwitch(){
		if (this.editMode){
			this.editMode = !this.editMode;
			this.editModeButton.style.background = OPE_VARIABLES.bgMain;
		}else{
			this.editMode = !this.editMode;
			this.editModeButton.style.background = WC_VARIABLES.selectedColor;
		}
	}
}

wc = new WC(qs('body'));
ope = new OPE(qs('body'));
ncp = new NCP(qs('body'));