You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
277 lines
9.1 KiB
277 lines
9.1 KiB
# -*- coding: utf-8 -*-
|
|
#############################################################################
|
|
#
|
|
# Cybrosys Technologies Pvt. Ltd.
|
|
#
|
|
# Copyright (C) 2025-TODAY Cybrosys Technologies(<https://www.cybrosys.com>).
|
|
# Author: Cybrosys Techno Solutions(<https://www.cybrosys.com>)
|
|
#
|
|
# You can modify it under the terms of the GNU AFFERO
|
|
# GENERAL PUBLIC LICENSE (AGPL v3), Version 3.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU AFFERO GENERAL PUBLIC LICENSE (AGPL v3) for more details.
|
|
#
|
|
# You should have received a copy of the GNU AFFERO GENERAL PUBLIC LICENSE
|
|
# (AGPL v3) along with this program.
|
|
# If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
#############################################################################
|
|
def post_init_hook(env):
|
|
|
|
env.cr.execute("""
|
|
CREATE OR REPLACE FUNCTION log_update_data()
|
|
RETURNS TRIGGER AS $$
|
|
DECLARE
|
|
changed_fields JSONB = '{}'::JSONB;
|
|
old_row JSONB;
|
|
new_row JSONB;
|
|
column_name TEXT;
|
|
BEGIN
|
|
old_row = to_jsonb(OLD);
|
|
new_row = to_jsonb(NEW);
|
|
|
|
FOR column_name IN SELECT jsonb_object_keys(old_row)
|
|
LOOP
|
|
IF old_row ->> column_name IS DISTINCT FROM new_row ->> column_name THEN
|
|
changed_fields = jsonb_set(
|
|
changed_fields,
|
|
array[column_name],
|
|
old_row -> column_name
|
|
);
|
|
END IF;
|
|
END LOOP;
|
|
|
|
IF jsonb_typeof(changed_fields) != 'null' AND changed_fields != '{}'::JSONB THEN
|
|
BEGIN
|
|
INSERT INTO undo_redo (user_id,
|
|
table_name,
|
|
record_id,
|
|
updated_data,mode
|
|
)
|
|
VALUES (
|
|
OLD.write_uid,
|
|
TG_TABLE_NAME,
|
|
OLD.id,
|
|
changed_fields,'undo'
|
|
);
|
|
EXCEPTION WHEN OTHERS THEN
|
|
RETURN NEW;
|
|
END;
|
|
END IF;
|
|
|
|
RETURN NEW;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
""")
|
|
env.cr.execute("""
|
|
CREATE OR REPLACE FUNCTION log_delete_data()
|
|
RETURNS TRIGGER AS $$
|
|
BEGIN
|
|
DELETE FROM undo_redo
|
|
WHERE record_id = OLD.id
|
|
AND table_name = TG_TABLE_NAME;
|
|
RETURN OLD;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
""")
|
|
env.cr.execute("""
|
|
DO $$
|
|
DECLARE
|
|
rec RECORD;
|
|
BEGIN
|
|
FOR rec IN
|
|
SELECT n.nspname, c.relname
|
|
FROM pg_class c
|
|
JOIN pg_namespace n ON c.relnamespace = n.oid
|
|
WHERE c.relkind = 'r'
|
|
AND n.nspname NOT IN ('pg_catalog', 'information_schema') AND c.relname != 'undo_redo'
|
|
AND EXISTS (
|
|
SELECT 1
|
|
FROM information_schema.columns
|
|
WHERE table_schema = n.nspname
|
|
AND table_name = c.relname
|
|
AND column_name = 'id'
|
|
)
|
|
LOOP
|
|
EXECUTE format('
|
|
CREATE TRIGGER capture_delete_dynamic
|
|
AFTER UPDATE ON %I.%I
|
|
FOR EACH ROW
|
|
EXECUTE FUNCTION log_update_data();', rec.nspname, rec.relname);
|
|
EXECUTE format('
|
|
CREATE TRIGGER log_delete_trigger
|
|
AFTER DELETE ON %I.%I
|
|
FOR EACH ROW
|
|
EXECUTE FUNCTION log_delete_data();',
|
|
rec.nspname, rec.relname);
|
|
|
|
END LOOP;
|
|
END $$;
|
|
""")
|
|
env.cr.execute("""
|
|
CREATE OR REPLACE FUNCTION add_update_trigger_to_new_tables()
|
|
RETURNS VOID AS $$
|
|
DECLARE
|
|
rec RECORD;
|
|
BEGIN
|
|
FOR rec IN
|
|
SELECT n.nspname, c.relname
|
|
FROM pg_class c
|
|
JOIN pg_namespace n ON c.relnamespace = n.oid
|
|
WHERE c.relkind = 'r'
|
|
AND n.nspname NOT IN ('pg_catalog', 'information_schema')
|
|
AND EXISTS (
|
|
SELECT 1
|
|
FROM information_schema.columns
|
|
WHERE table_schema = n.nspname
|
|
AND table_name = c.relname
|
|
AND column_name = 'id'
|
|
)
|
|
LOOP
|
|
IF NOT EXISTS (
|
|
SELECT 1
|
|
FROM pg_trigger
|
|
WHERE tgrelid = (SELECT oid FROM pg_class WHERE relname = rec.relname AND relnamespace = (SELECT oid FROM pg_namespace WHERE nspname = rec.nspname))
|
|
AND tgname = 'capture_delete_dynamic'
|
|
) THEN
|
|
IF rec.relname != 'undo_redo' THEN
|
|
EXECUTE format('
|
|
CREATE TRIGGER capture_delete_dynamic
|
|
AFTER UPDATE ON %I.%I
|
|
FOR EACH ROW
|
|
EXECUTE FUNCTION log_update_data();', rec.nspname, rec.relname);
|
|
EXECUTE format('
|
|
CREATE TRIGGER log_delete_trigger
|
|
AFTER DELETE ON %I.%I
|
|
FOR EACH ROW
|
|
EXECUTE FUNCTION log_delete_data();',
|
|
rec.nspname, rec.relname);
|
|
|
|
END IF;
|
|
END IF;
|
|
END LOOP;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
""")
|
|
env.cr.execute("""
|
|
CREATE OR REPLACE FUNCTION trigger_on_table_creation()
|
|
RETURNS EVENT_TRIGGER AS $$
|
|
BEGIN
|
|
PERFORM add_update_trigger_to_new_tables();
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
""")
|
|
env.cr.execute("""
|
|
CREATE EVENT TRIGGER auto_add_delete_triggers
|
|
ON ddl_command_end
|
|
WHEN TAG IN ('CREATE TABLE')
|
|
EXECUTE FUNCTION trigger_on_table_creation();
|
|
""")
|
|
env.cr.execute("""
|
|
CREATE OR REPLACE FUNCTION handle_log_reinsert()
|
|
RETURNS TRIGGER AS $$
|
|
DECLARE
|
|
target_table TEXT;
|
|
column_list TEXT;
|
|
value_list TEXT;
|
|
column_count INT;
|
|
new_mode TEXT;
|
|
latest_id INT;
|
|
BEGIN
|
|
IF TG_TABLE_NAME = 'undo_redo' THEN
|
|
target_table := OLD.table_name;
|
|
|
|
column_list := array_to_string(
|
|
array(SELECT jsonb_object_keys(OLD.updated_data)),
|
|
', '
|
|
);
|
|
value_list := array_to_string(
|
|
array(
|
|
SELECT format('%L', OLD.updated_data->>key)
|
|
FROM jsonb_object_keys(OLD.updated_data) AS key
|
|
),
|
|
', '
|
|
);
|
|
|
|
new_mode := CASE OLD.mode
|
|
WHEN 'undo' THEN 'redo'
|
|
ELSE 'undo'
|
|
END;
|
|
column_count := array_length(string_to_array(column_list, ','), 1);
|
|
IF column_count > 1 THEN
|
|
EXECUTE format(
|
|
'UPDATE %I SET (%s) = (%s) WHERE id = %L',
|
|
target_table,
|
|
column_list,
|
|
value_list,
|
|
OLD.record_id
|
|
);
|
|
ELSIF column_count = 1 THEN
|
|
EXECUTE format(
|
|
'UPDATE %I SET %s = %s WHERE id = %L',
|
|
target_table,
|
|
column_list,
|
|
value_list,
|
|
OLD.record_id
|
|
);
|
|
END IF;
|
|
SELECT id INTO latest_id
|
|
FROM undo_redo
|
|
WHERE table_name = OLD.table_name
|
|
AND record_id = OLD.record_id
|
|
ORDER BY id DESC
|
|
LIMIT 1;
|
|
|
|
UPDATE undo_redo
|
|
SET mode = new_mode
|
|
WHERE id = latest_id;
|
|
|
|
RETURN OLD;
|
|
END IF;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
""")
|
|
env.cr.execute("""
|
|
CREATE TRIGGER after_delete_trigger
|
|
AFTER DELETE ON undo_redo
|
|
FOR EACH ROW
|
|
EXECUTE FUNCTION handle_log_reinsert();
|
|
""")
|
|
|
|
def uninstall_hook(env):
|
|
env.cr.execute("""
|
|
DROP EVENT TRIGGER IF EXISTS auto_add_delete_triggers;
|
|
""")
|
|
env.cr.execute("""
|
|
DROP TRIGGER IF EXISTS after_delete_trigger ON undo_redo;
|
|
""")
|
|
env.cr.execute("""
|
|
DO $$
|
|
DECLARE
|
|
rec RECORD;
|
|
BEGIN
|
|
FOR rec IN
|
|
SELECT n.nspname, c.relname
|
|
FROM pg_class c
|
|
JOIN pg_namespace n ON c.relnamespace = n.oid
|
|
WHERE c.relkind = 'r'
|
|
AND n.nspname NOT IN ('pg_catalog', 'information_schema')
|
|
LOOP
|
|
EXECUTE format('DROP TRIGGER IF EXISTS capture_delete_dynamic ON %I.%I;',
|
|
rec.nspname, rec.relname);
|
|
EXECUTE format('DROP TRIGGER IF EXISTS log_delete_trigger ON %I.%I;',
|
|
rec.nspname, rec.relname);
|
|
END LOOP;
|
|
END $$;
|
|
""")
|
|
env.cr.execute("""
|
|
DROP FUNCTION IF EXISTS log_update_data() CASCADE;
|
|
DROP FUNCTION IF EXISTS log_delete_data() CASCADE;
|
|
DROP FUNCTION IF EXISTS add_update_trigger_to_new_tables() CASCADE;
|
|
DROP FUNCTION IF EXISTS trigger_on_table_creation() CASCADE;
|
|
DROP FUNCTION IF EXISTS handle_log_reinsert() CASCADE;
|
|
""")
|