invoiceninja/app/Http/Controllers/VendorController.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 Cache;

use App\Models\Activity;
use App\Models\Vendor;
use App\Models\Account;
use App\Models\VendorContact;
use App\Models\Size;
use App\Models\PaymentTerm;
use App\Models\Industry;
use App\Models\Currency;
use App\Models\Country;
use App\Ninja\Repositories\VendorRepository;
use App\Services\VendorService;

use App\Http\Requests\CreateVendorRequest;
use App\Http\Requests\UpdateVendorRequest;
// vendor
class VendorController extends BaseController
{
    protected $vendorService;
    protected $vendorRepo;
    protected $model = 'App\Models\Vendor';

    public function __construct(VendorRepository $vendorRepo, VendorService $vendorService)
    {
        //parent::__construct();

        $this->vendorRepo = $vendorRepo;
        $this->vendorService = $vendorService;


    }

    /**
     * Display a listing of the resource.
     *
     * @return Response
     */
    public function index()
    {
        return View::make('list', array(
            'entityType' => 'vendor',
            'title' => trans('texts.vendors'),
            'sortCol' => '4',
            'columns' => Utils::trans([
              'checkbox',
              'vendor',
              'city',
              'phone',
              'email',
              'date_created',
              ''
            ]),
        ));
    }

    public function getDatatable()
    {
        return $this->vendorService->getDatatable(Input::get('sSearch'));
    }

    /**
     * Store a newly created resource in storage.
     *
     * @return Response
     */
    public function store(CreateVendorRequest $request)
    {
        $data = $request->input();
        
        if(!$this->checkUpdatePermission($data, $response)){
            return $response;
        }
                
        $vendor = $this->vendorService->save($data);

        Session::flash('message', trans('texts.created_vendor'));

        return redirect()->to($vendor->getRoute());
    }

    /**
     * Display the specified resource.
     *
     * @param  int      $id
     * @return Response
     */
    public function show($publicId)
    {
        $vendor = Vendor::withTrashed()->scope($publicId)->with('vendorcontacts', 'size', 'industry')->firstOrFail();
        
        if(!$this->checkViewPermission($vendor, $response)){
            return $response;
        }
        
        Utils::trackViewed($vendor->getDisplayName(), 'vendor');

        $actionLinks = [
            ['label' => trans('texts.new_vendor'), 'url' => '/vendors/create/' . $vendor->public_id]
        ];

        $data = array(
            'actionLinks'           => $actionLinks,
            'showBreadcrumbs'       => false,
            'vendor'                => $vendor,
            'totalexpense'          => $vendor->getTotalExpense(),
            'title'                 => trans('texts.view_vendor'),
            'hasRecurringInvoices'  => false,
            'hasQuotes'             => false,
            'hasTasks'          => false,
        );

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

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

        $data = [
            'vendor' => null,
            'method' => 'POST',
            'url' => 'vendors',
            'title' => trans('texts.new_vendor'),
        ];

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

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

    /**
     * Show the form for editing the specified resource.
     *
     * @param  int      $id
     * @return Response
     */
    public function edit($publicId)
    {
        $vendor = Vendor::scope($publicId)->with('vendorcontacts')->firstOrFail();
        
        if(!$this->checkEditPermission($vendor, $response)){
            return $response;
        }
        
        $data = [
            'vendor' => $vendor,
            'method' => 'PUT',
            'url' => 'vendors/'.$publicId,
            'title' => trans('texts.edit_vendor'),
        ];

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

        if (Auth::user()->account->isNinjaAccount()) {
            if ($account = Account::whereId($vendor->public_id)->first()) {
                $data['proPlanPaid'] = $account['pro_plan_paid'];
            }
        }

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

    private static function getViewModel()
    {
        return [
            'data' => Input::old('data'),
            'account' => Auth::user()->account,
            'currencies' => Cache::get('currencies'),
            'countries' => Cache::get('countries'),
        ];
    }

    /**
     * Update the specified resource in storage.
     *
     * @param  int      $id
     * @return Response
     */
    public function update(UpdateVendorRequest $request)
    {
        $data = $request->input();
        
        if(!$this->checkUpdatePermission($data, $response)){
            return $response;
        }
                
        $vendor = $this->vendorService->save($data);

        Session::flash('message', trans('texts.updated_vendor'));

        return redirect()->to($vendor->getRoute());
    }

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

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

        if ($action == 'restore' && $count == 1) {
            return Redirect::to('vendors/' . Utils::getFirst($ids));
        } else {
            return Redirect::to('vendors');
        }
    }
}