agents_controller.rb 7.4KB

    class AgentsController < ApplicationController include DotHelper include ActionView::Helpers::TextHelper include SortableTable def index set_table_sort sorts: %w[name last_check_at last_event_at last_receive_at], default: { name: :asc } @agents = current_user.agents.preload(:scenarios, :controllers).reorder(table_sort).page(params[:page]) respond_to do |format| format.html format.json { render json: @agents } end end def handle_details_post @agent = current_user.agents.find(params[:id]) if @agent.respond_to?(:handle_details_post) render :json => @agent.handle_details_post(params) || {} else @agent.error "#handle_details_post called on an instance of #{@agent.class} that does not define it." head 500 end end def run @agent = current_user.agents.find(params[:id]) Agent.async_check(@agent.id) respond_to do |format| format.html { redirect_back "Agent run queued for '#{@agent.name}'" } format.json { head :ok } end end def dry_run attrs = params[:agent] || {} if agent = current_user.agents.find_by(id: params[:id]) # POST /agents/:id/dry_run if attrs.present? type = agent.type agent = Agent.build_for_type(type, current_user, attrs) end else # POST /agents/dry_run type = attrs.delete(:type) agent = Agent.build_for_type(type, current_user, attrs) end agent.name ||= '(Untitled)' if agent.valid? if event_payload = params[:event] dummy_agent = Agent.build_for_type('ManualEventAgent', current_user, name: 'Dry-Runner') dummy_agent.readonly! event = dummy_agent.events.build(user: current_user, payload: event_payload) end results = agent.dry_run!(event) render json: { log: results[:log], events: Utils.pretty_print(results[:events], false), memory: Utils.pretty_print(results[:memory] || {}, false), } else render json: { log: [ "#{pluralize(agent.errors.count, "error")} prohibited this Agent from being saved:", *agent.errors.full_messages ].join("\n- "), events: '', memory: '', } end end def type_details @agent = Agent.build_for_type(params[:type], current_user, {}) initialize_presenter render json: { can_be_scheduled: @agent.can_be_scheduled?, default_schedule: @agent.default_schedule, can_receive_events: @agent.can_receive_events?, can_create_events: @agent.can_create_events?, can_control_other_agents: @agent.can_control_other_agents?, can_dry_run: @agent.can_dry_run?, options: @agent.default_options, description_html: @agent.html_description, oauthable: render_to_string(partial: 'oauth_dropdown', locals: { agent: @agent }), form_options: render_to_string(partial: 'options', locals: { agent: @agent }) } end def event_descriptions html = current_user.agents.find(params[:ids].split(",")).group_by(&:type).map { |type, agents| agents.map(&:html_event_description).uniq.map { |desc| "<p><strong>#{type}</strong><br />" + desc + "</p>" } }.flatten.join() render :json => { :description_html => html } end def remove_events @agent = current_user.agents.find(params[:id]) @agent.events.delete_all respond_to do |format| format.html { redirect_back "All emitted events removed for '#{@agent.name}'" } format.json { head :ok } end end def propagate details = Agent.receive! # Eventually this should probably be scoped to the current_user. respond_to do |format| format.html { redirect_back "Queued propagation calls for #{details[:event_count]} event(s) on #{details[:agent_count]} agent(s)" } format.json { head :ok } end end def destroy_memory @agent = current_user.agents.find(params[:id]) @agent.update!(memory: {}) respond_to do |format| format.html { redirect_back "Memory erased for '#{@agent.name}'" } format.json { head :ok } end end def show @agent = current_user.agents.find(params[:id]) respond_to do |format| format.html format.json { render json: @agent } end end def new agents = current_user.agents if id = params[:id] @agent = agents.build_clone(agents.find(id)) else @agent = agents.build end @agent.scenario_ids = [params[:scenario_id]] if params[:scenario_id] && current_user.scenarios.find_by(id: params[:scenario_id]) initialize_presenter respond_to do |format| format.html format.json { render json: @agent } end end def edit @agent = current_user.agents.find(params[:id]) initialize_presenter end def create build_agent respond_to do |format| if @agent.save format.html { redirect_back "'#{@agent.name}' was successfully created.", return: agents_path } format.json { render json: @agent, status: :ok, location: agent_path(@agent) } else initialize_presenter format.html { render action: "new" } format.json { render json: @agent.errors, status: :unprocessable_entity } end end end def update @agent = current_user.agents.find(params[:id]) respond_to do |format| if @agent.update_attributes(params[:agent]) format.html { redirect_back "'#{@agent.name}' was successfully updated.", return: agents_path } format.json { render json: @agent, status: :ok, location: agent_path(@agent) } else initialize_presenter format.html { render action: "edit" } format.json { render json: @agent.errors, status: :unprocessable_entity } end end end def leave_scenario @agent = current_user.agents.find(params[:id]) @scenario = current_user.scenarios.find(params[:scenario_id]) @agent.scenarios.destroy(@scenario) respond_to do |format| format.html { redirect_back "'#{@agent.name}' removed from '#{@scenario.name}'" } format.json { head :no_content } end end def destroy @agent = current_user.agents.find(params[:id]) @agent.destroy respond_to do |format| format.html { redirect_back "'#{@agent.name}' deleted" } format.json { head :no_content } end end def validate build_agent if @agent.validate_option(params[:attribute]) render text: 'ok' else render text: 'error', status: 403 end end def complete build_agent render json: @agent.complete_option(params[:attribute]) end protected # Sanitize params[:return] to prevent open redirect attacks, a common security issue. def redirect_back(message, options = {}) case ret = params[:return] || options[:return] when "show" if @agent && !@agent.destroyed? path = agent_path(@agent) else path = agents_path end when /\A#{Regexp::escape scenarios_path}\/\d+\z/, agents_path path = ret end if path redirect_to path, notice: message else super agents_path, notice: message end end def build_agent @agent = Agent.build_for_type(params[:agent].delete(:type), current_user, params[:agent]) end def initialize_presenter if @agent.present? && @agent.is_form_configurable? @agent = FormConfigurableAgentPresenter.new(@agent, view_context) end end end