(pre Kotlin -> Java) unnest Kotlin classes

partiql-ast/src/main/kotlin/org/partiql/ast/v1/CaseSensitivity.kt

@@ -0,0 +1,10 @@
+package org.partiql.ast.v1
+
+/**
+ * TODO docs, equals, hashcode
+ */
+public enum class CaseSensitivity {
+    SENSITIVE,
+    INSENSITIVE,
+    OTHER,
+}

partiql-ast/src/main/kotlin/org/partiql/ast/v1/Column.kt

@@ -0,0 +1,25 @@
+package org.partiql.ast.v1
+
+import org.partiql.ast.v1.type.Type
+
+/**
+ * TODO docs, equals, hashcode
+ */
+public class Column(
+    @JvmField
+    public var name: String,
+    @JvmField
+    public var type: Type,
+    @JvmField
+    public var constraints: List<Constraint>,
+) : AstNode() {
+    public override fun children(): Collection<AstNode> {
+        val kids = mutableListOf<AstNode?>()
+        kids.add(type)
+        kids.addAll(constraints)
+        return kids.filterNotNull()
+    }
+
+    public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+        visitor.visitColumn(this, ctx)
+}

partiql-ast/src/main/kotlin/org/partiql/ast/v1/Constraint.kt

@@ -0,0 +1,20 @@
+package org.partiql.ast.v1
+
+/**
+ * TODO docs, equals, hashcode
+ */
+public class Constraint(
+    @JvmField
+    public var name: String?,
+    @JvmField
+    public var constraintBody: ConstraintBody,
+) : AstNode() {
+    public override fun children(): Collection<AstNode> {
+        val kids = mutableListOf<AstNode?>()
+        kids.add(constraintBody)
+        return kids.filterNotNull()
+    }
+
+    public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+        visitor.visitConstraint(this, ctx)
+}

partiql-ast/src/main/kotlin/org/partiql/ast/v1/ConstraintBody.kt

@@ -0,0 +1,52 @@
+package org.partiql.ast.v1
+
+import org.partiql.ast.v1.expr.Expr
+
+/**
+ * TODO docs, equals, hashcode
+ */
+public abstract class ConstraintBody : AstNode() {
+    public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R = when (this) {
+        is Nullable -> visitor.visitConstraintBodyNullable(this, ctx)
+        is NotNull -> visitor.visitConstraintBodyNotNull(this, ctx)
+        is Check -> visitor.visitConstraintBodyCheck(this, ctx)
+        else -> throw NotImplementedError()
+    }
+
+    /**
+     * TODO docs, equals, hashcode
+     */
+    public object Nullable : ConstraintBody() {
+        public override fun children(): Collection<AstNode> = emptyList()
+
+        public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+            visitor.visitConstraintBodyNullable(this, ctx)
+    }
+
+    /**
+     * TODO docs, equals, hashcode
+     */
+    public object NotNull : ConstraintBody() {
+        public override fun children(): Collection<AstNode> = emptyList()
+
+        public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+            visitor.visitConstraintBodyNotNull(this, ctx)
+    }
+
+    /**
+     * TODO docs, equals, hashcode
+     */
+    public class Check(
+        @JvmField
+        public var expr: Expr,
+    ) : ConstraintBody() {
+        public override fun children(): Collection<AstNode> {
+            val kids = mutableListOf<AstNode?>()
+            kids.add(expr)
+            return kids.filterNotNull()
+        }
+
+        public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+            visitor.visitConstraintBodyCheck(this, ctx)
+    }
+}

partiql-ast/src/main/kotlin/org/partiql/ast/v1/CreateIndex.kt

@@ -0,0 +1,24 @@
+package org.partiql.ast.v1
+
+/**
+ * TODO docs, equals, hashcode
+ */
+public class CreateIndex(
+    @JvmField
+    public var index: Identifier?,
+    @JvmField
+    public var table: Identifier,
+    @JvmField
+    public var fields: List<PathLit>,
+) : DDL() {
+    public override fun children(): Collection<AstNode> {
+        val kids = mutableListOf<AstNode?>()
+        index?.let { kids.add(it) }
+        kids.add(table)
+        kids.addAll(fields)
+        return kids.filterNotNull()
+    }
+
+    public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+        visitor.visitCreateIndex(this, ctx)
+}

