203 lines
7.7 KiB
TypeScript
203 lines
7.7 KiB
TypeScript
import path from "node:path";
|
|
import {saveData} from "../db/database";
|
|
import {Answers} from "../model/answers";
|
|
import {ifTrue} from "../util/utils";
|
|
import {AiProvider} from "../model/ai-provider";
|
|
import {ImageHandleFallbackPolicy, ImageHandlePolicy, RateLimitFallbackPolicy} from "./policies";
|
|
|
|
export class Environment {
|
|
static BOT_TOKEN: string;
|
|
static TEST_ENVIRONMENT: boolean;
|
|
static ADMIN_IDS: Set<number> = new Set<number>();
|
|
static MUTED_IDS: Set<number> = new Set<number>();
|
|
static CHAT_IDS_WHITELIST: Set<number> = new Set<number>();
|
|
static BOT_PREFIX: string;
|
|
static CREATOR_ID: number;
|
|
static IS_DOCKER: boolean;
|
|
static DATA_PATH: string;
|
|
static DB_FILE_NAME: string = "database.db";
|
|
static DB_PATH: string;
|
|
|
|
static ONLY_FOR_CREATOR_MODE: boolean;
|
|
|
|
static ENABLE_UNSAFE_EVAL: boolean;
|
|
|
|
static ANSWERS: Answers;
|
|
|
|
static USE_NAMES_IN_PROMPT: boolean;
|
|
|
|
static MAX_PHOTO_SIZE: number;
|
|
|
|
static PROCESS_LINKS: boolean;
|
|
|
|
static DEFAULT_AI_PROVIDER: AiProvider;
|
|
|
|
static RATE_LIMIT_FALLBACK_POLICY: RateLimitFallbackPolicy;
|
|
static IMAGE_HANDLE_POLICY: ImageHandlePolicy;
|
|
static IMAGE_HANDLE_FALLBACK_POLICY: ImageHandleFallbackPolicy;
|
|
|
|
static SYSTEM_PROMPT?: string;
|
|
static SEND_TIME_TOOK: boolean;
|
|
|
|
static OLLAMA_ADDRESS?: string;
|
|
static OLLAMA_MODEL?: string;
|
|
static OLLAMA_IMAGE_MODEL?: string;
|
|
static OLLAMA_THINK_MODEL?: string;
|
|
static OLLAMA_API_KEY?: string;
|
|
|
|
static GEMINI_API_KEY?: string;
|
|
static GEMINI_MODEL: string;
|
|
static GEMINI_IMAGE_MODEL: string;
|
|
|
|
static MISTRAL_API_KEY?: string;
|
|
static MISTRAL_MODEL: string;
|
|
|
|
static OPENAI_BASE_URL?: string;
|
|
static OPENAI_API_KEY?: string;
|
|
static OPENAI_MODEL: string;
|
|
static OPENAI_IMAGE_MODEL: string;
|
|
|
|
static errorText = "⚠️ Произошла ошибка.";
|
|
static waitText = "⏳ Секунду...";
|
|
static waitThinkText = "⏳ Дайте-ка подумать...";
|
|
static analyzingPictureText = "🔍 Внимательно изучаю изображение...";
|
|
static analyzingPicturesText = "🔍 Внимательно изучаю изображения...";
|
|
static genImageText = "👨🎨 Генерирую изображение...";
|
|
static ollamaCancelledText = "```Ollama\n❌ Отменено```";
|
|
|
|
static load() {
|
|
Environment.BOT_TOKEN = process.env.BOT_TOKEN;
|
|
Environment.TEST_ENVIRONMENT = ifTrue(process.env.TEST_ENVIRONMENT);
|
|
Environment.CHAT_IDS_WHITELIST = new Set(process.env.CHAT_IDS_WHITELIST?.split(",")?.map(e => parseInt(e.trim(), 10)) || []);
|
|
Environment.BOT_PREFIX = process.env.BOT_PREFIX || "";
|
|
Environment.CREATOR_ID = parseInt(process.env.CREATOR_ID || "");
|
|
Environment.IS_DOCKER = ifTrue(process.env.IS_DOCKER);
|
|
Environment.DATA_PATH = Environment.IS_DOCKER ? "/" + path.join("config", "data") : "data";
|
|
Environment.DB_PATH = "file:" + path.join(Environment.DATA_PATH, Environment.DB_FILE_NAME);
|
|
|
|
Environment.ONLY_FOR_CREATOR_MODE = ifTrue(process.env.ONLY_FOR_CREATOR_MODE);
|
|
|
|
Environment.ENABLE_UNSAFE_EVAL = ifTrue(process.env.ENABLE_UNSAFE_EVAL);
|
|
|
|
Environment.USE_NAMES_IN_PROMPT = ifTrue(process.env.USE_NAMES_IN_PROMPT);
|
|
|
|
Environment.MAX_PHOTO_SIZE = Number(process.env.MAX_PHOTO_SIZE || "1280");
|
|
|
|
Environment.PROCESS_LINKS = ifTrue(process.env.PROCESS_LINKS);
|
|
|
|
const aiProvider = process.env.DEFAULT_AI_PROVIDER || "OLLAMA";
|
|
if (Object.values(AiProvider).includes(aiProvider as AiProvider)) {
|
|
Environment.DEFAULT_AI_PROVIDER = aiProvider as AiProvider;
|
|
} else {
|
|
Environment.DEFAULT_AI_PROVIDER = AiProvider.OLLAMA;
|
|
}
|
|
|
|
const rateLimitFallbackPolicy = process.env.RATE_LIMIT_FALLBACK_POLICY || "NOTIFY_USER";
|
|
if (Object.values(RateLimitFallbackPolicy).includes(rateLimitFallbackPolicy as RateLimitFallbackPolicy)) {
|
|
Environment.RATE_LIMIT_FALLBACK_POLICY = rateLimitFallbackPolicy as RateLimitFallbackPolicy;
|
|
} else {
|
|
Environment.RATE_LIMIT_FALLBACK_POLICY = RateLimitFallbackPolicy.NOTIFY_USER;
|
|
}
|
|
|
|
const imageHandlePolicy = process.env.IMAGE_HANDLE_POLICY || "HANDLE_IF_CAPABLE";
|
|
if (Object.values(ImageHandlePolicy).includes(imageHandlePolicy as ImageHandlePolicy)) {
|
|
Environment.IMAGE_HANDLE_POLICY = imageHandlePolicy as ImageHandlePolicy;
|
|
} else {
|
|
Environment.IMAGE_HANDLE_POLICY = ImageHandlePolicy.HANDLE_IF_CAPABLE;
|
|
}
|
|
|
|
const imageHandleFallbackPolicy = process.env.IMAGE_HANDLE_FALLBACK_POLICY || "NOTIFY_USER";
|
|
if (Object.values(ImageHandleFallbackPolicy).includes(imageHandleFallbackPolicy as ImageHandleFallbackPolicy)) {
|
|
Environment.IMAGE_HANDLE_FALLBACK_POLICY = imageHandleFallbackPolicy as ImageHandleFallbackPolicy;
|
|
} else {
|
|
Environment.IMAGE_HANDLE_FALLBACK_POLICY = ImageHandleFallbackPolicy.NOTIFY_USER;
|
|
}
|
|
|
|
Environment.SEND_TIME_TOOK = ifTrue(process.env.SEND_TOOK_TIME || false);
|
|
|
|
Environment.OLLAMA_ADDRESS = process.env.OLLAMA_ADDRESS;
|
|
Environment.OLLAMA_MODEL = process.env.OLLAMA_MODEL || "gemma3:4b";
|
|
Environment.OLLAMA_IMAGE_MODEL = process.env.OLLAMA_IMAGE_MODEL || Environment.OLLAMA_MODEL;
|
|
Environment.OLLAMA_THINK_MODEL = process.env.OLLAMA_THINK_MODEL || Environment.OLLAMA_MODEL;
|
|
Environment.OLLAMA_API_KEY = process.env.OLLAMA_API_KEY;
|
|
|
|
Environment.GEMINI_API_KEY = process.env.GEMINI_API_KEY;
|
|
Environment.GEMINI_MODEL = process.env.GEMINI_MODEL || "gemini-2.5-flash-lite";
|
|
Environment.GEMINI_IMAGE_MODEL = process.env.GEMINI_IMAGE_MODEL || "gemini-2.5-flash-image";
|
|
|
|
Environment.MISTRAL_API_KEY = process.env.MISTRAL_API_KEY;
|
|
Environment.MISTRAL_MODEL = process.env.MISTRAL_MODEL || "mistral-tiny-latest";
|
|
|
|
Environment.OPENAI_BASE_URL = process.env.OPENAI_BASE_URL;
|
|
Environment.OPENAI_API_KEY = process.env.OPENAI_API_KEY;
|
|
Environment.OPENAI_MODEL = process.env.OPENAI_MODEL || "gpt-4.1-nano";
|
|
Environment.OPENAI_IMAGE_MODEL = process.env.OPENAI_IMAGE_MODEL || "gpt-image-1-mini";
|
|
}
|
|
|
|
static setSystemPrompt(prompt: string) {
|
|
this.SYSTEM_PROMPT = prompt;
|
|
}
|
|
|
|
static setAdmins(admins: Set<number>) {
|
|
this.ADMIN_IDS = admins;
|
|
}
|
|
|
|
static async addAdmin(id: number): Promise<boolean> {
|
|
const has = this.ADMIN_IDS.has(id);
|
|
if (!has) {
|
|
this.ADMIN_IDS.add(id);
|
|
await saveData();
|
|
}
|
|
|
|
return !has;
|
|
}
|
|
|
|
static async removeAdmin(id: number): Promise<boolean> {
|
|
const has = this.ADMIN_IDS.has(id);
|
|
if (has) {
|
|
this.ADMIN_IDS.delete(id);
|
|
await saveData();
|
|
}
|
|
|
|
return has;
|
|
}
|
|
|
|
static setMuted(muted: Set<number>) {
|
|
this.MUTED_IDS = muted;
|
|
}
|
|
|
|
static async addMute(id: number): Promise<boolean> {
|
|
if (this.MUTED_IDS.has(id)) return Promise.resolve(false);
|
|
|
|
this.MUTED_IDS.add(id);
|
|
await saveData();
|
|
return Promise.resolve(true);
|
|
}
|
|
|
|
static async removeMute(id: number): Promise<boolean> {
|
|
if (!this.MUTED_IDS.has(id)) return Promise.resolve(false);
|
|
this.MUTED_IDS.delete(id);
|
|
await saveData();
|
|
return Promise.resolve(true);
|
|
}
|
|
|
|
static setAnswers(answers: Answers) {
|
|
this.ANSWERS = answers;
|
|
}
|
|
|
|
static setOllamaModel(newModel: string) {
|
|
Environment.OLLAMA_MODEL = newModel;
|
|
}
|
|
|
|
static setGeminiModel(newModel: string) {
|
|
Environment.GEMINI_MODEL = newModel;
|
|
}
|
|
|
|
static setMistralModel(newModel: string) {
|
|
Environment.MISTRAL_MODEL = newModel;
|
|
}
|
|
|
|
static setOpenAIModel(newModel: string) {
|
|
Environment.OPENAI_MODEL = newModel;
|
|
}
|
|
} |