From 97f653bd42795245bd1c67ee4f984536c55f6558 Mon Sep 17 00:00:00 2001
From: longjuan <769022681@qq.com>
Date: Wed, 31 Jul 2024 12:40:03 +0800
Subject: [PATCH] feat: support selecting groups to join when link (#165)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

fixes https://github.com/halo-dev/plugin-s3/issues/152
```release-note
S3关联时支持选择加入的分组
```
![image](https://github.com/user-attachments/assets/2380688c-3e04-4485-986d-b9118e05fe95)
---
 console/src/api/models/link-request.ts        |  6 ++
 console/src/views/S3Link.vue                  | 81 +++++++++++++++++--
 src/main/java/run/halo/s3os/LinkRequest.java  |  1 +
 .../java/run/halo/s3os/S3LinkController.java  |  2 +-
 .../java/run/halo/s3os/S3LinkService.java     |  8 +-
 .../java/run/halo/s3os/S3LinkServiceImpl.java | 14 ++--
 6 files changed, 97 insertions(+), 15 deletions(-)

diff --git a/console/src/api/models/link-request.ts b/console/src/api/models/link-request.ts
index d9bc67c..60d85fd 100644
--- a/console/src/api/models/link-request.ts
+++ b/console/src/api/models/link-request.ts
@@ -20,6 +20,12 @@
  * @interface LinkRequest
  */
 export interface LinkRequest {
+    /**
+     * 
+     * @type {string}
+     * @memberof LinkRequest
+     */
+    'groupName'?: string;
     /**
      * 
      * @type {Array<string>}
diff --git a/console/src/views/S3Link.vue b/console/src/views/S3Link.vue
index 15a577f..e0c6027 100644
--- a/console/src/views/S3Link.vue
+++ b/console/src/views/S3Link.vue
@@ -1,6 +1,7 @@
 <script setup lang="ts">
-import { computed, onMounted, ref, watch } from "vue";
+import { computed, getCurrentInstance, onMounted, ref, watch } from "vue";
 import {
+  IconCheckboxCircle,
   IconRefreshLine,
   Toast,
   VButton,
@@ -12,25 +13,47 @@ import {
   VModal,
   VPageHeader,
   VSpace,
+  VStatusDot,
   VTag,
 } from "@halo-dev/components";
 import CarbonFolderDetailsReference from "~icons/carbon/folder-details-reference";
 import IconErrorWarning from "~icons/ri/error-warning-line";
-import { axiosInstance } from "@halo-dev/api-client";
+import { axiosInstance, coreApiClient, type Group } from "@halo-dev/api-client";
 import { S3LinkControllerApi } from "@/api";
 import type { S3ListResult, LinkResultItem, Policy, ObjectVo } from "@/api";
 
 const s3LinkControllerApi = new S3LinkControllerApi(undefined, axiosInstance.defaults.baseURL, axiosInstance);
 
+const componentInstance = getCurrentInstance();
+const t = (key: string) => {
+  // @ts-ignore
+  if (typeof componentInstance?.proxy?.$t === 'function') {
+    // @ts-ignore
+    return componentInstance.proxy.$t(key);
+  }
+  return key;
+};
+
 const selectedFiles = ref<string[]>([]);
 const policyName = ref<string>("");
 const page = ref(1);
 const size = ref(50);
+const selectedGroup = ref("")
 const policyOptions = ref<{ label: string; value: string; attrs: any }[]>([{
   label: "请选择存储策略",
   value: "",
   attrs: {disabled: true}
 }]);
+const defaultGroup = ref<Group>({
+  apiVersion: "",
+  kind: "",
+  metadata: {
+    name: "",
+  },
+  spec: {
+    displayName: t("core.attachment.common.text.ungrouped"),
+  },
+});
 // update when fetch first page
 const filePrefix = ref<string>("");
 // update when user input
@@ -43,6 +66,7 @@ const s3Objects = ref<S3ListResult>({
   currentContinuationObject: "",
   nextContinuationObject: "",
 });
+const customGroups = ref<Group[]>([]);
 // view state
 const isFetching = ref(false);
 const isShowModal = ref(false);
@@ -91,7 +115,7 @@ const handleCheckAllChange = (e: Event) => {
 const fetchPolicies = async () => {
   try {
     const {status, data} = await s3LinkControllerApi.listS3Policies();
-    if (status == 200) {
+    if (status === 200) {
       policyOptions.value = [{
         label: "请选择存储策略",
         value: "",
@@ -174,7 +198,8 @@ const handleLink = async () => {
   const linkResult = await s3LinkControllerApi.addAttachmentRecord({
     linkRequest: {
       policyName: policyName.value,
-      objectKeys: selectedFiles.value
+      objectKeys: selectedFiles.value,
+      groupName: selectedGroup.value,
     }
   });
   const items = linkResult.data.items || [];
@@ -219,7 +244,20 @@ const handleModalClose = () => {
   fetchObjects();
 };
 
-onMounted(fetchPolicies);
+const fetchCustomGroups = async () => {
+  const {status, data} = await coreApiClient.storage.group.listGroup({
+    labelSelector: ["!halo.run/hidden"],
+    sort: ["metadata.creationTimestamp,asc"],
+  });
+  if (status === 200) {
+    customGroups.value = data.items;
+  }
+};
+
+onMounted(() => {
+  fetchPolicies();
+  fetchCustomGroups();
+});
 
 watch(selectedFiles, (newValue) => {
   checkedAll.value = s3Objects.value.objects?.filter(file => !file.isLinked)
@@ -232,7 +270,7 @@ watch(selectedLinkedStatusItem, handleFirstPage);
 </script>
 
 <template>
-  <VPageHeader title="关联S3文件(Beta)">
+  <VPageHeader title="关联S3文件">
     <template #icon>
       <CarbonFolderDetailsReference class="mr-2 self-center"/>
     </template>
@@ -327,6 +365,37 @@ watch(selectedLinkedStatusItem, handleFirstPage);
           class="box-border h-full w-full divide-y divide-gray-100"
           role="list"
         >
+          <li style="padding: 0.5rem 1rem 0">
+            <span class="ml-1 mb-1 block text-sm text-gray-500">关联后所加入的分组</span>
+            <div class="mb-5 grid grid-cols-2 gap-x-2 gap-y-3 md:grid-cols-3 lg:grid-cols-4 2xl:grid-cols-6">
+              <button
+                type="button"
+                class="inline-flex h-full w-full items-center gap-2 rounded-md border border-gray-200 bg-white px-3 py-2.5 text-sm font-medium text-gray-800 hover:bg-gray-50 hover:shadow-sm"
+                v-for="(group, index) in [defaultGroup, ...customGroups]"
+                :key="index"
+                :class="{ '!bg-gray-100 shadow-sm': group.metadata.name === selectedGroup }"
+                @click="selectedGroup = group.metadata.name"
+              >
+                <div class="inline-flex w-full flex-1 gap-x-2 break-all text-left">
+                  <slot name="text">
+                    {{ group?.spec.displayName }}
+                  </slot>
+                  <VStatusDot
+                    v-if="group?.metadata.deletionTimestamp"
+                    v-tooltip="$t('core.common.status.deleting')"
+                    state="warning"
+                    animate
+                  />
+                </div>
+                <div class="flex-none">
+                  <IconCheckboxCircle
+                    v-if="group.metadata.name === selectedGroup"
+                    class="text-primary"
+                  />
+                </div>
+              </button>
+            </div>
+          </li>
           <li v-for="(file, index) in s3Objects.objects" :key="index">
             <VEntity :is-selected="checkSelection(file)">
               <template
diff --git a/src/main/java/run/halo/s3os/LinkRequest.java b/src/main/java/run/halo/s3os/LinkRequest.java
index ee73edf..b6990f7 100644
--- a/src/main/java/run/halo/s3os/LinkRequest.java
+++ b/src/main/java/run/halo/s3os/LinkRequest.java
@@ -9,4 +9,5 @@
 public class LinkRequest {
     private String policyName;
     private Set<String> objectKeys;
+    private String groupName;
 }
\ No newline at end of file
diff --git a/src/main/java/run/halo/s3os/S3LinkController.java b/src/main/java/run/halo/s3os/S3LinkController.java
index 0baa5ee..23b9145 100644
--- a/src/main/java/run/halo/s3os/S3LinkController.java
+++ b/src/main/java/run/halo/s3os/S3LinkController.java
@@ -41,6 +41,6 @@ public Mono<S3ListResult> listObjects(@PathVariable(value = "policyName") String
     @PostMapping("/attachments/link")
     public Mono<LinkResult> addAttachmentRecord(@RequestBody LinkRequest linkRequest) {
         return s3LinkService.addAttachmentRecords(linkRequest.getPolicyName(),
-            linkRequest.getObjectKeys());
+            linkRequest.getObjectKeys(), linkRequest.getGroupName());
     }
 }
diff --git a/src/main/java/run/halo/s3os/S3LinkService.java b/src/main/java/run/halo/s3os/S3LinkService.java
index 2a1bb45..ce38620 100644
--- a/src/main/java/run/halo/s3os/S3LinkService.java
+++ b/src/main/java/run/halo/s3os/S3LinkService.java
@@ -10,10 +10,12 @@ public interface S3LinkService {
     Flux<Policy> listS3Policies();
 
     Mono<S3ListResult> listObjects(String policyName, String continuationToken,
-        Integer pageSize, String filePrefix);
+                                   Integer pageSize, String filePrefix);
 
-    Mono<LinkResult> addAttachmentRecords(String policyName, Set<String> objectKeys);
+    Mono<LinkResult> addAttachmentRecords(String policyName, Set<String> objectKeys,
+                                          String groupName);
 
     Mono<S3ListResult> listObjectsUnlinked(String policyName, String continuationToken,
-        String continuationObject, Integer pageSize, String filePrefix);
+                                           String continuationObject, Integer pageSize,
+                                           String filePrefix);
 }
diff --git a/src/main/java/run/halo/s3os/S3LinkServiceImpl.java b/src/main/java/run/halo/s3os/S3LinkServiceImpl.java
index cd3cf27..ba6b6d1 100644
--- a/src/main/java/run/halo/s3os/S3LinkServiceImpl.java
+++ b/src/main/java/run/halo/s3os/S3LinkServiceImpl.java
@@ -113,11 +113,11 @@ public Mono<S3ListResult> listObjects(String policyName, String continuationToke
     }
 
     @Override
-    public Mono<LinkResult> addAttachmentRecords(String policyName, Set<String> objectKeys) {
+    public Mono<LinkResult> addAttachmentRecords(String policyName, Set<String> objectKeys, String groupName) {
         return getOperableObjectKeys(objectKeys, policyName)
             .flatMap(operableObjectKeys -> getExistingAttachments(objectKeys, policyName)
                 .flatMap(existingAttachments -> getLinkResultItems(objectKeys, operableObjectKeys,
-                    existingAttachments, policyName)
+                    existingAttachments, policyName, groupName)
                     .collectList()
                     .map(LinkResult::new)));
     }
@@ -147,12 +147,13 @@ private Mono<Set<String>> getExistingAttachments(Set<String> objectKeys,
     private Flux<LinkResult.LinkResultItem> getLinkResultItems(Set<String> objectKeys,
                                                                Set<String> operableObjectKeys,
                                                                Set<String> existingAttachments,
-                                                               String policyName) {
+                                                               String policyName,
+                                                               String groupName) {
         return Flux.fromIterable(objectKeys)
             .flatMap((objectKey) -> {
                 if (operableObjectKeys.contains(objectKey) &&
                     !existingAttachments.contains(objectKey)) {
-                    return addAttachmentRecord(policyName, objectKey)
+                    return addAttachmentRecord(policyName, objectKey, groupName)
                         .onErrorResume((throwable) -> Mono.just(
                             new LinkResult.LinkResultItem(objectKey, false,
                                 throwable.getMessage())));
@@ -223,7 +224,7 @@ record TokenState(String currToken, String nextToken) {
     }
 
     public Mono<LinkResult.LinkResultItem> addAttachmentRecord(String policyName,
-        String objectKey) {
+        String objectKey, String groupName) {
         return authenticationConsumer(authentication -> client.fetch(Policy.class, policyName)
             .flatMap((policy) -> {
                 var configMapName = policy.getSpec().getConfigMapName();
@@ -255,6 +256,9 @@ public Mono<LinkResult.LinkResultItem> addAttachmentRecord(String policyName,
                         }
                         spec.setOwnerName(authentication.getName());
                         spec.setPolicyName(policyName);
+                        if (StringUtils.isNotBlank(groupName)){
+                            spec.setGroupName(groupName);
+                        }
                     })
                     .flatMap(client::create)
                     .thenReturn(new LinkResult.LinkResultItem(objectKey, true, null));