From af95fd2b4e222acbce8f69277005b2decda61d14 Mon Sep 17 00:00:00 2001
From: weslee <chinauser1208@gmail.com>
Date: Sat, 27 May 2023 13:54:18 +0800
Subject: [PATCH] update

---
 .../create_stable_diffusion_results_table.php |  2 +-
 src/StableDiffusion.php                       | 37 ++++++++++++-------
 2 files changed, 25 insertions(+), 14 deletions(-)

diff --git a/publish/migrations/create_stable_diffusion_results_table.php b/publish/migrations/create_stable_diffusion_results_table.php
index d77df61..6bcfa8a 100644
--- a/publish/migrations/create_stable_diffusion_results_table.php
+++ b/publish/migrations/create_stable_diffusion_results_table.php
@@ -22,7 +22,7 @@ public function up(): void
         Schema::create('stable_diffusion_results', function (Blueprint $table) {
             $table->bigIncrements('id');
             $table->string('replicate_id')->unique();
-            $table->string('platform')->comment('with platform');
+            $table->string('platform')->comment('wich platform');
             $table->text('user_prompt');
             $table->mediumText('full_prompt');
             $table->string('url');
diff --git a/src/StableDiffusion.php b/src/StableDiffusion.php
index 363a902..1a813a7 100644
--- a/src/StableDiffusion.php
+++ b/src/StableDiffusion.php
@@ -11,6 +11,7 @@
 namespace Imactool\HyperfStableDiffusion;
 
     use Exception;
+    use Hyperf\Codec\Json;
     use Hyperf\Collection\Arr;
     use Hyperf\Context\ApplicationContext;
     use Hyperf\Guzzle\ClientFactory;
@@ -27,25 +28,36 @@ class StableDiffusion
 
         private static $platform = 'stablediffusionapi';
 
-        private function __construct()
+        private function __construct(protected $apiVersion)
         {
+            $class = '\\Imactool\\HyperfStableDiffusion\\Uri\\' . $apiVersion;
+            if (! class_exists($class)) {
+                throw new Exception("Class {$class} not found");
+            }
+            $this->apiBase = ApplicationContext::getContainer()->get($class);
         }
 
-        public static function make(): self
+        // 获取( https://stablediffusionapi.com)支持的已被本SDK适配的API集合
+        public function supportApi(string $api = ''): array|string
         {
-            return new self();
-        }
+            $aiApis = [
+                'StableDiffusionApiV3', // Stable Diffusion V3 APIs comes with below features https://documenter.getpostman.com/view/18679074/2s83zdwReZ#c7e3c6a0-b57d-4d17-ad5a-c4eb8571021f
+                'DreamboothApiV4', // [Beta] DreamBooth API https://documenter.getpostman.com/view/18679074/2s83zdwReZ#27db9713-6068-41c2-8431-ada0d08d3cd5
+            ];
 
-        public function useStableDiffusionApiV3(): StableDiffusion
-        {
-            $this->apiBase = ApplicationContext::getContainer()->get(StableDiffusionApiV3::class);
-            return $this;
+            if ($api) {
+                if (in_array($api, $aiApis)) {
+                    return $aiApis[$api];
+                }
+                throw new Exception('无效 ' . $api . '。仅支持这些API' . Json::encode($aiApis));
+            }
+
+            return $aiApis;
         }
 
-        public function useDreamboothApiV4(): StableDiffusion
+        public static function make(string $witchAPi = 'StableDiffusionApiV3'): self
         {
-            $this->apiBase = ApplicationContext::getContainer()->get(DreamboothApiV4::class);
-            return $this;
+            return new StableDiffusion($witchAPi);
         }
 
        /**
@@ -83,7 +95,6 @@ public function text2img()
             );
 
             $result = json_decode($response->getBody()->getContents(), true);
-            var_dump(['$result 请求结果' => $result]);
             $this->saveResult($result, $this->apiBase->text2imgUrl());
             return $result;
         }
@@ -102,7 +113,7 @@ public function img2img()
             );
 
             $result = json_decode($response->getBody()->getContents(), true);
-            var_dump(['$result 请求结果' => $result]);
+
             $this->saveResult($result, $this->apiBase->img2imgUrl());
             return $result;
         }