mirror of
https://github.com/leon-ai/leon.git
synced 2024-12-03 02:45:21 +03:00
78 lines
1.8 KiB
JavaScript
78 lines
1.8 KiB
JavaScript
import fs from 'node:fs'
|
|
import path from 'node:path'
|
|
import crypto from 'node:crypto'
|
|
|
|
import dotenv from 'dotenv'
|
|
import { prompt } from 'inquirer'
|
|
|
|
import { LogHelper } from '@/helpers/log-helper'
|
|
import { StringHelper } from '@/helpers/string-helper'
|
|
|
|
dotenv.config()
|
|
|
|
/**
|
|
* Generate HTTP API key script
|
|
* save it in the .env file
|
|
*/
|
|
const generateHTTPAPIKey = () =>
|
|
new Promise(async (resolve, reject) => {
|
|
LogHelper.info('Generating the HTTP API key...')
|
|
|
|
try {
|
|
const shasum = crypto.createHash('sha1')
|
|
const str = StringHelper.random(11)
|
|
const dotEnvPath = path.join(process.cwd(), '.env')
|
|
const envVarKey = 'LEON_HTTP_API_KEY'
|
|
let content = await fs.promises.readFile(dotEnvPath, 'utf8')
|
|
|
|
shasum.update(str)
|
|
const sha1 = shasum.digest('hex')
|
|
|
|
let lines = content.split('\n')
|
|
lines = lines.map((line) => {
|
|
if (line.indexOf(`${envVarKey}=`) !== -1) {
|
|
line = `${envVarKey}=${sha1}`
|
|
}
|
|
|
|
return line
|
|
})
|
|
|
|
content = lines.join('\n')
|
|
|
|
await fs.promises.writeFile(dotEnvPath, content)
|
|
LogHelper.success('HTTP API key generated')
|
|
|
|
resolve()
|
|
} catch (e) {
|
|
LogHelper.error(e.message)
|
|
reject(e)
|
|
}
|
|
})
|
|
|
|
export default () =>
|
|
new Promise(async (resolve, reject) => {
|
|
try {
|
|
if (
|
|
!process.env.LEON_HTTP_API_KEY ||
|
|
process.env.LEON_HTTP_API_KEY === ''
|
|
) {
|
|
await generateHTTPAPIKey()
|
|
} else if (!process.env.IS_DOCKER) {
|
|
const answer = await prompt({
|
|
type: 'confirm',
|
|
name: 'generate.httpAPIKey',
|
|
message: 'Do you want to regenerate the HTTP API key?',
|
|
default: false
|
|
})
|
|
|
|
if (answer.generate.httpAPIKey === true) {
|
|
await generateHTTPAPIKey()
|
|
}
|
|
}
|
|
|
|
resolve()
|
|
} catch (e) {
|
|
reject(e)
|
|
}
|
|
})
|