invoiceninja/app/controllers/AccountController.php

<?php

use ninja\repositories\AccountRepository;
use ninja\mailers\UserMailer;
use ninja\mailers\ContactMailer;

class AccountController extends \BaseController {

	protected $accountRepo;
	protected $userMailer;
	protected $contactMailer;

	public function __construct(AccountRepository $accountRepo, UserMailer $userMailer, ContactMailer $contactMailer)
	{
		parent::__construct();

		$this->accountRepo = $accountRepo;
		$this->userMailer = $userMailer;
		$this->contactMailer = $contactMailer;
	}	

	public function getStarted()
	{	
		if (Auth::check())
		{
			return Redirect::to('invoices/create');	
		}

		$user = false;	
		$guestKey = Input::get('guest_key');

		if ($guestKey) 
		{
			$user = User::where('password', '=', $guestKey)->first();

			if ($user && $user->registered)
			{
				return Redirect::to('/');				
			}
		}

		if (!$user)
		{
			$account = $this->accountRepo->create();
			$user = $account->users()->first();
			
			Session::forget(RECENTLY_VIEWED);
		}

		Auth::login($user, true);
		Event::fire('user.login');		

		return Redirect::to('invoices/create');
	}

	public function enableProPlan()
	{		
		$invoice = $this->accountRepo->enableProPlan();

		if ($invoice)
		{
			$this->contactMailer->sendInvoice($invoice);
		}

		return RESULT_SUCCESS;		
	}

	public function setTrashVisible($entityType, $visible)
	{
		Session::put("show_trash:{$entityType}", $visible == 'true');
		return Redirect::to("{$entityType}s");
	}

	public function getSearchData()
	{
		$data = $this->accountRepo->getSearchData();

		return Response::json($data);
	}

