app/controllers/pulitzer/versions_controller.rb in pulitzer-0.2.2 vs app/controllers/pulitzer/versions_controller.rb in pulitzer-0.3.0

@@ -1,17 +1,26 @@
 class Pulitzer::VersionsController < Pulitzer::ApplicationController
   before_action :set_version
 
+  def edit
+    if @version.preview?
+      route                       = "pulitzer_preview_#{@post.post_type.name.parameterize('_')}_path"
+      @preview_path               = main_app.public_send(route, @post.slug) if main_app.respond_to?(route)
+    end
+    render_ajax locals: {version: @version}
+  end
+
   def update
-    Pulitzer::UpdateVersionStatus.new(@version,@status).call
+    processing_version = Pulitzer::UpdateVersionStatus.new(@version,@status).call
 
-    redirect_to post_content_elements_path(@post), notice: "Post #{@status}"
+    render json: {html: render_to_string(partial: '/pulitzer/versions/edit', locals: {version: processing_version}),
+                  flash_message: "The new version of #{@post.title} has been activated."}
   end
 
 private
   def set_version
     @version  = Pulitzer::Version.find params[:id]
-    @post     = @version.post
-    @status   = params[:status].to_sym
+    @status = params[:status]
+    @post = @version.post
   end
 
 end