89 lines
3.8 KiB
JavaScript
89 lines
3.8 KiB
JavaScript
var { ExtensionCommon } = ChromeUtils.importESModule("resource://gre/modules/ExtensionCommon.sys.mjs");
|
|
var { Services } = globalThis || ChromeUtils.importESModule("resource://gre/modules/Services.sys.mjs");
|
|
var { MailServices } = ChromeUtils.importESModule("resource:///modules/MailServices.sys.mjs");
|
|
|
|
console.log("[ai-filter][api] Experiment API module loaded");
|
|
|
|
var resProto = Cc["@mozilla.org/network/protocol;1?name=resource"]
|
|
.getService(Ci.nsISubstitutingProtocolHandler);
|
|
|
|
function registerResourceUrl(extension, namespace) {
|
|
console.log(`[ai-filter][api] registerResourceUrl called for namespace="${namespace}"`);
|
|
if (resProto.hasSubstitution(namespace)) {
|
|
console.log(`[ai-filter][api] namespace="${namespace}" already registered, skipping`);
|
|
return;
|
|
}
|
|
let uri = Services.io.newURI(".", null, extension.rootURI);
|
|
console.log(`[ai-filter][api] setting substitution for "${namespace}" → ${uri.spec}`);
|
|
resProto.setSubstitutionWithFlags(namespace, uri, resProto.ALLOW_CONTENT_ACCESS);
|
|
}
|
|
|
|
var gTerm;
|
|
var AIFilterMod;
|
|
|
|
var aiFilter = class extends ExtensionCommon.ExtensionAPI {
|
|
async onStartup() {
|
|
console.log("[ai-filter][api] onStartup()");
|
|
let { extension } = this;
|
|
|
|
registerResourceUrl(extension, "aifilter");
|
|
|
|
|
|
try {
|
|
console.log("[ai-filter][api] importing ExpressionSearchFilter.jsm");
|
|
AIFilterMod = ChromeUtils.import("resource://aifilter/modules/ExpressionSearchFilter.jsm");
|
|
console.log("[ai-filter][api] ExpressionSearchFilter.jsm import succeeded");
|
|
}
|
|
catch (err) {
|
|
console.error("[ai-filter][api] failed to import ExpressionSearchFilter.jsm:", err);
|
|
}
|
|
}
|
|
|
|
onShutdown(isAppShutdown) {
|
|
console.log("[ai-filter][api] onShutdown(), isAppShutdown =", isAppShutdown);
|
|
if (!isAppShutdown && resProto.hasSubstitution("aifilter")) {
|
|
console.log("[ai-filter][api] removing substitution for namespace='aifilter'");
|
|
resProto.setSubstitution("aifilter", null);
|
|
}
|
|
}
|
|
|
|
getAPI(context) {
|
|
console.log("[ai-filter][api] getAPI()");
|
|
return {
|
|
aiFilter: {
|
|
initConfig: async (config) => {
|
|
try {
|
|
if (AIFilterMod?.AIFilter?.setConfig) {
|
|
AIFilterMod.AIFilter.setConfig(config);
|
|
console.log("[ai-filter][api] configuration applied", config);
|
|
}
|
|
} catch (err) {
|
|
console.error("[ai-filter][api] failed to apply config:", err);
|
|
}
|
|
},
|
|
classify: (msg) => {
|
|
console.log("[ai-filter][api] classify() called with msg:", msg);
|
|
try {
|
|
if (!gTerm) {
|
|
console.log("[ai-filter][api] instantiating new ClassificationTerm");
|
|
let mod = AIFilterMod || ChromeUtils.import("resource://aifilter/modules/ExpressionSearchFilter.jsm");
|
|
gTerm = new mod.ClassificationTerm();
|
|
}
|
|
console.log("[ai-filter][api] calling gTerm.match()");
|
|
let matchResult = gTerm.match(
|
|
msg.msgHdr,
|
|
msg.value,
|
|
Ci.nsMsgSearchOp.Contains
|
|
);
|
|
console.log("[ai-filter][api] gTerm.match() returned:", matchResult);
|
|
return matchResult;
|
|
}
|
|
catch (err) {
|
|
console.error("[ai-filter][api] error in classify():", err);
|
|
throw err;
|
|
}
|
|
}
|
|
}
|
|
};
|
|
}
|
|
};
|