-
Notifications
You must be signed in to change notification settings - Fork 10
/
index.js
128 lines (104 loc) · 3.67 KB
/
index.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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
import color from 'picocolors'
import { globSync } from 'glob'
import path from 'path'
import OSS from 'ali-oss'
import { URL } from 'node:url'
import { normalizePath } from 'vite'
export default function vitePluginAliOss (options) {
let baseConfig = '/'
let buildConfig = {}
if (options.enabled !== void 0 && !options.enabled) {
return
}
return {
name: 'vite-plugin-ali-oss',
enforce: 'post',
apply: 'build',
configResolved (config) {
baseConfig = config.base
buildConfig = config.build
},
closeBundle: {
sequential: true,
order: 'post',
async handler () {
if (!/^http/i.test(baseConfig)) {
throw Error('[vite-plugin-ali-oss] base must be a url')
}
const outDirPath = normalizePath(path.resolve(normalizePath(buildConfig.outDir)))
const {pathname: ossBasePath, origin: ossOrigin} = new URL(baseConfig)
const createOssOption = Object.assign({}, options)
delete createOssOption.overwrite
delete createOssOption.ignore
delete createOssOption.headers
delete createOssOption.test
delete createOssOption.enabled
const client = new OSS(createOssOption)
const ssrClient = buildConfig.ssrManifest
const ssrServer = buildConfig.ssr
const files = globSync(
outDirPath + '/**/*',
{
nodir: true,
dot: true,
ignore:
// custom ignore
options.ignore !== undefined ? options.ignore :
// ssr client ignore
ssrClient ? ['**/ssr-manifest.json', '**/*.html'] :
// ssr server ignore
ssrServer ? ['**'] :
// default ignore
'**/*.html'
}
)
console.log('')
console.log('ali oss upload start' + (ssrClient ? ' (ssr client)' : ssrServer ? ' (ssr server)' : ''))
console.log('')
const startTime = new Date().getTime()
for (const fileFullPath of files) {
const filePath = normalizePath(fileFullPath).split(outDirPath)[1] // eg: '/assets/vendor.bfb92b77.js'
const ossFilePath = ossBasePath.replace(/\/$/, '') + filePath // eg: '/base/assets/vendor.bfb92b77.js'
const completePath = ossOrigin + ossFilePath // eg: 'https://foo.com/base/assets/vendor.bfb92b77.js'
const output = `${buildConfig.outDir + filePath} => ${color.green(completePath)}`
if (options.test) {
console.log(`test upload path: ${output}`)
continue
}
if (options.overwrite) {
await client.put(
ossFilePath,
fileFullPath,
{
headers: options.headers || {}
}
)
console.log(`upload complete: ${output}`)
} else {
try {
await client.head(ossFilePath);
console.log(`${color.gray('files exists')}: ${output}`)
} catch (error) {
if (error.code === 'NoSuchKey') {
await client.put(
ossFilePath,
fileFullPath,
{
headers: Object.assign(options.headers || {}, { 'x-oss-forbid-overwrite': true })
}
)
console.log(`upload complete: ${output}`)
} else {
throw new Error(error)
}
}
}
}
const duration = (new Date().getTime() - startTime) / 1000
console.log('')
console.log(`ali oss upload complete ^_^, cost ${duration.toFixed(2)}s`)
console.log('')
}
}
}
}