wehub-resource-sync
58 lines
1.5 KiB
TypeScript
58 lines
1.5 KiB
TypeScript
import {
|
|
HookEventType,
|
|
PluginContext,
|
|
PluginHandler,
|
|
PluginParameters,
|
|
} from '../types';
|
|
|
|
export const handler: PluginHandler = async (
|
|
context: PluginContext,
|
|
parameters: PluginParameters,
|
|
eventType: HookEventType
|
|
) => {
|
|
let error = null;
|
|
let verdict = false;
|
|
let data: any = null;
|
|
|
|
try {
|
|
const modelList = parameters.models;
|
|
const not = parameters.not || false;
|
|
let requestModel = context.request?.json.model;
|
|
|
|
if (!modelList || !Array.isArray(modelList)) {
|
|
throw new Error('Missing or invalid model whitelist');
|
|
}
|
|
|
|
if (!requestModel) {
|
|
throw new Error('Missing model in request');
|
|
}
|
|
|
|
const inList = modelList.includes(requestModel);
|
|
verdict = not ? !inList : inList;
|
|
|
|
data = {
|
|
verdict,
|
|
not,
|
|
explanation: verdict
|
|
? not
|
|
? `Model "${requestModel}" is not in the allowed list as expected.`
|
|
: `Model "${requestModel}" is allowed.`
|
|
: not
|
|
? `Model "${requestModel}" is in the allowed list when it should not be.`
|
|
: `Model "${requestModel}" is not in the allowed list.`,
|
|
requestedModel: requestModel,
|
|
allowedModels: modelList,
|
|
};
|
|
} catch (e: any) {
|
|
error = e;
|
|
data = {
|
|
explanation: `An error occurred while checking model whitelist: ${e.message}`,
|
|
requestedModel: context.request?.json.model || 'No model specified',
|
|
not: parameters.not || false,
|
|
allowedModels: parameters.models || [],
|
|
};
|
|
}
|
|
|
|
return { error, verdict, data };
|
|
};
|