	public function showSection($section = ACCOUNT_DETAILS, $subSection = false)  
	{
		if ($section == ACCOUNT_DETAILS)
		{			
			$data = [
				'account' => Account::with('users')->findOrFail(Auth::user()->account_id),
				'countries' => Country::remember(DEFAULT_QUERY_CACHE)->orderBy('name')->get(),
				'sizes' => Size::remember(DEFAULT_QUERY_CACHE)->orderBy('id')->get(),
				'industries' => Industry::remember(DEFAULT_QUERY_CACHE)->orderBy('name')->get(),				
				'timezones' => Timezone::remember(DEFAULT_QUERY_CACHE)->orderBy('location')->get(),
				'dateFormats' => DateFormat::remember(DEFAULT_QUERY_CACHE)->get(),
				'datetimeFormats' => DatetimeFormat::remember(DEFAULT_QUERY_CACHE)->get(),
				'currencies' => Currency::remember(DEFAULT_QUERY_CACHE)->orderBy('name')->get(),
				'languages' => Language::remember(DEFAULT_QUERY_CACHE)->orderBy('name')->get(),
			];

			return View::make('accounts.details', $data);
		}
		else if ($section == ACCOUNT_PAYMENTS)
		{
			$account = Account::with('account_gateways')->findOrFail(Auth::user()->account_id);
			$accountGateway = null;
			$config = null;
			$configFields = null;
      $selectedCards = 0;

			if (count($account->account_gateways) > 0)
			{
				$accountGateway = $account->account_gateways[0];
				$config = $accountGateway->config;
                $selectedCards = $accountGateway->accepted_credit_cards;
                
				$configFields = json_decode($config);
                
				foreach($configFields as $configField => $value)
				{
					$configFields->$configField = str_repeat('*', strlen($value));
				}
			}
			
			$recommendedGateways = Gateway::remember(DEFAULT_QUERY_CACHE)
					->where('recommended', '=', '1')
					->orderBy('sort_order')
					->get();
			$recommendedGatewayArray = array();
			
			foreach($recommendedGateways as $recommendedGateway)
			{
				$arrayItem = array(
					'value' => $recommendedGateway->id,
					'other' => 'false',
					'data-imageUrl' => asset($recommendedGateway->getLogoUrl()),
					'data-siteUrl' => $recommendedGateway->site_url
				);
				$recommendedGatewayArray[$recommendedGateway->name] = $arrayItem;
			}
            
      $creditCardsArray = unserialize(CREDIT_CARDS);
      $creditCards = [];
			foreach($creditCardsArray as $card => $name)
			{
        if($selectedCards > 0 && ($selectedCards & $card) == $card)
            $creditCards[$name['text']] = ['value' => $card, 'data-imageUrl' => asset($name['card']), 'checked' => 'checked'];
        else
            $creditCards[$name['text']] = ['value' => $card, 'data-imageUrl' => asset($name['card'])];
			}

			$otherItem = array(
				'value' => 1000000,
				'other' => 'true',
				'data-imageUrl' => '',
				'data-siteUrl' => ''
			);
			$recommendedGatewayArray['Other Options'] = $otherItem;
      
			$data = [
				'account' => $account,
				'accountGateway' => $accountGateway,
				'config' => $configFields,
				'gateways' => Gateway::remember(DEFAULT_QUERY_CACHE)
					->orderBy('name')
					->get(),
				'dropdownGateways' => Gateway::remember(DEFAULT_QUERY_CACHE)
					->where('recommended', '=', '0')
					->orderBy('name')
					->get(),
				'recommendedGateways' => $recommendedGatewayArray,
        'creditCardTypes' => $creditCards, 
			];
			
			foreach ($data['gateways'] as $gateway)
			{
				$paymentLibrary =  $gateway->paymentlibrary;

				$gateway->fields = $gateway->getFields();	
	
				if ($accountGateway && $accountGateway->gateway_id == $gateway->id)
				{
					$accountGateway->fields = $gateway->fields;						
				}
			}	

			return View::make('accounts.payments', $data);
		}
		else if ($section == ACCOUNT_NOTIFICATIONS)
		{
			$data = [
				'account' => Account::with('users')->findOrFail(Auth::user()->account_id),
			];

			return View::make('accounts.notifications', $data);
		}
		else if ($section == ACCOUNT_IMPORT_EXPORT)
		{
			return View::make('accounts.import_export');	
		}	
		else if ($section == ACCOUNT_ADVANCED_SETTINGS)
		{
			$data = [
				'account' => Auth::user()->account,
				'feature' => $subSection
			];

			return View::make("accounts.{$subSection}", $data);	
		}
		else if ($section == ACCOUNT_PRODUCTS)
		{
			$data = [
				'account' => Auth::user()->account
			];

			return View::make('accounts.products', $data);		
		}
	}

	public function doSection($section = ACCOUNT_DETAILS, $subSection = false)
	{
		if ($section == ACCOUNT_DETAILS)
		{
			return AccountController::saveDetails();
		}
		else if ($section == ACCOUNT_PAYMENTS)
		{
			return AccountController::savePayments();
		}
		else if ($section == ACCOUNT_IMPORT_EXPORT)
		{
			return AccountController::importFile();
		}
		else if ($section == ACCOUNT_MAP)
		{
			return AccountController::mapFile();
		}
		else if ($section == ACCOUNT_NOTIFICATIONS)
		{
			return AccountController::saveNotifications();
		}		
		else if ($section == ACCOUNT_EXPORT)
		{
			return AccountController::export();
		}		
		else if ($section == ACCOUNT_ADVANCED_SETTINGS)
		{
			if ($subSection == ACCOUNT_CUSTOM_FIELDS) 
			{
				return AccountController::saveCustomFields();
			} 
			else if ($subSection == ACCOUNT_INVOICE_DESIGN)
			{
				return AccountController::saveInvoiceDesign();
			}
		}
		else if ($section == ACCOUNT_PRODUCTS)
		{
			return AccountController::saveProducts();
		}
	}

