Module: BetterRecord::Migration
- Defined in:
- lib/better_record/migration.rb
Instance Method Summary collapse
- #audit_table(table_name, rows = nil, query_text = nil, skip_columns = nil) ⇒ Object
- #login_triggers(table_name, password_col = 'password', email_col = 'email', function_name = nil, in_reverse = false) ⇒ Object
Instance Method Details
#audit_table(table_name, rows = nil, query_text = nil, skip_columns = nil) ⇒ Object
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# File 'lib/better_record/migration.rb', line 5 def audit_table(table_name, rows = nil, query_text = nil, skip_columns = nil) reversible do |d| d.up do if(rows && rows.is_a?(Array)) skip_columns = rows rows = true query_text = true end if(rows.nil?) execute "SELECT #{BetterRecord.db_audit_schema}.audit_table('#{table_name}');" else rows = !!rows ? 't' : 'f' query_text = !!query_text ? 't' : 'f' skip_columns = skip_columns.present? ? "'{#{skip_columns.join(',')}}'" : 'ARRAY[]' execute "SELECT #{BetterRecord.db_audit_schema}.audit_table('#{table_name}', BOOLEAN '#{rows}', BOOLEAN '#{query_text}', #{skip_columns}::text[]);" end end d.down do execute "DROP TRIGGER audit_trigger_row ON #{table_name};" execute "DROP TRIGGER audit_trigger_stm ON #{table_name};" end end end |
#login_triggers(table_name, password_col = 'password', email_col = 'email', function_name = nil, in_reverse = false) ⇒ Object
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
# File 'lib/better_record/migration.rb', line 31 def login_triggers(table_name, password_col = 'password', email_col = 'email', function_name = nil, in_reverse = false) table_name = table_name.to_s reversible do |d| d.__send__(in_reverse ? :down : :up) do password_text = '' if !!password_col create_pwd_txt = ->(col) { "IF (NEW.\#{col} IS NOT NULL)\nAND (\n(TG_OP = 'INSERT') OR ( NEW.\#{col} IS DISTINCT FROM OLD.\#{col} )\n) THEN\nIF (NEW.\#{col} IS DISTINCT FROM 'CLEAR_EXISTING_PASSWORD_FOR_ROW') THEN\nNEW.\#{col} = hash_password(NEW.\#{col});\nELSE\nNEW.\#{col} = NULL;\nEND IF;\nELSE\nIF (TG_OP IS DISTINCT FROM 'INSERT') THEN\nNEW.\#{col} = OLD.\#{col};\nELSE\nNEW.\#{col} = NULL;\nEND IF;\nEND IF;\n\n" } password_text = password_col.is_a?(Array) ? (password_col.map {|pwd| create_pwd_txt.call(pwd)}).join("\n") : create_pwd_txt.call(password_col) end email_text = '' if !!email_col email_text = "IF (TG_OP = 'INSERT') OR ( NEW.\#{email_col} IS DISTINCT FROM OLD.\#{email_col} ) THEN\nNEW.\#{email_col} = validate_email(NEW.\#{email_col});\nEND IF;\n\n" end execute "CREATE OR REPLACE FUNCTION \#{function_name.presence || table_name.singularize}_changed()\nRETURNS TRIGGER AS\n$BODY$\nBEGIN\n\#{password_text}\n\#{email_text}\nRETURN NEW;\nEND;\n$BODY$\nlanguage 'plpgsql';\n" execute "CREATE TRIGGER \#{function_name.presence || table_name}_on_insert\nBEFORE INSERT ON \#{table_name}\nFOR EACH ROW\nEXECUTE PROCEDURE \#{function_name.presence || table_name.singularize}_changed();\n" execute "CREATE TRIGGER \#{function_name.presence || table_name}_on_update\nBEFORE UPDATE ON \#{table_name}\nFOR EACH ROW\nEXECUTE PROCEDURE \#{function_name.presence || table_name.singularize}_changed();\n\n" end d.__send__(in_reverse ? :up : :down) do execute "DROP TRIGGER IF EXISTS #{function_name.presence || table_name}_on_insert ON #{table_name};" execute "DROP TRIGGER IF EXISTS #{function_name.presence || table_name}_on_update ON #{table_name};" end end end |