mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-09-22 08:12:00 +02:00
a2fc7d312a
* refactor: move model definitions and database-related methods to packages/data-schemas * ci: update tests due to new DB structure fix: disable mocking `librechat-data-provider` feat: Add schema exports to data-schemas package - Introduced a new schema module that exports various schemas including action, agent, and user schemas. - Updated index.ts to include the new schema exports for better modularity and organization. ci: fix appleStrategy tests fix: Agent.spec.js ci: refactor handleTools tests to use MongoMemoryServer for in-memory database fix: getLogStores imports ci: update banViolation tests to use MongoMemoryServer and improve session mocking test: refactor samlStrategy tests to improve mock configurations and user handling ci: fix crypto mock in handleText tests for improved accuracy ci: refactor spendTokens tests to improve model imports and setup ci: refactor Message model tests to use MongoMemoryServer and improve database interactions * refactor: streamline IMessage interface and move feedback properties to types/message.ts * refactor: use exported initializeRoles from `data-schemas`, remove api workspace version (this serves as an example of future migrations that still need to happen) * refactor: update model imports to use destructuring from `~/db/models` for consistency and clarity * refactor: remove unused mongoose imports from model files for cleaner code * refactor: remove unused mongoose imports from Share, Prompt, and Transaction model files for cleaner code * refactor: remove unused import in Transaction model for cleaner code * ci: update deploy workflow to reference new Docker Dev Branch Images Build and add new workflow for building Docker images on dev branch * chore: cleanup imports
91 lines
2.3 KiB
JavaScript
91 lines
2.3 KiB
JavaScript
const path = require('path');
|
|
const mongoose = require(path.resolve(__dirname, '..', 'api', 'node_modules', 'mongoose'));
|
|
const { User } = require('@librechat/data-schemas').createModels(mongoose);
|
|
require('module-alias')({ base: path.resolve(__dirname, '..', 'api') });
|
|
const { askQuestion, silentExit } = require('./helpers');
|
|
const banViolation = require('~/cache/banViolation');
|
|
const connect = require('./connect');
|
|
|
|
(async () => {
|
|
await connect();
|
|
|
|
console.purple('---------------------');
|
|
console.purple('Ban a user account!');
|
|
console.purple('---------------------');
|
|
|
|
let email = '';
|
|
let duration = '';
|
|
|
|
if (process.argv.length >= 4) {
|
|
// Check if there are enough command-line arguments.
|
|
email = process.argv[2];
|
|
duration = parseInt(process.argv[3]); // Parse the duration as an integer.
|
|
} else {
|
|
console.orange('Usage: npm run ban-user <email> <duration>');
|
|
console.orange('Note: if you do not pass in the arguments, you will be prompted for them.');
|
|
console.purple('--------------------------');
|
|
}
|
|
|
|
if (!email) {
|
|
email = await askQuestion('Email:');
|
|
}
|
|
|
|
if (!duration) {
|
|
const durationInMinutes = await askQuestion('Duration (in minutes):');
|
|
duration = parseInt(durationInMinutes) * 60000;
|
|
}
|
|
|
|
if (isNaN(duration) || duration <= 0) {
|
|
console.red('Error: Invalid duration!');
|
|
silentExit(1);
|
|
}
|
|
|
|
if (!email.includes('@')) {
|
|
console.red('Error: Invalid email address!');
|
|
silentExit(1);
|
|
}
|
|
|
|
const user = await User.findOne({ email }).lean();
|
|
if (!user) {
|
|
console.red('Error: No user with that email was found!');
|
|
silentExit(1);
|
|
} else {
|
|
console.purple(`Found user: ${user.email}`);
|
|
}
|
|
|
|
const req = {};
|
|
const res = {
|
|
clearCookie: () => {},
|
|
status: function () {
|
|
return this;
|
|
},
|
|
json: function () {
|
|
return this;
|
|
},
|
|
};
|
|
|
|
const errorMessage = {
|
|
type: 'concurrent',
|
|
violation_count: 20,
|
|
user_id: user._id,
|
|
prev_count: 0,
|
|
duration: duration,
|
|
};
|
|
|
|
await banViolation(req, res, errorMessage);
|
|
|
|
silentExit(0);
|
|
})();
|
|
|
|
process.on('uncaughtException', (err) => {
|
|
if (!err.message.includes('fetch failed')) {
|
|
console.error('There was an uncaught error:');
|
|
console.error(err);
|
|
}
|
|
|
|
if (err.message.includes('fetch failed')) {
|
|
return;
|
|
} else {
|
|
process.exit(1);
|
|
}
|
|
});
|