class BodyTrackersController < BodyTrackingPluginController
  before_action :find_project_by_project_id, only: [:index, :defaults]
  before_action :authorize

  def index
  end

  def defaults
    # Units
    available_units = @project.units.pluck(:shortname, :id).to_h
    defaults = Unit.where(project: nil).map do |u|
      u.attributes.except('id', 'project_id', 'created_at', 'updated_at')
    end
    defaults.delete_if { |u| available_units.has_key?(u['shortname']) }
    new_units = @project.units.create(defaults).map { |u| [u.shortname, u.id] }.to_h
    available_units.merge(new_units)

    flash[:notice] = "Loaded #{new_units.length > 0 ? new_units.length : "no" } new" \
      " #{'unit'.pluralize(new_units.length)}"

    # Quantities
    available_quantities = @project.quantities.map { |q| [[q.name, q.domain], q] }.to_h
    quantities_count = available_quantities.length
    defaults = Quantity.where(project: nil)
    Quantity.each_with_level(defaults) do |q, level|
      unless available_quantities.has_key?([q.name, q.domain])
        attrs = q.attributes.except('id', 'project_id', 'parent_id', 'lft', 'rgt',
                                    'created_at', 'updated_at')
        if q.parent
          attrs['parent_id'] = available_quantities[[q.parent.name, q.parent.domain]].id
        end
        if q.formula
          attrs['formula_attributes'] = q.formula.attributes
            .except('id', 'quantity_id', 'unit_id', 'created_at', 'updated_at')
          attrs['formula_attributes']['unit_id'] = available_units[q.formula.unit.shortname]
        end
        obj = @project.quantities.create(attrs)
        available_quantities[[q.name, q.domain]] = obj if obj.persisted?
      end
    end

    new_quantities_count = available_quantities.length - quantities_count
    flash[:notice] += ", #{new_quantities_count > 0 ? new_quantities_count : "no" } new" \
      " #{'quantity'.pluralize(new_quantities_count)}"

    ncv = @project.nutrients_column_view
    if ncv.quantities.count == 0
      ncv.quantities.append(@project.quantities.roots.first(6))
      ncv.save!
    end

    # Sources
    available_sources = @project.sources.pluck(:name, :id).to_h
    defaults = Source.where(project: nil).map do |s|
      s.attributes.except('id', 'project_id', 'created_at', 'updated_at')
    end
    defaults.delete_if { |s| available_sources.has_key?(s['name']) }
    new_sources = @project.sources.create(defaults).map { |s| [s.name, s.id] }.to_h
    available_sources.merge(new_sources)

    flash[:notice] += " and #{new_sources.length > 0 ? new_sources.length : "no" } new" \
      " #{'source'.pluralize(new_sources.length)}"

    redirect_to :back
  end
end