From ec746c021a82a902cab32b80bbafa49d5584f592 Mon Sep 17 00:00:00 2001
From: zhengkunwang <31820853+zhengkunwang223@users.noreply.github.com>
Date: Tue, 25 Feb 2025 14:16:10 +0800
Subject: [PATCH] feat: merge ai from dev (#7986)

---
 agent/app/dto/ai.go                          |  11 +-
 agent/app/service/ai.go                      |  24 +--
 frontend/src/api/interface/ai.ts             |   1 +
 frontend/src/components/app-status/index.vue |  11 +-
 frontend/src/views/ai/model/domain/index.vue | 176 +++++++++----------
 5 files changed, 105 insertions(+), 118 deletions(-)

diff --git a/agent/app/dto/ai.go b/agent/app/dto/ai.go
index 6c2065cae..193af5eeb 100644
--- a/agent/app/dto/ai.go
+++ b/agent/app/dto/ai.go
@@ -36,9 +36,10 @@ type OllamaBindDomainReq struct {
 }
 
 type OllamaBindDomainRes struct {
-	Domain    string   `json:"domain"`
-	SSLID     uint     `json:"sslID"`
-	AllowIPs  []string `json:"allowIPs"`
-	WebsiteID uint     `json:"websiteID"`
-	ConnUrl   string   `json:"connUrl"`
+	Domain        string   `json:"domain"`
+	SSLID         uint     `json:"sslID"`
+	AllowIPs      []string `json:"allowIPs"`
+	WebsiteID     uint     `json:"websiteID"`
+	ConnUrl       string   `json:"connUrl"`
+	AcmeAccountID uint     `json:"acmeAccountID"`
 }
diff --git a/agent/app/service/ai.go b/agent/app/service/ai.go
index f00b1d151..9bc576334 100644
--- a/agent/app/service/ai.go
+++ b/agent/app/service/ai.go
@@ -61,7 +61,7 @@ func (u *AIToolService) Search(req dto.SearchWithPage) (int64, []dto.OllamaModel
 		}
 		dtoLists = append(dtoLists, item)
 	}
-	return int64(total), dtoLists, err
+	return total, dtoLists, err
 }
 
 func (u *AIToolService) LoadDetail(name string) (string, error) {
@@ -244,14 +244,18 @@ func (u *AIToolService) BindDomain(req dto.OllamaBindDomain) error {
 		}
 	}
 	createWebsiteReq := request.WebsiteCreate{
-		Domains:      []request.WebsiteDomain{{Domain: req.Domain}},
+		Domains:      []request.WebsiteDomain{{Domain: req.Domain, Port: 80}},
 		Alias:        strings.ToLower(req.Domain),
 		Type:         constant.Deployment,
 		AppType:      constant.InstalledApp,
 		AppInstallID: req.AppInstallID,
 	}
+	if req.SSLID > 0 {
+		createWebsiteReq.WebsiteSSLID = req.SSLID
+		createWebsiteReq.EnableSSL = true
+	}
 	websiteService := NewIWebsiteService()
-	if err := websiteService.CreateWebsite(createWebsiteReq); err != nil {
+	if err = websiteService.CreateWebsite(createWebsiteReq); err != nil {
 		return err
 	}
 	website, err := websiteRepo.GetFirst(websiteRepo.WithAlias(strings.ToLower(req.Domain)))
@@ -263,18 +267,6 @@ func (u *AIToolService) BindDomain(req dto.OllamaBindDomain) error {
 			return err
 		}
 	}
-	if req.SSLID > 0 {
-		sslReq := request.WebsiteHTTPSOp{
-			WebsiteID:    website.ID,
-			Enable:       true,
-			Type:         "existed",
-			WebsiteSSLID: req.SSLID,
-			HttpConfig:   "HTTPSOnly",
-		}
-		if _, err = websiteService.OpWebsiteHTTPS(context.Background(), sslReq); err != nil {
-			return err
-		}
-	}
 	if err = ConfigAIProxy(website); err != nil {
 		return err
 	}
