nginx-ui/app/src/components/ChatGPT/ChatGPT.vue

<script setup lang="ts">
import Icon, { SendOutlined } from '@ant-design/icons-vue'
import { storeToRefs } from 'pinia'
import { marked } from 'marked'
import hljs from 'highlight.js'
import type { Ref } from 'vue'
import { urlJoin } from '@/lib/helper'
import { useSettingsStore, useUserStore } from '@/pinia'
import 'highlight.js/styles/vs2015.css'

import type { ChatComplicationMessage } from '@/api/openai'
import openai from '@/api/openai'
import ChatGPT_logo from '@/assets/svg/ChatGPT_logo.svg?component'

const props = defineProps<{
  content: string
  path?: string
  historyMessages?: ChatComplicationMessage[]
}>()

const emit = defineEmits(['update:history_messages'])

const { language: current } = storeToRefs(useSettingsStore())

const history_messages = computed(() => props.historyMessages)
const messages = ref([]) as Ref<ChatComplicationMessage[] | undefined >

onMounted(() => {
  messages.value = props.historyMessages
})

watch(history_messages, () => {
  messages.value = props.historyMessages
})

const loading = ref(false)
const ask_buffer = ref('')

// eslint-disable-next-line sonarjs/cognitive-complexity
async function request() {
  loading.value = true

  const t = ref({
    role: 'assistant',
    content: '',
  })

  const user = useUserStore()

  const { token } = storeToRefs(user)

  messages.value?.push(t.value)

  emit('update:history_messages', messages.value)

  const res = await fetch(urlJoin(window.location.pathname, '/api/chatgpt'), {
    method: 'POST',
    headers: { Accept: 'text/event-stream', Authorization: token.value },
    body: JSON.stringify({ filepath: props.path, messages: messages.value?.slice(0, messages.value?.length - 1) }),
  })

  const reader = res.body!.getReader()

  let buffer = ''

  let hasCodeBlockIndicator = false

  while (true) {
    const { done, value } = await reader.read()
    if (done) {
      setTimeout(() => {
        scrollToBottom()
      }, 500)
      loading.value = false
      store_record()
      break
    }
    apply(value!)
  }

  function apply(input: Uint8Array) {
    const decoder = new TextDecoder('utf-8')
    const raw = decoder.decode(input)

    // console.log(input, raw)

    const line = raw.split('\n\n')

    line?.forEach(v => {
      const data = v.slice('event:message\ndata:'.length)
      if (!data)
        return

      const content = JSON.parse(data).content

      if (!hasCodeBlockIndicator)
        hasCodeBlockIndicator = content.includes('`')

      for (const c of content) {
        buffer += c
        if (hasCodeBlockIndicator) {
          if (isCodeBlockComplete(buffer)) {
            t.value.content = buffer
            hasCodeBlockIndicator = false
          }
          else {
            t.value.content = `${buffer}\n\`\`\``
          }
        }
        else {
          t.value.content = buffer
        }
      }

      // keep container scroll to bottom
      scrollToBottom()
    })
  }

  function isCodeBlockComplete(text: string) {
    const codeBlockRegex = /```/g
    const matches = text.match(codeBlockRegex)
    if (matches)
      return matches.length % 2 === 0
    else
      return true
  }

  function scrollToBottom() {
    const container = document.querySelector('.right-settings .ant-card-body')
    if (container)
      container.scrollTop = container.scrollHeight
  }
}

async function send() {
  if (!messages.value)
    messages.value = []

  if (messages.value.length === 0) {
    console.log(current.value)
    messages.value.push({
      role: 'user',
      content: `${props.content}\n\nCurrent Language Code: ${current.value}`,
    })
  }
  else {
    messages.value.push({
      role: 'user',
      content: ask_buffer.value,
    })
    ask_buffer.value = ''
  }
  await request()
}

const renderer = new marked.Renderer()

renderer.code = (code, lang: string) => {
  const language = hljs.getLanguage(lang) ? lang : 'nginx'
  const highlightedCode = hljs.highlight(code, { language }).value

  return `<pre><code class="hljs ${language}">${highlightedCode}</code></pre>`
}

marked.setOptions({
  renderer,
  pedantic: false,
  gfm: true,
  breaks: false,
})

function store_record() {
  openai.store_record({
    file_name: props.path,
    messages: messages.value,
  })
}

function clear_record() {
  openai.store_record({
    file_name: props.path,
    messages: [],
  })
  messages.value = []
  emit('update:history_messages', [])
}
const editing_idx = ref(-1)
async function regenerate(index: number) {
  editing_idx.value = -1
  messages.value = messages.value?.slice(0, index)
  await request()
}

const show = computed(() => !messages.value || messages.value?.length === 0)
</script>

<template>
  <div
    v-if="show"
    class="chat-start"
  >
    <AButton
      :loading="loading"
      @click="send"
    >
      <Icon
        v-if="!loading"
        :component="ChatGPT_logo"
      />
      {{ $gettext('Ask ChatGPT for Help') }}
    </AButton>
  </div>
  <div
    v-else
    class="chatgpt-container"
  >
    <AList
      class="chatgpt-log"
      item-layout="horizontal"
      :data-source="messages"
    >
      <template #renderItem="{ item, index }">
        <AListItem>
          <AComment :author="item.role === 'assistant' ? $gettext('Assistant') : $gettext('User')">
            <template #content>
              <div
                v-if="item.role === 'assistant' || editing_idx !== index"
                class="content"
                v-html="marked.parse(item.content)"
              />
              <AInput
                v-else
                v-model:value="item.content"
                style="padding: 0"
                :bordered="false"
              />
            </template>
            <template #actions>
              <span
                v-if="item.role === 'user' && editing_idx !== index"
                @click="editing_idx = index"
              >
                {{ $gettext('Modify') }}
              </span>
              <template v-else-if="editing_idx === index">
                <span @click="regenerate(index + 1)">{{ $gettext('Save') }}</span>
                <span @click="editing_idx = -1">{{ $gettext('Cancel') }}</span>
              </template>
              <span
                v-else-if="!loading"
                @click="regenerate(index)"
              >
                {{ $gettext('Reload') }}
              </span>
            </template>
          </AComment>
        </AListItem>
      </template>
    </AList>
    <div class="input-msg">
      <div class="control-btn">
        <ASpace v-show="!loading">
          <APopconfirm
            :cancel-text="$gettext('No')"
            :ok-text="$gettext('OK')"
            :title="$gettext('Are you sure you want to clear the record of chat?')"
            @confirm="clear_record"
          >
            <AButton type="text">
              {{ $gettext('Clear') }}
            </AButton>
          </APopconfirm>
          <AButton
            type="text"
            @click="regenerate((messages?.length ?? 1) - 1)"
          >
            {{ $gettext('Regenerate response') }}
          </AButton>
        </ASpace>
      </div>
      <ATextarea
        v-model:value="ask_buffer"
        auto-size
      />
      <div class="send-btn">
        <AButton
          size="small"
          type="text"
          :loading="loading"
          @click="send"
        >
          <SendOutlined />
        </AButton>
      </div>
    </div>
  </div>
</template>

<style lang="less" scoped>
.chatgpt-container {
  margin: 0 auto;
  max-width: 800px;

  .chatgpt-log {
    .content {
      width: 100%;

      :deep(.hljs) {
        border-radius: 5px;
      }
    }

    :deep(.ant-list-item) {
      padding: 0;
    }

    :deep(.ant-comment-content) {
      width: 100%;
    }

    :deep(.ant-comment) {
      width: 100%;
    }

    :deep(.ant-comment-content-detail) {
      width: 100%;

      p {
        margin-bottom: 10px;
      }
    }

    :deep(.ant-list-item:first-child) {
      display: none;
    }
  }

  .input-msg {
    position: relative;

    .control-btn {
      display: flex;
      justify-content: center;
    }

    .send-btn {
      position: absolute;
      right: 0;
      bottom: 3px;
    }
  }
}
</style>