From e194754dae4d3d0034070f71a24e95f2e4e11949 Mon Sep 17 00:00:00 2001
From: Kuingsmile <96409857+Kuingsmile@users.noreply.github.com>
Date: Wed, 28 Jan 2026 20:41:35 +0800
Subject: [PATCH] :sparkles: Feature(custom): support read .env file in script
 system

---
 src/main/utils/runScript.ts          | 17 +++++++++++++++
 src/renderer/i18n/locales/en.json    |  1 +
 src/renderer/i18n/locales/zh-CN.json |  1 +
 src/renderer/i18n/locales/zh-TW.json |  1 +
 src/renderer/pages/ScriptPage.vue    | 31 ++++++++++++++++++++++++----
 src/renderer/utils/static.ts         |  2 ++
 6 files changed, 49 insertions(+), 4 deletions(-)

diff --git a/src/main/utils/runScript.ts b/src/main/utils/runScript.ts
index 3f7b789b..d6f644c0 100644
--- a/src/main/utils/runScript.ts
+++ b/src/main/utils/runScript.ts
@@ -7,9 +7,24 @@ import { scriptsDir } from '@core/datastore/dirs'
 import picgo from '@core/picgo'
 import logger from '@core/picgo/logger'
 import axios from 'axios'
+import dotenv from 'dotenv'
 import fs from 'fs-extra'
 import { IPicGo } from 'piclist'
 
