From ce52582f49e618729720a057ae5029d2a1d15da4 Mon Sep 17 00:00:00 2001
From: liach
Date: Wed, 4 Sep 2019 11:56:09 -0400
Subject: Add tinyv2 save/load

---
 .../cuchaz/enigma/command/MappingCommandsUtil.java    | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

(limited to 'src/main/java/cuchaz/enigma/command/MappingCommandsUtil.java')

diff --git a/src/main/java/cuchaz/enigma/command/MappingCommandsUtil.java b/src/main/java/cuchaz/enigma/command/MappingCommandsUtil.java
index bacb8ff..fc68edf 100644
--- a/src/main/java/cuchaz/enigma/command/MappingCommandsUtil.java
+++ b/src/main/java/cuchaz/enigma/command/MappingCommandsUtil.java
@@ -91,7 +91,11 @@ public final class MappingCommandsUtil {
         MappingFormat format = null;
         try {
             format = MappingFormat.valueOf(type.toUpperCase());
-        } catch (IllegalArgumentException ignored) {}
+        } catch (IllegalArgumentException ignored) {
+            if (type.equals("tinyv2")) {
+                format = MappingFormat.TINY_V2;
+            }
+        }
 
         if (format != null) {
             return format.getReader().read(path, ProgressListener.none(), saveParameters);
@@ -106,7 +110,18 @@ public final class MappingCommandsUtil {
             return;
         }
 
-        if (type.startsWith("tiny")) {
+        if (type.startsWith("tinyv2:") || type.startsWith("tiny_v2:")) {
+            String[] split = type.split(":");
+
+            if (split.length != 3) {
+                throw new IllegalArgumentException("specify column names as 'tinyv2:from_namespace:to_namespace'");
+            }
+
+            new TinyV2Writer(split[1], split[2]).write(mappings, path, ProgressListener.none(), saveParameters);
+            return;
+        }
+
+        if (type.startsWith("tiny:")) {
             String[] split = type.split(":");
 
             if (split.length != 3) {
-- 
cgit v1.2.3