preferences.js

Zotero_Preferences.ZeNotes = {
	async init(){
		if(!Zotero.getMainWindow())
		{
			window.close();
		}
		this.savevalues = [];
		this.tableutils = Zotero_Preferences.ZNTable;
		this.defaulthiddentags = ["Id", "itemid", "key", "filekey"];
		var includes = document.getElementsByClassName("zn-include");
		
		for(include of includes)
		{
			await Zotero_Preferences.ZeNotes.readxhtml(include, include.src);
		}
		
		document.getElementById("zn-refresh").addEventListener("click", function(){
			Zotero.ZeNotes.Ui.reload();
		});	
	},
	
	loadpreference(prefid, elid, mode="")
	{
		var el = document.getElementById(elid)
		if(el!=null) {
			if(el.type.toUpperCase()=="CHECKBOX")
			{
				el.checked = Zotero.ZeNotes.Prefs.get(prefid)=="true" || Zotero.ZeNotes.Prefs.get(prefid)==true;
			}
			else {
				if(mode=="encrypt")
				{
					var v = Zotero.ZeNotes.Prefs.getb(prefid);
					el.value = v;
				}
				else
				{
					el.value = Zotero.ZeNotes.Prefs.get(prefid);
				}
			}
		}
	},
	
	setpreference(e, id, mode="") {
		var value = "";
		if(e.target.type.toUpperCase()=="CHECKBOX")
		{
			value = e.target.checked;
		}
		else
		{
			value = e.target.value;
		}
		
		if(mode=="encrypt")
		{
			Zotero.ZeNotes.Prefs.setb(id, value);
		}
		else
		{
			Zotero.ZeNotes.Prefs.set(id, value);
		}
		
		if(Zotero.ZeNotes.Prefs.get("load-on-change")=="true" || Zotero.ZeNotes.Prefs.get("load-on-change")==true){
			Zotero.ZeNotes.Ui.reload();
		}
	},
	
	updatecolumnwidth(e)
	{
		var sample = document.getElementById("zn-header-size-val");
		sample.value = e.target.value;
	},
	
	updatedisplay(e, id)
	{
		var sample = document.getElementById(id);
		sample.value = e.target.value;
	},
	
	initopacity()
	{
		var sample = document.getElementById("zn-bg-sample");
		var value = Zotero.ZeNotes.Prefs.get("bg-opacity");
		var el = document.getElementById("zn-bg-slider");
		if(el!=null)
		{
			var color = Zotero.ZeNotes.Utils.addopacity(sample.style.backgroundColor, value);
			el.value = parseInt(value);
			sample.style.backgroundColor = color;
		}
	},
	
	updateopacity(e)
	{
		var sample = document.getElementById("zn-bg-sample");
		var color = Zotero.ZeNotes.Utils.addopacity(sample.style.backgroundColor, e.target.value);
		sample.style.backgroundColor = color;
	},

	importpref(e)
	{
		var lsettings = document.getElementById("zn-load-settings");
		var collectionid = undefined;
		if (Zotero.platformMajorVersion < 102) {			
			collectionid = lsettings.options[lsettings.selectedIndex].value;
		}
		else
		{
			collectionid = lsettings.dataset.collectionid;
		}
		
		if(typeof collectionid=="undefined")
		{
			console.log("Please choose a source collection first!");
			return;
		}
		
		if(!confirm("Do you really want to overwrite current collection settings?\nThis change is irreversible!"))
		{
			return;
		}
		Zotero.ZeNotes.Database.copysettings(collectionid, this.collectionid, this.collection).then(()=>{
			console.log("Collection settings imported!");
		})
	},
	
	updateprefimport(e)
	{
		var lsettings = document.getElementById("zn-load-settings");
		lsettings.value = e.target.label;
		lsettings.dataset.collectionid = e.target.value;
	},
	
	targetlanguages()
	{
		var tl = Zotero.ZeNotes.Prefs.get("target-language");
		var options = Zotero.ZeNotes.Languages.list().map(function(e){return {label: e.name, value:e.code}});
		var combo = this.addcombobox("zn-target-language", options, tl);
		combo.addEventListener("change", function(e){
			Zotero.ZeNotes.Prefs.set("target-language", e.target.value);
		});
		combo.addEventListener("command", function(e){
			Zotero.ZeNotes.Prefs.set("target-language", e.target.value);
		})
	},
	
	addcombobox(itemid, options, value){
		var menulist = document.getElementById(itemid);
		if(Zotero.platformMajorVersion < 102)
		{
			if(menulist.tagName.toUpperCase()!="SELECT")
			{
				var p = menulist.parentNode.parentNode;
				menulist.parentNode.remove();
				var sel = document.createElement("select");
				p.insertBefore(sel, p.firstChild);
				sel.setAttribute("id", itemid);
				menulist = sel;
			}
			else
			{
				menulist.innerHTML="";
			}
		}
		else
		{
			menulist.innerHTML="";
		}
		
		if(menulist==undefined)
		{
			return;
		}
						
		for(s of options)
		{
			if(Zotero.platformMajorVersion < 102){
				var opt = document.createElement("option");
				opt.innerHTML = s.label;
				opt.setAttribute("label", s.label);
				opt.setAttribute("value", s.value);
				menulist.appendChild(opt);
				if(s.value==value)
				{
					opt.selected = "true";
				}
			}
			else
			{
				var opt = document.createXULElement("menuitem");
				opt.setAttribute("label", s.label);
				opt.setAttribute("value", s.value);
				menulist.appendChild(opt);
				if(s.value==value)
				{
					menulist.parentNode.selectedItem = opt;
				}
			}
		}
		return menulist;
	},
	
	loadpreferences()
	{		
		var lsettings = document.getElementById("zn-load-settings");
		if(lsettings==null)
		{
			return;
		}
		if (Zotero.platformMajorVersion < 102)
		{
			if(lsettings.tagName.toUpperCase()!="SELECT")
			{
				var p = lsettings.parentNode.parentNode;
				lsettings.parentNode.remove();
				var sel = document.createElement("select");
				p.insertBefore(sel, p.firstChild);
				sel.setAttribute("id", "zn-load-settings");
				lsettings = sel;
			}
			else
			{
				lsettings.innerHTML="";
			}
		}
		else
		{
			lsettings.innerHTML="";
		}
		
		if(lsettings==undefined)
		{
			return;
		}
								
		Zotero.ZeNotes.Database.getsettings().then(settings=>{
			settings = settings.sort(function(a, b) {
			   return a.label.localeCompare(b.label);
			});
			
			for(s of settings)
			{
				if(this.combovalues(lsettings).includes(s.label))
				{
					continue;
				}
				
				if(s.collectionid==this.collectionid)
				{
					continue;
				}
				
				if (Zotero.platformMajorVersion < 102) {
					var opt = document.createElement("option");
					opt.innerHTML = s.label;
					opt.setAttribute("label", s.label);
					opt.setAttribute("value", s.collectionid);
					lsettings.appendChild(opt);
				}
				else
				{
					var opt = document.createXULElement("menuitem");
					opt.setAttribute("label", s.label);
					opt.setAttribute("value", s.collectionid);
					lsettings.appendChild(opt);
				}
			}
		})
	},
	
	combovalues(combo){
		var values = [];
		if(combo.tagName.toUpperCase()=="SELECT")
		{
			combo.childNodes.forEach(n=>{
				values.push(n.innerHTML)
			})
		}
		else
		{
			combo.childNodes.forEach(n=>{
				values.push(n.label)
			})
		}
		return values;
	},
	
	async readxhtml(include, filename)
	{
		return fetch(Zotero.ZeNotes.rootURI+"content/settings/"+filename)
		.then(response => response.text())
		.then(content => {
			const parser = new DOMParser();
			const doc = parser.parseFromString(content, 'application/xhtml+xml');
			const importedNode = document.importNode(doc.documentElement, true);
			include.appendChild(importedNode);
			Zotero_Preferences.ZeNotes.loadtables();
			Zotero_Preferences.ZeNotes.loadpreferences();
			Zotero_Preferences.ZeNotes.targetlanguages();
			Zotero_Preferences.ZeNotes.initopacity();
			Zotero_Preferences.ZeNotes.loadpreference("html-filter", "zn-html-filter");
			Zotero_Preferences.ZeNotes.loadpreference("html-filter-replacement", "zn-html-filter-replacement");
			
			Zotero_Preferences.ZeNotes.loadpreference("add-to-menu", "zn-add-to-menu");
			Zotero_Preferences.ZeNotes.loadpreference("load-on-change", "zn-reload-on-change");
			Zotero_Preferences.ZeNotes.loadpreference("vertical-table", "zn-vertical-table");
			
			Zotero_Preferences.ZeNotes.loadpreference("header-size", "zn-header-size-val");
			Zotero_Preferences.ZeNotes.loadpreference("header-size", "zn-header-size");
			
			Zotero_Preferences.ZeNotes.loadpreference("column-width", "zn-column-width-val");
			Zotero_Preferences.ZeNotes.loadpreference("column-width", "zn-column-width");
			
			Zotero_Preferences.ZeNotes.loadpreference("bard-api-key", "zn-bard-api-key", "encrypt");
			Zotero_Preferences.ZeNotes.loadpreference("bard-model", "zn-bard-model");
			
			Zotero_Preferences.ZeNotes.loadpreference("openai-api-key", "zn-openai-api-key", "encrypt");
			Zotero_Preferences.ZeNotes.loadpreference("openai-model", "zn-openai-model");
			Zotero_Preferences.ZeNotes.loadpreference("openai-max-token", "zn-openai-max-token");
			
			Zotero_Preferences.ZeNotes.loadpreference("google-translate-key", "zn-google-translate-key", "encrypt");
			Zotero_Preferences.ZeNotes.loadpreference("deepl-api-key", "zn-deepl-api-key", "encrypt");
			
			
			Zotero_Preferences.ZeNotes.loadpreference("paraphrase-custom-prompt", "zn-paraphrase-custom-prompt");
			Zotero_Preferences.ZeNotes.loadpreference("cell-custom-prompt", "zn-cell-custom-prompt");
			Zotero_Preferences.ZeNotes.loadpreference("row-custom-prompt", "zn-row-custom-prompt");
			Zotero_Preferences.ZeNotes.loadpreference("table-custom-prompt", "zn-table-custom-prompt");
		})
		.catch(error => {
			console.log('Error loading content: ' + error);
		});
	},
	
	loadtables(){
		// get collection first
		var c = Zotero.getActiveZoteroPane().getSelectedCollection();
		this.collection = "All documents";
		this.collectionid = "all-documents";
		if(c!=undefined && c.name!=undefined)
		{
			this.collection = c.name;
			this.collectionid = c.id;
		}
		
		var titleel = document.getElementById("zn-settings-main-title");
		if(titleel!=null)
		{
			titleel.innerHTML = "Settings for 「"+this.collection+"」";
		}
		
		var table1 = document.getElementById("table-manage-tags-body");
		var table2 = document.getElementById("table-sort-tags-body");
				
		var buttonlist1 = ["up", "down", "first", "last", "visible"];
		var buttonlist2 = ["sort", "up", "down", "first", "last"];
		if(table1!=null)
		{
			Zotero_Preferences.ZeNotes.loadtagsfromdb(table1, ["value", "type", "status", "width", "actions"], buttonlist1);
		}
		if(table2!=null)
		{
			Zotero_Preferences.ZeNotes.loadtagsfromdb(table2, ["value", "actions"], buttonlist2);
		}
	},

	async saveusersettings()
	{
		var data = {
			"hidden": Zotero_Preferences.ZNTable.getuserhiddentags(),
			"order": Zotero_Preferences.ZNTable.getusertagorder(),
			"sort": Zotero_Preferences.ZNTable.getusersortorder(),
			"reverse": Zotero_Preferences.ZNTable.getuserreverseorder(),
			"width": Zotero_Preferences.ZNTable.getusercolumnwidth()
		}
		return Zotero.ZeNotes.Database.addsetting(this.collectionid, this.collection, JSON.stringify(data));
	},
	
	saveandreload()
	{
		this.saveusersettings().then(()=>{
			if(Zotero.ZeNotes.Prefs.get("load-on-change")=="true" || Zotero.ZeNotes.Prefs.get("load-on-change")==true){
				Zotero.ZeNotes.Ui.reload();
			}
		});
	},
		
	async loadtagsfromdb(table, columns, buttonlist)
	{
		var tags = [];
		var hiddentags = this.defaulthiddentags;
		
		var usersettings = {
			"hidden": [],
			"order": [],
			"sort": [],
			"reverse": [],
			"width": {},
		}
		
		try {
			var dbdata = await Zotero.ZeNotes.Database.getsetting(this.collectionid);
			usersettings = JSON.parse(dbdata);
		}
		catch(e){
			console.log(e);
		}
		
		if(!Object.keys(usersettings).includes("width"))
		{
			usersettings["width"] = {};
		}
		
		if(usersettings["hidden"].length>0)
		{
			hiddentags = usersettings["hidden"];
		}

		Zotero.ZeNotes.Data.get().then(data=> {
			var width = "";
			for(t of data["selected_tags"])
			{
				let status = "visible";
				let direction = "down";
				if(hiddentags.includes(t))
				{
					status = "hidden";
				}
				
				if(usersettings["reverse"].includes(t))
				{
					direction = "up";
				}
				
				if(Object.keys(usersettings["width"]).includes(t))
				{
					width = usersettings["width"][t];
				}
				
				tags.push({
					type: "tag",
					value: t,
					status: status,
					direction: direction,
					width: "<input onchange=\"Zotero_Preferences.ZeNotes.saveandreload();\" value=\""+width+"\" class=\"tag-width\" data-tag=\""+t+"\"/>",
					actions: this.tableutils.actions(t, status, direction, buttonlist),
				});
				width = "";
			}
			
			for(t of data["info_columns"])
			{
				let status = "visible";
				let direction = "down";
				if(hiddentags.includes(t))
				{
					status = "hidden";
				}
				
				if(usersettings["reverse"].includes(t))
				{
					direction = "up";
				}
				
				if(Object.keys(usersettings["width"]).includes(t))
				{
					width = usersettings["width"][t];
				}
				
				tags.push({
					type: "info",
					value: t,
					status: status,
					direction: direction,
					width: "<input onchange=\"Zotero_Preferences.ZeNotes.saveandreload();\" value=\""+width+"\" class=\"tag-width\" data-tag=\""+t+"\"/>",
					actions: this.tableutils.actions(t, status, direction, buttonlist),
				});
				width = "";
			}
			this.tableutils.rendertags(table, tags, columns, usersettings);
		});
	}
}

/**
Refresh after losing focus
*/

this.addEventListener("focus", function() 
{
	Zotero_Preferences.ZeNotes.loadtables();
	Zotero_Preferences.ZeNotes.loadpreferences();
});

this.addEventListener("blur", function() 
{
	Zotero.ZeNotes.Ui.reload();
});