Update build command for edition

Author: timkelty

src/Helper.php

@@ -177,16 +177,6 @@ public static function base64UrlEncode(string $data): string
         return rtrim($base64Url, '=');
     }
 
-    public static function validateProjectTypeForEdition(string $projectType, string $edition): bool
-    {
-        // TODO: replace with enums
-        if (!$edition || ($projectType === 'team' && $edition === 'pro')) {
-            return false;
-        }
-
-        return true;
-    }
-
     public static function makeGatewayApiRequest(iterable $headers): ResponseInterface
     {
         if (!Helper::isCraftCloud()) {

src/cli/controllers/BuildController.php

@@ -2,36 +2,56 @@
 
 namespace craft\cloud\cli\controllers;
 
+use Composer\Semver\Semver;
 use Craft;
 use craft\console\Controller;
-use Illuminate\Support\Collection;
-use samdark\log\PsrMessage;
-use yii\console\ExitCode;
+use craft\helpers\App;
+use yii\console\Exception;
 
 class BuildController extends Controller
 {
-    public function actionIndex(string $json): int
+    public $defaultAction = 'build';
+    public ?string $publishAssetBundlesTo = null;
+    public string $craftEdition = '';
+
+    public function options($actionID): array
+    {
+        return array_merge(parent::options($actionID), [
+            'publish-asset-bundles-to',
+            'craft-edition',
+        ]);
+    }
+
+    public function actionBuild(): int
     {
-        $options = json_decode($json, true);
-        $exitCode = ExitCode::OK;
-
-        Collection::make([
-            'cloud/validate/project-type',
-            'cloud/asset-bundles/publish',
-        ])->each(function(string $command) use ($options, &$exitCode) {
-            $params = $options[$command] ?? [];
-            $exitCode = $this->run("/$command", $params);
-
-            if ($exitCode !== ExitCode::OK) {
-                Craft::error(new PsrMessage('Command failed.', [
-                    'command' => $command,
-                    'params' => $params,
-                ]));
-
-                return false;
-            }
-        });
-
-        return $exitCode;
+        if (!$this->isEditionValid($this->craftEdition)) {
+            throw new Exception('Invalid Craft CMS edition.');
+        }
+
+        return $this->run('/cloud/asset-bundles/publish', [
+            'to' => $this->publishAssetBundlesTo,
+        ]);
+    }
+
+    private function isEditionValid(string $edition): bool
+    {
+        $craftVersion = Craft::$app->getInfo()->version;
+
+        // CRAFT_EDITION is enforced in these versions, so we don't need to validate
+        if (App::env('CRAFT_EDITION') && Semver::satisfies($craftVersion, '^4.10 || ^5.2')) {
+            return true;
+        }
+
+        $editionFromProjectConfig = Craft::$app->getProjectConfig()->get('system.edition', true);
+
+        if (!$editionFromProjectConfig) {
+            throw new Exception('Unable to determine the Craft CMS edition.');
+        }
+
+        if ($edition !== $editionFromProjectConfig) {
+            return false;
+        }
+
+        return true;
     }
 }