Fixes broken code (eatonphil#10)

datager · Apr 16, 2020 · afd6424 · afd6424
1 parent e7ff8cb
commit afd6424
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"strings"
 
 	"github.com/eatonphil/gosql"
 
@@ -100,6 +101,11 @@ repl:
 			continue repl
 		}
 
+		trimmedLine := strings.TrimRight(line, " ")
+		if trimmedLine == "quit" || trimmedLine == "exit" || strings.TrimLeft(trimmedLine, " ") == "\\q" {
+			break
+		}
+
 		ast, err := gosql.Parse(line)
 		if err != nil {
 			fmt.Println("Error while parsing:", err)
@@ -114,6 +120,12 @@ repl:
 					fmt.Println("Error creating table", err)
 					continue repl
 				}
+			case gosql.DropTableKind:
+				err = mb.DropTable(ast.Statements[0].DropTableStatement)
+				if err != nil {
+					fmt.Println("Error dropping table", err)
+					continue repl
+				}
 			case gosql.InsertKind:
 				err = mb.Insert(stmt.InsertStatement)
 				if err != nil {