partiql-ast/src/main/kotlin/org/partiql/ast/v1/CreateTable.kt

@@ -0,0 +1,21 @@
+package org.partiql.ast.v1
+
+/**
+ * TODO docs, equals, hashcode
+ */
+public class CreateTable(
+    @JvmField
+    public var name: Identifier,
+    @JvmField
+    public var definition: TableDefinition?,
+) : DDL() {
+    public override fun children(): Collection<AstNode> {
+        val kids = mutableListOf<AstNode?>()
+        kids.add(name)
+        definition?.let { kids.add(it) }
+        return kids.filterNotNull()
+    }
+
+    public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+        visitor.visitCreateTable(this, ctx)
+}

partiql-ast/src/main/kotlin/org/partiql/ast/v1/DDL.kt

@@ -0,0 +1,14 @@
+package org.partiql.ast.v1
+
+/**
+ * TODO docs, equals, hashcode
+ */
+public abstract class DDL : Statement() {
+    public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R = when (this) {
+        is CreateTable -> visitor.visitCreateTable(this, ctx)
+        is CreateIndex -> visitor.visitCreateIndex(this, ctx)
+        is DropTable -> visitor.visitDropTable(this, ctx)
+        is DropIndex -> visitor.visitDropIndex(this, ctx)
+        else -> throw NotImplementedError()
+    }
+}

partiql-ast/src/main/kotlin/org/partiql/ast/v1/DropIndex.kt

@@ -0,0 +1,21 @@
+package org.partiql.ast.v1
+
+/**
+ * TODO docs, equals, hashcode
+ */
+public class DropIndex(
+    @JvmField
+    public var index: Identifier,
+    @JvmField
+    public var table: Identifier,
+) : DDL() {
+    public override fun children(): Collection<AstNode> {
+        val kids = mutableListOf<AstNode?>()
+        kids.add(index)
+        kids.add(table)
+        return kids.filterNotNull()
+    }
+
+    public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+        visitor.visitDropIndex(this, ctx)
+}

partiql-ast/src/main/kotlin/org/partiql/ast/v1/DropTable.kt

@@ -0,0 +1,18 @@
+package org.partiql.ast.v1
+
+/**
+ * TODO docs, equals, hashcode
+ */
+public class DropTable(
+    @JvmField
+    public var table: Identifier,
+) : DDL() {
+    public override fun children(): Collection<AstNode> {
+        val kids = mutableListOf<AstNode?>()
+        kids.add(table)
+        return kids.filterNotNull()
+    }
+
+    public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+        visitor.visitDropTable(this, ctx)
+}

partiql-ast/src/main/kotlin/org/partiql/ast/v1/Exclude.kt

