# frozen_string_literal: true

module BetterRecord
  module Migration
    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

    def login_triggers(table_name, password_col = 'password', email_col = 'email')
      table_name = table_name.to_s

      reversible do |d|
        d.up do
          password_text = ''

          if !!password_col
            password_text = <<-SQL
              IF (NEW.#{password_col} IS NOT NULL)
              AND (
                (TG_OP = 'INSERT') OR ( NEW.#{password_col} IS DISTINCT FROM OLD.#{password_col} )
              ) THEN
                NEW.#{password_col} = hash_password(NEW.#{password_col});
              ELSE
                IF (TG_OP IS DISTINCT FROM 'INSERT') THEN
                  NEW.#{password_col} = OLD.#{password_col};
                ELSE
                  NEW.#{password_col} = NULL;
                END IF;
              END IF;

            SQL
          end

          email_text = ''

          if !!email_col
            email_text = <<-SQL
              IF (TG_OP = 'INSERT') OR ( NEW.#{email_col} IS DISTINCT FROM OLD.#{email_col} ) THEN
                NEW.#{email_col} = validate_email(NEW.#{email_col});
              END IF;

            SQL
          end

          execute <<-SQL
            CREATE OR REPLACE FUNCTION #{table_name.singularize}_changed()
              RETURNS TRIGGER AS
            $BODY$
            BEGIN
              #{password_text}
              #{email_text}
              RETURN NEW;
            END;
            $BODY$
            language 'plpgsql';
          SQL

          execute <<-SQL
            CREATE TRIGGER #{table_name}_on_insert
            BEFORE INSERT ON #{table_name}
            FOR EACH ROW
            EXECUTE PROCEDURE #{table_name.singularize}_changed();
          SQL

          execute <<-SQL
            CREATE TRIGGER #{table_name}_on_update
            BEFORE UPDATE ON #{table_name}
            FOR EACH ROW
            EXECUTE PROCEDURE #{table_name.singularize}_changed();

          SQL
        end

        d.down do
          execute "DROP TRIGGER IF EXISTS #{table_name}_on_insert ON #{table_name};"
          execute "DROP TRIGGER IF EXISTS #{table_name}_on_update ON #{table_name};"
        end
      end
    end
  end
end