otwarchive-symphonyarchive/app/controllers/chapters_controller.rb

class ChaptersController < ApplicationController
  # only registered users and NOT admin should be able to create new chapters
  before_action :users_only, except: [:index, :show, :destroy, :confirm_delete]
  before_action :check_user_status, only: [:new, :create, :update, :update_positions]
  before_action :check_user_not_suspended, only: [:edit, :confirm_delete, :destroy]
  before_action :load_work
  # only authors of a work should be able to edit its chapters
  before_action :check_ownership, except: [:index, :show]
  before_action :check_visibility, only: [:show]
  before_action :load_chapter, only: [:show, :edit, :update, :preview, :post, :confirm_delete, :destroy]

  cache_sweeper :feed_sweeper

  # GET /work/:work_id/chapters
  # GET /work/:work_id/chapters.xml
  def index
    # this route is never used
    redirect_to work_path(params[:work_id])
  end

  # GET /work/:work_id/chapters/manage
  def manage
    @chapters = @work.chapters_in_order(include_content: false,
                                        include_drafts: true)
  end

  # GET /work/:work_id/chapters/:id
  # GET /work/:work_id/chapters/:id.xml
  def show
    @tag_groups = @work.tag_groups

    redirect_to url_for(controller: :chapters, action: :show, work_id: @work.id, id: params[:selected_id]) and return if params[:selected_id]

    @chapters = @work.chapters_in_order(
      include_content: false,
      include_drafts: (logged_in_as_admin? ||
                       @work.user_is_owner_or_invited?(current_user))
    )

    unless @chapters.include?(@chapter)
      access_denied
      return
    end
 
    chapter_position = @chapters.index(@chapter)
    if @chapters.length > 1
      @previous_chapter = @chapters[chapter_position - 1] unless chapter_position.zero?
      @next_chapter = @chapters[chapter_position + 1]
    end

    if @work.unrevealed?
      @page_title = t(".unrevealed") + t(".chapter_position", position: @chapter.position.to_s)
    else
      fandoms = @tag_groups["Fandom"]
      fandom = fandoms.empty? ? t(".unspecified_fandom") : fandoms[0].name
      title_fandom = fandoms.size > 3 ? t(".multifandom") : fandom
      author = @work.anonymous? ? t(".anonymous") : @work.pseuds.sort.collect(&:byline).join(", ")
      @page_title = get_page_title(title_fandom, author, @work.title + t(".chapter_position", position: @chapter.position.to_s))
    end

    if params[:view_adult]
      cookies[:view_adult] = "true"
    elsif @work.adult? && !see_adult?
      render "works/_adult", layout: "application" and return
    end

    @kudos = @work.kudos.with_user.includes(:user)

    if current_user.respond_to?(:subscriptions)
      @subscription = current_user.subscriptions.where(subscribable_id: @work.id,
                                                       subscribable_type: "Work").first ||
                      current_user.subscriptions.build(subscribable: @work)
    end
    # update the history.
    Reading.update_or_create(@work, current_user) if current_user

    respond_to do |format|
      format.html
      format.js
    end
  end

  # GET /work/:work_id/chapters/new
  # GET /work/:work_id/chapters/new.xml
  def new
    @chapter = @work.chapters.build(position: @work.number_of_chapters + 1)
  end

  # GET /work/:work_id/chapters/1/edit
  def edit
    return unless params["remove"] == "me"

    @chapter.creatorships.for_user(current_user).destroy_all
    if @work.chapters.any? { |c| current_user.is_author_of?(c) }
      flash[:notice] = ts("You have been removed as a creator from the chapter.")
      redirect_to @work
    else # remove from work if no longer co-creator on any chapter
      redirect_to edit_work_path(@work, remove: "me")
    end
  end

  def draft_flash_message(work)
    flash[:notice] = work.posted ? t("chapters.draft_flash.posted_work") : t("chapters.draft_flash.unposted_work_html", deletion_date: view_context.date_in_zone(work.created_at + 29.days)).html_safe
  end

  # POST /work/:work_id/chapters
  # POST /work/:work_id/chapters.xml
  def create
    if params[:cancel_button]
      redirect_back_or_default(root_path)
      return
    end

    @chapter = @work.chapters.build(chapter_params)
    @work.wip_length = params[:chapter][:wip_length]

    if params[:edit_button] || chapter_cannot_be_saved?
      render :new
    else # :post_without_preview or :preview
      @chapter.posted = true if params[:post_without_preview_button]
      @work.set_revised_at_by_chapter(@chapter)
      if @chapter.save && @work.save
        if @chapter.posted
          post_chapter
          redirect_to [@work, @chapter]
        else
          draft_flash_message(@work)
          redirect_to preview_work_chapter_path(@work, @chapter)
        end
      else
        render :new
      end
    end
  end

  # PUT /work/:work_id/chapters/1
  # PUT /work/:work_id/chapters/1.xml
  def update
    if params[:cancel_button]
      # Not quite working yet - should send the user back to wherever they were before they hit edit
      redirect_back_or_default(root_path)
      return
    end

    @chapter.attributes = chapter_params
    @work.wip_length = params[:chapter][:wip_length]

    if params[:edit_button] || chapter_cannot_be_saved?
      render :edit
    elsif params[:preview_button]
      @preview_mode = true
      if @chapter.posted?
        flash[:notice] = ts("This is a preview of what this chapter will look like after your changes have been applied. You should probably read the whole thing to check for problems before posting.")
      else
        draft_flash_message(@work)
      end
      render :preview
    else
      @chapter.posted = true if params[:post_button] || params[:post_without_preview_button]
      posted_changed = @chapter.posted_changed?
      @work.set_revised_at_by_chapter(@chapter)
      if @chapter.save && @work.save
        flash[:notice] = ts("Chapter was successfully #{posted_changed ? 'posted' : 'updated'}.")
        redirect_to work_chapter_path(@work, @chapter)
      else
        render :edit
      end
    end
  end

  def update_positions
    if params[:chapters]
      @work.reorder_list(params[:chapters])
      flash[:notice] = ts("Chapter order has been successfully updated.")
    elsif params[:chapter]
      params[:chapter].each_with_index do |id, position|
        @work.chapters.update(id, position: position + 1)
        (@chapters ||= []) << Chapter.find(id)
      end
    end
    respond_to do |format|
      format.html { redirect_to(@work) and return }
      format.js { head :ok }
    end
  end

  # GET /chapters/1/preview
  def preview
    @preview_mode = true
  end

  # POST /chapters/1/post
  def post
    if params[:cancel_button]
      redirect_to @work
    elsif params[:edit_button]
      redirect_to [:edit, @work, @chapter]
    else
      @chapter.posted = true
      @work.set_revised_at_by_chapter(@chapter)
      if @chapter.save && @work.save
        post_chapter
        redirect_to(@work)
      else
        render :preview
      end
    end
  end

  # GET /work/:work_id/chapters/1/confirm_delete
  def confirm_delete
  end

  # DELETE /work/:work_id/chapters/1
  # DELETE /work/:work_id/chapters/1.xml
  def destroy
    if @chapter.is_only_chapter? || @chapter.only_non_draft_chapter?
      flash[:error] = t(".only_chapter")
      redirect_to(edit_work_path(@work))
      return
    end

    was_draft = !@chapter.posted?
    if @chapter.destroy
      @work.minor_version = @work.minor_version + 1 unless was_draft
      @work.set_revised_at
      @work.save
      flash[:notice] = ts("The chapter #{was_draft ? 'draft ' : ''}was successfully deleted.")
    else
      flash[:error] = ts("Something went wrong. Please try again.")
    end
    redirect_to controller: "works", action: "show", id: @work
  end

  private

  # Check whether we should display :new or :edit instead of previewing or
  # saving the user's changes.
  def chapter_cannot_be_saved?
    # The chapter can only be saved if the work can be saved:
    if @work.invalid?
      @work.errors.full_messages.each do |message|
        @chapter.errors.add(:base, message)
      end
    end

    @chapter.errors.any? || @chapter.invalid?
  end

  # fetch work these chapters belong to from db
  def load_work
    @work = params[:work_id] ? Work.find_by(id: params[:work_id]) : Chapter.find_by(id: params[:id]).try(:work)
    if @work.blank?
      flash[:error] = ts("Sorry, we couldn't find the work you were looking for.")
      redirect_to root_path and return
    end
    @check_ownership_of = @work
    @check_visibility_of = @work
  end

  # Loads the specified chapter from the database. Redirects to the work if no
  # chapter is specified, or if the specified chapter doesn't exist.
  def load_chapter
    @chapter = @work.chapters.find_by(id: params[:id])

    return if @chapter

    flash[:error] = ts("Sorry, we couldn't find the chapter you were looking for.")
    redirect_to work_path(@work)
  end

  def post_chapter
    @work.update_attribute(:posted, true) unless @work.posted
    flash[:notice] = ts("Chapter has been posted!")
  end

  private

  def chapter_params
    params.require(:chapter).permit(:title, :position, :wip_length, :"published_at(3i)",
                                    :"published_at(2i)", :"published_at(1i)", :summary,
                                    :notes, :endnotes, :content, :published_at,
                                    author_attributes: [:byline, ids: [], coauthors: []])
  end
end