@riavzon/bot-detector 1.0.5 → 1.0.8

package/dist/main.cjs

@@ -3120,6 +3120,83 @@ async function runGeneration() {
 	return runCycle();
 }
 
+//#endregion
+//#region src/botDetector/db/schema.ts
+function visitorIdDefault(db) {
+	if (isMySQL(db)) return "NOT NULL DEFAULT (UUID())";
+	if (db.dialect === "postgresql") return "NOT NULL DEFAULT gen_random_uuid()";
+	return "NOT NULL";
+}
+function lastSeenDef(db) {
+	if (isMySQL(db)) return "TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP";
+	if (isSQLite(db)) return "TEXT DEFAULT CURRENT_TIMESTAMP";
+	return "TIMESTAMP DEFAULT CURRENT_TIMESTAMP";
+}
+function tableOptions(db) {
+	return isMySQL(db) ? "ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci" : "";
+}
+function timestampType(db) {
+	return isSQLite(db) ? "TEXT" : "TIMESTAMP";
+}
+async function createTables(db) {
+	const visitorId = visitorIdDefault(db);
+	const lastSeen = lastSeenDef(db);
+	const tblOpts = tableOptions(db);
+	const createVisitorsTable = `
+        CREATE TABLE IF NOT EXISTS visitors (
+            visitor_id CHAR(36) ${visitorId},
+            canary_id VARCHAR(64) PRIMARY KEY,
+            ip_address VARCHAR(45),
+            user_agent TEXT,
+            country VARCHAR(64),
+            region VARCHAR(64),
+            region_name VARCHAR(350),
+            city VARCHAR(64),
+            district VARCHAR(260),
+            lat VARCHAR(150),
+            lon VARCHAR(150),
+            timezone VARCHAR(64),
+            currency VARCHAR(64),
+            isp VARCHAR(64),
+            org VARCHAR(64),
+            as_org VARCHAR(64),
+            device_type VARCHAR(64),
+            browser VARCHAR(64),
+            proxy BOOLEAN,
+            hosting BOOLEAN,
+            is_bot BOOLEAN DEFAULT false,
+            first_seen ${timestampType(db)} DEFAULT CURRENT_TIMESTAMP,
+            last_seen ${lastSeen},
+            request_count INT DEFAULT 1,
+            deviceVendor VARCHAR(64) DEFAULT 'unknown',
+            deviceModel VARCHAR(64) DEFAULT 'unknown',
+            browserType VARCHAR(64) DEFAULT 'unknown',
+            browserVersion VARCHAR(64) DEFAULT 'unknown',
+            os VARCHAR(64) DEFAULT 'unknown',
+            suspicious_activity_score INT DEFAULT 0
+        ) ${tblOpts}
+    `;
+	const createBannedTable = `
+        CREATE TABLE IF NOT EXISTS banned (
+            canary_id VARCHAR(64) PRIMARY KEY,
+            ip_address VARCHAR(45),
+            country VARCHAR(64),
+            user_agent TEXT,
+            reason TEXT,
+            score INT DEFAULT NULL,
+            FOREIGN KEY (canary_id) REFERENCES visitors(canary_id)
+        )
+    `;
+	try {
+		await db.exec(createVisitorsTable);
+		await db.exec(createBannedTable);
+		consola.default.success("Tables created successfully.");
+	} catch (error) {
+		consola.default.error("Error creating tables:", error);
+		throw error;
+	}
+}
+
 //#endregion
 //#region src/botDetector/db/warmUp.ts
 async function warmUp() {
@@ -3221,10 +3298,12 @@ exports.BadBotDetected = BadBotDetected;
 exports.CheckerRegistry = CheckerRegistry;
 exports.GoodBotDetected = GoodBotDetected;
 exports.banIp = banIp;
+exports.createTables = createTables;
 exports.defineConfiguration = configuration;
 exports.detectBots = validator;
 exports.getBatchQueue = getBatchQueue;
 exports.getDataSources = getDataSources;
+exports.getDb = getDb;
 exports.getGeoData = getData;
 exports.getStorage = getStorage;
 exports.parseUA = parseUA;