invoiceninja/app/Http/Controllers/ClientController.php

<?php namespace App\Http\Controllers;

use Auth;
use Datatable;
use Utils;
use View;
use URL;
use Validator;
use Input;
use Session;
use Redirect;

use App\Models\Activity;
use App\Models\Client;
use App\Models\Contact;
use App\Models\Invoice;
use App\Models\Size;
use App\Models\PaymentTerm;
use App\Models\Industry;
use App\Models\Currency;
use App\Models\Country;

use App\Ninja\Repositories\ClientRepository;

class ClientController extends BaseController
{
    protected $clientRepo;

    public function __construct(ClientRepository $clientRepo)
    {
        parent::__construct();

        $this->clientRepo = $clientRepo;
    }

    /**
     * Display a listing of the resource.
     *
     * @return Response
     */
    public function index()
    {
        return View::make('list', array(
            'entityType' => ENTITY_CLIENT,
            'title' => trans('texts.clients'),
            'columns' => Utils::trans(['checkbox', 'client', 'contact', 'email', 'date_created', 'last_login', 'balance', 'action']),
        ));
    }

    public function getDatatable()
    {
        $clients = $this->clientRepo->find(Input::get('sSearch'));

        return Datatable::query($clients)
            ->addColumn('checkbox', function ($model) { return '<input type="checkbox" name="ids[]" value="'.$model->public_id.'" '.Utils::getEntityRowClass($model).'>'; })
            ->addColumn('name', function ($model) { return link_to('clients/'.$model->public_id, $model->name); })
            ->addColumn('first_name', function ($model) { return link_to('clients/'.$model->public_id, $model->first_name.' '.$model->last_name); })
            ->addColumn('email', function ($model) { return link_to('clients/'.$model->public_id, $model->email); })
            ->addColumn('created_at', function ($model) { return Utils::timestampToDateString(strtotime($model->created_at)); })
            ->addColumn('last_login', function ($model) { return Utils::timestampToDateString(strtotime($model->last_login)); })
            ->addColumn('balance', function ($model) { return Utils::formatMoney($model->balance, $model->currency_id); })
            ->addColumn('dropdown', function ($model) {
            if ($model->is_deleted) {
                return '<div style="height:38px"/>';
            }

                $str = '<div class="btn-group tr-action" style="visibility:hidden;">
  							<button type="button" class="btn btn-xs btn-default dropdown-toggle" data-toggle="dropdown">
    							'.trans('texts.select').' <span class="caret"></span>
  							</button>
  							<ul class="dropdown-menu" role="menu">';

                    if (!$model->deleted_at || $model->deleted_at == '0000-00-00') {
                        $str .= '<li><a href="'.URL::to('clients/'.$model->public_id.'/edit').'">'.trans('texts.edit_client').'</a></li>
						    <li class="divider"></li>
						    <li><a href="'.URL::to('invoices/create/'.$model->public_id).'">'.trans('texts.new_invoice').'</a></li>
						    <li><a href="'.URL::to('payments/create/'.$model->public_id).'">'.trans('texts.new_payment').'</a></li>
						    <li><a href="'.URL::to('credits/create/'.$model->public_id).'">'.trans('texts.new_credit').'</a></li>
						    <li class="divider"></li>
						    <li><a href="javascript:archiveEntity('.$model->public_id.')">'.trans('texts.archive_client').'</a></li>';
                    } else {
                        $str .= '<li><a href="javascript:restoreEntity('.$model->public_id.')">'.trans('texts.restore_client').'</a></li>';
                    }

                        return $str.'<li><a href="javascript:deleteEntity('.$model->public_id.')">'.trans('texts.delete_client').'</a></li></ul>
							</div>';
            })
            ->make();
    }

    /**
     * Store a newly created resource in storage.
     *
     * @return Response
     */
    public function store()
    {
        return $this->save();
    }

    /**
     * Display the specified resource.
     *
     * @param  int      $id
     * @return Response
     */
    public function show($publicId)
    {
        $client = Client::withTrashed()->scope($publicId)->with('contacts', 'size', 'industry')->firstOrFail();
        Utils::trackViewed($client->getDisplayName(), ENTITY_CLIENT);

        $actionLinks = [
            ['label' => trans('texts.create_invoice'), 'url' => URL::to('invoices/create/'.$client->public_id)],
            ['label' => trans('texts.enter_payment'), 'url' => URL::to('payments/create/'.$client->public_id)],
            ['label' => trans('texts.enter_credit'), 'url' => URL::to('credits/create/'.$client->public_id)],
        ];

        if (Utils::isPro()) {
            array_unshift($actionLinks, ['label' => trans('texts.create_quote'), 'url' => URL::to('quotes/create/'.$client->public_id)]);
        }

        $data = array(
            'actionLinks' => $actionLinks,
            'showBreadcrumbs' => false,
            'client' => $client,
            'credit' => $client->getTotalCredit(),
            'title' => trans('texts.view_client'),
            'hasRecurringInvoices' => Invoice::scope()->where('is_recurring', '=', true)->whereClientId($client->id)->count() > 0,
            'gatewayLink' => $client->getGatewayLink(),
        );

        return View::make('clients.show', $data);
    }

