Adds ON DELETE SET DEFAULT to schema designer

Fixes #372
digitallyinduced · Nov 20, 2020 · c84a7af · c84a7af
1 parent 831a124
commit c84a7af
Show file tree

Hide file tree

Showing 8 changed files with 33 additions and 2 deletions.
diff --git a/IHP/IDE/SchemaDesigner/Compiler.hs b/IHP/IDE/SchemaDesigner/Compiler.hs
@@ -46,6 +46,7 @@ compileOnDelete Nothing = ""
 compileOnDelete (Just NoAction) = "ON DELETE NO ACTION"
 compileOnDelete (Just Restrict) = "ON DELETE RESTRICT"
 compileOnDelete (Just SetNull) = "ON DELETE SET NULL"
+compileOnDelete (Just SetDefault) = "ON DELETE SET DEFAULT"
 compileOnDelete (Just Cascade) = "ON DELETE CASCADE"
 
 compileColumn :: PrimaryKeyConstraint -> Column -> Text

diff --git a/IHP/IDE/SchemaDesigner/Controller/Columns.hs b/IHP/IDE/SchemaDesigner/Controller/Columns.hs
@@ -136,6 +136,7 @@ instance Controller ColumnsController where
             Just NoAction -> do pure "NoAction"
             Just Restrict -> do pure "Restrict"
             Just SetNull -> do pure "SetNull"
+            Just SetDefault -> do pure "SetDefault"
             Just Cascade -> do pure "Cascade"
             Nothing -> do pure "NoAction"
         render EditForeignKeyView { .. }
@@ -157,6 +158,7 @@ instance Controller ColumnsController where
         let onDelete = case onDeleteParam of
                 "Restrict" -> Restrict
                 "SetNull" -> SetNull
+                "SetDefault" -> SetDefault
                 "Cascade" -> Cascade
                 _ -> NoAction
         case constraintId of

diff --git a/IHP/IDE/SchemaDesigner/Parser.hs b/IHP/IDE/SchemaDesigner/Parser.hs
@@ -162,7 +162,7 @@ parseUniqueConstraint = do
 parseOnDelete = choice
         [ (lexeme "NO" >> lexeme "ACTION") >> pure NoAction
         , (lexeme "RESTRICT" >> pure Restrict)
-        , (lexeme "SET" >> lexeme "NULL") >> pure SetNull
+        , (lexeme "SET" >> ((lexeme "NULL" >> pure SetNull) <|> (lexeme "DEFAULT" >> pure SetDefault)))
         , (lexeme "CASCADE" >> pure Cascade)
         ]
 

diff --git a/IHP/IDE/SchemaDesigner/Types.hs b/IHP/IDE/SchemaDesigner/Types.hs
@@ -43,6 +43,7 @@ data OnDelete
     = NoAction
     | Restrict
     | SetNull
+    | SetDefault
     | Cascade
     deriving (Show, Eq)
 

diff --git a/IHP/IDE/SchemaDesigner/View/Columns/EditForeignKey.hs b/IHP/IDE/SchemaDesigner/View/Columns/EditForeignKey.hs
@@ -56,6 +56,7 @@ instance View EditForeignKeyView where
                                 {onDeleteSelector "NoAction"}
                                 {onDeleteSelector "Restrict"}
                                 {onDeleteSelector "SetNull"}
+                                {onDeleteSelector "SetDefault"}
                                 {onDeleteSelector "Cascade"}
                             </select>
                         </div>

diff --git a/IHP/IDE/SchemaDesigner/View/Columns/NewForeignKey.hs b/IHP/IDE/SchemaDesigner/View/Columns/NewForeignKey.hs
@@ -53,6 +53,7 @@ instance View NewForeignKeyView where
                                 {onDeleteSelector "NoAction"}
                                 {onDeleteSelector "Restrict"}
                                 {onDeleteSelector "SetNull"}
+                                {onDeleteSelector "SetDefault"}
                                 {onDeleteSelector "Cascade"}
                             </select>
                         </div>

diff --git a/Test/IDE/SchemaDesigner/CompilerSpec.hs b/Test/IDE/SchemaDesigner/CompilerSpec.hs
@@ -115,7 +115,20 @@ tests = do
                         }
                     }
             compileSql [statement] `shouldBe` "ALTER TABLE users ADD CONSTRAINT users_ref_company_id FOREIGN KEY (company_id) REFERENCES companies (id) ON DELETE CASCADE;\n"
-
+
+        it "should compile ALTER TABLE .. ADD FOREIGN KEY .. ON DELETE SET DEFAULT" do
+            let statement = AddConstraint
+                    { tableName = "users"
+                    , constraintName = "users_ref_company_id"
+                    , constraint = ForeignKeyConstraint
+                        { columnName = "company_id"
+                        , referenceTable = "companies"
+                        , referenceColumn = "id"
+                        , onDelete = Just SetDefault
+                        }
+                    }
+            compileSql [statement] `shouldBe` "ALTER TABLE users ADD CONSTRAINT users_ref_company_id FOREIGN KEY (company_id) REFERENCES companies (id) ON DELETE SET DEFAULT;\n"
+
         it "should compile ALTER TABLE .. ADD FOREIGN KEY .. ON DELETE SET NULL" do
             let statement = AddConstraint
                     { tableName = "users"

diff --git a/Test/IDE/SchemaDesigner/ParserSpec.hs b/Test/IDE/SchemaDesigner/ParserSpec.hs
@@ -116,6 +116,18 @@ tests = do
                         }
                     }
 
+        it "should parse ALTER TABLE .. ADD FOREIGN KEY .. ON DELETE SET DEFAULT" do
+            parseSql "ALTER TABLE users ADD CONSTRAINT users_ref_company_id FOREIGN KEY (company_id) REFERENCES companies (id) ON DELETE SET DEFAULT;" `shouldBe` AddConstraint
+                    { tableName = "users"
+                    , constraintName = "users_ref_company_id"
+                    , constraint = ForeignKeyConstraint
+                        { columnName = "company_id"
+                        , referenceTable = "companies"
+                        , referenceColumn = "id"
+                        , onDelete = Just SetDefault
+                        }
+                    }
+
         it "should parse ALTER TABLE .. ADD FOREIGN KEY .. ON DELETE SET NULL" do
             parseSql "ALTER TABLE users ADD CONSTRAINT users_ref_company_id FOREIGN KEY (company_id) REFERENCES companies (id) ON DELETE SET NULL;" `shouldBe` AddConstraint
                     { tableName = "users"