invoiceninja/app/Repositories/InvoiceRepository.php

<?php
/**
 * Invoice Ninja (https://invoiceninja.com)
 *
 * @link https://github.com/invoiceninja/invoiceninja source repository
 *
 * @copyright Copyright (c) 2020. Invoice Ninja LLC (https://invoiceninja.com)
 *
 * @license https://opensource.org/licenses/AAL
 */

namespace App\Repositories;

use App\Factory\InvoiceInvitationFactory;

use App\Jobs\Company\UpdateCompanyLedgerWithInvoice;
use App\Jobs\Product\UpdateOrCreateProduct;

use App\Models\ClientContact;
use App\Models\Invoice;
use App\Models\InvoiceInvitation;
use App\Utils\Traits\MakesHash;

/**
 * InvoiceRepository
 */

class InvoiceRepository extends BaseRepository {
	use MakesHash;

	/**
	 * Gets the class name.
	 *
	 * @return     string  The class name.
	 */
	public function getClassName() {
		return Invoice::class ;
	}

	/**
	 * Saves the invoices
	 *
	 * @param      array.                                        $data     The invoice data
	 * @param      InvoiceSum|\App\Models\Invoice               $invoice  The invoice
	 *
	 * @return     Invoice|InvoiceSum|\App\Models\Invoice|null  Returns the invoice object
	 */
	public function save($data, Invoice $invoice):?Invoice {

		/* Always carry forward the initial invoice amount this is important for tracking client balance changes later......*/
		$starting_amount = $invoice->amount;

		$invoice->fill($data);

		$invoice->save();

		if (isset($data['client_contacts'])) {
			foreach ($data['client_contacts'] as $contact) {
				if ($contact['send_email'] == 1 && is_string($contact['id'])) {
					$client_contact = ClientContact::find($this->decodePrimaryKey($contact['id']));
					$client_contact->send_email = true;
					$client_contact->save();
				}
			}
		}

		if (isset($data['invitations'])) {
			$invitations = collect($data['invitations']);

			/* Get array of Keys which have been removed from the invitations array and soft delete each invitation */
			collect($invoice->invitations->pluck('key'))->diff($invitations->pluck('key'))->each(function ($invitation) {
					InvoiceInvitation::whereRaw("BINARY `key`= ?", [$invitation])->delete();
				});

			foreach ($data['invitations'] as $invitation) {
				$inv = false;

				if (array_key_exists('key', $invitation)) {
					// $inv = InvoiceInvitation::whereKey($invitation['key'])->first();
					$inv = InvoiceInvitation::whereRaw("BINARY `key`= ?", [$invitation['key']])->first();
				}

				if (!$inv) {

					if (isset($invitation['id'])) {
						unset($invitation['id']);
					}

					$new_invitation = InvoiceInvitationFactory::create($invoice->company_id, $invoice->user_id);
					//$new_invitation->fill($invitation);
					$new_invitation->invoice_id        = $invoice->id;
					$new_invitation->client_contact_id = $this->decodePrimaryKey($invitation['client_contact_id']);
					$new_invitation->save();

				}
			}
		}

		/* If no invitations have been created, this is our fail safe to maintain state*/
		if ($invoice->invitations->count() == 0) {
			$invoice->service()->createInvitations();
		}

		$invoice = $invoice->calc()->getInvoice();

		$invoice->save();

		$finished_amount = $invoice->amount;

		/**/
		if (($finished_amount != $starting_amount) && ($invoice->status_id != Invoice::STATUS_DRAFT)) {
			UpdateCompanyLedgerWithInvoice::dispatchNow($invoice, ($finished_amount-$starting_amount), $invoice->company);
		}

		$invoice = $invoice->service()->applyNumber()->save();

		if ($invoice->company->update_products !== false) {
			UpdateOrCreateProduct::dispatch($invoice->line_items, $invoice, $invoice->company);
		}

		return $invoice->fresh();
	}

	/**
	 * Mark the invoice as sent.
	 *
	 * @param      \App\Models\Invoice               $invoice  The invoice
	 *
	 * @return     Invoice|\App\Models\Invoice|null  Return the invoice object
	 */
	public function markSent(Invoice $invoice):?Invoice {
		return $invoice->service()->markSent()->save();
	}

	public function getInvitationByKey($key)
	{
		return InvoiceInvitation::whereRaw("BINARY `key`= ?", [$key])->first();
	}
}