Compare commits

..

No commits in common. "Intops_API" and "master" have entirely different histories.

4 changed files with 71 additions and 345 deletions

View File

@ -1,352 +1,81 @@
'use strict'; 'use strict';
const util = require('util');
var mysql = require('mysql'); var mysql = require('mysql');
var fs = require('fs'); const random = require('random');
const config = { var fs = require("fs");
var check;
var con = mysql.createConnection({
host: "localhost", host: "localhost",
user: "root", user: "root",
password: "", password: "",
database: "intops" 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) { // console.log(sql_levenshtein_cmd);
if (a.length === 0) con.connect(function (err) {
return b.length; if (err) throw err;
if (b.length === 0) console.log("Connected!!!")
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) { exports.SaveLogs = async function (req, res) {
var sql_cmd = "SELECT * FROM vehicle WHERE plate=\"" + req.body.plate + "\"";
var sql_query_cmd = "select * from vehicle where levenshtein('" + req.body.plate + "',`plate`) BETWEEN 0 AND 2";
console.log("req: ", req.body.plate);
con.connect(function (err) {
// if (err)
// throw err;
if (check != req.body.plate) {
con.query(sql_query_cmd, function (err, result, fields) {
console.log("levenshtein");
console.log(result);
var result_vehicle_id;
var sql_insert_logs_cmd, sql_insert_logs_unknow_cmd;
var io = req.app.get('socketio'); //if levenshtein not found plate
if (result == "") {
console.log("Nothing to show");
//timestamp in seconds
var today = new Date();
var currentTimeInInt = Math.round(today.getTime() / 1000);
sql_insert_logs_unknow_cmd = "INSERT INTO logs_unknow(`id`,`plate`,`plate_image_in`,`frame_image_in`,`time_in`,`plate_image_out`,`frame_image_out`,`time_out`,`seal_no`,`note`,`factory`) VALUES (" + random.int(1, 1000) + ",'" + req.body.plate + "','1','1'," + currentTimeInInt + ",'1','1',11,'1','1','1')";
con.query(sql_insert_logs_unknow_cmd, function (err, result, fields) {
if (err)
throw err;
});
console.log("Log unknown done");
}
var today = new Date(); //if levenshtein found plate
var currentDate = today.getDate(); else {
var currentMonth = today.getMonth() + 1; console.log("Log to logs table");
var currentYear = today.getFullYear(); //get vehicle id in vehicle table
var rootDir = "/mnt/d/xampp/htdocs/AIParking_Intops_Server/web/data/uploads/"; con.query("SELECT id FROM vehicle where `plate`='" + result[0].plate + "'", function (err, result_vehicle_id) {
if (err)
if (!fs.existsSync(rootDir + currentYear)) throw err;
fs.mkdirSync(rootDir + currentYear); console.log("result_vehicle_id");
if (!fs.existsSync(rootDir + currentYear + "/" + currentMonth)) console.log(result_vehicle_id[0].id);
fs.mkdirSync(rootDir + currentYear + "/" + currentMonth); var today = new Date();
if (!fs.existsSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate)) var currentTimeInInt = Math.round(today.getTime() / 1000);
fs.mkdirSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate); sql_insert_logs_cmd = "INSERT INTO logs(`id`,`vehicle_id`,`plate_image_in`,`frame_image_in`,`time_in`,`plate_image_out`,`frame_image_out`,`time_out`,`seal_no`,`note`,`factory`) VALUES (" + random.int(1, 1000) + "," + result_vehicle_id[0].id + ",'1','1'," + currentTimeInInt + ",'1','1',11,'1','1','1')";
if (!fs.existsSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type)) con.query(sql_insert_logs_cmd, function (err, result, fields) {
fs.mkdirSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type); if (err)
if (!fs.existsSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type + "/plate")) throw err;
fs.mkdirSync(rootDir + currentYear + "/" + currentMonth + "/" + currentDate + "/" + req.body.type + "/plate"); console.log("Log done");
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"); const base64Image = req.body.plate_image;
const imageBuffer = new Buffer(base64Image, "base64");
fs.writeFileSync("image.jpg", imageBuffer);
console.log("wrote image");
check = req.body.plate;
} }
if (req.body.type == "out" && isSavePlate(req.body.type, currentTimestamp, currentPlate)) { else {
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"); console.log("conflic plate");
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 { res.send({ "status": "Done!" });
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);
}; };

View File

@ -8,7 +8,11 @@
}, },
"dependencies": { "dependencies": {
"cookie-parser": "^1.4.3", "cookie-parser": "^1.4.3",
"express": "^4.13.3", "cors": "^2.8.5",
"express": "^4.17.1",
"jsonwebtoken": "^8.5.1",
"mysql": "^2.18.1",
"random": "^2.1.1",
"socket.io": "^2.1.1" "socket.io": "^2.1.1"
}, },
"engines": { "engines": {

View File

@ -6,6 +6,7 @@ var express = require('express'),
port = 4004, port = 4004,
bodyParser = require('body-parser'), bodyParser = require('body-parser'),
jsonwebtoken = require("jsonwebtoken"); jsonwebtoken = require("jsonwebtoken");
app.use(bodyParser.urlencoded({extended: true})); app.use(bodyParser.urlencoded({extended: true}));
app.use(bodyParser.json({limit: '50mb'})); app.use(bodyParser.json({limit: '50mb'}));
@ -33,14 +34,6 @@ app.use(function (req, res) {
const server = app.listen(port); const server = app.listen(port);
const io = require('socket.io').listen(server);
app.set('socketio', io);
io.sockets.on('connection', function (socket) {
socket.on('logs', function (msg) {
socket.broadcast.emit(msg); // Send message to everyone BUT sender
});
});
console.log('AIParking API Server Started On Port: ' + port); console.log('AIParking API Server Started On Port: ' + port);
module.exports = app; module.exports = app;