From 18e87cf9678d624f6990c58cd50d97d236a459c7 Mon Sep 17 00:00:00 2001
From: hhhero <krishuang136@gmail.com>
Date: Sun, 7 Jul 2024 18:14:03 +0800
Subject: [PATCH] =?UTF-8?q?[=E6=96=B0=E5=A2=9E]ai=E5=86=99=E4=BD=9C?=
 =?UTF-8?q?=E9=A1=B5=E9=9D=A2=E5=AE=8C=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/ai/writer/index.ts               |  53 +++++++
 src/views/ai/writer/components/Left.vue  | 175 +++++++++++++++++++++++
 src/views/ai/writer/components/Right.vue |  25 ++++
 src/views/ai/writer/components/Tag.vue   |  32 +++++
 src/views/ai/writer/index.vue            |  20 +++
 5 files changed, 305 insertions(+)
 create mode 100644 src/api/ai/writer/index.ts
 create mode 100644 src/views/ai/writer/components/Left.vue
 create mode 100644 src/views/ai/writer/components/Right.vue
 create mode 100644 src/views/ai/writer/components/Tag.vue
 create mode 100644 src/views/ai/writer/index.vue

diff --git a/src/api/ai/writer/index.ts b/src/api/ai/writer/index.ts
new file mode 100644
index 00000000..4ce4e2d0
--- /dev/null
+++ b/src/api/ai/writer/index.ts
@@ -0,0 +1,53 @@
+import request from '@/config/axios'
+import { fetchEventSource } from '@microsoft/fetch-event-source'
+
+import { getAccessToken } from '@/utils/auth'
+import { config } from '@/config/axios/config'
+
+export interface WriteParams {
+  /**
+   * 1:撰写 2:回复
+   */
+  type: 1 | 2
+  /**
+   * 写作内容提示 1。撰写 2回复
+   */
+  prompt: string
+  /**
+   *  原文
+   */
+  originalContent: string
+  /**
+   * 长度
+   */
+  length: number
+  /**
+   * 格式
+   */
+  format: number
+  /**
+   * 语气
+   */
+  tone: number
+  /**
+   * 语言
+   */
+  language: number
+}
+export const writeStream = (data: WriteParams, onMessage, onError, onClose, ctrl) => {
+  // return request.post({ url: '/ai/write/generate-stream', data })
+  const token = getAccessToken()
+  return fetchEventSource(`${config.base_url}/ai/write/generate-stream`, {
+    method: 'post',
+    headers: {
+      'Content-Type': 'application/json',
+      Authorization: `Bearer ${token}`
+    },
+    openWhenHidden: true,
+    body: JSON.stringify(data),
+    onmessage: onMessage,
+    onerror: onError,
+    onclose: onClose,
+    signal: ctrl.signal
+  })
+}
diff --git a/src/views/ai/writer/components/Left.vue b/src/views/ai/writer/components/Left.vue
new file mode 100644
index 00000000..9c3031ae
--- /dev/null
+++ b/src/views/ai/writer/components/Left.vue
@@ -0,0 +1,175 @@
+<template>
+  <!-- 定义tab组件 -->
+  <DefineTab v-slot="{ active, text, itemClick }">
+    <span
+      class="inline-block w-1/2 rounded-full cursor-pointer text-center leading-[30px] relative z-1 text-[5C6370] hover:text-black"
+      :class="active ? 'text-black shadow-md' : 'hover:bg-[#DDDFE3]'"
+      @click="itemClick"
+    >
+      {{ text }}
+    </span>
+  </DefineTab>
+  <!-- 定义label组件 -->
+  <DefineLabel v-slot="{ label, hint, hintClick }">
+    <h3 class="mt-5 mb-3 flex items-center justify-between text-[14px]">
+      <span>{{ label }}</span>
+      <span @click="hintClick" v-if="hint" class="flex items-center text-[12px] text-[#846af7]">
+        <Icon icon="ep:question-filled" />
+        {{ hint }}
+      </span>
+    </h3>
+  </DefineLabel>
+  <!-- TODO 小屏幕的时候是定位在左边的，大屏是分开的 -->
+  <div class="relative" v-bind="$attrs">
+    <!-- tab -->
+    <div
+      class="absolute left-1/2 top-2 -translate-x-1/2 w-[303px] rounded-full bg-[#DDDFE3] p-1 z-10"
+    >
+      <div
+        class="flex items-center relative after:content-[''] after:block after:bg-white after:h-[30px] after:w-1/2 after:absolute after:top-0 after:left-0 after:transition-transform after:rounded-full"
+        :class="selectedTab === 2 && 'after:transform after:translate-x-[100%]'"
+      >
+        <ReuseTab
+          v-for="tab in tabs"
+          :key="tab.value"
+          :text="tab.text"
+          :active="tab.value === selectedTab"
+          :itemClick="() => switchTab(tab.value)"
+        />
+      </div>
+    </div>
+    <div
+      class="px-7 pb-2 pt-[46px] overflow-y-auto lg:block w-[380px] box-border bg-[#ECEDEF] h-full"
+    >
+      <div>
+        <template v-if="selectedTab === 1">
+          <ReuseLabel label="写作内容" hint="示例" />
+          <el-input
+            type="textarea"
+            :rows="5"
+            :maxlength="500"
+            v-model="writeForm.prompt"
+            placeholder="请输入写作内容"
+            showWordLimit
+          />
+        </template>
+
+        <template v-else>
+          <ReuseLabel label="原文" hint="示例" />
+          <el-input
+            type="textarea"
+            :rows="5"
+            :maxlength="500"
+            v-model="writeForm.originalContent"
+            placeholder="请输入原文"
+            showWordLimit
+          />
+
+          <ReuseLabel label="回复内容" />
+          <el-input
+            type="textarea"
+            :rows="5"
+            :maxlength="500"
+            v-model="writeForm.prompt"
+            placeholder="请输入回复内容"
+            showWordLimit
+          />
+        </template>
+
+        <ReuseLabel label="长度" />
+        <Tag v-model="writeForm.length" :tags="writeTags.lenTags" />
+        <ReuseLabel label="格式" />
+        <Tag v-model="writeForm.format" :tags="writeTags.formatTags" />
+        <ReuseLabel label="语气" />
+        <Tag v-model="writeForm.tone" :tags="writeTags.toneTags" />
+        <ReuseLabel label="语言" />
+        <Tag v-model="writeForm.language" :tags="writeTags.langTags" />
+
+        <div class="flex items-center justify-center mt-3">
+          <el-button>重置</el-button>
+          <el-button @click="submit" color="#846af7">生成</el-button>
+        </div>
+      </div>
+    </div>
+  </div>
+</template>
+
+<script setup lang="ts">
+  import { createReusableTemplate } from '@vueuse/core'
+  import { ref } from 'vue'
+  import { ElMessage } from 'element-plus'
+  import Tag from './Tag.vue'
+  import { WriteParams } from '@/api/ai/writer'
+  import { omit } from 'lodash-es'
+  import { getIntDictOptions } from '@/utils/dict'
+
+  type TabType = WriteParams['type']
+
+  const emits = defineEmits<{
+    (e: 'submit', params: Partial<WriteParams>)
+  }>()
+
+  const selectedTab = ref<TabType>(1)
+  const tabs: {
+    text: string
+    value: TabType
+  }[] = [
+    { text: '撰写', value: 1 },
+    { text: '回复', value: 2 }
+  ]
+  const [DefineTab, ReuseTab] = createReusableTemplate<{
+    active?: boolean
+    text: string
+    itemClick: () => void
+  }>()
+
+  const initData: WriteParams = {
+    type: 1,
+    prompt: '',
+    originalContent: '',
+    tone: 1,
+    language: 1,
+    length: 100,
+    format: 1
+  }
+  const writeForm = ref<WriteParams>({ ...initData })
+
+  const writeTags = {
+    lenTags: getIntDictOptions('ai_write_length'),
+    // 格式
+
+    formatTags: getIntDictOptions('ai_write_format'),
+    // 语气
+
+    toneTags: getIntDictOptions('ai_write_tone'),
+    // 语言
+    langTags: getIntDictOptions('ai_write_language')
+    //
+  }
+
+  const [DefineLabel, ReuseLabel] = createReusableTemplate<{
+    label: string
+    class?: string
+    hint?: string
+    hintClick?: () => void
+  }>()
+
+  const switchTab = (value: TabType) => {
+    selectedTab.value = value
+    writeForm.value = { ...initData }
+  }
+
+  const submit = () => {
+    if (selectedTab.value === 2 && !writeForm.value.originalContent) {
+      ElMessage.warning('请输入原文')
+      return
+    } else if (!writeForm.value.prompt) {
+      ElMessage.warning(`请输入${selectedTab.value === 1 ? '写作' : '回复'}内容`)
+      return
+    }
+    emits('submit', {
+      ...(selectedTab.value === 1 ? omit(writeForm.value, ['originalContent']) : writeForm.value),
+      type: selectedTab.value
+    })
+  }
+</script>
diff --git a/src/views/ai/writer/components/Right.vue b/src/views/ai/writer/components/Right.vue
new file mode 100644
index 00000000..a11b7c24
--- /dev/null
+++ b/src/views/ai/writer/components/Right.vue
@@ -0,0 +1,25 @@
+<template>
+  <div class="h-full box-border py-6 px-7">
+    <div class="w-full h-full bg-white box-border p-3 sm:p-16 overflow-y-auto">
+      <el-input
+        type="textarea"
+        :value="msg"
+        autosize
+        :input-style="{ boxShadow: 'none' }"
+        resize="none"
+        placeholder="生成的内容……"
+      />
+    </div>
+  </div>
+</template>
+
+<script setup lang="ts">
+  defineProps({
+    msg: {
+      type: String,
+      default: ''
+    }
+  })
+</script>
+
+<style scoped></style>
diff --git a/src/views/ai/writer/components/Tag.vue b/src/views/ai/writer/components/Tag.vue
new file mode 100644
index 00000000..4a32e572
--- /dev/null
+++ b/src/views/ai/writer/components/Tag.vue
@@ -0,0 +1,32 @@
+<template>
+  <div class="flex flex-wrap gap-[8px]">
+    <span
+      v-for="tag in props.tags"
+      :key="tag.value"
+      class="tag mb-2 border-[2px] border-solid border-[#DDDFE3] px-2 leading-6 text-[12px] bg-[#DDDFE3] rounded-[4px] cursor-pointer"
+      :class="modelValue === tag.value && '!border-[#846af7] text-[#846af7]'"
+      @click="emits('update:modelValue', tag.value)"
+    >
+      {{ tag.label }}
+    </span>
+  </div>
+</template>
+
+<script setup lang="ts">
+  const props = withDefaults(
+    defineProps<{
+      tags: { label: string; value: string }[]
+      modelValue: string
+      [k: string]: any
+    }>(),
+    {
+      tags: () => []
+    }
+  )
+
+  const emits = defineEmits<{
+    (e: 'update:modelValue', value: string): void
+  }>()
+</script>
+
+<style scoped></style>
diff --git a/src/views/ai/writer/index.vue b/src/views/ai/writer/index.vue
new file mode 100644
index 00000000..3cb4a7d7
--- /dev/null
+++ b/src/views/ai/writer/index.vue
@@ -0,0 +1,20 @@
+<template>
+  <div class="h-[calc(100vh-var(--top-tool-height)-var(--app-footer-height)-40px)] -m-5 flex">
+    <Left class="h-full" @submit="submit" />
+    <Right class="flex-grow" :msg="msg" />
+  </div>
+</template>
+
+<script setup lang="ts">
+  import Left from './components/Left.vue'
+  import Right from './components/Right.vue'
+  import { writeStream } from '@/api/ai/writer'
+
+  const msg = ref('')
+
+  const submit = async (params) => {
+    const res = await writeStream(params)
+  }
+</script>
+
+<style scoped></style>