class DirectoryController < ApplicationController
  before_filter :login_required, :except => [:show, :list, :index, :html]
  
  def index
    list
    render :action => 'list'
  end
  
  def html
    @companies = Company.find :all, :order => "city"
    render :action => 'html', :layout => false
  end
  
  def create
    @company = Company.new params[:company]
    @company.account = session[:user]    
    
    if @company.save
      flash[:notice] = 'Company was successfully created.'
      redirect_to :action => 'show', :id => @company.id
    else
      render :action => 'new'
    end
  end
  
  def edit
    @company = Company.find(params[:id])
    security_check @company
  end
  
  def freshen
    @company = Company.find(params[:id])
    security_check @company
    
    if @company.update_attributes(params[:company])
      flash[:notice] = 'Company was successfully updated.'
      redirect_to :action => 'show', :id => @company
    else
      render :action => 'edit'
    end
  end
  
  def list
    @directory_pages, @companies =
      paginate :companies, :per_page => 20
  end
  
  def new
    if session[:user].company
      redirect_to :action => 'edit', :id => session[:user].company.id
    else
      @company = Company.new
    end    
  end
  
  def show
    @company = Company.find(params[:id])
  end
  
  def remove
    @company = Company.find(params[:id])
    security_check @company
    @company.destroy
    
    redirect_to :action => 'list'
  end
end