	private function saveProducts()
	{
		$account = Auth::user()->account;

		$account->fill_products = Input::get('fill_products') ? true : false;
		$account->update_products = Input::get('update_products') ? true : false;
		$account->save();

		Session::flash('message', trans('texts.updated_settings'));
		return Redirect::to('company/products');		
	}

	private function saveCustomFields()
	{
		if (Auth::user()->account->isPro())
		{
			$account = Auth::user()->account;
			$account->custom_label1 = trim(Input::get('custom_label1'));
			$account->custom_value1 = trim(Input::get('custom_value1'));
			$account->custom_label2 = trim(Input::get('custom_label2'));
			$account->custom_value2 = trim(Input::get('custom_value2'));
			$account->custom_client_label1 = trim(Input::get('custom_client_label1'));
			$account->custom_client_label2 = trim(Input::get('custom_client_label2'));		
			$account->custom_invoice_label1 = trim(Input::get('custom_invoice_label1'));
			$account->custom_invoice_label2 = trim(Input::get('custom_invoice_label2'));
			$account->custom_invoice_taxes1 = Input::get('custom_invoice_taxes1') ? true : false;
			$account->custom_invoice_taxes2 = Input::get('custom_invoice_taxes2') ? true : false;			
			$account->save();

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

		return Redirect::to('company/advanced_settings/custom_fields');		
	}

	private function saveInvoiceDesign()
	{
		if (Auth::user()->account->isPro())
		{
			$account = Auth::user()->account;
			$account->hide_quantity = Input::get('hide_quantity') ? true : false;
			$account->hide_paid_to_date = Input::get('hide_paid_to_date') ? true : false;
			$account->primary_color = Input::get('primary_color');// ? Input::get('primary_color') : null;
			$account->secondary_color = Input::get('secondary_color');// ? Input::get('secondary_color') : null;
			$account->save();

			Session::flash('message', trans('texts.updated_settings'));
		}
		
		return Redirect::to('company/advanced_settings/invoice_design');		
	}

	private function export()
	{
		$output = fopen('php://output','w') or Utils::fatalError();
		header('Content-Type:application/csv'); 
		header('Content-Disposition:attachment;filename=export.csv');
		
		$clients = Client::scope()->get();
		AccountController::exportData($output, $clients->toArray());

		$contacts = Contact::scope()->get();
		AccountController::exportData($output, $contacts->toArray());

		$invoices = Invoice::scope()->get();
		AccountController::exportData($output, $invoices->toArray());

		$invoiceItems = InvoiceItem::scope()->get();
		AccountController::exportData($output, $invoiceItems->toArray());

		$payments = Payment::scope()->get();
		AccountController::exportData($output, $payments->toArray());

		$credits = Credit::scope()->get();
		AccountController::exportData($output, $credits->toArray());

		fclose($output);
		exit;
	}

	private function exportData($output, $data)
	{
		if (count($data) > 0)
		{
			fputcsv($output, array_keys($data[0]));
		}

		foreach($data as $record) 
		{
		    fputcsv($output, $record);
		}

		fwrite($output, "\n");
	}

	private function importFile()
	{
		$data = Session::get('data');
		Session::forget('data');

		$map = Input::get('map');
		$count = 0;
		$hasHeaders = Input::get('header_checkbox');
		
		$countries = Country::remember(DEFAULT_QUERY_CACHE)->get();
		$countryMap = [];

		foreach ($countries as $country) 
		{
			$countryMap[strtolower($country->name)] = $country->id;
		}		

		foreach ($data as $row)
		{
			if ($hasHeaders)
			{
				$hasHeaders = false;
				continue;
			}

			$client = Client::createNew();		
			$contact = Contact::createNew();
			$contact->is_primary = true;
			$contact->send_invoice = true;
			$count++;

			foreach ($row as $index => $value)
			{
				$field = $map[$index];
				$value = trim($value);

				if ($field == Client::$fieldName && !$client->name)
				{
					$client->name = $value;
				}			
				else if ($field == Client::$fieldPhone && !$client->work_phone)
				{
					$client->work_phone = $value;
				}
				else if ($field == Client::$fieldAddress1 && !$client->address1)
				{
					$client->address1 = $value;
				}
				else if ($field == Client::$fieldAddress2 && !$client->address2)
				{
					$client->address2 = $value;
				}
				else if ($field == Client::$fieldCity && !$client->city)
				{
					$client->city = $value;
				}
				else if ($field == Client::$fieldState && !$client->state)
				{
					$client->state = $value;
				}
				else if ($field == Client::$fieldPostalCode && !$client->postal_code)
				{
					$client->postal_code = $value;
				}
				else if ($field == Client::$fieldCountry && !$client->country_id)
				{
					$value = strtolower($value);
					$client->country_id = isset($countryMap[$value]) ? $countryMap[$value] : null;
				}
				else if ($field == Client::$fieldNotes && !$client->private_notes)
				{
					$client->private_notes = $value;
				}
				else if ($field == Contact::$fieldFirstName && !$contact->first_name)
				{
					$contact->first_name = $value;
				}
				else if ($field == Contact::$fieldLastName && !$contact->last_name)
				{
					$contact->last_name = $value;
				}
				else if ($field == Contact::$fieldPhone && !$contact->phone)
				{
					$contact->phone = $value;
				}
				else if ($field == Contact::$fieldEmail && !$contact->email)
				{
					$contact->email = strtolower($value);
				}				
			}

			$client->save();
			$client->contacts()->save($contact);		
			Activity::createClient($client, false);
		}

		$message = Utils::pluralize('created_client', $count);
		Session::flash('message', $message);
		return Redirect::to('clients');
	}

	private function mapFile()
	{		
		$file = Input::file('file');

		if ($file == null)
		{
			Session::flash('error', trans('texts.select_file'));
			return Redirect::to('company/import_export');			
		}

		$name = $file->getRealPath();

		require_once(app_path().'/includes/parsecsv.lib.php');
		$csv = new parseCSV();
		$csv->heading = false;
		$csv->auto($name);
		
		if (count($csv->data) + Client::scope()->count() > Auth::user()->getMaxNumClients())
		{
      $message = trans('texts.limit_clients', ['count' => Auth::user()->getMaxNumClients()]);
			Session::flash('error', $message);
			return Redirect::to('company/import_export');
		}

		Session::put('data', $csv->data);

		$headers = false;
		$hasHeaders = false;
		$mapped = array();
		$columns = array('',
			Client::$fieldName,
			Client::$fieldPhone,
			Client::$fieldAddress1,
			Client::$fieldAddress2,
			Client::$fieldCity,
			Client::$fieldState,
			Client::$fieldPostalCode,
			Client::$fieldCountry,
			Client::$fieldNotes,
			Contact::$fieldFirstName,
			Contact::$fieldLastName,
			Contact::$fieldPhone,
			Contact::$fieldEmail
		);

		if (count($csv->data) > 0) 
		{
			$headers = $csv->data[0];
			foreach ($headers as $title) 
			{
				if (strpos(strtolower($title),'name') > 0)
				{
					$hasHeaders = true;
					break;
				}
			}

			for ($i=0; $i<count($headers); $i++)
			{
				$title = strtolower($headers[$i]);
				$mapped[$i] = '';

				if ($hasHeaders)
				{
					$map = array(
						'first' => Contact::$fieldFirstName,
						'last' => Contact::$fieldLastName,
						'email' => Contact::$fieldEmail,
						'mobile' => Contact::$fieldPhone,
						'phone' => Client::$fieldPhone,
						'name|organization' => Client::$fieldName,
						'street|address|address1' => Client::$fieldAddress1,	
						'street2|address2' => Client::$fieldAddress2,						
						'city' => Client::$fieldCity,
						'state|province' => Client::$fieldState,
						'zip|postal|code' => Client::$fieldPostalCode,
						'country' => Client::$fieldCountry,
						'note' => Client::$fieldNotes,
					);

					foreach ($map as $search => $column)
					{
						foreach(explode("|", $search) as $string)
						{
							if (strpos($title, 'sec') === 0)
							{
								continue;
							}

							if (strpos($title, $string) !== false)
							{
								$mapped[$i] = $column;
								break(2);
							}
						}
					}
				}
			}
		}

		$data = array(
			'data' => $csv->data, 
			'headers' => $headers,
			'hasHeaders' => $hasHeaders,
			'columns' => $columns,
			'mapped' => $mapped
		);

		return View::make('accounts.import_map', $data);
	}

	private function saveNotifications()
	{
		$account = Auth::user()->account;
		$account->invoice_terms = Input::get('invoice_terms');
		$account->email_footer = Input::get('email_footer');
		$account->save();

		$user = Auth::user();
		$user->notify_sent = Input::get('notify_sent');
		$user->notify_viewed = Input::get('notify_viewed');
		$user->notify_paid = Input::get('notify_paid');
		$user->save();
		
		Session::flash('message', trans('texts.updated_settings'));
		return Redirect::to('company/notifications');
	}

	private function savePayments()
	{  
		$rules = array();
		$recommendedId = Input::get('recommendedGateway_id');

		if ($gatewayId = $recommendedId == 1000000 ? Input::get('gateway_id') : $recommendedId) 
		{
			$gateway = Gateway::findOrFail($gatewayId);
			
			$paymentLibrary = $gateway->paymentlibrary;
			
			$fields = $gateway->getFields();
			
			foreach ($fields as $field => $details)
			{
				if (!in_array($field, ['testMode', 'developerMode', 'headerImageUrl', 'solutionType', 'landingPage', 'brandName']))
				{
					if(strtolower($gateway->name) == 'beanstream')
					{
						if(in_array($field, ['merchant_id', 'passCode']))
						{
							$rules[$gateway->id.'_'.$field] = 'required';
						}
					} 
					else 
					{
						$rules[$gateway->id.'_'.$field] = 'required';
					}
				}				
			}			
		}
        
    $creditcards = Input::get('creditCardTypes');
		$validator = Validator::make(Input::all(), $rules);

		if ($validator->fails()) 
		{
			return Redirect::to('company/payments')
				->withErrors($validator)
				->withInput();
		} 
		else 
		{
			$account = Account::findOrFail(Auth::user()->account_id);						

			if ($gatewayId) 
			{
				$accountGateway = AccountGateway::createNew();
				$accountGateway->gateway_id = $gatewayId;

				$config = new stdClass;
				foreach ($fields as $field => $details)
				{
					$value = trim(Input::get($gateway->id.'_'.$field));

					if ($value && $value === str_repeat('*', strlen($value)))
					{
						Session::flash('error', trans('validation.notmasked'));
						return Redirect::to('company/payments');			
					}

					$config->$field = $value;
				}
                
		        $cardCount = 0;
		        if ($creditcards) 
		        {
			        foreach($creditcards as $card => $value)
			        {
			            $cardCount += intval($value);
			        }			
		        }
			
				$accountGateway->config = json_encode($config);
				$accountGateway->accepted_credit_cards = $cardCount;
	
				$account->account_gateways()->delete();
				$account->account_gateways()->save($accountGateway);

				Session::flash('message', trans('texts.updated_settings'));
			}
			else
			{
				Session::flash('error', trans('validation.required', ['attribute' => 'gateway']));
			}
			
			return Redirect::to('company/payments');
		}				
	}

	private function saveDetails()
	{
		$rules = array(
			'name' => 'required',
			'email' => 'email|required|unique:users,email,' . Auth::user()->id . ',id'
		);

		$validator = Validator::make(Input::all(), $rules);

		if ($validator->fails()) 
		{
			return Redirect::to('company/details')
				->withErrors($validator)
				->withInput();
		} 
		else 
		{
			$account = Auth::user()->account;
			$account->name = trim(Input::get('name'));
			$account->work_email = trim(Input::get('work_email'));
			$account->work_phone = trim(Input::get('work_phone'));
			$account->address1 = trim(Input::get('address1'));
			$account->address2 = trim(Input::get('address2'));
			$account->city = trim(Input::get('city'));
			$account->state = trim(Input::get('state'));
			$account->postal_code = trim(Input::get('postal_code'));
			$account->country_id = Input::get('country_id') ? Input::get('country_id') : null;			
			$account->size_id = Input::get('size_id') ? Input::get('size_id') : null;
			$account->industry_id = Input::get('industry_id') ? Input::get('industry_id') : null;
			$account->timezone_id = Input::get('timezone_id') ? Input::get('timezone_id') : null;
			$account->date_format_id = Input::get('date_format_id') ? Input::get('date_format_id') : null;
			$account->datetime_format_id = Input::get('datetime_format_id') ? Input::get('datetime_format_id') : null;
			$account->currency_id = Input::get('currency_id') ? Input::get('currency_id') : 1; // US Dollar
			$account->language_id = Input::get('language_id') ? Input::get('language_id') : 1; // English
			$account->save();

			$user = Auth::user();
			$user->first_name = trim(Input::get('first_name'));
			$user->last_name = trim(Input::get('last_name'));
			$user->username = trim(Input::get('email'));
			$user->email = trim(strtolower(Input::get('email')));
			$user->phone = trim(Input::get('phone'));				
			$user->save();

			/* Logo image file */
			if ($file = Input::file('logo'))
			{
				$path = Input::file('logo')->getRealPath();
				File::delete('logo/' . $account->account_key . '.jpg');				
								
				$image = Image::make($path)->resize(200, 120, true, false);
				Image::canvas($image->width, $image->height, '#FFFFFF')->insert($image)->save($account->getLogoPath());
			}

			Event::fire('user.refresh');

			Session::flash('message', trans('texts.updated_settings'));
			return Redirect::to('company/details');
		}
	}

	public function removeLogo() {

		File::delete('logo/' . Auth::user()->account->account_key . '.jpg');

		Session::flash('message', trans('texts.removed_logo'));
		return Redirect::to('company/details');		
	}

	public function checkEmail()
	{		
		$email = User::withTrashed()->where('email', '=', Input::get('email'))->where('id', '<>', Auth::user()->id)->first();

		if ($email) 
		{
			return "taken";
		} 
		else 
		{
			return "available";
		}
	}

	public function submitSignup()
	{
		$rules = array(
			'new_first_name' => 'required',
			'new_last_name' => 'required',
			'new_password' => 'required|min:6',
			'new_email' => 'email|required|unique:users,email,' . Auth::user()->id . ',id'
		);

		$validator = Validator::make(Input::all(), $rules);

		if ($validator->fails()) 
		{
			return '';
		} 

		$user = Auth::user();
		$user->first_name = trim(Input::get('new_first_name'));
		$user->last_name = trim(Input::get('new_last_name'));
		$user->email = trim(strtolower(Input::get('new_email')));
		$user->username = $user->email;
		$user->password = trim(Input::get('new_password'));
		$user->password_confirmation = trim(Input::get('new_password'));
		$user->registered = true;
		$user->amend();

		$this->userMailer->sendConfirmation($user);

		$activities = Activity::scope()->get();
		foreach ($activities as $activity) 
		{
			$activity->message = str_replace('Guest', $user->getFullName(), $activity->message);
			$activity->save();
		}

		if (Input::get('go_pro') == 'true')
		{
			Session::set(REQUESTED_PRO_PLAN, true);
		}

		Session::set(SESSION_COUNTER, -1);

		return "{$user->first_name} {$user->last_name}";
	}
}