0
0
mirror of https://github.com/Maks1mS/free-ozon-dpr.git synced 2024-12-23 18:42:59 +03:00
free-ozon-dpr/scripts/merge-data.js

78 lines
1.8 KiB
JavaScript
Raw Permalink Normal View History

2024-06-05 14:56:19 +03:00
import fs from "node:fs/promises";
import { asyncMap } from "modern-async";
import { getDistance } from "ol/sphere.js";
import { extractIDFromURL } from "./utils.js";
2024-06-05 14:56:19 +03:00
function removeDuplicatesByUrl(points) {
const uniquePoints = [];
const urlSet = new Set();
points.forEach((points) => {
let isDuplicate = false;
const link = points.link;
if (urlSet.has(link)) {
isDuplicate = true;
}
if (!isDuplicate) {
uniquePoints.push(points);
urlSet.add(link);
}
});
return uniquePoints;
}
function removeDuplicatesByRadius(points, radius) {
const uniquePoints = [];
const coordinatesSet = new Set();
points.forEach((point) => {
const lonLat = point.coordinates;
let isDuplicate = false;
coordinatesSet.forEach((setCoordinates) => {
if (getDistance(setCoordinates, lonLat) <= radius) {
isDuplicate = true;
}
});
if (!isDuplicate) {
uniquePoints.push(point);
coordinatesSet.add(lonLat);
}
});
return uniquePoints;
}
function convert(data) {
return data.points.map((p) => ({
...p,
source: data.source,
provider: data.name,
}));
}
async function main() {
const dataDir = await fs.readdir("./data");
let data = await asyncMap(dataDir, async (filename) => {
const fileContent = await fs.readFile(`./data/${filename}`);
const data = JSON.parse(fileContent);
return convert(data);
});
data = data.flatMap((v) => v);
data = removeDuplicatesByUrl(data);
data = removeDuplicatesByRadius(data, 10);
// pick all except url
data = data.map(x => {
const { link, ...rest } = x;
return { id: extractIDFromURL(link), ...rest };
});
2024-06-05 14:56:19 +03:00
await fs.writeFile("merged-data.json", JSON.stringify(data, undefined, 2));
}
main();