improved company identification for web

This commit is contained in:
Denis Duliçi 2021-09-30 16:52:33 +03:00
parent c96b801199
commit 5ade223eec

View File

@ -10,6 +10,8 @@ class IdentifyCompany
{ {
use Users; use Users;
public $request;
/** /**
* Handle an incoming request. * Handle an incoming request.
* *
@ -22,16 +24,18 @@ class IdentifyCompany
*/ */
public function handle($request, Closure $next, ...$guards) public function handle($request, Closure $next, ...$guards)
{ {
$company_id = $request->isApi() $this->request = $request;
? $this->getCompanyIdFromApi($request)
: $this->getCompanyIdFromWeb($request); $company_id = $this->request->isApi()
? $this->getCompanyIdFromApi()
: $this->getCompanyIdFromWeb();
if (empty($company_id)) { if (empty($company_id)) {
abort(500, 'Missing company'); abort(500, 'Missing company');
} }
// Check if user can access company // Check if user can access company
if ($request->isNotSigned($company_id) && $this->isNotUserCompany($company_id)) { if ($this->request->isNotSigned($company_id) && $this->isNotUserCompany($company_id)) {
throw new AuthenticationException('Unauthenticated.', $guards); throw new AuthenticationException('Unauthenticated.', $guards);
} }
@ -42,21 +46,38 @@ class IdentifyCompany
config(['filesystems.disks.' . config('filesystems.default') . '.url' => url('/' . $company_id) . '/uploads']); config(['filesystems.disks.' . config('filesystems.default') . '.url' => url('/' . $company_id) . '/uploads']);
// Fix routes // Fix routes
app('url')->defaults(['company_id' => $company_id]); if ($this->request->isNotApi()) {
$request->route()->forgetParameter('company_id'); app('url')->defaults(['company_id' => $company_id]);
$this->request->route()->forgetParameter('company_id');
}
return $next($request); return $next($this->request);
} }
protected function getCompanyIdFromWeb($request) protected function getCompanyIdFromWeb()
{ {
return (int) $request->route('company_id'); return $this->getCompanyIdFromRoute() ?: ($this->getCompanyIdFromQuery() ?: $this->getCompanyIdFromHeader());
} }
protected function getCompanyIdFromApi($request) protected function getCompanyIdFromApi()
{ {
$company_id = $request->get('company_id', $request->header('X-Company')); $company_id = $this->getCompanyIdFromQuery() ?: $this->getCompanyIdFromHeader();
return $company_id ?: optional($this->getFirstCompanyOfUser())->id; return $company_id ?: optional($this->getFirstCompanyOfUser())->id;
} }
protected function getCompanyIdFromRoute()
{
return (int) $this->request->route('company_id');
}
protected function getCompanyIdFromQuery()
{
return (int) $this->request->query('company_id');
}
protected function getCompanyIdFromHeader()
{
return (int) $this->request->header('X-Company');
}
} }