+function getFreshEnv() {
+  const envPath = path.join(scriptsDir(), '.env')
+
+  if (fs.existsSync(envPath)) {
+    const buf = fs.readFileSync(envPath)
+    const config = dotenv.parse(buf)
+    for (const k in config) {
+      process.env[k] = config[k]
+    }
+    return config
+  }
+  return {}
+}
+
 export const scriptLifecycleStages = [
   'onSoftwareOpen',
   'onSoftwareClose',
@@ -38,6 +53,7 @@ function format(data: unknown): string {
 
 export async function runScript(ctx: IPicGo, script: string, extra: Record<string, any>): Promise<any> {
   try {
+    const env = getFreshEnv()
     const base64Decode = (str: string): string => Buffer.from(str, 'base64').toString('utf-8')
     const base64Encode = (data: Buffer | string): string =>
       (Buffer.isBuffer(data) ? data : Buffer.from(String(data))).toString('base64')
@@ -62,6 +78,7 @@ export async function runScript(ctx: IPicGo, script: string, extra: Record<strin
       base64Encode,
       os,
       Buffer,
+      env,
     }
     vm.createContext(exposedAPI)
     ctx.log.info('start to run script')
diff --git a/src/renderer/i18n/locales/en.json b/src/renderer/i18n/locales/en.json
index fd42613d..dc710cd9 100644
--- a/src/renderer/i18n/locales/en.json
+++ b/src/renderer/i18n/locales/en.json
@@ -731,6 +731,7 @@
       "disabled": "Disabled",
       "disableScript": "Disable Script",
       "duplicateScriptNameError": "Script name already exists, please use a different name",
+      "editENVFile": "Edit Env File",
       "editScript": "Edit Script",
       "emptyScriptList": "Script list is empty",
       "enabled": "Enabled",
diff --git a/src/renderer/i18n/locales/zh-CN.json b/src/renderer/i18n/locales/zh-CN.json
index a574a68e..fbe61c20 100644
--- a/src/renderer/i18n/locales/zh-CN.json
+++ b/src/renderer/i18n/locales/zh-CN.json
@@ -731,6 +731,7 @@
       "disabled": "已禁用",
       "disableScript": "禁用脚本",
       "duplicateScriptNameError": "脚本名称已存在，请使用不同的名称",
+      "editENVFile": "编辑环境变量",
       "editScript": "编辑脚本",
       "emptyScriptList": "脚本列表为空",
       "enabled": "已启用",
diff --git a/src/renderer/i18n/locales/zh-TW.json b/src/renderer/i18n/locales/zh-TW.json
index 46160df3..b6f4ee70 100644
--- a/src/renderer/i18n/locales/zh-TW.json
+++ b/src/renderer/i18n/locales/zh-TW.json
@@ -731,6 +731,7 @@
       "disabled": "已禁用",
       "disableScript": "禁用腳本",
       "duplicateScriptNameError": "腳本名稱已存在，請使用不同的名稱",
+      "editENVFile": "編輯環境變量",
       "editScript": "編輯腳本",
       "emptyScriptList": "腳本列表為空",
       "enabled": "已啟用",
diff --git a/src/renderer/pages/ScriptPage.vue b/src/renderer/pages/ScriptPage.vue
index 7ddaf42e..41c092ca 100644
--- a/src/renderer/pages/ScriptPage.vue
+++ b/src/renderer/pages/ScriptPage.vue
@@ -11,6 +11,18 @@
             <p class="m-0 text-sm text-secondary">{{ t('pages.scripts.description') }}</p>
           </div>
         </div>
+        <div class="flex flex-wrap gap-3 overflow-visible">
+          <CustomButton
+            type="primary"
+            :icon="FolderOpen"
+            :text="t('pages.scripts.openScriptFolder')"
+            @click="handleOpenScriptFolder"
+          />
+        </div>
+      </div>
+      <div
+        class="flex w-full flex-row items-center justify-between gap-4 overflow-visible rounded-2xl border border-border-secondary px-6 py-2 shadow-md max-md:items-stretch max-md:p-5"
+      >
         <div class="flex flex-wrap gap-3 overflow-visible">
           <div class="flex max-w-[220px] min-w-[180px] flex-1 flex-col gap-1">
             <MultiSelect
@@ -21,9 +33,9 @@
           </div>
           <CustomButton
             type="primary"
-            :icon="FolderOpen"
-            :text="t('pages.scripts.openScriptFolder')"
-            @click="handleOpenScriptFolder"
+            :icon="Edit2Icon"
+            :text="t('pages.scripts.editENVFile')"
+            @click="openEditPage(['.env'])"
           />
         </div>
       </div>
@@ -200,7 +212,18 @@
 
 <script lang="ts" setup>
 import dayjs from 'dayjs'
-import { CheckCircle2, Clock, FileCode, FolderOpen, Pencil, Play, Plus, Trash2, XIcon } from 'lucide-vue-next'
+import {
+  CheckCircle2,
+  Clock,
+  Edit2Icon,
+  FileCode,
+  FolderOpen,
+  Pencil,
+  Play,
+  Plus,
+  Trash2,
+  XIcon,
+} from 'lucide-vue-next'
 import { computed, onBeforeMount, onBeforeUnmount, ref, watch } from 'vue'
 import { useI18n } from 'vue-i18n'
 
diff --git a/src/renderer/utils/static.ts b/src/renderer/utils/static.ts
index 35d9bf8f..c1e10b27 100644
--- a/src/renderer/utils/static.ts
+++ b/src/renderer/utils/static.ts
@@ -70,6 +70,7 @@ export const picBedManualUrlList: IStringKeyMap = {
 
 export const defaultScriptTemplate = `
 // ctx 为 核心PicList实例, extra为额外参数, 其中extra.galleryItem为当前删除的相册对象
+// env 变量中有脚本目录下的 .env 文件内容
 // 可用额外API: axios, crypto, fs, path, os, setTimeout, setInterval, clearTimeout, clearInterval, base64Decode, base64Encode
 // 使用console.log或ctx.log输出的信息在软件日志文件piclist.log中可以查询到
 
@@ -81,6 +82,7 @@ async function main(ctx, extra) {
 
 export const defaultScriptTemplateEn = `
 // ctx is the core PicList instance, extra is additional parameters, among which extra.galleryItem is the currently deleted album object
+// env variable contains the contents of the .env file in the script directory
 // Available additional APIs: axios, crypto, fs, path, os, setTimeout, setInterval, clearTimeout, clearInterval, base64Decode, base64Encode
 // Use console.log or ctx.log to output information that can be found in the software log file piclist.log