103 lines
4.1 KiB
TypeScript
103 lines
4.1 KiB
TypeScript
import { Command, Flags } from '@oclif/core'
|
|
import path from 'path'
|
|
import * as Server from '../index'
|
|
import * as DataSource from '../DataSource'
|
|
import dotenv from 'dotenv'
|
|
import logger from '../utils/logger'
|
|
|
|
dotenv.config({ path: path.join(__dirname, '..', '..', '.env'), override: true })
|
|
|
|
enum EXIT_CODE {
|
|
SUCCESS = 0,
|
|
FAILED = 1
|
|
}
|
|
let processExitCode = EXIT_CODE.SUCCESS
|
|
|
|
export default class Start extends Command {
|
|
static args = []
|
|
static flags = {
|
|
FLOWISE_USERNAME: Flags.string(),
|
|
FLOWISE_PASSWORD: Flags.string(),
|
|
PORT: Flags.string(),
|
|
DEBUG: Flags.string(),
|
|
APIKEY_PATH: Flags.string(),
|
|
LOG_PATH: Flags.string(),
|
|
LOG_LEVEL: Flags.string(),
|
|
EXECUTION_MODE: Flags.string(),
|
|
TOOL_FUNCTION_BUILTIN_DEP: Flags.string(),
|
|
TOOL_FUNCTION_EXTERNAL_DEP: Flags.string(),
|
|
OVERRIDE_DATABASE: Flags.string(),
|
|
DATABASE_TYPE: Flags.string(),
|
|
DATABASE_PATH: Flags.string(),
|
|
DATABASE_PORT: Flags.string(),
|
|
DATABASE_HOST: Flags.string(),
|
|
DATABASE_NAME: Flags.string(),
|
|
DATABASE_USER: Flags.string(),
|
|
DATABASE_PASSWORD: Flags.string()
|
|
}
|
|
|
|
async stopProcess() {
|
|
logger.info('Shutting down Flowise...')
|
|
try {
|
|
// Shut down the app after timeout if it ever stuck removing pools
|
|
setTimeout(() => {
|
|
logger.info('Flowise was forced to shut down after 30 secs')
|
|
process.exit(processExitCode)
|
|
}, 30000)
|
|
|
|
// Removing pools
|
|
const serverApp = Server.getInstance()
|
|
if (serverApp) await serverApp.stopApp()
|
|
} catch (error) {
|
|
logger.error('There was an error shutting down Flowise...', error)
|
|
}
|
|
process.exit(processExitCode)
|
|
}
|
|
|
|
async run(): Promise<void> {
|
|
process.on('SIGTERM', this.stopProcess)
|
|
process.on('SIGINT', this.stopProcess)
|
|
|
|
// Prevent throw new Error from crashing the app
|
|
// TODO: Get rid of this and send proper error message to ui
|
|
process.on('uncaughtException', (err) => {
|
|
logger.error('uncaughtException: ', err)
|
|
})
|
|
|
|
const { flags } = await this.parse(Start)
|
|
if (flags.FLOWISE_USERNAME) process.env.FLOWISE_USERNAME = flags.FLOWISE_USERNAME
|
|
if (flags.FLOWISE_PASSWORD) process.env.FLOWISE_PASSWORD = flags.FLOWISE_PASSWORD
|
|
if (flags.PORT) process.env.PORT = flags.PORT
|
|
if (flags.APIKEY_PATH) process.env.APIKEY_PATH = flags.APIKEY_PATH
|
|
if (flags.LOG_PATH) process.env.LOG_PATH = flags.LOG_PATH
|
|
if (flags.LOG_LEVEL) process.env.LOG_LEVEL = flags.LOG_LEVEL
|
|
if (flags.EXECUTION_MODE) process.env.EXECUTION_MODE = flags.EXECUTION_MODE
|
|
if (flags.DEBUG) process.env.DEBUG = flags.DEBUG
|
|
if (flags.TOOL_FUNCTION_BUILTIN_DEP) process.env.TOOL_FUNCTION_BUILTIN_DEP = flags.TOOL_FUNCTION_BUILTIN_DEP
|
|
if (flags.TOOL_FUNCTION_EXTERNAL_DEP) process.env.TOOL_FUNCTION_EXTERNAL_DEP = flags.TOOL_FUNCTION_EXTERNAL_DEP
|
|
|
|
// Database config
|
|
if (flags.OVERRIDE_DATABASE) process.env.OVERRIDE_DATABASE = flags.OVERRIDE_DATABASE
|
|
if (flags.DATABASE_TYPE) process.env.DATABASE_TYPE = flags.DATABASE_TYPE
|
|
if (flags.DATABASE_PATH) process.env.DATABASE_PATH = flags.DATABASE_PATH
|
|
if (flags.DATABASE_PORT) process.env.DATABASE_PORT = flags.DATABASE_PORT
|
|
if (flags.DATABASE_HOST) process.env.DATABASE_HOST = flags.DATABASE_HOST
|
|
if (flags.DATABASE_NAME) process.env.DATABASE_NAME = flags.DATABASE_NAME
|
|
if (flags.DATABASE_USER) process.env.DATABASE_USER = flags.DATABASE_USER
|
|
if (flags.DATABASE_PASSWORD) process.env.DATABASE_PASSWORD = flags.DATABASE_PASSWORD
|
|
|
|
await (async () => {
|
|
try {
|
|
logger.info('Starting Flowise...')
|
|
await DataSource.init()
|
|
await Server.start()
|
|
} catch (error) {
|
|
logger.error('There was an error starting Flowise...', error)
|
|
processExitCode = EXIT_CODE.FAILED
|
|
// @ts-ignore
|
|
process.emit('SIGINT')
|
|
}
|
|
})()
|
|
}
|
|
}
|