Mirrors/invoiceninja
1
0
Fork 0
mirror of https://github.com/invoiceninja/invoiceninja.git synced 2024-11-14 23:22:52 +01:00
Code Issues Releases Activity
2d091f3cb6
invoiceninja/app/Libraries/MultiDB.php

667 lines
16 KiB
PHP
Raw Blame History

<?php
/**
* Invoice Ninja (https://invoiceninja.com).
*
* @link https://github.com/invoiceninja/invoiceninja source repository
*
* @copyright Copyright (c) 2024. Invoice Ninja LLC (https://invoiceninja.com)
*
* @license https://www.elastic.co/licensing/elastic-license
*/
namespace App\Libraries;
use App\Models\User;
use App\Models\Client;
use App\Models\Account;
use App\Models\Company;
use App\Models\Document;
use App\Models\PaymentHash;
use Illuminate\Support\Str;
use App\Models\CompanyToken;
use App\Models\ClientContact;
use App\Models\VendorContact;
use App\DataProviders\SMSNumbers;
use Illuminate\Support\Facades\DB;
/**
* Class MultiDB.
*
* Caution!
*
* When we perform scans across databases,
* we need to remember that if we don't
* return a DB 'HIT' the DB connection will
* be set to the last DB in the chain,
*
* So for these cases, we need to reset the
* DB connection to the default connection.
*
* Even that may be problematic, and we
* may need to know the current DB connection
* so that we can fall back gracefully.
*/
class MultiDB
{
public const DB_PREFIX = 'db-ninja-';
public static $dbs = ['db-ninja-01', 'db-ninja-02'];
private static $protected_domains = [
'www',
'app',
'ninja',
'sentry',
'sentry2',
'staging',
'pdf',
'demo',
'docs',
'client_domain',
'custom_domain',
'preview',
'invoiceninja',
'cname',
'sandbox',
'stage',
'html',
'lb',
'shopify',
'beta',
'prometh',
'license',
'socket',
];
private static $protected_expense_mailboxes = [];
/**
* @return array
*/
public static function getDbs(): array
{
return self::$dbs;
}
public static function checkDomainAvailable($subdomain): bool
{
if (!config('ninja.db.multi_db_enabled')) {
return Company::whereSubdomain($subdomain)->count() == 0;
}
if (in_array($subdomain, self::$protected_domains)) {
return false;
}
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (Company::on($db)->whereSubdomain($subdomain)->exists()) {
self::setDb($current_db);
return false;
}
}
self::setDb($current_db);
return true;
}
public static function checkExpenseMailboxAvailable($expense_mailbox): bool
{
if (!config('ninja.db.multi_db_enabled')) {
return Company::where("expense_mailbox", $expense_mailbox)->withTrashed()->exists();
}
if (in_array($expense_mailbox, self::$protected_expense_mailboxes)) {
return false;
}
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (Company::on($db)->where("expense_mailbox", $expense_mailbox)->withTrashed()->exists()) {
self::setDb($current_db);
return false;
}
}
self::setDb($current_db);
return true;
}
public static function checkUserEmailExists($email): bool
{
if (!config('ninja.db.multi_db_enabled')) {
return User::where(['email' => $email])->withTrashed()->exists();
} // true >= 1 emails found / false -> == emails found
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (User::on($db)->where(['email' => $email])->withTrashed()->exists()) { // if user already exists, validation will fail
self::setDb($current_db);
return true;
}
}
self::setDb($current_db);
return false;
}
/**
* A user and company must co exists on the same database.
*
* This function will check that if a user exists on the system,
* the company is also located on the same database.
*
* If no user is found, then we also return true as this must be
* a new user request.
*
* @param string $email The user email
* @param string $company_key The company key
* @return bool True|False
*/
public static function checkUserAndCompanyCoExist($email, $company_key): bool
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (User::on($db)->where(['email' => $email])->withTrashed()->exists()) {
if (Company::on($db)->where(['company_key' => $company_key])->exists()) {
self::setDb($current_db);
return true;
} else {
self::setDb($current_db);
return false;
}
}
}
self::setDb($current_db);
return true;
}
/**
* @param array $data
* @return User|null
*/
public static function hasUser(array $data): ?User
{
if (!config('ninja.db.multi_db_enabled')) {
return User::where($data)->withTrashed()->first();
}
$current_db = config('database.default');
foreach (self::$dbs as $db) {
self::setDB($db);
if ($user = User::where($data)->withTrashed()->first()) {
return $user;
}
}
self::setDb($current_db);
return null;
}
/**
* @param string $email
* @return ClientContact|null
*/
public static function hasContact(string $email): ?ClientContact
{
if (!config('ninja.db.multi_db_enabled')) {
return ClientContact::where('email', $email)->withTrashed()->first();
}
$current_db = config('database.default');
foreach (self::$dbs as $db) {
$user = ClientContact::on($db)->where('email', $email)->withTrashed()->first();
if ($user) {
self::setDb($db);
return $user;
}
}
self::setDB($current_db);
return null;
}
/**
* @param array $search
* @return ClientContact|null
*/
public static function findContact(array $search): ?ClientContact
{
if (!config('ninja.db.multi_db_enabled')) {
return ClientContact::where($search)->first();
}
$current_db = config('database.default');
foreach (self::$dbs as $db) {
$user = ClientContact::on($db)->where($search)->first();
if ($user) {
self::setDb($db);
return $user;
}
}
self::setDB($current_db);
return null;
}
public static function contactFindAndSetDb($token): bool
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (ClientContact::on($db)->where('token', $token)->exists()) {
self::setDb($db);
return true;
}
}
self::setDB($current_db);
return false;
}
public static function userFindAndSetDb($email): bool
{
$current_db = config('database.default');
//multi-db active
foreach (self::$dbs as $db) {
if (User::on($db)->where('email', $email)->withTrashed()->exists()) {
self::setDb($db);
return true;
}
}
self::setDB($current_db);
return false;
}
public static function documentFindAndSetDb($hash): bool
{
$current_db = config('database.default');
//multi-db active
foreach (self::$dbs as $db) {
if (Document::on($db)->where('hash', $hash)->exists()) {
self::setDb($db);
return true;
}
}
self::setDB($current_db);
return false;
}
public static function findAndSetDb($token): bool
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (CompanyToken::on($db)->where('token', $token)->exists()) {
self::setDb($db);
return true;
}
}
self::setDB($current_db);
return false;
}
public static function findAndSetDbByCompanyKey($company_key): bool
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (Company::on($db)->where('company_key', $company_key)->exists()) {
self::setDb($db);
return true;
}
}
self::setDB($current_db);
return false;
}
public static function findAndSetDbByCompanyId($company_id): ?Company
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if ($company = Company::on($db)->where('id', $company_id)->first()) {
self::setDb($db);
return $company;
}
}
self::setDB($current_db);
return null;
}
public static function findAndSetDbByShopifyName($shopify_name): ?Company
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if ($company = Company::on($db)->with('tokens')->where('shopify_name', $shopify_name)->first()) {
self::setDb($db);
return $company;
}
}
self::setDB($current_db);
return null;
}
public static function findAndSetDbByAccountKey($account_key): bool
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (Account::on($db)->where('key', $account_key)->exists()) {
self::setDb($db);
return true;
}
}
self::setDB($current_db);
return false;
}
public static function findAndSetDbByInappTransactionId($transaction_id): bool
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (Account::on($db)->where('inapp_transaction_id', $transaction_id)->exists()) {
self::setDb($db);
return true;
}
}
self::setDB($current_db);
return false;
}
public static function findAndSetDbByContactKey($contact_key): bool
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (ClientContact::on($db)->where('contact_key', $contact_key)->exists()) {
self::setDb($db);
return true;
}
}
self::setDB($current_db);
return false;
}
public static function findAndSetDbByVendorContactKey($contact_key): bool
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (VendorContact::on($db)->where('contact_key', $contact_key)->exists()) {
self::setDb($db);
return true;
}
}
self::setDB($current_db);
return false;
}
public static function findAndSetDbByClientHash($client_hash): bool
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if (Client::on($db)->where('client_hash', $client_hash)->exists()) {
self::setDb($db);
return true;
}
}
self::setDB($current_db);
return false;
}
public static function findAndSetDbByClientId($client_id): ?Client
{
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if ($client = Client::on($db)->where('id', $client_id)->first()) {
self::setDb($db);
return $client;
}
}
self::setDB($current_db);
return null;
}
public static function findAndSetDbByDomain($query_array)
{
if (!config('ninja.db.multi_db_enabled')) {
return Company::where($query_array)->first();
}
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if ($company = Company::on($db)->where($query_array)->first()) {
self::setDb($db);
return $company;
}
}
self::setDB($current_db);
return false;
}
public static function findAndSetDbByExpenseMailbox($expense_mailbox)
{
if (!config('ninja.db.multi_db_enabled')) {
return Company::where("expense_mailbox", $expense_mailbox)->first();
}
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if ($company = Company::on($db)->where("expense_mailbox", $expense_mailbox)->first()) {
self::setDb($db);
return $company;
}
}
self::setDB($current_db);
return false;
}
public static function findAndSetByPaymentHash(string $hash)
{
if (!config('ninja.db.multi_db_enabled')) {
return PaymentHash::with('fee_invoice')->where('hash', $hash)->first();
}
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if ($payment_hash = PaymentHash::on($db)->where('hash', $hash)->first()) {
self::setDb($db);
return $payment_hash;
}
}
self::setDB($current_db);
return false;
}
public static function findAndSetDbByInvitation($entity, $invitation_key)
{
$class = 'App\Models\\' . ucfirst(Str::camel($entity)) . 'Invitation';
$current_db = config('database.default');
foreach (self::$dbs as $db) {
if ($invite = $class::on($db)->where('key', $invitation_key)->exists()) {
self::setDb($db);
return true;
}
}
self::setDB($current_db);
return false;
}
/**
* @param string $phone
* @return bool
*/
public static function hasPhoneNumber(string $phone): bool
{
if (!config('ninja.db.multi_db_enabled')) {
return Account::where('account_sms_verification_number', $phone)->where('account_sms_verified', true)->exists();
}
$current_db = config('database.default');
if (SMSNumbers::hasNumber($phone)) {
return true;
}
foreach (self::$dbs as $db) {
self::setDB($db);
if ($exists = Account::where('account_sms_verification_number', $phone)->where('account_sms_verified', true)->exists()) {
self::setDb($current_db);
return true;
}
}
self::setDb($current_db);
return false;
}
public static function randomSubdomainGenerator(): string
{
$current_db = config('database.default');
do {
$length = 10;
$string = '';
$vowels = ['a', 'e', 'i', 'o', 'u', 'y'];
$consonants = [
'b',
'c',
'd',
'f',
'g',
'h',
'j',
'k',
'l',
'm',
'n',
'p',
'r',
's',
't',
'v',
'w',
'x',
'y',
'z',
];
$max = $length / 2;
for ($i = 1; $i <= $max; $i++) {
$string .= $consonants[rand(0, 19)];
$string .= $vowels[rand(0, 5)];
}
} while (!self::checkDomainAvailable($string));
self::setDb($current_db);
return $string;
}
/**
* @param $database
* @return void
*/
public static function setDB(string $database): void
{
/* This will set the database connection for the request */
config(['database.default' => $database]);
}
public static function setDefaultDatabase()
{
config(['database.default' => config('ninja.db.default')]);
}
}
View Git Blame Copy Permalink