archtechx/tenancy
1
0
Fork 0
mirror of https://github.com/archtechx/tenancy.git synced 2026-02-05 13:54:04 +00:00
Code Activity

Move reusable methods to Tenancy, get rid of tenancy.models.rls

Browse source
This commit is contained in:
lukinovec 2023-06-05 16:35:16 +02:00
parent bad5c693ab
commit cc28d1813a
3 changed files with 49 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

29
src/Commands/CreateRLSPoliciesForTenantTables.php
View file

@ -21,21 +21,13 @@ class CreateRLSPoliciesForTenantTables extends Command
public function handle(): int
{
foreach ($this->getModels() as $model) {
foreach (tenancy()->getModels() as $model) {
DB::transaction(fn () => $this->useRlsOnModel($model));
}
return Command::SUCCESS;
}
protected function getModels(): array
{
$tables = array_map(fn ($table) => $table->tablename, Schema::getAllTables());
$models = array_map(fn (string $table) => $this->getModelFromTable($table), $tables);
return array_filter($models);
}
/**
* Make model use RLS if it belongs to a tenant directly, or through a parent primary model.
*/
@ -46,9 +38,9 @@ class CreateRLSPoliciesForTenantTables extends Command
DB::statement("DROP POLICY IF EXISTS {$table}_rls_policy ON {$table}");
if (! Schema::hasColumn($table, $tenantKey)) {
if (! tenancy()->modelBelongsToTenant($model)) {
// Table is not directly related to a tenant
if (in_array(BelongsToPrimaryModel::class, class_uses_recursive($model::class))) {
if (tenancy()->modelBelongsToTenantIndirectly($model)) {
$this->makeSecondaryModelUseRls($model);
} else {
$this->components->info("Skipping RLS policy creation – table '$table' is not related to a tenant.");
@ -93,19 +85,4 @@ class CreateRLSPoliciesForTenantTables extends Command
DB::statement("ALTER TABLE {$table} ENABLE ROW LEVEL SECURITY");
DB::statement("ALTER TABLE {$table} FORCE ROW LEVEL SECURITY");
}
protected function getModelFromTable(string $table): Model|null
{
foreach (get_declared_classes() as $class) {
if (is_subclass_of($class, Model::class)) {
$model = new $class;
if ($model->getTable() === $table) {
return $model;
}
}
}
return null;
}
}