From 938019e90e1f23af0a925bb05ec30c8815c13077 Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Thu, 9 Dec 2021 11:33:47 +0100
Subject: [PATCH] postgre new object templates

---
 .../src/frontend/drivers.js                   | 24 +++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/plugins/dbgate-plugin-postgres/src/frontend/drivers.js b/plugins/dbgate-plugin-postgres/src/frontend/drivers.js
index 9795c7ac..065434bf 100644
--- a/plugins/dbgate-plugin-postgres/src/frontend/drivers.js
+++ b/plugins/dbgate-plugin-postgres/src/frontend/drivers.js
@@ -68,6 +68,30 @@ const postgresDriverBase = {
   __analyserInternals: {
     refTableCond: '',
   },
+
+  getNewObjectTemplates() {
+    return [
+      { label: 'New view', sql: 'CREATE VIEW myview\nAS\nSELECT * FROM table1' },
+      {
+        label: 'New procedure',
+        sql: `CREATE PROCEDURE myproc (arg1 INT)
+LANGUAGE SQL 
+AS $$
+  SELECT * FROM table1;
+$$`,
+      },
+      {
+        label: 'New function (plpgsql)',
+        sql: `CREATE FUNCTION myfunc (arg1 INT)
+RETURNS INT
+AS $$
+BEGIN
+  RETURN 1;
+END
+$$ LANGUAGE plpgsql;`,
+      },
+    ];
+  },
 };
 
 /** @type {import('dbgate-types').EngineDriver} */