module ForemanSalt
  module Api
    module V2
      class SaltStatesController < ::ForemanSalt::Api::V2::BaseController
        include StateImporter

        before_filter :find_resource, :except => [:index, :create, :import]
        before_filter :find_proxy, :only => :import
        before_filter :find_environment, :only => :index

        api :GET, '/salt_states', N_('List all Salt states')
        param :salt_environment_id, :identifier_dottable, :required => false, :desc => N_('Limit to a specific environment')
        def index
          if @salt_environment
            @salt_states = resource_scope_for_index.joins(:salt_environments).where('salt_module_environments.salt_environment_id' => @salt_environment)
          else
            @salt_states = resource_scope_for_index
          end

          @subtotal = @salt_states.count
        end

        api :GET, '/salt_states/:id/', N_('Show a state')
        param :id, :identifier_dottable, :required => true
        def show
        end

        def_param_group :state do
          param :state, Hash, :required => true, :action_aware => true do
            param :name, String, :required => true, :desc => N_('Name of the Salt state')
          end
        end

        api :POST, '/salt_states', N_('Create a state')
        param_group :state, :as => :create
        def create
          @salt_state = SaltModule.new(params[:state])
          process_response @salt_state.save
        end

        api :DELETE, '/salt_states/:id/', N_('Destroy a state')
        param :id, :identifier_dottable, :required => true
        def destroy
          process_response @salt_state.destroy
        end

        api :POST, '/salt_states/import/:smart_proxy_id', N_('Import states from a salt master')
        param :smart_proxy_id, :identifier_dottable, :required => true, :desc => N_('Salt Smart Proxy ID')
        param :salt_environment_ids, Array, :required => false, :desc => N_('Limit to a specific environments')
        param :actions, Array, :required => false, :desc => N_('Limit to specific actions: i.e. add, remove')
        param :dryrun, :bool, :required => false, :desc => N_('Dryrun only')
        def import
          states = fetch_states_from_proxy(@proxy, params[:salt_environment_ids])
          unless params[:dryrun]
            states[:changes].each do |environment, state|
              if state[:add].any? && (params[:actions].blank? || params[:actions].include?('add'))
                add_to_environment(state[:add], environment)
              end

              if state[:remove].any? && (params[:actions].blank? || params[:actions].include?('remove'))
                remove_from_environment(state[:remove], environment)
              end
            end
            clean_orphans
          end
          render :text => states.to_json
        end

        def controller_permission
          'salt_modules'
        end

        def resource_class
          ForemanSalt::SaltModule
        end

        def action_permission
          case params[:action]
          when 'import'
            :import
          else
            super
          end
        end

        private

        def find_environment
          if params[:salt_environment_id]
            @salt_environment = ForemanSalt::SaltEnvironment.find(params[:salt_environment_id])
            fail _('Could not find salt environment with id %s') % params[:salt_environment_id] unless @salt_environment
          end
        end
      end
    end
  end
end