mirror of
https://github.com/invoiceninja/invoiceninja.git
synced 2024-09-20 08:21:34 +02:00
380 lines
14 KiB
PHP
380 lines
14 KiB
PHP
<?php
|
|
|
|
namespace App\Console\Commands;
|
|
|
|
use Illuminate\Console\Command;
|
|
use DB;
|
|
use Mail;
|
|
use Exception;
|
|
use App\Models\DbServer;
|
|
use App\Models\LookupCompany;
|
|
use App\Models\LookupAccount;
|
|
use App\Models\LookupUser;
|
|
use App\Models\LookupContact;
|
|
use App\Models\LookupAccountToken;
|
|
use App\Models\LookupInvitation;
|
|
|
|
class InitLookup extends Command
|
|
{
|
|
/**
|
|
* The name and signature of the console command.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $signature = 'ninja:init-lookup {--truncate=} {--subdomain} {--validate=} {--update=} {--company_id=} {--page_size=100} {--database=db-ninja-1}';
|
|
|
|
/**
|
|
* The console command description.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $description = 'Initialize lookup tables';
|
|
|
|
protected $log = '';
|
|
protected $isValid = true;
|
|
|
|
/**
|
|
* Create a new command instance.
|
|
*
|
|
* @return void
|
|
*/
|
|
public function __construct()
|
|
{
|
|
parent::__construct();
|
|
}
|
|
|
|
/**
|
|
* Execute the console command.
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function handle()
|
|
{
|
|
$this->logMessage('Running InitLookup...');
|
|
|
|
config(['database.default' => DB_NINJA_LOOKUP]);
|
|
|
|
$database = $this->option('database');
|
|
$dbServer = DbServer::whereName($database)->first();
|
|
|
|
if ($this->option('subdomain')) {
|
|
$this->logMessage('Updating subdomains...');
|
|
$this->popuplateSubdomains();
|
|
} else if ($this->option('truncate')) {
|
|
$this->logMessage('Truncating data...');
|
|
$this->truncateTables();
|
|
} else {
|
|
config(['database.default' => $this->option('database')]);
|
|
|
|
$count = DB::table('companies')
|
|
->where('id', '>=', $this->option('company_id') ?: 1)
|
|
->count();
|
|
|
|
for ($i=0; $i<$count; $i += (int) $this->option('page_size')) {
|
|
$this->initCompanies($dbServer->id, $i);
|
|
}
|
|
}
|
|
|
|
$this->logMessage('Results: ' . ($this->isValid ? RESULT_SUCCESS : RESULT_FAILURE));
|
|
|
|
if ($this->option('validate')) {
|
|
if ($errorEmail = env('ERROR_EMAIL')) {
|
|
Mail::raw($this->log, function ($message) use ($errorEmail, $database) {
|
|
$message->to($errorEmail)
|
|
->from(CONTACT_EMAIL)
|
|
->subject("Check-Lookups [{$database}]: " . strtoupper($this->isValid ? RESULT_SUCCESS : RESULT_FAILURE));
|
|
});
|
|
} elseif (! $this->isValid) {
|
|
throw new Exception('Check lookups failed!!');
|
|
}
|
|
}
|
|
}
|
|
|
|
private function popuplateSubdomains()
|
|
{
|
|
$data = [];
|
|
|
|
config(['database.default' => $this->option('database')]);
|
|
|
|
$accounts = DB::table('accounts')
|
|
->orderBy('id')
|
|
->where('subdomain', '!=', '')
|
|
->get(['account_key', 'subdomain']);
|
|
foreach ($accounts as $account) {
|
|
$data[$account->account_key] = $account->subdomain;
|
|
}
|
|
|
|
config(['database.default' => DB_NINJA_LOOKUP]);
|
|
|
|
$validate = $this->option('validate');
|
|
$update = $this->option('update');
|
|
|
|
foreach ($data as $accountKey => $subdomain) {
|
|
LookupAccount::whereAccountKey($accountKey)->update(['subdomain' => $subdomain]);
|
|
}
|
|
}
|
|
|
|
private function initCompanies($dbServerId, $offset = 0)
|
|
{
|
|
$data = [];
|
|
|
|
config(['database.default' => $this->option('database')]);
|
|
|
|
$companies = DB::table('companies')
|
|
->offset($offset)
|
|
->limit((int) $this->option('page_size'))
|
|
->orderBy('id')
|
|
->where('id', '>=', $this->option('company_id') ?: 1)
|
|
->get(['id']);
|
|
foreach ($companies as $company) {
|
|
$data[$company->id] = $this->parseCompany($company->id);
|
|
}
|
|
|
|
config(['database.default' => DB_NINJA_LOOKUP]);
|
|
|
|
$validate = $this->option('validate');
|
|
$update = $this->option('update');
|
|
|
|
foreach ($data as $companyId => $company) {
|
|
|
|
$lookupCompany = false;
|
|
if ($validate || $update) {
|
|
$lookupCompany = LookupCompany::whereDbServerId($dbServerId)->whereCompanyId($companyId)->first();
|
|
}
|
|
if ($validate && ! $lookupCompany) {
|
|
$this->logError("LookupCompany - dbServerId: {$dbServerId}, companyId: {$companyId} | Not found!");
|
|
continue;
|
|
}
|
|
if (! $lookupCompany) {
|
|
$lookupCompany = LookupCompany::create([
|
|
'db_server_id' => $dbServerId,
|
|
'company_id' => $companyId,
|
|
]);
|
|
}
|
|
|
|
foreach ($company as $accountKey => $account) {
|
|
|
|
$lookupAccount = false;
|
|
if ($validate || $update) {
|
|
$lookupAccount = LookupAccount::whereLookupCompanyId($lookupCompany->id)->whereAccountKey($accountKey)->first();
|
|
}
|
|
if ($validate && ! $lookupAccount) {
|
|
$this->logError("LookupAccount - lookupCompanyId: {$lookupCompany->id}, accountKey {$accountKey} | Not found!");
|
|
continue;
|
|
}
|
|
if (! $lookupAccount) {
|
|
$lookupAccount = LookupAccount::create([
|
|
'lookup_company_id' => $lookupCompany->id,
|
|
'account_key' => $accountKey
|
|
]);
|
|
}
|
|
|
|
foreach ($account['users'] as $user) {
|
|
$lookupUser = false;
|
|
if ($validate || $update) {
|
|
$lookupUser = LookupUser::whereLookupAccountId($lookupAccount->id)->whereUserId($user['user_id'])->first();
|
|
}
|
|
if ($validate) {
|
|
if (! $lookupUser) {
|
|
$this->logError("LookupUser - lookupAccountId: {$lookupAccount->id}, userId: {$user['user_id']} | Not found!");
|
|
continue;
|
|
} elseif ($user['email'] != $lookupUser->email || $user['oauth_user_key'] != $lookupUser->oauth_user_key || $user['referral_code'] != $lookupUser->referral_code) {
|
|
$this->logError("LookupUser - lookupAccountId: {$lookupAccount->id}, userId: {$user['user_id']} | Out of date!");
|
|
continue;
|
|
}
|
|
}
|
|
if ($update && $lookupUser) {
|
|
if ($user['email'] != $lookupUser->email || $user['oauth_user_key'] != $lookupUser->oauth_user_key || $user['referral_code'] != $lookupUser->referral_code) {
|
|
$lookupUser->email = $user['email'];
|
|
$lookupUser->oauth_user_key = $user['oauth_user_key'];
|
|
$lookupUser->referral_code = $user['referral_code'];
|
|
$lookupUser->save();
|
|
}
|
|
} elseif (! $lookupUser) {
|
|
LookupUser::create([
|
|
'lookup_account_id' => $lookupAccount->id,
|
|
'email' => $user['email'] ?: null,
|
|
'user_id' => $user['user_id'],
|
|
'oauth_user_key' => $user['oauth_user_key'],
|
|
'referral_code' => $user['referral_code'],
|
|
]);
|
|
}
|
|
}
|
|
|
|
foreach ($account['contacts'] as $contact) {
|
|
$lookupContact = false;
|
|
if ($validate || $update) {
|
|
$lookupContact = LookupContact::whereLookupAccountId($lookupAccount->id)->whereContactKey($contact['contact_key'])->first();
|
|
}
|
|
if ($validate && ! $lookupContact) {
|
|
$this->logError("LookupContact - lookupAccountId: {$lookupAccount->id}, contactKey: {$contact['contact_key']} | Not found!");
|
|
continue;
|
|
}
|
|
if (! $lookupContact) {
|
|
LookupContact::create([
|
|
'lookup_account_id' => $lookupAccount->id,
|
|
'contact_key' => $contact['contact_key'],
|
|
]);
|
|
}
|
|
}
|
|
|
|
foreach ($account['invitations'] as $invitation) {
|
|
$lookupInvitation = false;
|
|
if ($validate || $update) {
|
|
$lookupInvitation = LookupInvitation::whereLookupAccountId($lookupAccount->id)->whereInvitationKey($invitation['invitation_key'])->first();
|
|
}
|
|
if ($validate) {
|
|
if (! $lookupInvitation) {
|
|
$this->logError("LookupInvitation - lookupAccountId: {$lookupAccount->id}, invitationKey: {$invitation['invitation_key']} | Not found!");
|
|
continue;
|
|
} elseif ($invitation['message_id'] && $lookupInvitation->message_id != $invitation['message_id']) {
|
|
$this->logError("LookupInvitation - lookupAccountId: {$lookupAccount->id}, invitationKey: {$invitation['invitation_key']} | Not the same!");
|
|
continue;
|
|
}
|
|
}
|
|
if ($update && $lookupInvitation) {
|
|
if ($invitation['message_id'] && $lookupInvitation->message_id != $invitation['message_id']) {
|
|
$lookupInvitation->message_id = $invitation['message_id'];
|
|
$lookupInvitation->save();
|
|
}
|
|
} elseif (! $lookupInvitation) {
|
|
LookupInvitation::create([
|
|
'lookup_account_id' => $lookupAccount->id,
|
|
'invitation_key' => $invitation['invitation_key'],
|
|
'message_id' => $invitation['message_id'] ?: null,
|
|
]);
|
|
}
|
|
}
|
|
|
|
foreach ($account['tokens'] as $token) {
|
|
$lookupToken = false;
|
|
if ($validate || $update) {
|
|
$lookupToken = LookupAccountToken::whereLookupAccountId($lookupAccount->id)->whereToken($token['token'])->first();
|
|
}
|
|
if ($validate && ! $lookupToken) {
|
|
$this->logError("LookupAccountToken - lookupAccountId: {$lookupAccount->id}, token: {$token['token']} | Not found!");
|
|
continue;
|
|
}
|
|
if (! $lookupToken) {
|
|
LookupAccountToken::create([
|
|
'lookup_account_id' => $lookupAccount->id,
|
|
'token' => $token['token'],
|
|
]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private function parseCompany($companyId)
|
|
{
|
|
$data = [];
|
|
|
|
config(['database.default' => $this->option('database')]);
|
|
|
|
$accounts = DB::table('accounts')->whereCompanyId($companyId)->orderBy('id')->get([
|
|
'id', 'account_key'
|
|
]);
|
|
foreach ($accounts as $account) {
|
|
$data[$account->account_key] = $this->parseAccount($account->id);
|
|
}
|
|
|
|
return $data;
|
|
}
|
|
|
|
private function parseAccount($accountId)
|
|
{
|
|
$data = [
|
|
'users' => [],
|
|
'contacts' => [],
|
|
'invitations' => [],
|
|
'tokens' => [],
|
|
];
|
|
|
|
$users = DB::table('users')->whereAccountId($accountId)->orderBy('id')->get([
|
|
'email',
|
|
'id',
|
|
'oauth_user_id',
|
|
'oauth_provider_id',
|
|
'referral_code',
|
|
]);
|
|
foreach ($users as $user) {
|
|
$data['users'][] = [
|
|
'email' => $user->email,
|
|
'user_id' => $user->id,
|
|
'oauth_user_key' => ($user->oauth_provider_id && $user->oauth_user_id) ? ($user->oauth_provider_id . '-' . $user->oauth_user_id) : null,
|
|
'referral_code' => $user->referral_code,
|
|
];
|
|
}
|
|
|
|
$contacts = DB::table('contacts')->whereAccountId($accountId)->orderBy('id')->get([
|
|
'contact_key'
|
|
]);
|
|
foreach ($contacts as $contact) {
|
|
$data['contacts'][] = [
|
|
'contact_key' => $contact->contact_key,
|
|
];
|
|
}
|
|
|
|
$invitations = DB::table('invitations')->whereAccountId($accountId)->orderBy('id')->get([
|
|
'invitation_key',
|
|
'message_id'
|
|
]);
|
|
foreach ($invitations as $invitation) {
|
|
$data['invitations'][] = [
|
|
'invitation_key' => $invitation->invitation_key,
|
|
'message_id' => $invitation->message_id,
|
|
];
|
|
}
|
|
|
|
$tokens = DB::table('account_tokens')->whereAccountId($accountId)->orderBy('id')->get([
|
|
'token'
|
|
]);
|
|
foreach ($tokens as $token) {
|
|
$data['tokens'][] = [
|
|
'token' => $token->token,
|
|
];
|
|
}
|
|
|
|
return $data;
|
|
}
|
|
|
|
private function logMessage($str)
|
|
{
|
|
$str = date('Y-m-d h:i:s') . ' ' . $str;
|
|
$this->info($str);
|
|
$this->log .= $str . "\n";
|
|
}
|
|
|
|
private function logError($str)
|
|
{
|
|
$this->isValid = false;
|
|
$this->logMessage($str);
|
|
}
|
|
|
|
private function truncateTables()
|
|
{
|
|
DB::statement('SET FOREIGN_KEY_CHECKS = 0');
|
|
DB::statement('truncate lookup_companies');
|
|
DB::statement('truncate lookup_accounts');
|
|
DB::statement('truncate lookup_users');
|
|
DB::statement('truncate lookup_contacts');
|
|
DB::statement('truncate lookup_invitations');
|
|
DB::statement('truncate lookup_account_tokens');
|
|
DB::statement('SET FOREIGN_KEY_CHECKS = 1');
|
|
}
|
|
|
|
protected function getOptions()
|
|
{
|
|
return [
|
|
['subdomain', null, InputOption::VALUE_OPTIONAL, 'Subdomain', null],
|
|
['truncate', null, InputOption::VALUE_OPTIONAL, 'Truncate', null],
|
|
['company_id', null, InputOption::VALUE_OPTIONAL, 'Company Id', null],
|
|
['page_size', null, InputOption::VALUE_OPTIONAL, 'Page Size', null],
|
|
['database', null, InputOption::VALUE_OPTIONAL, 'Database', null],
|
|
['validate', null, InputOption::VALUE_OPTIONAL, 'Validate', null],
|
|
];
|
|
}
|
|
|
|
}
|