LibreChat/api/server/services/PluginService.js
Danny Avila 5f3ecef575
fix(config/scripts): Enhance User Creation and Ban Handling, Standardize Imports (#1144)
* chore: use relative imports for scripts

* fix(create-user): newUser.save() now properly awaited, double-check user creation, use relative imports, catch exception

* fix(ban-user): catch exception, handle case where IP is undefined, proper check of user ban on login
2023-11-06 09:19:43 -05:00

84 lines
1.9 KiB
JavaScript

const PluginAuth = require('../../models/schema/pluginAuthSchema');
const { encrypt, decrypt } = require('../utils/');
const getUserPluginAuthValue = async (user, authField) => {
try {
const pluginAuth = await PluginAuth.findOne({ user, authField }).lean();
if (!pluginAuth) {
return null;
}
const decryptedValue = decrypt(pluginAuth.value);
return decryptedValue;
} catch (err) {
console.log(err);
return err;
}
};
// const updateUserPluginAuth = async (userId, authField, pluginKey, value) => {
// try {
// const encryptedValue = encrypt(value);
// const pluginAuth = await PluginAuth.findOneAndUpdate(
// { userId, authField },
// {
// $set: {
// value: encryptedValue,
// pluginKey
// }
// },
// {
// new: true,
// upsert: true
// }
// );
// return pluginAuth;
// } catch (err) {
// console.log(err);
// return err;
// }
// };
const updateUserPluginAuth = async (userId, authField, pluginKey, value) => {
try {
const encryptedValue = encrypt(value);
const pluginAuth = await PluginAuth.findOne({ userId, authField }).lean();
if (pluginAuth) {
const pluginAuth = await PluginAuth.updateOne(
{ userId, authField },
{ $set: { value: encryptedValue } },
);
return pluginAuth;
} else {
const newPluginAuth = await new PluginAuth({
userId,
authField,
value: encryptedValue,
pluginKey,
});
await newPluginAuth.save();
return newPluginAuth;
}
} catch (err) {
console.log(err);
return err;
}
};
const deleteUserPluginAuth = async (userId, authField) => {
try {
const response = await PluginAuth.deleteOne({ userId, authField });
return response;
} catch (err) {
console.log(err);
return err;
}
};
module.exports = {
getUserPluginAuthValue,
updateUserPluginAuth,
deleteUserPluginAuth,
};