✨ Feature: add backupDomain for sm.ms

PicGo · Aug 27, 2022 · c6d54f1 · c6d54f1
1 parent c0107f1
commit c6d54f1
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 3 deletions.
diff --git a/src/i18n/en.ts b/src/i18n/en.ts
@@ -12,6 +12,8 @@ export const EN: ILocales = {
   // smms
   PICBED_SMMS: 'SM.MS',
   PICBED_SMMS_TOKEN: 'Set Token',
+  PICBED_SMMS_BACKUP_DOMAIN: 'Set Backup Domain',
+  PICBED_SMMS_MESSAGE_BACKUP_DOMAIN: 'Ex. smms.app',
 
   // Ali-cloud
   PICBED_ALICLOUD: 'Ali Cloud',

diff --git a/src/i18n/zh-CN.ts b/src/i18n/zh-CN.ts
@@ -10,6 +10,8 @@ export const ZH_CN = {
   // smms
   PICBED_SMMS: 'SM.MS',
   PICBED_SMMS_TOKEN: '设定Token',
+  PICBED_SMMS_BACKUP_DOMAIN: '备用域名',
+  PICBED_SMMS_MESSAGE_BACKUP_DOMAIN: '例如 smms.app',
 
   // Ali-cloud
   PICBED_ALICLOUD: '阿里云OSS',

diff --git a/src/plugins/uploader/smms.ts b/src/plugins/uploader/smms.ts
@@ -3,10 +3,11 @@ import { Options } from 'request-promise-native'
 import { IBuildInEvent } from '../../utils/enum'
 import { ILocalesKey } from '../../i18n/zh-CN'
 
-const postOptions = (fileName: string, image: Buffer, apiToken: string): Options => {
+const postOptions = (fileName: string, image: Buffer, apiToken: string, backupDomain = ''): Options => {
+  const domain = backupDomain || 'sm.ms'
   return {
     method: 'POST',
-    url: 'https://sm.ms/api/v2/upload',
+    url: `https://${domain}/api/v2/upload`,
     headers: {
       contentType: 'multipart/form-data',
       'User-Agent': 'PicGo',
@@ -33,7 +34,7 @@ const handle = async (ctx: IPicGo): Promise<IPicGo> => {
       if (!image && img.base64Image) {
         image = Buffer.from(img.base64Image, 'base64')
       }
-      const postConfig = postOptions(img.fileName, image, smmsConfig?.token)
+      const postConfig = postOptions(img.fileName, image, smmsConfig?.token, smmsConfig.backupDomain)
       let body = await ctx.Request.request(postConfig)
       body = JSON.parse(body)
       if (body.code === 'success') {
@@ -66,6 +67,16 @@ const config = (ctx: IPicGo): IPluginConfig[] => {
       get alias () { return ctx.i18n.translate<ILocalesKey>('PICBED_SMMS_TOKEN') },
       default: userConfig.token || '',
       required: true
+    },
+    {
+      name: 'backupDomain',
+      get message () {
+        return ctx.i18n.translate<ILocalesKey>('PICBED_SMMS_BACKUP_DOMAIN')
+      },
+      type: 'input',
+      get alias () { return ctx.i18n.translate<ILocalesKey>('PICBED_SMMS_BACKUP_DOMAIN') },
+      default: userConfig.backupDomain || '',
+      required: false
     }
   ]
   return config

diff --git a/src/types/index.ts b/src/types/index.ts
@@ -196,6 +196,7 @@ export interface ICLIConfigs {
 /** SM.MS 图床配置项 */
 export interface ISmmsConfig {
   token: string
+  backupDomain?: string
 }
 /** 七牛云图床配置项 */
 export interface IQiniuConfig {