feat: Snowflake time travel syntax

Signed-off-by: Andreas Reichel <[email protected]>

Author: manticore-projects

src/main/java/net/sf/jsqlparser/schema/Table.java

@@ -10,6 +10,7 @@
 package net.sf.jsqlparser.schema;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
 
@@ -28,11 +29,9 @@
 /**
  * A table. It can have an alias and the schema name it belongs to.
  */
-public class Table extends ASTNodeAccessImpl implements ErrorDestination, FromItem, MultiPartName {
+public class Table extends ASTNodeAccessImpl
+        implements ErrorDestination, FromItem, MultiPartName, Cloneable {
 
-    // private Database database;
-    // private String schemaName;
-    // private String name;
     private static final int NAME_IDX = 0;
 
     private static final int SCHEMA_IDX = 1;
@@ -45,6 +44,9 @@ public class Table extends ASTNodeAccessImpl implements ErrorDestination, FromIt
 
     private List<String> partDelimiters = new ArrayList<>();
 
+    // holds the various `time travel` syntax for BigQuery, RedShift, Snowflake or RedShift
+    private String timeTravelStr = null;
+
     private Alias alias;
 
     private SampleClause sampleClause;
@@ -74,6 +76,10 @@ public Table(String name) {
         setName(name);
     }
 
+    public Table(String name, boolean splitNamesOnDelimiter) {
+        setName(name, splitNamesOnDelimiter);
+    }
+
     public Table(String schemaName, String name) {
         setSchemaName(schemaName);
         setName(name);
@@ -197,12 +203,20 @@ public void setName(String name) {
                         .of("0", "N", "n", "FALSE", "false", "OFF", "off")
                         .contains(System.getProperty("SPLIT_NAMES_ON_DELIMITER"));
 
+        setName(name, splitNamesOnDelimiter);
+    }
+
+    public void setName(String name, boolean splitNamesOnDelimiter) {
         if (MultiPartName.isQuoted(name) && name.contains(".") && splitNamesOnDelimiter) {
             partItems.clear();
             for (String unquotedIdentifier : MultiPartName.unquote(name).split("\\.")) {
                 partItems.add("\"" + unquotedIdentifier + "\"");
             }
             Collections.reverse(partItems);
+        } else if (name.contains(".") && splitNamesOnDelimiter) {
+            partItems.clear();
+            partItems.addAll(Arrays.asList(MultiPartName.unquote(name).split("\\.")));
+            Collections.reverse(partItems);
         } else {
             setIndex(NAME_IDX, name);
         }
@@ -294,6 +308,15 @@ public <T, S> T accept(IntoTableVisitor<T> intoTableVisitor, S context) {
         return intoTableVisitor.visit(this, context);
     }
 
+    public String getTimeTravel() {
+        return timeTravelStr;
+    }
+
+    public Table setTimeTravel(String timeTravelStr) {
+        this.timeTravelStr = timeTravelStr;
+        return this;
+    }
+
     @Override
     public Pivot getPivot() {
         return pivot;
@@ -346,6 +369,11 @@ public Table setSampleClause(SampleClause sampleClause) {
 
     public StringBuilder appendTo(StringBuilder builder) {
         builder.append(getFullyQualifiedName());
+
+        if (timeTravelStr != null) {
+            builder.append(" ").append(timeTravelStr);
+        }
+
         if (alias != null) {
             builder.append(alias);
         }
@@ -422,7 +450,9 @@ public Table getResolvedTable() {
      */
     public Table setResolvedTable(Table resolvedTable) {
         // clone, not reference
-        this.resolvedTable = new Table(resolvedTable.getFullyQualifiedName());
+        if (resolvedTable != null) {
+            this.resolvedTable = new Table(resolvedTable.getFullyQualifiedName());
+        }
         return this;
     }
 
@@ -465,4 +495,16 @@ public static Table[] setUnsetCatalogAndSchema(String currentCatalogName,
         }
         return tables;
     }
+
+    @Override
+    public Table clone() {
+        try {
+            Table clone = (Table) super.clone();
+            clone.setName(this.getFullyQualifiedName());
+            clone.setResolvedTable(this.resolvedTable);
+            return clone;
+        } catch (CloneNotSupportedException e) {
+            throw new AssertionError();
+        }
+    }
 }

src/main/java/net/sf/jsqlparser/util/deparser/SelectDeParser.java

@@ -449,25 +449,28 @@ public <S> StringBuilder visit(SelectItem<?> selectItem, S context) {
 
 
     @Override
-    public <S> StringBuilder visit(Table tableName, S context) {
-        builder.append(tableName.getFullyQualifiedName());
-        Alias alias = tableName.getAlias();
+    public <S> StringBuilder visit(Table table, S context) {
+        builder.append(table.getFullyQualifiedName());
+        if (table.getTimeTravel() != null) {
+            builder.append(" ").append(table.getTimeTravel());
+        }
+        Alias alias = table.getAlias();
         if (alias != null) {
             builder.append(alias);
         }
-        Pivot pivot = tableName.getPivot();
+        Pivot pivot = table.getPivot();
         if (pivot != null) {
             pivot.accept(this, context);
         }
-        UnPivot unpivot = tableName.getUnPivot();
+        UnPivot unpivot = table.getUnPivot();
         if (unpivot != null) {
             unpivot.accept(this, context);
         }
-        MySQLIndexHint indexHint = tableName.getIndexHint();
+        MySQLIndexHint indexHint = table.getIndexHint();
         if (indexHint != null) {
             builder.append(indexHint);
         }
-        SQLServerHints sqlServerHints = tableName.getSqlServerHints();
+        SQLServerHints sqlServerHints = table.getSqlServerHints();
         if (sqlServerHints != null) {
             builder.append(sqlServerHints);
         }