-
Notifications
You must be signed in to change notification settings - Fork 0
/
preload.js
38 lines (35 loc) · 971 Bytes
/
preload.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
/*
* @Author: 陈德立*******419287484@qq.com
* @Date: 2022-03-14 16:24:24
* @LastEditTime: 2022-04-14 17:10:25
* @LastEditors: 陈德立*******419287484@qq.com
* @Github: https://github.com/Alan1034
* @Description:
* @FilePath: \desktop-map\preload.js
*
*/
const { contextBridge } = require('electron')
const fs = require('fs')
const os = require('os');
const path = require('path');
contextBridge.exposeInMainWorld(
'electron',
{
createWriteStream: (name, path) => {
const writerStream = fs.createWriteStream(name);
writerStream.write(JSON.stringify(path), 'UTF8');
writerStream.end()
return (channel = "close", func) => writerStream.on(channel, () => {
func(writerStream)
})
},
getHomeDesktopDir: (filename) => {
const homedir = os.homedir()
const dirPath = path.format({
dir: path.join(homedir, "Desktop"),
base: `${filename}`
});
return dirPath
}
}
)