feat: 给请求加上同源referer

app/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lanzouyun-pan",
-  "version": "3.5.1",
+  "version": "3.5.2",
   "private": true,
   "main": "build/main/index.js",
   "description": "蓝奏云网盘客户端",

package.json

@@ -1,6 +1,6 @@
 {
   "name": "lanzouyun-pan",
-  "version": "3.5.1",
+  "version": "3.5.2",
   "private": true,
   "main": "build/main/index.js",
   "description": "蓝奏云网盘客户端",

src/common/core/profile.ts

@@ -2,6 +2,7 @@ import * as http from '../http'
 import cheerio from 'cheerio'
 import {Matcher} from './matcher'
 import type {Config} from '../../renderer/store/Config'
+import store from '../store'
 
 enum PROFILE_EL {
   个性域名 = '个性域名', // domain
@@ -39,7 +40,7 @@ export async function profile() {
           return {...prev, verification: $my('#phone_id', el).text().trim()}
       }
       return prev
-    }, {} as Pick<Config, 'domain' | 'lastLogin' | 'maxSize' | 'verification'> & {supportList: string[]})
+    }, {} as Pick<Config, 'referer' | 'domain' | 'lastLogin' | 'maxSize' | 'verification'> & {supportList: string[]})
 
   const iframe = $main('iframe').attr('src')
   const mainPage = await http.request.get(iframe).text()
@@ -48,5 +49,6 @@ export async function profile() {
   return {
     ...profiles,
     more: ajaxData,
+    referer: new URL(iframe, store.get('lanzouUrl')).href,
   }
 }

src/common/http.ts

@@ -1,10 +1,10 @@
 import got from 'got'
 import {cookieJar, shareCookieJar} from './cookie'
-import config from '../project.config'
 import store from './store'
 import {delay} from './util'
 import electronApi from '../renderer/electronApi'
 import {message} from 'antd'
+import {config} from '../renderer/store/Config'
 
 const base = got.extend({
   headers: {
@@ -78,6 +78,18 @@ const base = got.extend({
 export const request = got.extend(base, {
   cookieJar,
   prefixUrl: store.get('lanzouUrl'),
+  hooks: {
+    beforeRequest: [
+      options => {
+        if (config.referer && (!options.headers['referer'] || !options.headers['Referer'])) {
+          const url = typeof options.url === 'string' ? new URL(options.url) : options.url
+          if (url.origin === new URL(config.referer).origin) {
+            options.headers['referer'] = config.referer
+          }
+        }
+      },
+    ],
+  },
 })
 
 export const share = got.extend(base, {

src/renderer/store/Config.ts

@@ -5,6 +5,7 @@ type BaseProps<T> = Partial<Pick<T, {[P in keyof T]: T[P] extends (...args: any)
 
 export class Config {
   @persist domain = '' // https://wwn.lanzouf.com
+  @persist referer = ''
   @persist lastLogin = '' // "2022-04-23 21:03:29"
   // @persist('list') supportList: string[] = [] // ['tar']
   @persist verification = '' // 188****8888

src/renderer/store/task/UploadTask.ts

@@ -152,7 +152,9 @@ export class UploadTask implements BaseTask {
     const encoder = new FormDataEncoder(form)
     return {
       from: Readable.from(encoder),
-      to: http.request.stream.post('html5up.php', {headers: encoder.headers}),
+      to: http.request.stream.post('html5up.php', {
+        headers: {Referer: config.referer, ...encoder.headers},
+      }),
     }
   }