module ForemanPuppet
  module Api
    module V2
      class HostClassesController < V2::BaseController
        before_action :find_host, only: %i[index create destroy]
        before_action :find_puppetclass, only: %i[create destroy]

        api :GET, '/hosts/:host_id/puppetclass_ids/', N_('List all Puppet class IDs for host')

        def index
          render json: { root_node_name => HostClass.authorized(:edit_classes).where(host_puppet_facet_id: @host.puppet.id).pluck('puppetclass_id') }
        end

        api :POST, '/hosts/:host_id/puppetclass_ids', N_('Add a Puppet class to host')
        param :host_id, String, required: true, desc: N_('ID of host')
        param :puppetclass_id, String, required: true, desc: N_('ID of Puppet class')

        def create
          @host_class = HostClass.create!(host_puppet_facet_id: @host.puppet.id, puppetclass_id: @puppetclass.id)
          render json: { host_id: @host.id, puppetclass_id: @host_class.puppetclass_id }
        end

        api :DELETE, '/hosts/:host_id/puppetclass_ids/:id/', N_('Remove a Puppet class from host')
        param :host_id, String, required: true, desc: N_('ID of host')
        param :id, String, required: true, desc: N_('ID of Puppet class')

        def destroy
          @host_class = HostClass.authorized(:edit_classes).where(host_puppet_facet_id: @host.puppet.id, puppetclass_id: @puppetclass.id)
          process_response @host_class.destroy_all
        end

        def resource_class
          ForemanPuppet::HostClass
        end

        private

        # overwrite resource_name so it's host and and not host_class, since we want to return @host
        def find_host
          if params[:host_id].blank?
            not_found
            return false
          end
          if ::Host::Managed.respond_to?(:authorized) &&
             ::Host::Managed.authorized('view_host', Host::Managed)
            @host = resource_finder(::Host.authorized(:view_hosts), params[:host_id])
            @host.puppet || @host.create_puppet
            @host
          end
        end

        def find_puppetclass
          puppetclass_id = if params[:action] == 'create'
                             params[:puppetclass_id]
                           else
                             params[:id]
                           end
          @puppetclass = resource_finder(Puppetclass.authorized(:view_puppetclasses), puppetclass_id)
        end
      end
    end
  end
end