ManageIQ/manageiq-ui-classic

module Mixins
  module Actions
    module VmActions
      module Ownership
        extend ActiveSupport::Concern

        def ownership_items_hash
          return [] unless @ownershipitems

          @ownershipitems.map { |item| {:id => item.id.to_s, :kind => Api::CollectionConfig.new.name_for_subclass(item.class)} }
        end

        included do
          helper_method :ownership_items_hash
        end

        # Set Ownership selected db records
        def set_ownership
          assert_privileges(params[:pressed])
          # check to see if coming from show_list or drilled into vms from another CI
          controller = if request.parameters[:controller] == "vm" || %w[all_vms vms instances].include?(params[:display])
                         "vm"
                       elsif %w[miq_templates images].include?(params[:display]) || params[:pressed].starts_with?("miq_template_")
                         "miq_template"
                       else
                         request.parameters[:controller]
                       end
          @edit ||= {}
          @edit[:controller] = controller

          recs = session[:checked_items] || checked_or_params

          if recs.empty?
            add_flash(_("One or more %{model} must be selected to Set Ownership") % {
              :model => Dictionary.gettext(db.to_s, :type => :model, :notfound => :titleize, :plural => true)
            }, :error)
            @refresh_div = "flash_msg_div"
            @refresh_partial = "layouts/flash_msg"
            return
          end

          @origin_ownership_items = recs.collect(&:to_i)
          @ownershipitems = filter_ownership_items(get_class_from_controller_param(controller), recs)

          if @ownershipitems.blank?
            add_flash(_('None of the selected items allow ownership changes'), :error)
            @refresh_div = "flash_msg_div"
            @refresh_partial = "layouts/flash_msg"
            return
          end

          if @explorer
            @sb[:explorer] = true
            ownership(@origin_ownership_items)
          elsif params[:pressed].starts_with?("service_")
            ownership(@origin_ownership_items)
          elsif role_allows?(:feature => "vm_ownership")
            drop_breadcrumb(:name => _("Set Ownership"), :url => "/vm_common/ownership")
            # redirect to build the ownership screen
            javascript_redirect(:controller => controller, :action => 'ownership', :rec_ids => @origin_ownership_items, :escape => false)
          else
            head :ok
          end
        end

        alias image_ownership set_ownership
        alias instance_ownership set_ownership
        alias vm_ownership set_ownership
        alias miq_template_ownership set_ownership

        def get_class_from_controller_param(controller)
          case controller
          when "catalog"
            ServiceTemplate
          when "orchestration_stack"
            OrchestrationStack
          when "service"
            Service
          when "vm_or_template", "vm_infra", "vm_cloud", "vm"
            VmOrTemplate
          when "miq_template"
            MiqTemplate
          when "auth_key_pair_cloud"
            ManageIQ::Providers::CloudManager::AuthKeyPair
          end
        end

        # Assign/unassign ownership to a set of objects
        def ownership(ownership_ids = [])
          @sb[:explorer] = true if @explorer
          @in_a_form = @ownershipedit = true
          drop_breadcrumb(:name => _("Set Ownership"), :url => "/vm_common/ownership")
          ownership_ids = params[:rec_ids] if params[:rec_ids]
          @origin_ownership_items = ownership_ids
          build_ownership_info(ownership_ids)
          return if @ownershipitems.empty?

          build_targets_hash(@ownershipitems)
          if @sb[:explorer]
            @refresh_partial = "shared/views/ownership"
          else
            render :action => "show"
          end
        end

        def build_ownership_info(ownership_ids)
          @edit ||= {}
          klass = get_class_from_controller_param(params[:controller])
          load_user_group_items(ownership_ids, klass)

          @groups = {} # Create new entries hash (2nd pulldown)
          Rbac.filtered(MiqGroup.non_tenant_groups).each { |g| @groups[g.description] = g.id.to_s }

          @view = get_db_view(klass, :clickable => false) # Instantiate the MIQ Report view object
          session[:edit] = @edit
        end

        def header_for_ownership
          raise _("Items are required for Set Ownership screen") if @ownershipitems.nil? || @ownershipitems.count == 0

          if @ownershipitems.count == 1
            case @ownershipitems.first
            when ManageIQ::Providers::InfraManager::Vm
              _('Set Ownership for Virtual Machine')
            when ManageIQ::Providers::CloudManager::Vm
              _('Set Ownership for Instance')
            when ManageIQ::Providers::InfraManager::Template
              _('Set Ownership for Template')
            when ManageIQ::Providers::CloudManager::Template
              _('Set Ownership for Image')
            else
              _('Set Ownership')
            end
          else
            _('Set Ownership for selected items')
          end
        end

        def valid_items_for(klass, param_ids)
          scope = klass.respond_to?(:with_ownership) ? klass.with_ownership : klass
          checked_ids = Rbac.filtered(scope.where(:id => param_ids)).pluck(:id)
          checked_ids.to_set == param_ids.to_set
        end

        def ownership_update
          case params[:button]
          when "cancel" then ownership_handle_cancel_button
          when "save" then ownership_handle_save_button
          end
        end

        private

        def load_user_group_items(ownership_ids, klass)
          @ownershipitems ||= filter_ownership_items(klass, ownership_ids)
          if @ownershipitems.length > 1
            @user = @group = 'dont-change'
          else
            record = @ownershipitems.first
            @user = record.evm_owner.id.to_s if record.evm_owner
            @group = record.miq_group.id.to_s if record.miq_group
          end
        end

        def ownership_handle_cancel_button
          add_flash(_("Set Ownership was cancelled by the user"))
          if @sb[:explorer]
            @edit = @sb[:action] = nil
            replace_right_cell
          else
            session[:flash_msgs] = @flash_array
            ownership_handle_redirect
          end
        end

        def ownership_handle_save_button
          klass = get_class_from_controller_param(request.parameters[:controller])
          object_types = object_types_for_flash_message(klass, params[:objectIds])

          flash = _("Ownership saved for selected %{object_types}") % {:object_types => object_types}
          add_flash(flash)
          if @sb[:explorer]
            @sb[:action] = nil
            replace_right_cell
          else
            session[:flash_msgs] = @flash_array
            ownership_handle_redirect
          end
        end

        # Passing the selected item id to the previous_breadcrumb_url if its a show page.
        def ownership_handle_redirect
          object_ids = params[:objectIds].instance_of?(Array) ? params[:objectIds] : params[:objectIds].split(',')
          show_url = "/#{params[:controller]}/show"
          redirect_url = previous_breadcrumb_url == show_url ? "#{show_url}/#{object_ids.first}" : previous_breadcrumb_url
          javascript_redirect(redirect_url)
        end

        def filter_ownership_items(klass, ownership_ids)
          records = find_records_with_rbac(klass.order(:name), ownership_ids)
          records.try(:with_ownership) || records
        end
      end
    end
  end
end