From fec09652b77b6bb319063ab21f074ba753812fd6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=80=97=E5=AD=90?= <i@haozi.net>
Date: Thu, 28 Nov 2024 20:45:22 +0800
Subject: [PATCH] fix: base64 decode

---
 web/src/components/common/CodeEditor.vue |  3 ++-
 web/src/utils/common/base64.ts           | 17 +++++++++++++++++
 web/src/utils/common/index.ts            |  1 +
 web/src/views/task/CronView.vue          |  4 ++--
 4 files changed, 22 insertions(+), 3 deletions(-)
 create mode 100644 web/src/utils/common/base64.ts

diff --git a/web/src/components/common/CodeEditor.vue b/web/src/components/common/CodeEditor.vue
index 6a76a846..2ebad9ea 100644
--- a/web/src/components/common/CodeEditor.vue
+++ b/web/src/components/common/CodeEditor.vue
@@ -1,5 +1,6 @@
 <script setup lang="ts">
 import file from '@/api/panel/file'
+import { decodeBase64 } from '@/utils'
 import { languageByPath } from '@/utils/file'
 import Editor from '@guolao/vue-monaco-editor'
 
@@ -21,7 +22,7 @@ const get = async () => {
   await file
     .content(props.path)
     .then((res) => {
-      data.value = atob(res.data.content)
+      data.value = decodeBase64(res.data.content)
       window.$message.success('获取成功')
     })
     .catch(() => {
diff --git a/web/src/utils/common/base64.ts b/web/src/utils/common/base64.ts
new file mode 100644
index 00000000..fad398b5
--- /dev/null
+++ b/web/src/utils/common/base64.ts
@@ -0,0 +1,17 @@
+export function decodeBase64(base64: string): string {
+  const binary = atob(base64)
+  const bytes = new Uint8Array(binary.length)
+  for (let i = 0; i < binary.length; i++) {
+    bytes[i] = binary.charCodeAt(i)
+  }
+  return new TextDecoder('utf-8').decode(bytes)
+}
+
+export function encodeBase64(str: string): string {
+  const bytes = new TextEncoder().encode(str)
+  let binary = ''
+  for (let i = 0; i < bytes.length; i++) {
+    binary += String.fromCharCode(bytes[i])
+  }
+  return btoa(binary)
+}
diff --git a/web/src/utils/common/index.ts b/web/src/utils/common/index.ts
index c6d8d118..97af185a 100644
--- a/web/src/utils/common/index.ts
+++ b/web/src/utils/common/index.ts
@@ -1,3 +1,4 @@
+export * from './base64'
 export * from './color'
 export * from './common'
 export * from './icon'
diff --git a/web/src/views/task/CronView.vue b/web/src/views/task/CronView.vue
index 77735616..bafe2105 100644
--- a/web/src/views/task/CronView.vue
+++ b/web/src/views/task/CronView.vue
@@ -7,7 +7,7 @@ import { NButton, NDataTable, NInput, NPopconfirm, NSwitch, NTag } from 'naive-u
 
 import cron from '@/api/panel/cron'
 import file from '@/api/panel/file'
-import { formatDateTime, renderIcon } from '@/utils'
+import { decodeBase64, formatDateTime, renderIcon } from '@/utils'
 import type { CronTask } from '@/views/task/types'
 import { CronNaive } from '@vue-js-cron/naive-ui'
 
@@ -209,7 +209,7 @@ const handleEdit = async (row: any) => {
       editTask.value.id = row.id
       editTask.value.name = row.name
       editTask.value.time = row.time
-      editTask.value.script = atob(res.data.content)
+      editTask.value.script = decodeBase64(res.data.content)
       editModal.value = true
     })
   })