add csv/json import/export for sheets

Author: dazKind

castle.hxml

@@ -5,6 +5,7 @@
 -lib hx3compat:git:https://github.com/HaxeFoundation/hx3compat
 -lib format:git:https://github.com/HaxeFoundation/format
 -lib hxbit:git:https://github.com/HeapsIO/hxbit
+-lib thx.csv
 -D lz4js
 -D analyzer-optimize
 -dce std

src/Main.hx

@@ -918,8 +918,26 @@ class Main extends Model {
 		var ndel = new MenuItem( { label : "Delete" } );
 		var nindex = new MenuItem( { label : "Add Index", type : MenuItemType.checkbox } );
 		var ngroup = new MenuItem( { label : "Add Group", type : MenuItemType.checkbox } );
-		for( m in [nins, nleft, nright, nren, ndel, nindex, ngroup] )
+		
+		var ncsv = new MenuItem( { label : "CSV..." } );
+		var csv = new Menu();
+		var importSheetCSV = new MenuItem( { label : "Import...", type : MenuItemType.checkbox } );
+		var exportSheetCSV = new MenuItem( { label : "Export...", type : MenuItemType.checkbox } );
+		csv.append(importSheetCSV);
+		csv.append(exportSheetCSV);
+		ncsv.submenu = csv;
+
+		var njson = new MenuItem( { label : "JSON..." } );
+		var json = new Menu();
+		var importSheetJSON = new MenuItem( { label : "Import...", type : MenuItemType.checkbox } );
+		var exportSheetJSON = new MenuItem( { label : "Export...", type : MenuItemType.checkbox } );
+		json.append(importSheetJSON);
+		json.append(exportSheetJSON);
+		njson.submenu = json;
+		
+		for( m in [nins, nleft, nright, nren, ndel, nindex, ngroup, njson, ncsv] )
 			n.append(m);
+		
 		nleft.click = function() {
 			var prev = -1;
 			for( i in 0...base.sheets.length ) {
@@ -1005,6 +1023,49 @@ class Main extends Model {
 		nren.click = function() {
 			li.dblclick();
 		};
+
+		importSheetJSON.click = function() {
+			var i = J("<input>").attr("type", "file").css("display","none").change(function(e) {
+				var j = JTHIS;
+				this.importSheetJSON(s, j.val());
+				initContent();
+				j.remove();
+			});
+			i.appendTo(J("body"));
+			i.click();
+		};
+		exportSheetJSON.click = function() {
+			var i = J("<input>").attr("type", "file").attr("nwsaveas",'${s.name}.json').css("display","none").change(function(e) {
+				var j = JTHIS;
+				this.exportSheetJSON(s, j.val());
+				initContent();
+				j.remove();
+			});
+			i.appendTo(J("body"));
+			i.click();
+		};
+		
+		importSheetCSV.click = function() {
+			var i = J("<input>").attr("type", "file").css("display","none").change(function(e) {
+				var j = JTHIS;
+				this.importSheetCSV(s, j.val());
+				initContent();
+				j.remove();
+			});
+			i.appendTo(J("body"));
+			i.click();
+		};
+		exportSheetCSV.click = function() {
+			var i = J("<input>").attr("type", "file").attr("nwsaveas",'${s.name}.csv').css("display","none").change(function(e) {
+				var j = JTHIS;
+				this.exportSheetCSV(s, j.val());
+				initContent();
+				j.remove();
+			});
+			i.appendTo(J("body"));
+			i.click();
+		};
+		
 		if( s.isLevel() || (s.hasColumn("width", [TInt]) && s.hasColumn("height", [TInt]) && s.hasColumn("props",[TDynamic])) ) {
 			var nlevel = new MenuItem( { label : "Level", type : MenuItemType.checkbox } );
 			nlevel.checked = s.isLevel();

src/Model.hx

@@ -15,6 +15,8 @@
  */
 import cdb.Data;
 
+import thx.csv.*;
+
 typedef Prefs = {
 	windowPos : { x : Int, y : Int, w : Int, h : Int, max : Bool },
 	curFile : String,
@@ -189,4 +191,76 @@ class Model {
 		js.Browser.getLocalStorage().setItem("prefs", haxe.Serializer.run(prefs));
 	}
 
+	@:access(cdb.Sheet)
+	function exportSheetJSON(_s:cdb.Sheet, _filePath:String) {
+		sys.io.File.saveContent(_filePath, haxe.Json.stringify(_s.sheet, null, "    "));
+	}
+
+	@:access(cdb.Sheet)
+	function importSheetJSON(_s:cdb.Sheet, _filePath:String) {
+		trace(_s);
+
+		// TODO: do import validation!!!
+		var s = haxe.Json.parse(sys.io.File.getContent(_filePath));
+		for( c in cast(s.columns, Array<Dynamic>) ) {
+			c.type = cdb.Parser.getType(c.typeStr);
+			c.typeStr = null;
+		}
+	
+		_s.sheet = s;
+		_s.sync();
+		_s.base.updateSheets();
+		_s.base.sync();
+	}
+
+	function exportSheetCSV(_s:cdb.Sheet, _filePath:String) {
+		var lines = [[ for (c in _s.columns) c.name]];		
+
+		for (l in _s.getLines()) {
+			var ol = [];
+			for (c in _s.columns) {
+				switch (c.type) {
+					case TId, TString, TFile: 
+						ol.push(Reflect.field(l, c.name));
+					default:
+						ol.push(haxe.Json.stringify(Reflect.field(l, c.name)));
+				}
+			}
+			lines.push(ol);
+		}
+		sys.io.File.saveContent(_filePath, Csv.encode(lines));
+	}
+
+	@:access(cdb.Sheet)
+	function importSheetCSV(_s:cdb.Sheet, _filePath:String) {
+		var d = Csv.decode(sys.io.File.getContent(_filePath));
+		var header = d.shift(); // get rid off first line
+
+		var colMapSheet = new Map<String, Column>();
+		for (c in _s.columns)
+			colMapSheet.set(c.name, c);
+
+		var lines = [];
+		for (l in d) {
+			var line = {};
+			for (i in 0...l.length) {
+				var c = colMapSheet.get(header[i]);
+				if (c == null)
+					continue;
+
+				var val = l[i];
+				switch (c.type) {
+					case TId, TString, TFile: 
+						Reflect.setField(line, c.name, val);
+					default:
+						Reflect.setField(line, c.name, haxe.Json.parse(val));
+				}
+			}
+			lines.push(line);
+		}
+
+		_s.sheet.lines = lines;
+		_s.sync();
+	}
+
 }