# frozen_string_literal: true

module EE
  module Gitlab
    module Auth
      module Ldap
        module Access
          extend ActiveSupport::Concern
          extend ::Gitlab::Utils::Override

          override :update_user
          def update_user
            return if ::Gitlab::Database.read_only?

            update_user_attributes
            update_memberships
            update_identity
            update_ssh_keys if sync_ssh_keys?
            update_kerberos_identity if import_kerberos_identities?
          end

          private

          override :find_ldap_user
          def find_ldap_user
            found_user = super
            return found_user if found_user

            if ldap_identity
              ::Gitlab::Auth::Ldap::Person.find_by_email(user.email, adapter)
            end
          end

          # Update user ssh keys if they changed in LDAP
          def update_ssh_keys
            remove_old_ssh_keys
            add_new_ssh_keys
          end

          # Add ssh keys that are in LDAP but not in GitLab
          # rubocop: disable CodeReuse/ActiveRecord
          def add_new_ssh_keys
            keys = ldap_user.ssh_keys - user.keys.ldap.pluck(:key)

            keys.each do |key|
              ::Gitlab::AppLogger.info "#{self.class.name}: adding LDAP SSH key #{key.inspect} to #{user.name} (#{user.id})"
              new_key = ::LDAPKey.new(title: "LDAP - #{ldap_config.sync_ssh_keys}", key: key)
              new_key.user = user

              unless new_key.save
                ::Gitlab::AppLogger.error "#{self.class.name}: failed to add LDAP SSH key #{key.inspect} to #{user.name} (#{user.id})\n"\
                  "error messages: #{new_key.errors.messages}"
              end
            end
          end
          # rubocop: enable CodeReuse/ActiveRecord

          # Remove ssh keys that do not exist in LDAP any more
          # rubocop: disable CodeReuse/ActiveRecord
          def remove_old_ssh_keys
            keys = user.keys.ldap.where.not(key: ldap_user.ssh_keys)

            keys.each do |deleted_key|
              ::Gitlab::AppLogger.info "#{self.class.name}: removing LDAP SSH key #{deleted_key.key} from #{user.name} (#{user.id})"

              unless deleted_key.destroy
                ::Gitlab::AppLogger.error "#{self.class.name}: failed to remove LDAP SSH key #{key.inspect} from #{user.name} (#{user.id})"
              end
            end
          end
          # rubocop: enable CodeReuse/ActiveRecord

          # Update user Kerberos identity with Kerberos principal name from Active Directory
          # rubocop: disable CodeReuse/ActiveRecord
          def update_kerberos_identity
            # there can be only one Kerberos identity in GitLab; if the user has a Kerberos identity in AD,
            # replace any existing Kerberos identity for the user
            return unless ldap_user.kerberos_principal.present?

            kerberos_identity = user.identities.find_by(provider: :kerberos)
            return if kerberos_identity && kerberos_identity.extern_uid == ldap_user.kerberos_principal

            kerberos_identity ||= ::Identity.new(provider: :kerberos, user: user)
            kerberos_identity.extern_uid = ldap_user.kerberos_principal
            unless kerberos_identity.save
              ::Gitlab::AppLogger.error "#{self.class.name}: failed to add Kerberos principal #{ldap_user.kerberos_principal} to #{user.name} (#{user.id})\n"\
                "error messages: #{kerberos_identity.errors.messages}"
            end
          end
          # rubocop: enable CodeReuse/ActiveRecord

          # Update user attributes (name and email) if they changed in LDAP
          def update_user_attributes
            ldap_email = ldap_user.try(:email)&.last.to_s.downcase
            ldap_name = ldap_user.try(:name)

            return if ldap_email.blank? && ldap_name.blank?

            attrs = { user: user }
            attrs[:email] = ldap_email if ldap_email.present?
            attrs[:name] = ldap_name if ldap_name.present?

            ::Users::UpdateService.new(user, attrs).execute do |user|
              user.skip_reconfirmation!
            end
          end

          def update_identity
            return if ldap_user.dn.empty? || ldap_user.dn == ldap_identity.extern_uid

            unless ldap_identity.update(extern_uid: ldap_user.dn)
              ::Gitlab::AppLogger.error "Could not update DN for #{user.name} (#{user.id})\n"\
                                 "error messages: #{user.ldap_identity.errors.messages}"
            end
          end

          delegate :sync_ssh_keys?, to: :ldap_config

          def import_kerberos_identities?
            # Kerberos may be enabled for Git HTTP access and/or as an Omniauth provider
            ldap_config.active_directory && (::Gitlab.config.kerberos.enabled || ::AuthHelper.kerberos_enabled? )
          end

          # rubocop: disable CodeReuse/ActiveRecord
          def update_memberships
            return if ldap_user.nil? || ldap_user.group_cns.empty?

            # Only update membership for new users. After this, regular
            # LDAP group sync will take care of things.
            return unless user.created_at > 5.minutes.ago && !user.last_credential_check_at

            group_ids = ::LdapGroupLink.where(cn: ldap_user.group_cns, provider: provider)
                          .distinct(:group_id)
                          .pluck(:group_id)

            ::LdapGroupSyncWorker.perform_async(group_ids, provider) if group_ids.any?
          end
          # rubocop: enable CodeReuse/ActiveRecord
        end
      end
    end
  end
end
