From 0f0797d8a6a9036a4cc2aa4e71efbb0befbdf01e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ra=C3=BAl=20Mar=C3=ADn?= <git@rmr.ninja>
Date: Mon, 16 Nov 2020 12:25:25 +0100
Subject: [PATCH] CDB_RegenerateTable: Initial implementation to get commands

---
 scripts-available/CDB_RegenerateTable.sql | 40 +++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 scripts-available/CDB_RegenerateTable.sql

diff --git a/scripts-available/CDB_RegenerateTable.sql b/scripts-available/CDB_RegenerateTable.sql
new file mode 100644
index 0000000..447846b
--- /dev/null
+++ b/scripts-available/CDB_RegenerateTable.sql
@@ -0,0 +1,40 @@
+--
+-- Given a table
+-- Replace cartodb ==> @extschema@
+-- Replace plpython3u ==> @@plpythonu@@
+--
+CREATE OR REPLACE FUNCTION cartodb.__CDB_RegenerateTable_Get_Commands(tableoid OID)
+RETURNS text
+AS $$
+    import subprocess
+
+    query = "SELECT current_database()::text as dname"
+    rv = plpy.execute(query, 1)
+    database_name_string = str(rv[0]['dname'])
+
+    query = """SELECT concat(quote_ident(nspname), '.', quote_ident(relname)) as quoted_name
+                        FROM pg_catalog.pg_class AS c
+                        JOIN pg_catalog.pg_namespace AS ns
+                        ON c.relnamespace = ns.oid
+                        WHERE c.oid = '%s'""" % (tableoid)
+    rv = plpy.execute(query, 1)
+    full_tablename_string = str(rv[0]['quoted_name'])
+
+    # NOTE: We always use -s so data is never dumped!
+    # That would be a security issue that we would need to deal with (and we currently do not need it)
+    process_parameters = ["pg_dump", "-s", "-t", full_tablename_string, database_name_string]
+
+    proc = subprocess.Popen(process_parameters, stdout=subprocess.PIPE, shell=False)
+    (out, err) = proc.communicate()
+    if (err):
+        plpy.error(err)
+
+    lines = out.splitlines()
+    lines = [line.rstrip() for line in lines]
+    lines = [line for line in lines if line]
+    lines = [line for line in lines if not line.startswith(b'--')]
+    lines = [line.decode("utf-8") for line in lines]
+
+    return "".join(lines)
+$$
+LANGUAGE plpython3u VOLATILE PARALLEL UNSAFE;