@@ -295,6 +287,8 @@ func (u *AIToolService) GetBindDomain(req dto.OllamaBindDomainReq) (*dto.OllamaB
 	res.Domain = website.PrimaryDomain
 	if website.WebsiteSSLID > 0 {
 		res.SSLID = website.WebsiteSSLID
+		ssl, _ := websiteSSLRepo.GetFirst(repo.WithByID(website.WebsiteSSLID))
+		res.AcmeAccountID = ssl.AcmeAccountID
 	}
 	res.ConnUrl = fmt.Sprintf("%s://%s", strings.ToLower(website.Protocol), website.PrimaryDomain)
 	res.AllowIPs = GetAllowIps(website)
diff --git a/frontend/src/api/interface/ai.ts b/frontend/src/api/interface/ai.ts
index a120bea66..c1ebbdb4e 100644
--- a/frontend/src/api/interface/ai.ts
+++ b/frontend/src/api/interface/ai.ts
@@ -107,5 +107,6 @@ export namespace AI {
         allowIPs: string[];
         websiteID?: number;
         connUrl: string;
+        acmeAccountID: number;
     }
 }
diff --git a/frontend/src/components/app-status/index.vue b/frontend/src/components/app-status/index.vue
index b563982ba..6cf8032db 100644
--- a/frontend/src/components/app-status/index.vue
+++ b/frontend/src/components/app-status/index.vue
@@ -116,7 +116,15 @@ let refresh = ref(1);
 const httpPort = ref(0);
 const httpsPort = ref(0);
 
-const em = defineEmits(['setting', 'isExist', 'before', 'after', 'update:loading', 'update:maskShow']);
+const em = defineEmits([
+    'setting',
+    'isExist',
+    'before',
+    'after',
+    'update:loading',
+    'update:maskShow',
+    'update:appInstallID',
+]);
 const setting = () => {
     em('setting', false);
 };
