relations_controller.rb in avo-0.5.0.beta1

- old
+ new
@@ -1,34 +1,103 @@
-require_dependency 'avo/application_controller'
+require_dependency 'avo/base_controller'
 
 module Avo
-  class RelationsController < ApplicationController
-    before_action :authorize_user
+  class RelationsController < BaseController
+    before_action :set_model, only: [:show, :index, :new, :create, :destroy]
+    before_action :set_related_resource_name
+    before_action :set_related_resource
+    before_action :hydrate_related_resource
+    before_action :set_related_model, only: [:show]
+    before_action :set_attachment_class
+    before_action :set_attachment_resource
+    before_action :set_attachment_model, only: [:create, :destroy]
+    before_action :set_reflection, only: [:index]
 
-    def attach
-      resource.send(params[:attachment_name]) << attachment_model
+    def index
+      @parent_resource = @resource.dup
+      @resource = @related_resource
+      @parent_model = @parent_resource.model_class.find(params[:id])
+      @parent_resource.hydrate(model: @parent_model)
+      @query = @parent_model.public_send(params[:related_name])
 
-      render json: {
-        success: true,
-        message: I18n.t('avo.attachment_class_attached', attachment_class: attachment_class),
-      }
+      super
     end
 
-    def detach
-      resource.send(params[:attachment_name]).delete attachment_model
+    def show
+      @reflection = params[:related_name]
+      @resource, @model = @related_resource, @related_model
 
-      render json: {
-        success: true,
-        message: I18n.t('avo.attachment_class_detached', attachment_class: attachment_class),
-      }
+      super
     end
 
+    def new
+      query = @authorization.apply_policy @attachment_class
+
+      @options = query.all.map do |model|
+        {
+          value: model.id,
+          label: model.send(@attachment_resource.title)
+        }
+      end
+    end
+
+    def create
+      if reflection_class == 'HasManyReflection'
+        @model.send("#{params[:related_name]}") << @attachment_model
+      else
+        @model.send("#{params[:related_name]}=", @attachment_model)
+      end
+
+      respond_to do |format|
+        if @model.save
+          format.html { redirect_to resource_path(@model), notice: t('avo.attachment_class_attached', attachment_class: @attachment_class) }
+          format.json { render :show, status: :created, location: resource_path(@model) }
+        else
+          format.html { render :new }
+          format.json { render json: @model.errors, status: :unprocessable_entity }
+        end
+      end
+    end
+
+    def destroy
+      if reflection_class == 'HasManyReflection'
+        @model.send("#{params[:related_name]}").delete @attachment_model
+      else
+        @model.send("#{params[:related_name]}=", nil)
+      end
+
+      respond_to do |format|
+        format.html { redirect_to params[:referrer] || resources_path(@model), notice: t('avo.attachment_class_detached', attachment_class: @attachment_class) }
+      end
+    end
+
     private
-      def attachment_class
-        App.get_model_class_by_name params[:attachment_name].pluralize 1
+      def set_attachment_class
+        @attachment_class = @model._reflections[params[:related_name].to_s].klass
       end
 
-      def attachment_model
-        resource._reflections[params[:attachment_name].to_s].klass.find params[:attachment_id]
+      def set_attachment_resource
+        @attachment_resource = App.get_resource_by_model_name @attachment_class
+      end
+
+      def set_attachment_model
+        @attachment_model = @model._reflections[params[:related_name].to_s].klass.find attachment_id
+      end
+
+      def set_reflection
+        @reflection = @model._reflections[params[:related_name].to_s]
+      end
+
+      def attachment_id
+        params[:related_id] or params.require(:fields).permit(:related_id)[:related_id]
+      end
+
+      def reflection_class
+        reflection = @model._reflections[params[:related_name]]
+
+        klass = @model._reflections[params[:related_name]].class.name.demodulize.to_s
+        klass = @model._reflections[params[:related_name]].through_reflection.class.name.demodulize.to_s if klass == 'ThroughReflection'
+
+        klass
       end
   end
 end