mirror of
https://github.com/Kuingsmile/PicList.git
synced 2025-01-23 22:58:14 -05:00
104 lines
3.7 KiB
JavaScript
104 lines
3.7 KiB
JavaScript
// upload dist bundled-app to cos
|
|
require('dotenv').config()
|
|
const crypto = require('crypto')
|
|
const fs = require('fs')
|
|
const mime = require('mime-types')
|
|
const pkg = require('../package.json')
|
|
const configList = require('./config')
|
|
const axios = require('axios').default
|
|
const path = require('path')
|
|
const distPath = path.join(__dirname, '../dist_electron')
|
|
|
|
const BUCKET = 'picgo-1251750343'
|
|
const AREA = 'ap-chengdu'
|
|
const VERSION = pkg.version
|
|
const FILE_PATH = `${VERSION}/`
|
|
const SECRET_ID = process.env.PICGO_ENV_COS_SECRET_ID
|
|
const SECRET_KEY = process.env.PICGO_ENV_COS_SECRET_KEY
|
|
|
|
// https://cloud.tencent.com/document/product/436/7778#signature
|
|
/**
|
|
* @param {string} fileName
|
|
* @returns
|
|
*/
|
|
const generateSignature = (fileName, folder = FILE_PATH) => {
|
|
const secretKey = SECRET_KEY
|
|
const area = AREA
|
|
const bucket = BUCKET
|
|
const path = folder
|
|
const today = Math.floor(new Date().getTime() / 1000)
|
|
const tomorrow = today + 86400
|
|
const signTime = `${today};${tomorrow}`
|
|
const signKey = crypto.createHmac('sha1', secretKey).update(signTime).digest('hex')
|
|
const httpString = `put\n/${path}${fileName}\n\nhost=${bucket}.cos.${area}.myqcloud.com\n`
|
|
const sha1edHttpString = crypto.createHash('sha1').update(httpString).digest('hex')
|
|
const stringToSign = `sha1\n${signTime}\n${sha1edHttpString}\n`
|
|
const signature = crypto.createHmac('sha1', signKey).update(stringToSign).digest('hex')
|
|
return {
|
|
signature,
|
|
signTime
|
|
}
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param {string} fileName
|
|
* @param {Buffer} fileBuffer
|
|
* @param {{ signature: string, signTime: string }} signature
|
|
* @returns
|
|
*/
|
|
const getReqOptions = (fileName, fileBuffer, signature, folder = FILE_PATH) => {
|
|
return {
|
|
method: 'PUT',
|
|
url: `http://${BUCKET}.cos.${AREA}.myqcloud.com/${encodeURI(folder)}${encodeURI(fileName)}`,
|
|
headers: {
|
|
Host: `${BUCKET}.cos.${AREA}.myqcloud.com`,
|
|
Authorization: `q-sign-algorithm=sha1&q-ak=${SECRET_ID}&q-sign-time=${signature.signTime}&q-key-time=${signature.signTime}&q-header-list=host&q-url-param-list=&q-signature=${signature.signature}`,
|
|
contentType: mime.lookup(fileName),
|
|
useAgent: `PicGo;${pkg.version};null;null`
|
|
},
|
|
maxContentLength: Infinity,
|
|
maxBodyLength: Infinity,
|
|
data: fileBuffer,
|
|
resolveWithFullResponse: true
|
|
}
|
|
}
|
|
|
|
const uploadFile = async () => {
|
|
try {
|
|
const platform = process.platform
|
|
if (configList[platform]) {
|
|
let versionFileHasUploaded = false
|
|
for (const [index, config] of configList[platform].entries()) {
|
|
const fileName = `${config.appNameWithPrefix}${VERSION}${config.arch}${config.ext}`
|
|
const filePath = path.join(distPath, fileName)
|
|
const versionFilePath = path.join(distPath, config['version-file'])
|
|
let versionFileName = config['version-file']
|
|
if (VERSION.toLocaleLowerCase().includes('beta')) {
|
|
versionFileName = versionFileName.replace('.yml', '.beta.yml')
|
|
}
|
|
// upload dist file
|
|
const signature = generateSignature(fileName)
|
|
const reqOptions = getReqOptions(fileName, fs.readFileSync(filePath), signature)
|
|
console.log('[PicGo Dist] Uploading...', fileName, `${index + 1}/${configList[platform].length}`)
|
|
await axios.request(reqOptions)
|
|
|
|
// upload version file
|
|
if (!versionFileHasUploaded) {
|
|
const signature = generateSignature(versionFileName, '')
|
|
const reqOptions = getReqOptions(versionFileName, fs.readFileSync(versionFilePath), signature, '')
|
|
console.log('[PicGo Version File] Uploading...', versionFileName)
|
|
await axios.request(reqOptions)
|
|
versionFileHasUploaded = true
|
|
}
|
|
}
|
|
} else {
|
|
console.warn('platform not supported!', platform)
|
|
}
|
|
} catch (e) {
|
|
console.error(e)
|
|
}
|
|
}
|
|
|
|
uploadFile()
|