refactor code

This commit is contained in:
an-lee
2024-10-18 22:50:31 +08:00
parent 61ad5cca57
commit f0d82f3bfa
7 changed files with 19 additions and 15 deletions

View File

@@ -107,7 +107,7 @@ class AudiosHandler {
return audio.toJSON(); return audio.toJSON();
} catch (err) { } catch (err) {
logger.error(err); logger.error(err.message);
throw err; throw err;
} }
} }

View File

@@ -174,10 +174,6 @@ db.connect = async () => {
// migrate up to the latest state // migrate up to the latest state
await umzug.up(); await umzug.up();
await sequelize.query("PRAGMA foreign_keys = false;");
await sequelize.sync();
await sequelize.authenticate();
} catch (err) { } catch (err) {
logger.error(err); logger.error(err);
await sequelize.close(); await sequelize.close();
@@ -200,6 +196,10 @@ db.connect = async () => {
await db.backup(); await db.backup();
} }
await sequelize.query("PRAGMA foreign_keys = false;");
await sequelize.sync();
await sequelize.authenticate();
// vacuum the database // vacuum the database
logger.info("Vacuuming the database"); logger.info("Vacuuming the database");
await sequelize.query("VACUUM"); await sequelize.query("VACUUM");

View File

@@ -333,7 +333,8 @@ export class Audio extends Model<Audio> {
md5, md5,
}, },
}); });
if (existing) { if (!!existing) {
logger.warn("Audio already exists:", existing.id);
return existing; return existing;
} }
@@ -388,7 +389,7 @@ export class Audio extends Model<Audio> {
}); });
return record.save().catch((err) => { return record.save().catch((err) => {
logger.error(err); logger.error(err.message);
// Remove copied file // Remove copied file
fs.removeSync(destFile); fs.removeSync(destFile);
throw err; throw err;

View File

@@ -58,11 +58,13 @@ export class ChatMember extends Model<ChatMember> {
@BelongsTo(() => Chat, { @BelongsTo(() => Chat, {
foreignKey: "chatId", foreignKey: "chatId",
constraints: false,
}) })
chat: Chat; chat: Chat;
@BelongsTo(() => ChatAgent, { @BelongsTo(() => ChatAgent, {
foreignKey: "userId", foreignKey: "userId",
constraints: false,
}) })
agent: ChatAgent; agent: ChatAgent;

View File

@@ -349,7 +349,8 @@ export class Video extends Model<Video> {
md5, md5,
}, },
}); });
if (existing) { if (!!existing) {
logger.warn("Video already exists:", existing.id);
return existing; return existing;
} }

View File

@@ -347,7 +347,7 @@ export default class FfmpegWrapper {
resolve(output); resolve(output);
}) })
.on("error", (err) => { .on("error", (err) => {
logger.error(err); logger.error(err.message);
reject(err); reject(err);
}) })
.save(output); .save(output);

View File

@@ -133,7 +133,7 @@ class Youtubedr {
{ {
timeout: TEN_MINUTES, timeout: TEN_MINUTES,
signal: this.abortController.signal, signal: this.abortController.signal,
env: this.proxyEnv() env: this.proxyEnv(),
} }
); );
@@ -189,7 +189,7 @@ class Youtubedr {
command, command,
{ {
timeout: ONE_MINUTE, timeout: ONE_MINUTE,
env: this.proxyEnv() env: this.proxyEnv(),
}, },
(error, stdout, stderr) => { (error, stdout, stderr) => {
if (error) { if (error) {
@@ -249,7 +249,7 @@ class Youtubedr {
this.getYtVideoId(url); this.getYtVideoId(url);
return true; return true;
} catch (error) { } catch (error) {
logger.warn(error); logger.warn(error.message);
return false; return false;
} }
}; };
@@ -264,14 +264,14 @@ class Youtubedr {
*/ */
proxyEnv = () => { proxyEnv = () => {
// keep current environment variables // keep current environment variables
let env = {...process.env} let env = { ...process.env };
const proxyConfig = settings.getSync("proxy") as ProxyConfigType; const proxyConfig = settings.getSync("proxy") as ProxyConfigType;
if (proxyConfig.enabled && proxyConfig.url) { if (proxyConfig.enabled && proxyConfig.url) {
env["HTTP_PROXY"] = proxyConfig.url; env["HTTP_PROXY"] = proxyConfig.url;
env["HTTPS_PROXY"] = proxyConfig.url; env["HTTPS_PROXY"] = proxyConfig.url;
} }
return env return env;
} };
} }
export default new Youtubedr(); export default new Youtubedr();