    /**
     * Show the form for creating a new resource.
     *
     * @return Response
     */
    public function create()
    {
        if (Client::scope()->count() > Auth::user()->getMaxNumClients()) {
            return View::make('error', ['hideHeader' => true, 'error' => "Sorry, you've exceeded the limit of ".Auth::user()->getMaxNumClients()." clients"]);
        }

        $data = [
            'client' => null,
            'method' => 'POST',
            'url' => 'clients',
            'title' => trans('texts.new_client'),
        ];

        $data = array_merge($data, self::getViewModel());

        return View::make('clients.edit', $data);
    }

    /**
     * Show the form for editing the specified resource.
     *
     * @param  int      $id
     * @return Response
     */
    public function edit($publicId)
    {
        $client = Client::scope($publicId)->with('contacts')->firstOrFail();
        $data = [
            'client' => $client,
            'method' => 'PUT',
            'url' => 'clients/'.$publicId,
            'title' => trans('texts.edit_client'),
        ];

        $data = array_merge($data, self::getViewModel());

        return View::make('clients.edit', $data);
    }

    private static function getViewModel()
    {
        /* Remember function no longer works
        return [
            'sizes' => Size::remember(DEFAULT_QUERY_CACHE)->orderBy('id')->get(),
            'paymentTerms' => PaymentTerm::remember(DEFAULT_QUERY_CACHE)->orderBy('num_days')->get(['name', 'num_days']),
            'industries' => Industry::remember(DEFAULT_QUERY_CACHE)->orderBy('name')->get(),
            'currencies' => Currency::remember(DEFAULT_QUERY_CACHE)->orderBy('name')->get(),
            'countries' => Country::remember(DEFAULT_QUERY_CACHE)->orderBy('name')->get(),
            'customLabel1' => Auth::user()->account->custom_client_label1,
            'customLabel2' => Auth::user()->account->custom_client_label2,
        ];
        */
        return [
            'sizes' => Size::orderBy('id')->get(),
            'paymentTerms' => PaymentTerm::orderBy('num_days')->get(['name', 'num_days']),
            'industries' => Industry::orderBy('name')->get(),
            'currencies' => Currency::orderBy('name')->get(),
            'countries' => Country::orderBy('name')->get(),
            'customLabel1' => Auth::user()->account->custom_client_label1,
            'customLabel2' => Auth::user()->account->custom_client_label2,
        ];
    }

    /**
     * Update the specified resource in storage.
     *
     * @param  int      $id
     * @return Response
     */
    public function update($publicId)
    {
        return $this->save($publicId);
    }

    private function save($publicId = null)
    {
        $rules = array(
            'email' => 'required',
        );
        $validator = Validator::make(Input::all(), $rules);

        if ($validator->fails()) {
            $url = $publicId ? 'clients/'.$publicId.'/edit' : 'clients/create';

            return Redirect::to($url)
                ->withErrors($validator)
                ->withInput(Input::except('password'));
        } else {
            if ($publicId) {
                $client = Client::scope($publicId)->firstOrFail();
            } else {
                $client = Client::createNew();
            }

            $client->name = trim(Input::get('name'));
            $client->id_number = trim(Input::get('id_number'));
            $client->vat_number = trim(Input::get('vat_number'));
            $client->work_phone = trim(Input::get('work_phone'));
            $client->custom_value1 = trim(Input::get('custom_value1'));
            $client->custom_value2 = trim(Input::get('custom_value2'));
            $client->address1 = trim(Input::get('address1'));
            $client->address2 = trim(Input::get('address2'));
            $client->city = trim(Input::get('city'));
            $client->state = trim(Input::get('state'));
            $client->postal_code = trim(Input::get('postal_code'));
            $client->country_id = Input::get('country_id') ?: null;
            $client->private_notes = trim(Input::get('private_notes'));
            $client->size_id = Input::get('size_id') ?: null;
            $client->industry_id = Input::get('industry_id') ?: null;
            $client->currency_id = Input::get('currency_id') ?: null;
            $client->payment_terms = Input::get('payment_terms') ?: 0;
            $client->website = trim(Input::get('website'));

            $client->save();

            $data = json_decode(Input::get('data'));
            $contactIds = [];
            $isPrimary = true;

            foreach ($data->contacts as $contact) {
                if (isset($contact->public_id) && $contact->public_id) {
                    $record = Contact::scope($contact->public_id)->firstOrFail();
                } else {
                    $record = Contact::createNew();
                }

                $record->email = trim(strtolower($contact->email));
                $record->first_name = trim($contact->first_name);
                $record->last_name = trim($contact->last_name);
                $record->phone = trim($contact->phone);
                $record->is_primary = $isPrimary;
                $isPrimary = false;

                $client->contacts()->save($record);
                $contactIds[] = $record->public_id;
            }

            foreach ($client->contacts as $contact) {
                if (!in_array($contact->public_id, $contactIds)) {
                    $contact->delete();
                }
            }

            if ($publicId) {
                Session::flash('message', trans('texts.updated_client'));
            } else {
                Activity::createClient($client);
                Session::flash('message', trans('texts.created_client'));
            }

            return Redirect::to('clients/'.$client->public_id);
        }
    }

    public function bulk()
    {
        $action = Input::get('action');
        $ids = Input::get('id') ? Input::get('id') : Input::get('ids');
        $count = $this->clientRepo->bulk($ids, $action);

        $message = Utils::pluralize($action.'d_client', $count);
        Session::flash('message', $message);

        if ($action == 'restore' && $count == 1) {
            return Redirect::to('clients/'.$ids[0]);
        } else {
            return Redirect::to('clients');
        }
    }
}