@seasketch/geoprocessing/dist/scripts/dataPrep/precalcData.js

Geoprocessing and reporting framework for SeaSketch 2.0

import inquirer from "inquirer";
import { precalcDatasources, } from "../base/datasources/index.js";
import { getProjectClient } from "../base/project/projectClient.js";
// This is a standalone script used as a CLI command with a top-level function
const projectPath = process.argv[2];
const projectClient = getProjectClient(projectPath);
const numDs = projectClient.datasources.filter((d) => d.precalc === true).length;
const numGeos = projectClient.geographies.filter((g) => g.precalc === true).length;
if (numDs === 0) {
    console.error("No precalc-able datasources found, exiting");
    process.exit();
}
if (numGeos === 0) {
    console.error("No precalc-able geographies found, exiting");
    process.exit();
}
const subsetAnswer = await precalcSubsetQuestion();
const dsOptions = {
    datasourceMatcher: [],
    geographyMatcher: [],
};
if (["all", "both", "datasource"].includes(subsetAnswer.subset)) {
    if (subsetAnswer.subset === "all") {
        dsOptions.datasourceMatcher = ["*"];
    }
    else {
        const precalcDs = projectClient.datasources.filter((ds) => ds.precalc === true);
        // Ask user what they want to precalculate
        const precalcDsAnswers = await precalcWhichDsQuestion(numDs);
        if (precalcDsAnswers.precalcWhichDs === "list") {
            const dsAnswers = await datasourcesQuestion(precalcDs);
            dsOptions.datasourceMatcher = dsAnswers.datasources;
        }
    }
}
const geogOptions = {};
if (["all", "both", "geography"].includes(subsetAnswer.subset)) {
    if (subsetAnswer.subset === "all") {
        dsOptions.geographyMatcher = ["*"];
    }
    else {
        // Ask user what they want to precalculate
        const precalcGeosAnswers = await precalcWhichGeosQuestion();
        if (precalcGeosAnswers.precalcWhichGeos === "list") {
            const geogAnswers = await geographiesQuestion(projectClient.geographies);
            geogOptions.geographyMatcher = geogAnswers.geographies;
        }
    }
}
// Then precalculate
await precalcDatasources(projectClient, { ...dsOptions, ...geogOptions });
export async function precalcSubsetQuestion() {
    return inquirer.prompt([
        {
            type: "list",
            name: "subset",
            message: `Do you want to precalculate only a subset?`,
            default: "datasource",
            choices: [
                {
                    value: "datasource",
                    name: "Yes, by datasource",
                },
                {
                    value: "geography",
                    name: "Yes, by geography",
                },
                {
                    value: "both",
                    name: `Yes, by both`,
                },
                {
                    value: "all",
                    name: "No, just precalculate everything (may take a while)",
                },
            ],
        },
    ]);
}
export async function precalcWhichDsQuestion(numDs) {
    return inquirer.prompt([
        {
            type: "list",
            name: "precalcWhichDs",
            message: `Which datasources do you want to precalculate? (will precalculate for all geographies)`,
            default: "list",
            choices: [
                {
                    value: "list",
                    name: "Let me choose",
                },
                {
                    value: "all",
                    name: `All ${numDs} datasources`,
                },
            ],
        },
    ]);
}
export async function precalcWhichGeosQuestion() {
    return inquirer.prompt([
        {
            type: "list",
            name: "precalcWhichGeos",
            message: `Which geographies do you want to precalculate? (will precalculate for all datasources)`,
            default: "list",
            choices: [
                {
                    value: "list",
                    name: "Let me choose",
                },
                {
                    value: "all",
                    name: `All ${numGeos} geographies`,
                },
            ],
        },
    ]);
}
export async function datasourcesQuestion(datasources) {
    const datasourcesQuestion = await getDatasourcesQuestion(datasources);
    const answer = await inquirer.prompt([
        datasourcesQuestion,
    ]);
    return answer;
}
export async function getDatasourcesQuestion(datasources) {
    if (datasources.length === 0) {
        console.error("No datasources found, exiting");
        process.exit();
    }
    const datasourceQuestion = {
        type: "checkbox",
        name: "datasources",
        message: "What datasources would you like to precalculate? (select as many as you want)",
        choices: [],
    };
    return {
        ...datasourceQuestion,
        choices: [
            ...datasourceQuestion.choices,
            ...datasources.map((ds) => ({
                value: ds.datasourceId,
                name: `${ds.datasourceId} - ${ds.geo_type}`,
            })),
        ],
    };
}
export async function geographiesQuestion(geographies) {
    const geographiesQuestion = await getGeographiesQuestion(geographies);
    const answer = await inquirer.prompt([
        geographiesQuestion,
    ]);
    return answer;
}
export async function getGeographiesQuestion(geographies) {
    if (geographies.length === 0) {
        console.error("No geographies found, exiting");
        process.exit();
    }
    const geographyQuestion = {
        type: "checkbox",
        name: "geographies",
        message: "What geographies would you like to precalculate? (select as many as you want)",
        choices: [],
    };
    return {
        ...geographyQuestion,
        choices: [
            ...geographyQuestion.choices,
            ...geographies.map((geog) => ({
                value: geog.geographyId,
                name: `${geog.geographyId} - ${geog.display}`,
            })),
        ],
    };
}
//# sourceMappingURL=precalcData.js.map