352 lines
16 KiB
JavaScript
352 lines
16 KiB
JavaScript
'use strict';
|
|
const util = require('util');
|
|
var mysql = require('mysql');
|
|
var fs = require('fs');
|
|
const config = {
|
|
host: "localhost",
|
|
user: "root",
|
|
password: "",
|
|
database: "intops"
|
|
};
|
|
|
|
var currentIn = [];
|
|
var currentInOther = [];
|
|
var currentOut = [];
|
|
var currentOutOther = [];
|
|
var maxCompare = 5;
|
|
var maxFalse = 0;
|
|
var falseIn = 0;
|
|
var falseOut = 0;
|
|
|
|
function makeDb(config) {
|
|
const connection = mysql.createConnection(config);
|
|
return {
|
|
query(sql, args) {
|
|
return util.promisify(connection.query)
|
|
.call(connection, sql, args);
|
|
},
|
|
close() {
|
|
return util.promisify(connection.end).call(connection);
|
|
}
|
|
};
|
|
}
|
|
|
|
function levenshtein(a, b) {
|
|
if (a.length === 0)
|
|
return b.length;
|
|
if (b.length === 0)
|
|
return a.length;
|
|
|
|
var matrix = [];
|
|
|
|
// increment along the first column of each row
|
|
var i;
|
|
for (i = 0; i <= b.length; i++) {
|
|
matrix[i] = [i];
|
|
}
|
|
|
|
// increment each column in the first row
|
|
var j;
|
|
for (j = 0; j <= a.length; j++) {
|
|
matrix[0][j] = j;
|
|
}
|
|
|
|
// Fill in the rest of the matrix
|
|
for (i = 1; i <= b.length; i++) {
|
|
for (j = 1; j <= a.length; j++) {
|
|
if (b.charAt(i - 1) == a.charAt(j - 1)) {
|
|
matrix[i][j] = matrix[i - 1][j - 1];
|
|
} else {
|
|
matrix[i][j] = Math.min(matrix[i - 1][j - 1] + 1, // substitution
|
|
Math.min(matrix[i][j - 1] + 1, // insertion
|
|
matrix[i - 1][j] + 1)); // deletion
|
|
}
|
|
}
|
|
}
|
|
|
|
return matrix[b.length][a.length];
|
|
}
|
|
|
|
function isSavePlate(type, timestamp, plate) {
|
|
if (type === "in") {
|
|
for (var i = 0; i < currentIn.length; i++) {
|
|
if ((timestamp - currentIn[i].time) < 600) {
|
|
if (plate === currentIn[i].plate) {
|
|
return false;
|
|
} /*else if (levenshtein(plate, currentIn[i].plate) < 2) {
|
|
return false;
|
|
}*/
|
|
}
|
|
}
|
|
}
|
|
if (type === "out") {
|
|
for (var i = 0; i < currentOut.length; i++) {
|
|
if ((timestamp - currentOut[i].time) < 600) {
|
|
if (plate === currentOut[i].plate) {
|
|
return false;
|
|
} /*else if (levenshtein(plate, currentOut[i].plate) < 2) {
|
|
return false;
|
|
}*/
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
function isSavePlateOther(type, timestamp, plate) {
|
|
if (type === "in") {
|
|
for (var i = 0; i < currentInOther.length; i++) {
|
|
if ((timestamp - currentInOther[i].time) < 600) {
|
|
if (plate === currentInOther[i].plate) {
|
|
return false;
|
|
} else if (levenshtein(plate, currentInOther[i].plate) < 2) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
for (var i = 0; i < currentIn.length; i++) {
|
|
if ((timestamp - currentIn[i].time) < 600) {
|
|
if (plate === currentIn[i].plate) {
|
|
return false;
|
|
} else if (levenshtein(plate, currentIn[i].plate) < 3) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if (type === "out") {
|
|
for (var i = 0; i < currentOutOther.length; i++) {
|
|
if ((timestamp - currentOutOther[i].time) < 600) {
|
|
if (plate === currentOutOther[i].plate) {
|
|
return false;
|
|
} else if (levenshtein(plate, currentOutOther[i].plate) < 2) {
|
|
return false;
|
|
} else if (levenshtein(plate, currentOut[i].plate) < 3) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
for (var i = 0; i < currentOut.length; i++) {
|
|
if ((timestamp - currentOut[i].time) < 600) {
|
|
if (plate === currentOut[i].plate) {
|
|
return false;
|
|
} else if (levenshtein(plate, currentOut[i].plate) < 3) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
function consoleLogPlate(type, reqPlate, currentPlate, currentTimestamp, saveType) {
|
|
if (type === "in") {
|
|
var listsIn = [];
|
|
for (var i = 0; i < currentIn.length; i++) {
|
|
listsIn.push(currentIn[i].plate);
|
|
}
|
|
console.log("req: ", reqPlate, " ", currentPlate, " ", type, " ", saveType, " ", listsIn);
|
|
|
|
if (currentIn.length >= maxCompare) {
|
|
currentIn.shift();
|
|
}
|
|
currentIn.push({"plate": currentPlate, "time": currentTimestamp});
|
|
}
|
|
if (type === "out") {
|
|
var listsOut = [];
|
|
for (var i = 0; i < currentOut.length; i++) {
|
|
listsOut.push(currentOut[i].plate);
|
|
}
|
|
console.log("req: ", reqPlate, " ", currentPlate, " ", type, " ", saveType, " ", listsOut);
|
|
if (currentOut.length >= maxCompare) {
|
|
currentOut.shift();
|
|
}
|
|
currentOut.push({"plate": currentPlate, "time": currentTimestamp});
|
|
}
|
|
}
|
|
|
|
function consoleLogPlateOther(type, reqPlate, currentPlate, currentTimestamp, saveType) {
|
|
if (type === "in") {
|
|
var listsIn = [];
|
|
for (var i = 0; i < currentInOther.length; i++) {
|
|
listsIn.push(currentInOther[i].plate);
|
|
}
|
|
console.log("req: ", reqPlate, " ", currentPlate, " ", type, " ", saveType, " ", listsIn);
|
|
|
|
if (currentInOther.length >= maxCompare) {
|
|
currentInOther.shift();
|
|
}
|
|
currentInOther.push({"plate": currentPlate, "time": currentTimestamp});
|
|
}
|
|
if (type === "out") {
|
|
var listsOut = [];
|
|
for (var i = 0; i < currentOutOther.length; i++) {
|
|
listsOut.push(currentOutOther[i].plate);
|
|
}
|
|
console.log("req: ", reqPlate, " ", currentPlate, " ", type, " ", saveType, " ", listsOut);
|
|
if (currentOutOther.length >= maxCompare) {
|
|
currentOutOther.shift();
|
|
}
|
|
currentOutOther.push({"plate": currentPlate, "time": currentTimestamp});
|
|
}
|
|
}
|
|
|
|
const db = makeDb(config);
|
|
|
|
exports.SaveLogs = async function (req, res) {
|
|
|
|
var io = req.app.get('socketio');
|
|
|
|
var today = new Date();
|
|
var currentDate = today.getDate();
|
|
var currentMonth = today.getMonth() + 1;
|
|
var currentYear = today.getFullYear();
|
|
var rootDir = "/mnt/d/xampp/htdocs/AIParking_Intops_Server/web/data/uploads/";
|
|
|
|
if (!fs.existsSync(rootDir + currentYear))
|
|
fs.mkdirSync(rootDir + currentYear);
|
|
if (!fs.existsSync(rootDir + currentYear + "/" + currentMonth))
|
|
fs.mkdirSync(rootDir + currentYear + "/" + currentMonth);
|
|
if (!fs.existsSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate))
|
|
fs.mkdirSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate);
|
|
if (!fs.existsSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type))
|
|
fs.mkdirSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type);
|
|
if (!fs.existsSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type + "/plate"))
|
|
fs.mkdirSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type + "/plate");
|
|
if (!fs.existsSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type + "/frame"))
|
|
fs.mkdirSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type + "/frame");
|
|
var currentTimestamp = Math.round(today.getTime() / 1000);
|
|
|
|
var vehicleInfo = {"status": false};
|
|
|
|
var currentPlate = "";
|
|
var fileName = req.body.plate + "_" + currentTimestamp + ".png";
|
|
var plateSaved = currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type + "/plate/" + fileName;
|
|
var frameSaved = currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type + "/frame/" + fileName;
|
|
|
|
await fs.writeFileSync(rootDir + plateSaved, req.body.plate_image, 'base64');
|
|
await fs.writeFileSync(rootDir + frameSaved, req.body.frame_image, 'base64');
|
|
|
|
var query_plate_raw = await db.query("SELECT * FROM vehicle WHERE plate='" + req.body.plate + "'");
|
|
if (query_plate_raw.length > 0) {
|
|
vehicleInfo = {"status": true, "data": query_plate_raw[0]};
|
|
currentPlate = query_plate_raw[0].plate;
|
|
if (req.body.type == "in" && isSavePlate(req.body.type, currentTimestamp, currentPlate)) {
|
|
var newLogs = await db.query("INSERT INTO logs(`vehicle_id`,`plate_image_in`,`frame_image_in`,`time_in`) VALUES ('" + query_plate_raw[0].id + "','" + plateSaved + "','" + frameSaved + "'," + currentTimestamp + ")");
|
|
io.emit('logs', {
|
|
id: newLogs.insertId,
|
|
vehicleInfo: query_plate_raw[0],
|
|
type: "in",
|
|
image: plateSaved,
|
|
time: currentTimestamp
|
|
});
|
|
consoleLogPlate(req.body.type, req.body.plate, currentPlate, currentTimestamp, "absolute");
|
|
}
|
|
if (req.body.type == "out" && isSavePlate(req.body.type, currentTimestamp, currentPlate)) {
|
|
var checkIn = await db.query("SELECT * FROM logs WHERE vehicle_id=" + query_plate_raw[0].id + " AND time_out=0 ORDER BY time_in DESC LIMIT 0,1");
|
|
if (checkIn.length > 0) {
|
|
await db.query("UPDATE logs SET `plate_image_out`='" + plateSaved + "',`frame_image_out`='" + frameSaved + "',`time_out`=" + currentTimestamp + " WHERE id=" + checkIn[0].id);
|
|
io.emit('logs', {
|
|
id: checkIn[0].id,
|
|
vehicleInfo: query_plate_raw[0],
|
|
type: "out",
|
|
image: plateSaved,
|
|
time: currentTimestamp,
|
|
logs: checkIn[0]
|
|
});
|
|
} else {
|
|
var newLogs = await db.query("INSERT INTO logs(`vehicle_id`,`plate_image_out`,`frame_image_out`,`time_out`) VALUES ('" + query_plate_raw[0].id + "','" + plateSaved + "','" + frameSaved + "'," + currentTimestamp + ")");
|
|
io.emit('logs', {
|
|
id: newLogs.insertId,
|
|
vehicleInfo: query_plate_raw[0],
|
|
type: "out",
|
|
image: plateSaved,
|
|
time: currentTimestamp,
|
|
logs: false
|
|
});
|
|
}
|
|
consoleLogPlate(req.body.type, req.body.plate, currentPlate, currentTimestamp, "absolute");
|
|
}
|
|
} else {
|
|
var query_plate_levenshtein = await db.query("SELECT * FROM `vehicle` WHERE LENGTH(`plate`)=" + req.body.plate.length + " AND levenshtein('" + req.body.plate + "', `plate`) BETWEEN 0 AND 1 LIMIT 0,1");
|
|
if (query_plate_levenshtein.length > 0) {
|
|
vehicleInfo = {"status": true, "data": query_plate_levenshtein[0]};
|
|
currentPlate = query_plate_levenshtein[0].plate;
|
|
if (req.body.type == "in" && isSavePlate(req.body.type, currentTimestamp, currentPlate)) {
|
|
var newLogs = await db.query("INSERT INTO logs(`vehicle_id`,`plate_image_in`,`frame_image_in`,`time_in`) VALUES ('" + query_plate_levenshtein[0].id + "','" + plateSaved + "','" + frameSaved + "'," + currentTimestamp + ")");
|
|
io.emit('logs', {
|
|
id: newLogs.insertId,
|
|
vehicleInfo: query_plate_levenshtein[0],
|
|
type: "in",
|
|
image: plateSaved,
|
|
time: currentTimestamp
|
|
});
|
|
consoleLogPlate(req.body.type, req.body.plate, currentPlate, currentTimestamp, "levenshtein");
|
|
}
|
|
if (req.body.type == "out" && isSavePlate(req.body.type, currentTimestamp, currentPlate)) {
|
|
var checkIn = await db.query("SELECT * FROM logs WHERE vehicle_id=" + query_plate_levenshtein[0].id + " AND time_out=0 ORDER BY time_in DESC LIMIT 0,1");
|
|
if (checkIn.length > 0) {
|
|
await db.query("UPDATE logs SET `plate_image_out`='" + plateSaved + "',`frame_image_out`='" + frameSaved + "',`time_out`=" + currentTimestamp + " WHERE id=" + checkIn[0].id);
|
|
io.emit('logs', {
|
|
id: checkIn[0].id,
|
|
vehicleInfo: query_plate_levenshtein[0],
|
|
type: "out",
|
|
image: plateSaved,
|
|
time: currentTimestamp,
|
|
logs: checkIn[0]
|
|
});
|
|
} else {
|
|
var newLogs = await db.query("INSERT INTO logs(`vehicle_id`,`plate_image_out`,`frame_image_out`,`time_out`) VALUES ('" + query_plate_levenshtein[0].id + "','" + plateSaved + "','" + frameSaved + "'," + currentTimestamp + ")");
|
|
io.emit('logs', {
|
|
id: newLogs.insertId,
|
|
vehicleInfo: query_plate_levenshtein[0],
|
|
type: "out",
|
|
image: plateSaved,
|
|
time: currentTimestamp,
|
|
logs: false
|
|
});
|
|
}
|
|
consoleLogPlate(req.body.type, req.body.plate, currentPlate, currentTimestamp, "levenshtein");
|
|
}
|
|
|
|
} else {
|
|
currentPlate = req.body.plate;
|
|
if (req.body.type == "in" && isSavePlateOther(req.body.type, currentTimestamp, currentPlate)) {
|
|
if (falseIn < maxFalse) {
|
|
vehicleInfo = {"status": false};
|
|
falseIn++;
|
|
console.log("status: results in not found");
|
|
} else {
|
|
falseIn = 0;
|
|
vehicleInfo = {"status": true};
|
|
await db.query("INSERT INTO logs_unknow(`plate`,`plate_image_in`,`frame_image_in`,`time_in`) VALUES ('" + req.body.plate + "','" + plateSaved + "','" + frameSaved + "'," + currentTimestamp + ")");
|
|
consoleLogPlateOther(req.body.type, req.body.plate, currentPlate, currentTimestamp, "other");
|
|
}
|
|
}
|
|
if (req.body.type == "out" && isSavePlateOther(req.body.type, currentTimestamp, currentPlate)) {
|
|
if (falseOut < maxFalse) {
|
|
vehicleInfo = {"status": false};
|
|
falseOut++;
|
|
console.log("status: results out not found");
|
|
} else {
|
|
falseIn = 0;
|
|
vehicleInfo = {"status": true};
|
|
var checkIn_raw = await db.query("SELECT * FROM logs_unknow WHERE plate='" + req.body.plate + "' AND time_out=0 ORDER BY time_in DESC LIMIT 0,1");
|
|
if (checkIn_raw.length > 0) {
|
|
await db.query("UPDATE logs_unknow SET `plate_image_out`='" + plateSaved + "',`frame_image_out`='" + frameSaved + "',`time_out`=" + currentTimestamp + " WHERE id=" + checkIn_raw[0].id);
|
|
} else {
|
|
var checkIn_levenshtein = await db.query("SELECT * FROM `logs_unknow` WHERE levenshtein('" + req.body.plate + "', `plate`) BETWEEN 0 AND 1 AND time_out=0 ORDER BY time_in DESC LIMIT 0,1");
|
|
if (checkIn_levenshtein.length > 0) {
|
|
await db.query("UPDATE logs_unknow SET `plate_image_out`='" + plateSaved + "',`frame_image_out`='" + frameSaved + "',`time_out`=" + currentTimestamp + " WHERE id=" + checkIn_levenshtein[0].id);
|
|
} else {
|
|
await db.query("INSERT INTO logs_unknow(`plate`,`plate_image_out`,`frame_image_out`,`time_out`) VALUES ('" + req.body.plate + "','" + plateSaved + "','" + frameSaved + "'," + currentTimestamp + ")");
|
|
}
|
|
}
|
|
consoleLogPlateOther(req.body.type, req.body.plate, currentPlate, currentTimestamp, "other");
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
res.send(vehicleInfo);
|
|
}; |