@@ -5,89 +5,14 @@ package org.partiql.ast.v1
  */
 public class Exclude(
     @JvmField
-    public var items: List<Item>,
+    public var excludePaths: List<ExcludePath>,
 ) : AstNode() {
     public override fun children(): Collection<AstNode> {
         val kids = mutableListOf<AstNode?>()
-        kids.addAll(items)
+        kids.addAll(excludePaths)
         return kids.filterNotNull()
     }
 
     public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
         visitor.visitExclude(this, ctx)
-
-    /**
-     * TODO docs, equals, hashcode
-     */
-    public class Item(
-        @JvmField
-        public var root: Expr.Var,
-        @JvmField
-        public var steps: List<Step>,
-    ) : AstNode() {
-        public override fun children(): Collection<AstNode> {
-            val kids = mutableListOf<AstNode?>()
-            kids.add(root)
-            kids.addAll(steps)
-            return kids.filterNotNull()
-        }
-
-        public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
-            visitor.visitExcludeItem(this, ctx)
-    }
-
-    /**
-     * TODO docs, equals, hashcode
-     */
-    public abstract class Step : AstNode() {
-        /**
-         * TODO docs, equals, hashcode
-         */
-        public class StructField(
-            @JvmField
-            public var symbol: Identifier.Symbol,
-        ) : Step() {
-            public override fun children(): Collection<AstNode> {
-                val kids = mutableListOf<AstNode?>()
-                kids.add(symbol)
-                return kids.filterNotNull()
-            }
-
-            public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
-                visitor.visitExcludeStepStructField(this, ctx)
-        }
-
-        /**
-         * TODO docs, equals, hashcode
-         */
-        public class CollIndex(
-            @JvmField
-            public var index: Int,
-        ) : Step() {
-            public override fun children(): Collection<AstNode> = emptyList()
-
-            public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
-                visitor.visitExcludeStepCollIndex(this, ctx)
-        }
-
-        /**
-         * TODO docs, equals, hashcode
-         */
-        public object StructWildcard : Step() {
-            public override fun children(): Collection<AstNode> = emptyList()
-
-            public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
-                visitor.visitExcludeStepStructWildcard(this, ctx)
-        }
-
-        /**
-         * TODO docs, equals, hashcode
-         */
-        public object CollWildcard : Step() {
-            public override fun children(): Collection<AstNode> = emptyList()
-
-            public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
-                visitor.visitExcludeStepCollWildcard(this, ctx)
-        }
-    }
 }

partiql-ast/src/main/kotlin/org/partiql/ast/v1/ExcludePath.kt

@@ -0,0 +1,23 @@
+package org.partiql.ast.v1
+
+import org.partiql.ast.v1.expr.ExprVar
+
+/**
+ * TODO docs, equals, hashcode
+ */
+public class ExcludePath(
+    @JvmField
+    public var root: ExprVar,
+    @JvmField
+    public var excludeSteps: List<ExcludeStep>,
+) : AstNode() {
+    public override fun children(): Collection<AstNode> {
+        val kids = mutableListOf<AstNode?>()
+        kids.add(root)
+        kids.addAll(excludeSteps)
+        return kids.filterNotNull()
+    }
+
+    public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+        visitor.visitExcludePath(this, ctx)
+}

partiql-ast/src/main/kotlin/org/partiql/ast/v1/ExcludeStep.kt

@@ -0,0 +1,56 @@
+package org.partiql.ast.v1
+
+/**
+ * TODO docs, equals, hashcode
+ */
+public abstract class ExcludeStep : AstNode() {
+    /**
+     * TODO docs, equals, hashcode
+     */
+    public class StructField(
+        @JvmField
+        public var symbol: Identifier.Symbol,
+    ) : ExcludeStep() {
+        public override fun children(): Collection<AstNode> {
+            val kids = mutableListOf<AstNode?>()
+            kids.add(symbol)
+            return kids.filterNotNull()
+        }
+
+        public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+            visitor.visitExcludeStepStructField(this, ctx)
+    }
+
+    /**
+     * TODO docs, equals, hashcode
+     */
+    public class CollIndex(
+        @JvmField
+        public var index: Int,
+    ) : ExcludeStep() {
+        public override fun children(): Collection<AstNode> = emptyList()
+
+        public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+            visitor.visitExcludeStepCollIndex(this, ctx)
+    }
+
+    /**
+     * TODO docs, equals, hashcode
+     */
+    public object StructWildcard : ExcludeStep() {
+        public override fun children(): Collection<AstNode> = emptyList()
+
+        public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+            visitor.visitExcludeStepStructWildcard(this, ctx)
+    }
+
+    /**
+     * TODO docs, equals, hashcode
+     */
+    public object CollWildcard : ExcludeStep() {
+        public override fun children(): Collection<AstNode> = emptyList()
+
+        public override fun <R, C> accept(visitor: AstVisitor<R, C>, ctx: C): R =
+            visitor.visitExcludeStepCollWildcard(this, ctx)
+    }
+}