publiclab/plots2

class FeaturesController < ApplicationController
  before_action :require_user, except: [:embed]

  def index
    @title = "Features"
    @features = Node.where(type: 'feature')
    @sorted_features = Node.sort_features(@features, params[:sort]).paginate(page: params[:page])
  end

  def embed
    @node = Node.find_by(title: params[:id])
    render layout: false
  end

  def new
    @node = Node.new
    @revision = Revision.new
    unless current_user.admin?
      flash[:warning] = 'Only admins may edit features.'
      redirect_to '/features'
    end
  end

  def edit
    if !current_user.admin?
      flash[:warning] = 'Only admins may edit features.'
      redirect_to '/features'
    else
      @node = Node.find params[:id]
    end
  end

  def create
    if !current_user.admin?
      flash[:warning] = 'Only admins may edit features.'
      redirect_to "/features?_=#{Time.now.to_i}"
    else
      @node = Node.new(uid:   current_user.id,
                       title: params[:title],
                       type:  'feature')
      if @node.valid?
        saved = true
        @revision = false
        ActiveRecord::Base.transaction do
          @node.save!
          @revision = @node.new_revision(uid:   current_user.id,
                                         title: params[:title],
                                         body:  params[:body])
          if @revision.valid?
            @revision.save!
            @node.vid = @revision.vid
            @node.save!
          else
            saved = false
            @node.destroy
          end
        end
      end
      if saved
        flash[:notice] = 'Feature saved.'
        redirect_to "/features?_=#{Time.now.to_i}"
      else
        render template: 'features/new'
      end
    end
  end

  def update
    if !current_user.admin?
      flash[:warning] = 'Only admins may edit features.'
      redirect_to "/features?_=#{Time.now.to_i}"
    else
      @node = Node.find(params[:id])
      @revision = @node.new_revision(uid: current_user.uid)
      @revision.title = params[:title] || @node.latest.title
      @revision.body = params[:body] if params[:body]
      if @revision.valid?
        ActiveRecord::Base.transaction do
          @revision.save
          @node.vid = @revision.vid
          @node.title = @revision.title
          @node.save
        end
        ActionController::Base.new.expire_fragment("feature_#{params[:title]}")
        flash[:notice] = 'Edits saved and cache cleared.'
        redirect_to "/features?_=#{Time.now.to_i}"
      else
        flash[:error] = 'Your edit could not be saved.'
        render action: 'edit'
      end
    end
  end
end