Delete All Rows Menu Option

digitallyinduced · Jun 7, 2021 · da4a0c6 · da4a0c6
1 parent 53e6f51
commit da4a0c6
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 5 deletions.
diff --git a/IHP/IDE/Data/Controller.hs b/IHP/IDE/Data/Controller.hs
@@ -112,7 +112,7 @@ instance Controller DataController where
     action EditRowValueAction { tableName, targetName, id } = do
         connection <- connectToAppDb
         tableNames <- fetchTableNames connection
-        
+
         rows :: [[DynamicField]] <- fetchRows connection tableName
 
         let targetId = cs id
@@ -144,6 +144,12 @@ instance Controller DataController where
         PG.close connection
         redirectTo ShowTableRowsAction { .. }
 
+    action DeleteTableRowsAction { tableName } = do
+        connection <- connectToAppDb
+        let query = "TRUNCATE TABLE " <> tableName
+        PG.execute_ connection (PG.Query . cs $! query)
+        PG.close connection
+        redirectTo ShowTableRowsAction { .. }
 
 connectToAppDb :: (?context :: ControllerContext) => _
 connectToAppDb = PG.connectPostgreSQL $ fromConfig databaseUrl
@@ -188,7 +194,7 @@ fetchRowsPage :: FromRow r => PG.Connection -> Text -> Int -> Int -> IO [r]
 fetchRowsPage connection tableName page rows = do
     pkFields <- tablePrimaryKeyFields connection tableName
     let slice = " OFFSET " <> show (page * rows - rows) <> " ROWS FETCH FIRST " <> show rows <> " ROWS ONLY"
-    let query = "SELECT * FROM " <> tableName <> " ORDER BY " <> intercalate ", " pkFields <> slice 
+    let query = "SELECT * FROM " <> tableName <> " ORDER BY " <> intercalate ", " pkFields <> slice
 
     PG.query_ connection (PG.Query . cs $! query)
 

diff --git a/IHP/IDE/Data/View/ShowDatabase.hs b/IHP/IDE/Data/View/ShowDatabase.hs
@@ -45,6 +45,11 @@ renderTableSelector tableNames activeTableName = [hsx|
             </a>
             <div class="custom-menu menu-for-table shadow backdrop-blur" id={contextMenuId}>
                 <a href={pathTo (ShowTableAction name)}>Show Schema</a>
+                <a
+                    href={pathTo (DeleteTableRowsAction name)}
+                    class="js-delete"
+                    data-confirm={"Are you sure you want to delete all rows in '" <> name <> "' table?"}
+                >Delete All Rows</a>
             </div>
         |]
             where

diff --git a/IHP/IDE/ToolServer/Types.hs b/IHP/IDE/ToolServer/Types.hs
@@ -61,10 +61,10 @@ data EnumValuesController
     | DeleteEnumValueAction { enumName :: Text, valueId :: Int }
     deriving (Eq, Show, Data)
 
-
 data DataController
     = ShowDatabaseAction
     | ShowTableRowsAction { tableName :: Text }
+    | DeleteTableRowsAction { tableName :: Text }
     | ShowQueryAction
     | DeleteEntryAction { primaryKey :: Text, tableName :: Text }
     | CreateRowAction
@@ -117,7 +117,7 @@ data ColumnDefinition = ColumnDefinition
     } deriving (Show)
 
 
--- | Keeps track of all all available apps in the projects. Used to display 
+-- | Keeps track of all all available apps in the projects. Used to display
 -- the apps inside the sidebar navigation
 --
 -- Usually this list is like: @["Web"]@ or @["Web", "Admin"]@
@@ -130,4 +130,3 @@ newtype AppUrl = AppUrl Text
 -- | List of all controllers. Used inside e.g. the Schema Designer to decide whether to display
 -- the 'Generate Controller' option
 newtype WebControllers = WebControllers [Text]
-