@seasketch/geoprocessing/dist/scripts/base/datasources/metrics.js

Geoprocessing and reporting framework for SeaSketch 2.0

import fs from "fs-extra";
import path from "node:path";
import { metricsSchema } from "../../../src/types/index.js";
import { precalcConfig } from "../../../src/precalc/config.js";
import { sortMetrics, rekeyMetrics } from "../../../src/metrics/helpers.js";
/**
 * Manage a metrics datasource
 */
/**
 * Reads metrics from disk, validates them
 * If metrics file not exist then start a new one and ensure directory exists
 */
export function readMetrics(filePath) {
    const metrics = [];
    const diskPrecalc = (() => {
        try {
            const dsString = fs.readFileSync(filePath).toString();
            try {
                return JSON.parse(dsString);
            }
            catch {
                throw new Error(`Unable to parse JSON found in ${filePath}, fix it and try again`);
            }
        }
        catch {
            console.log(`Precalc file not found at ${filePath}, creating a new one`);
            fs.ensureDirSync(path.dirname(precalcConfig.defaultSrcPath));
            // fallback to default
            return metrics;
        }
    })();
    const result = metricsSchema.safeParse(diskPrecalc);
    if (result.success) {
        return result.data;
    }
    else {
        console.error(`${filePath} is invalid, either you need to delete it and run precalc, or manually fix it`);
        console.log(JSON.stringify(result.error.issues, null, 2));
        throw new Error("Please fix or report this issue");
    }
}
/**
 * Writes metrics to disk, sorting and rekeying before writing easier diffing
 */
export function writeMetrics(metrics, filePath) {
    const safeMetrics = metricsSchema.safeParse(metrics);
    if (safeMetrics.success) {
        fs.writeJSONSync(filePath, sortMetrics(rekeyMetrics(metrics)), {
            spaces: 2,
        });
    }
    else {
        console.error(`precalculated metrics are invalid, this is a bug with the precalc script`);
        console.log(JSON.stringify(safeMetrics.error.issues, null, 2));
        throw new Error("Please fix or report this issue");
    }
}
/**
 * Creates or updates metrics on disk, sorting and rekeying before writing for easier diffing
 * @param inputMetrics - new metrics to add
 * @param matcher - old metrics to replace
 * @param filePath - path to metrics file
 * @returns - full set of updated metrics
 */
export async function createOrUpdateMetrics(inputMetrics, matcher, filePath) {
    let metrics = readMetrics(filePath);
    if (matcher) {
        metrics = metrics.filter(matcher);
    }
    metrics = sortMetrics(rekeyMetrics(metrics.concat(inputMetrics)));
    writeMetrics(metrics, filePath);
    return metrics;
}
//# sourceMappingURL=metrics.js.map