WIP.

lib/antlr/create_table.go

@@ -143,6 +143,8 @@ func processCreateTable(ctx *generated.ColumnCreateTableContext) (Event, error)
 					}
 				}
 			}
+		default:
+			slog.Warn(fmt.Sprintf("Skipping unsupported create table types: %T", castedChild))
 		}
 	}
 

lib/antlr/create_table_test.go

@@ -7,13 +7,22 @@ import (
 	"testing"
 )
 
-func TestCreateTable(t *testing.T) {
+func TestCreateTablePoC(t *testing.T) {
 	{
-		// Create table LIKE (not currently supported)
 		events, err := Parse("CREATE TABLE table_name LIKE other_table;")
 		assert.NoError(t, err)
-		assert.Len(t, events, 0)
+		assert.Len(t, events, 1)
+
+		createTableEvent, isOk := events[0].(CreateTableEvent)
+		assert.True(t, isOk)
+
+		assert.Equal(t, "table_name", createTableEvent.GetTable())
+		assert.Len(t, createTableEvent.GetColumns(), 0)
+		assert.Equal(t, "other_table", createTableEvent.LikeTable.TableName)
 	}
+}
+
+func TestCreateTable(t *testing.T) {
 	{
 		// Create table with column as CHARACTER SET and collation specified at the column level
 		events, err := Parse("CREATE TABLE table_name (id INT, name VARCHAR(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci);")

lib/antlr/parse.go

@@ -41,7 +41,11 @@ func Parse(sqlCmd string) ([]Event, error) {
 	stream := antlr.NewCommonTokenStream(lexer, antlr.TokenDefaultChannel)
 	// This will go through our custom visit function. If you are trying to print out the AST, split this function into [sqlStatements] and [parser]
 	// Then have print [sqlStatements.ToStringTree(nil, parser)]
-	return visit(generated.NewMySqlParser(stream).SqlStatements())
+
+	parser := generated.NewMySqlParser(stream)
+	statements := parser.SqlStatements()
+	fmt.Println("##", statements.ToStringTree(nil, parser))
+	return visit(statements)
 }
 
 func visit(tree antlr.Tree) ([]Event, error) {
@@ -65,6 +69,9 @@ func visit(tree antlr.Tree) ([]Event, error) {
 		}
 
 		return events, nil
+	case *generated.CopyCreateTableContext:
+		fmt.Println("Here?")
+		return nil, nil
 	case *generated.ColumnCreateTableContext:
 		evt, err := processCreateTable(ctx)
 		if err != nil {
@@ -78,7 +85,6 @@ func visit(tree antlr.Tree) ([]Event, error) {
 		return processDropTable(ctx)
 	case
 		*generated.EmptyStatement_Context,
-		*generated.CopyCreateTableContext,
 		*generated.TruncateTableContext,
 		*generated.AdministrationStatementContext,
 		*generated.CreateDatabaseContext,

lib/antlr/types.go

@@ -55,8 +55,13 @@ func (a AfterPosition) Kind() string {
 	return "after"
 }
 
+type LikeTable struct {
+	TableName string
+}
+
 type CreateTableEvent struct {
 	TableName string
+	LikeTable *LikeTable
 	Columns   []Column
 }