feat: minRep and allowedGroups, closes #7 , closes #6

Author: barisusakli

languages/en-GB/openai.json

@@ -0,0 +1,3 @@
+{
+	"error.need-x-reputation-to-mention": "You need at least %1 reputation for mentioning chatgpt"
+}

languages/en-US/openai.json

@@ -0,0 +1,3 @@
+{
+	"error.need-x-reputation-to-mention": "You need at least %1 reputation for mentioning chatgpt"
+}

lib/controllers.js

@@ -1,9 +1,14 @@
 'use strict';
 
+const groups = require.main.require('./src/groups');
+
 const Controllers = module.exports;
 
-Controllers.renderAdminPage = function (req, res/* , next */) {
+Controllers.renderAdminPage = async function (req, res/* , next */) {
+	const groupsData = await groups.getNonPrivilegeGroups('groups:createtime', 0, -1);
+	groupsData.sort((a, b) => b.system - a.system);
 	res.render('admin/plugins/openai', {
 		title: 'OpenAI',
+		groups: groupsData,
 	});
 };

library.js

@@ -14,9 +14,21 @@ const user = require.main.require('./src/user');
 const messaging = require.main.require('./src/messaging');
 const api = require.main.require('./src/api');
 const privileges = require.main.require('./src/privileges');
+const groups = require.main.require('./src/groups');
+const sockets = require.main.require('./src/socket.io');
 
 const plugin = module.exports;
 
+const defaults = {
+	apikey: '',
+	'chatgpt-username': '',
+	enablePrivateMessages: 'off',
+	model: 'gpt-3.5-turbo',
+	minimumReputation: 0,
+	allowedGroups: '[]',
+};
+
+
 plugin.init = async (params) => {
 	const { router /* , middleware , controllers */ } = params;
 	const settings = await meta.settings.get('openai')
@@ -29,14 +41,23 @@ plugin.init = async (params) => {
 	routeHelpers.setupAdminPageRoute(router, '/admin/plugins/openai', controllers.renderAdminPage);
 };
 
+async function getSettings() {
+	const settings = await meta.settings.get('openai')
+	return {...defaults, ...settings };
+}
+
 plugin.actionMentionsNotify = async function (hookData) {
 	try {
 		const { notification } = hookData;
 		if (!notification) {
 			return;
 		}
 
-		const settings = (await meta.settings.get('openai')) || {};
+		const settings = await getSettings();
+		if (!await canUseOpenAI(notification.from, settings)) {
+			return;
+		}
+
 		const chatgptusername = settings['chatgpt-username'];
 		const chatgptUid = await user.getUidByUsername(chatgptusername);
 		if (!chatgptUid) {
@@ -79,7 +100,11 @@ plugin.actionMessagingSave = async function (hookData) {
 		if (message.system) {
 			return;
 		}
-		const settings = (await meta.settings.get('openai')) || {};
+		const settings = await getSettings();
+		if (!await canUseOpenAI(message.fromuid, settings)) {
+			return;
+		}
+
 		const chatgptusername = settings['chatgpt-username'];
 		const chatgptUid = await user.getUidByUsername(chatgptusername);
 		const { roomId, content, fromuid } = message;
@@ -140,6 +165,55 @@ plugin.actionMessagingSave = async function (hookData) {
 	}
 };
 
+async function canUseOpenAI(uid, settings) {
+	if (!await checkReputation(uid, settings)) {
+		return false;
+	}
+	if (!await checkGroupMembership(uid, settings)) {
+		return false;
+	}
+	return true;
+}
+
+async function checkReputation(uid, settings) {
+	const reputation = await user.getUserField(uid, 'reputation');
+	const hasEnoughRep = parseInt(settings.minimumReputation, 10) === 0 || parseInt(reputation, 10) >= parseInt(settings.minimumReputation, 10);
+
+	if (!hasEnoughRep) {
+		sockets.server.in(`uid_${uid}`).emit('event:alert', {
+			type: 'danger',
+			title: '[[global:alert.error]]',
+			message: `[[openai:error.need-x-reputation-to-mention, ${settings.minimumReputation}]]`,
+		});
+	}
+	return hasEnoughRep;
+}
+
+async function checkGroupMembership(uid, settings) {
+	let allowedGroups = [];
+	try {
+		allowedGroups = JSON.parse(settings.allowedGroups) || [];
+	} catch (err) {
+		console.error(err);
+		allowedGroups = [];
+	}
+
+	if (!allowedGroups.length) {
+		return true;
+	}
+
+	const isMembers = await groups.isMemberOfGroups(uid, allowedGroups);
+	const memberOfAny = isMembers.includes(true);
+	if (!memberOfAny) {
+		sockets.server.in(`uid_${uid}`).emit('event:alert', {
+			type: 'danger',
+			title: '[[global:alert.error]]',
+			message: `[[error:no-privileges]]`,
+		});
+	}
+	return memberOfAny;
+}
+
 async function getMessageIds(roomId, uid, start, stop) {
 	const isPublic = await db.getObjectField(`chat:room:${roomId}`, 'public');
 	if (parseInt(isPublic, 10) === 1) {

plugin.json

@@ -11,5 +11,6 @@
 	"modules": {
 		"../admin/plugins/openai.js": "./public/lib/admin.js"
 	},
-	"templates": "templates"
+	"templates": "templates",
+	"languages": "languages"
 }

templates/admin/plugins/openai.tpl

@@ -41,6 +41,28 @@
 						</select>
 					</div>
 				</div>
+				<div class="">
+					<h5 class="fw-bold tracking-tight settings-header">Restrictions</h5>
+
+					<div class="mb-3">
+						<label class="form-label" for="minimumReputation">Minimum Reputation</label>
+						<input type="text" id="minimumReputation" name="minimumReputation" title="Minimum Reputation" class="form-control">
+						<p class="form-text">
+							Minimum reputation required to mention chatgpt user. (0 to disable)
+						</p>
+					</div>
+					<div class="mb-3">
+						<label class="form-label" form="allowedGroups">Allowed Groups</label>
+						<select class="form-select" multiple id="allowedGroups" name="allowedGroups" size="10">
+							{{{ each groups }}}
+							<option value="{./displayName}">{./displayName}</option>
+							{{{ end }}}
+						</select>
+						<p class="form-text">
+							Only users in these groups will be able to mention the chatgpt user. Leave blank to allow all groups.
+						</p>
+					</div>
+				</div>
 			</form>
 		</div>