[2.2.0] Tenant aware commands (#221)

* wip

* Apply fixes from StyleCI

* Return value when a single tenant is passed

* array map -> foreach

* bug fixes

src/Traits/HasATenantArgument.php

@@ -0,0 +1,29 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Stancl\Tenancy\Traits;
+
+use Symfony\Component\Console\Input\InputArgument;
+
+trait HasATenantArgument
+{
+    protected function getArguments()
+    {
+        return array_merge([
+            ['tenant', InputArgument::REQUIRED, 'Tenant id', null],
+        ], parent::getArguments());
+    }
+
+    protected function getTenants(): array
+    {
+        return [tenancy()->find($this->argument('tenant'))];
+    }
+
+    public function __construct()
+    {
+        parent::__construct();
+
+        $this->specifyParameters();
+    }
+}

src/Traits/HasATenantsOption.php

@@ -14,4 +14,16 @@ trait HasATenantsOption
             ['tenants', null, InputOption::VALUE_IS_ARRAY | InputOption::VALUE_OPTIONAL, '', null],
         ], parent::getOptions());
     }
+
+    protected function getTenants(): array
+    {
+        return tenancy()->all($this->option('tenants'))->all();
+    }
+
+    public function __construct()
+    {
+        parent::__construct();
+
+        $this->specifyParameters();
+    }
 }

src/Traits/TenantAwareCommand.php

@@ -0,0 +1,37 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Stancl\Tenancy\Traits;
+
+use Stancl\Tenancy\Tenant;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+
+trait TenantAwareCommand
+{
+    /** @return mixed|void */
+    protected function execute(InputInterface $input, OutputInterface $output)
+    {
+        $tenants = $this->getTenants();
+
+        if (count($tenants) === 1) {
+            return $tenants[0]->run(function () {
+                return $this->laravel->call([$this, 'handle']);
+            });
+        }
+
+        foreach ($tenants as $tenant) {
+            $tenant->run(function () {
+                $this->laravel->call([$this, 'handle']);
+            });
+        }
+    }
+
+    /**
+     * Get an array of tenants for which the command should be executed.
+     *
+     * @return Tenant[]
+     */
+    abstract protected function getTenants(): array;
+}