theforeman/foreman

module Api
  module V2
    class SmartProxiesController < V2::BaseController
      include Api::Version2
      include Foreman::Controller::Parameters::SmartProxy

      before_action :find_resource, :only => %w{show update destroy refresh version logs}

      api :GET, "/smart_proxies/", N_("List all smart proxies")
      param_group :taxonomy_scope, ::Api::V2::BaseController
      param_group :search_and_pagination, ::Api::V2::BaseController
      param :include_status, :bool, N_("Flag to indicate whether to include status or not")
      add_scoped_search_description_for(SmartProxy)

      def index
        @smart_proxies = resource_scope_for_index.includes(:features)
        @status = params[:include_status]
      end

      api :GET, "/smart_proxies/:id/", N_("Show a smart proxy")
      param :id, :identifier, :required => true
      param :include_status, :bool, N_("Flag to indicate whether to include status or not")
      param :include_version, :bool, N_("Flag to indicate whether to include version or not")

      def show
        @status = params[:include_status]
        @version = params[:include_version]
      end

      def_param_group :smart_proxy do
        param :smart_proxy, Hash, :required => true, :action_aware => true do
          param :name, String, :required => true
          param :url, String, :required => true
          param_group :taxonomies, ::Api::V2::BaseController
        end
      end

      api :POST, "/smart_proxies/", N_("Create a smart proxy")
      param_group :smart_proxy, :as => :create

      def create
        @smart_proxy = SmartProxy.new(smart_proxy_params)
        process_response @smart_proxy.save
      end

      api :PUT, "/smart_proxies/:id/", N_("Update a smart proxy")
      param :id, String, :required => true
      param_group :smart_proxy

      def update
        process_response @smart_proxy.update(smart_proxy_params)
      end

      api :DELETE, "/smart_proxies/:id/", N_("Delete a smart proxy")
      param :id, String, :required => true

      def destroy
        process_response @smart_proxy.destroy
      end

      api :PUT, "/smart_proxies/:id/refresh", N_("Refresh smart proxy features")
      param :id, String, :required => true

      def refresh
        process_response @smart_proxy.refresh.blank? && @smart_proxy.save && @smart_proxy.reload
      end

      api :POST, "/smart_proxies/:id/import_subnets", N_("Import subnets from Smart proxy")
      param :id, String, :required => true

      def import_subnets
        proxy = SmartProxy.find(params[:id])
        @subnets = Subnet::Ipv4.import(proxy)
        if @subnets.nil?
          render_error :custom_error, :status => :unprocessable_entity, :locals => {:message => 'Proxy does not have a DHCP feature'}
        elsif @subnets.empty?
          render :import_subnets, :status => :no_content
        else
          render :import_subnets, :status => :created
        end
      end

      def version
        render :version, :locals => {:success => true, :result => @smart_proxy.statuses[:version].version}
      rescue Foreman::Exception => e
        render_error :custom_error, :status => :unprocessable_entity, :locals => {:message => e.message}
      end

      def logs
        render :logs, :locals => {:success => true, :result => @smart_proxy.statuses[:logs].logs}
      rescue Foreman::Exception => e
        render_error :custom_error, :status => :unprocessable_entity, :locals => {:message => e.message}
      end

      private

      def action_permission
        case params[:action]
        when 'refresh'
          :edit
        when 'version', 'logs'
          :view
        else
          super
        end
      end
    end
  end
end