@@ -128,6 +136,7 @@ const onCheck = async (key: any, name: any) => {
             em('isExist', res.data);
             em('update:maskShow', res.data.status !== 'Running');
             operateReq.installId = res.data.appInstallId;
+            em('update:appInstallID', res.data.appInstallId);
             httpPort.value = res.data.httpPort;
             httpsPort.value = res.data.httpsPort;
             refresh.value++;
diff --git a/frontend/src/views/ai/model/domain/index.vue b/frontend/src/views/ai/model/domain/index.vue
index 6e0f76c33..d0525f4a6 100644
--- a/frontend/src/views/ai/model/domain/index.vue
+++ b/frontend/src/views/ai/model/domain/index.vue
@@ -1,98 +1,81 @@
 <template>
-    <DrawerPro v-model="open" :title="$t('aiTools.proxy.proxy')" :back="handleClose" size="large">
+    <DrawerPro v-model="open" :header="$t('aiTools.proxy.proxy')" :back="handleClose" size="large">
         <div v-loading="loading">
             <el-form ref="formRef" label-position="top" @submit.prevent :model="req" :rules="rules">
-                <el-row type="flex" justify="center">
-                    <el-col :span="22">
-                        <el-alert class="common-prompt" :closable="false" type="warning">
-                            <template #default>
-                                <ul>
-                                    <li>{{ $t('aiTools.proxy.proxyHelper1') }}</li>
-                                    <li>{{ $t('aiTools.proxy.proxyHelper2') }}</li>
-                                    <li>{{ $t('aiTools.proxy.proxyHelper3') }}</li>
-                                </ul>
-                            </template>
-                        </el-alert>
-                        <el-form-item :label="$t('website.domain')" prop="domain">
-                            <el-input v-model.trim="req.domain" :disabled="operate === 'update'" />
-                            <span class="input-help">
-                                {{ $t('aiTools.proxy.proxyHelper4') }}
-                            </span>
-                            <span class="input-help">
-                                {{ $t('aiTools.proxy.proxyHelper6') }}
-                                <el-link
-                                    class="pageRoute"
-                                    icon="Position"
-                                    @click="toWebsite(req.websiteID)"
-                                    type="primary"
-                                >
-                                    {{ $t('firewall.quickJump') }}
-                                </el-link>
-                            </span>
-                        </el-form-item>
-                        <el-form-item :label="$t('xpack.waf.whiteList') + ' IP'" prop="ipList">
-                            <el-input
-                                :rows="3"
-                                type="textarea"
-                                clearable
-                                v-model="req.ipList"
-                                :placeholder="$t('xpack.waf.ipGroupHelper')"
-                            />
-                            <span class="input-help">
-                                {{ $t('aiTools.proxy.whiteListHelper') }}
-                            </span>
-                        </el-form-item>
-                        <el-form-item>
-                            <el-checkbox v-model="req.enableSSL" @change="changeSSL">
-                                {{ $t('website.enable') + ' ' + 'HTTPS' }}
-                            </el-checkbox>
-                        </el-form-item>
-                        <el-form-item
-                            :label="$t('website.acmeAccountManage')"
-                            prop="acmeAccountID"
-                            v-if="req.enableSSL"
+                <el-alert class="common-prompt" :closable="false" type="warning">
+                    <template #default>
+                        <ul>
+                            <li>{{ $t('aiTools.proxy.proxyHelper1') }}</li>
+                            <li>{{ $t('aiTools.proxy.proxyHelper2') }}</li>
+                            <li>{{ $t('aiTools.proxy.proxyHelper3') }}</li>
+                        </ul>
+                    </template>
+                </el-alert>
+                <el-form-item :label="$t('website.domain')" prop="domain">
+                    <el-input v-model.trim="req.domain" :disabled="operate === 'update'" />
+                    <span class="input-help">
+                        {{ $t('aiTools.proxy.proxyHelper4') }}
+                    </span>
+                    <span class="input-help">
+                        {{ $t('aiTools.proxy.proxyHelper6') }}
+                        <el-link class="pageRoute" icon="Position" @click="toWebsite(req.websiteID)" type="primary">
+                            {{ $t('firewall.quickJump') }}
+                        </el-link>
+                    </span>
+                </el-form-item>
+                <el-form-item :label="$t('xpack.waf.whiteList') + ' IP'" prop="ipList">
+                    <el-input
+                        :rows="3"
+                        type="textarea"
+                        clearable
+                        v-model="req.ipList"
+                        :placeholder="$t('xpack.waf.ipGroupHelper')"
+                    />
+                    <span class="input-help">
+                        {{ $t('aiTools.proxy.whiteListHelper') }}
+                    </span>
+                </el-form-item>
+                <el-form-item>
+                    <el-checkbox v-model="req.enableSSL" @change="changeSSL">
+                        {{ $t('website.enable') + ' ' + 'HTTPS' }}
+                    </el-checkbox>
+                </el-form-item>
+                <el-form-item :label="$t('website.acmeAccountManage')" prop="acmeAccountID" v-if="req.enableSSL">
+                    <el-select v-model="req.acmeAccountID" :placeholder="$t('website.selectAcme')" @change="loadSSL">
+                        <el-option :key="0" :label="$t('website.imported')" :value="0"></el-option>
+                        <el-option
+                            v-for="(acme, index) in acmeAccounts"
+                            :key="index"
+                            :label="acme.email"
+                            :value="acme.id"
                         >
-                            <el-select
-                                v-model="req.acmeAccountID"
-                                :placeholder="$t('website.selectAcme')"
-                                @change="listSSL"
-                            >
-                                <el-option :key="0" :label="$t('website.imported')" :value="0"></el-option>
-                                <el-option
-                                    v-for="(acme, index) in acmeAccounts"
-                                    :key="index"
-                                    :label="acme.email"
-                                    :value="acme.id"
-                                >
-                                    <span>
-                                        {{ acme.email }}
-                                        <el-tag class="ml-5">{{ getAccountName(acme.type) }}</el-tag>
-                                    </span>
-                                </el-option>
-                            </el-select>
-                        </el-form-item>
-                        <el-form-item :label="$t('website.ssl')" prop="sslID" v-if="req.enableSSL">
-                            <el-select
-                                v-model="req.sslID"
-                                :placeholder="$t('website.selectSSL')"
-                                @change="changeSSl(req.sslID)"
-                            >
-                                <el-option
-                                    v-for="(ssl, index) in ssls"
-                                    :key="index"
-                                    :label="ssl.primaryDomain"
-                                    :value="ssl.id"
-                                ></el-option>
-                            </el-select>
-                        </el-form-item>
-                        <el-alert :closable="false">
-                            {{ $t('aiTools.proxy.proxyHelper5') }}
-                            <el-link class="pageRoute" icon="Position" @click="toInstalled()" type="primary">
-                                {{ $t('firewall.quickJump') }}
-                            </el-link>
-                        </el-alert>
-                    </el-col>
-                </el-row>
+                            <span>
+                                {{ acme.email }}
+                                <el-tag class="ml-5">{{ getAccountName(acme.type) }}</el-tag>
+                            </span>
+                        </el-option>
+                    </el-select>
+                </el-form-item>
+                <el-form-item :label="$t('website.ssl')" prop="sslID" v-if="req.enableSSL">
+                    <el-select
+                        v-model="req.sslID"
+                        :placeholder="$t('website.selectSSL')"
+                        @change="changeSSl(req.sslID)"
+                    >
+                        <el-option
+                            v-for="(ssl, index) in ssls"
+                            :key="index"
+                            :label="ssl.primaryDomain"
+                            :value="ssl.id"
+                        ></el-option>
+                    </el-select>
+                </el-form-item>
+                <el-alert :closable="false">
+                    {{ $t('aiTools.proxy.proxyHelper5') }}
+                    <el-link class="pageRoute" icon="Position" @click="toInstalled()" type="primary">
+                        {{ $t('firewall.quickJump') }}
+                    </el-link>
+                </el-alert>
             </el-form>
         </div>
         <template #footer>
@@ -134,6 +117,9 @@ const req = ref({
     appInstallID: 0,
     websiteID: 0,
 });
+const sslReq = reactive({
+    acmeAccountID: '',
+});
 const rules = reactive<FormRules>({
     domain: [Rules.domainWithPort],
     sslID: [Rules.requiredSelectBusiness],
@@ -167,9 +153,7 @@ const changeSSL = () => {
 };
 
 const loadSSL = () => {
-    const sslReq = {
-        acmeAccountID: String(req.value.acmeAccountID),
-    };
+    sslReq.acmeAccountID = String(req.value.acmeAccountID);
     listSSL(sslReq).then((res) => {
         ssls.value = res.data || [];
         if (ssls.value.length > 0) {
@@ -193,9 +177,6 @@ const loadSSL = () => {
 const listAcmeAccount = () => {
     searchAcmeAccount({ page: 1, pageSize: 100 }).then((res) => {
         acmeAccounts.value = res.data.items || [];
-        if (acmeAccounts.value.length > 0) {
-            req.value.acmeAccountID = acmeAccounts.value[0].id;
-        }
         loadSSL();
     });
 };
@@ -229,6 +210,7 @@ const search = async (appInstallID: number) => {
             if (res.data.sslID > 0) {
                 req.value.enableSSL = true;
                 req.value.sslID = res.data.sslID;
+                req.value.acmeAccountID = res.data.acmeAccountID;
                 listAcmeAccount();
             }
         }