src/services/ImageRecognitionService.ts

/**
 * This file is part of SudoBot.
 *
 * Copyright (C) 2021-2024 OSN Developers.
 *
 * SudoBot is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * SudoBot is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
 */

import Tesseract, { createWorker } from "tesseract.js";
import Service from "../core/Service";
import { log } from "../utils/logger";

export const name = "imageRecognitionService";

export default class ImageRecognitionService extends Service {
    protected worker: Tesseract.Worker | null = null;
    protected timeout: NodeJS.Timeout | null = null;

    protected async createWorkerIfNeeded() {
        if (!this.worker && !this.timeout) {
            log("Spawning new tesseract worker for image recognition");
            this.worker = await createWorker("eng");
            this.setTimeout();
        } else if (this.worker && this.timeout) {
            log("Using existing tesseract worker for image recognition");
            clearTimeout(this.timeout);
            this.setTimeout();
        }
    }

    protected setTimeout() {
        this.timeout = setTimeout(() => {
            log("Terminating existing tesseract worker due to inactivity");
            this.worker?.terminate();
            this.timeout = null;
        }, 60_000);
    }

    async recognize(image: Tesseract.ImageLike) {
        await this.createWorkerIfNeeded();
        return this.worker!.recognize(image);
    }
}
1	/**
2	* This file is part of SudoBot.
3	*
4	* Copyright (C) 2021-2024 OSN Developers.
5	*
6	* SudoBot is free software; you can redistribute it and/or modify it
7	* under the terms of the GNU Affero General Public License as published by
8	* the Free Software Foundation, either version 3 of the License, or
9	* (at your option) any later version.
10	*
11	* SudoBot is distributed in the hope that it will be useful, but
12	* WITHOUT ANY WARRANTY; without even the implied warranty of
13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14	* GNU Affero General Public License for more details.
15	*
16	* You should have received a copy of the GNU Affero General Public License
17	* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
18	*/
19
20	import Tesseract, { createWorker } from "tesseract.js";
21	import Service from "../core/Service";
22	import { log } from "../utils/logger";
23
24	export const name = "imageRecognitionService";
25
26	export default class ImageRecognitionService extends Service {
27	protected worker: Tesseract.Worker \| null = null;
28	protected timeout: NodeJS.Timeout \| null = null;
29
30	protected async createWorkerIfNeeded() {
31	if (!this.worker && !this.timeout) {
32	log("Spawning new tesseract worker for image recognition");
33	this.worker = await createWorker("eng");
34	this.setTimeout();
35	} else if (this.worker && this.timeout) {
36	log("Using existing tesseract worker for image recognition");
37	clearTimeout(this.timeout);
38	this.setTimeout();
39	}
40	}
41
42	protected setTimeout() {
43	this.timeout = setTimeout(() => {
44	log("Terminating existing tesseract worker due to inactivity");
45	this.worker?.terminate();
46	this.timeout = null;
47	}, 60_000);
48	}
49
50	async recognize(image: Tesseract.ImageLike) {
51	await this.createWorkerIfNeeded();
52	return this.worker!.recognize(image);
53	}
54	}