Parser supports more variants of CREATE SEQUENCE statements now

Also allow parsing of ALTER SEQUENCE statements

IHP/IDE/SchemaDesigner/Parser.hs

@@ -65,7 +65,7 @@ statement = do
     let create = try createExtension <|> try (StatementCreateTable <$> createTable) <|> try createIndex <|> try createFunction <|> try createTrigger <|> try createEnumType <|> try createPolicy <|> try createSequence
     let alter = do
             lexeme "ALTER"
-            alterTable <|> alterType
+            alterTable <|> alterType <|> alterSequence
     s <- setStatement <|> create <|> alter <|> selectStatement <|> try dropTable <|> try dropIndex <|> try dropPolicy <|> dropType <|> commentStatement <|> comment
     space
     pure s
@@ -572,6 +572,11 @@ alterType = do
     typeName <- qualifiedIdentifier
     addValue typeName
 
+alterSequence = do
+    lexeme "SEQUENCE"
+    raw <- cs <$> someTill (anySingle) (char ';')
+    pure UnknownStatement { raw = "ALTER SEQUENCE " <> raw };
+
 -- | ALTER TABLE users ALTER COLUMN email DROP NOT NULL;
 --  ALTER TABLE users ALTER COLUMN email SET NOT NULL;
 --  ALTER TABLE users ALTER COLUMN email SET DEFAULT 'value';
@@ -734,6 +739,35 @@ createSequence = do
     lexeme "CREATE"
     lexeme "SEQUENCE"
     name <- qualifiedIdentifier
+
+    -- We accept all the following SEQUENCE attributes, but don't save them
+    -- This is mostly to void issues in migrations when parsing the pg_dump output
+    optional do
+        lexeme "AS"
+        sqlType
+
+    optional do
+        lexeme "START"
+        lexeme "WITH"
+        expression
+
+    optional do
+        lexeme "INCREMENT"
+        lexeme "BY"
+        expression
+
+    optional do
+        lexeme "NO"
+        lexeme "MINVALUE"
+
+    optional do
+        lexeme "NO"
+        lexeme "MAXVALUE"
+
+    optional do
+        lexeme "CACHE"
+        expression
+
     char ';'
     pure CreateSequence { name }
 

Test/IDE/SchemaDesigner/ParserSpec.hs

@@ -552,6 +552,18 @@ $$;
 
         it "should parse 'CREATE SEQUENCE ..' statements with qualified name" do
             parseSql "CREATE SEQUENCE public.a;" `shouldBe` CreateSequence { name = "a" }
+
+        it "should parse 'CREATE SEQUENCE .. AS .. START WITH .. INCREMENT BY .. NO MINVALUE NO MAXVALUE CACHE ..;'" do
+            let sql = [trimming|
+                CREATE SEQUENCE public.a
+                    AS integer
+                    START WITH 1
+                    INCREMENT BY 1
+                    NO MINVALUE
+                    NO MAXVALUE
+                    CACHE 1;
+            |]
+            parseSql sql `shouldBe` CreateSequence { name = "a" }
 
         it "should parse 'SET' statements" do
             parseSql "SET statement_timeout = 0;" `shouldBe` Set { name = "statement_timeout", value = IntExpression 0 }
@@ -735,6 +747,10 @@ COMMENT ON EXTENSION "uuid-ossp" IS 'generate universally unique identifiers (UU
                     , arguments = [TextExpression "hello"]
                     }
 
+        it "should parse 'ALTER SEQUENCE ..' statements" do
+            let sql = cs [plain|ALTER SEQUENCE public.a OWNED BY public.b.serial_number;|]
+            parseSql sql `shouldBe` UnknownStatement { raw = "ALTER SEQUENCE public.a OWNED BY public.b.serial_number" }
+
 col :: Column
 col = Column
     { name = ""