firebase-tools
Version:
Command-Line Interface for Firebase
99 lines (98 loc) • 4.49 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.validateRulesTool = void 0;
const zod_1 = require("zod");
const tool_js_1 = require("../../tool.js");
const util_js_1 = require("../../util.js");
const rules_js_1 = require("../../../gcp/rules.js");
const path_1 = require("path");
function formatRulesetIssues(issues, rulesSource) {
const sourceLines = rulesSource.split("\n");
const formattedOutput = [];
for (const issue of issues) {
const { sourcePosition, description, severity } = issue;
let issueString = `${severity}: ${description} [Ln ${sourcePosition.line}, Col ${sourcePosition.column}]`;
if (sourcePosition.line) {
const lineIndex = sourcePosition.line - 1;
if (lineIndex >= 0 && lineIndex < sourceLines.length) {
const errorLine = sourceLines[lineIndex];
issueString += `\n\`\`\`\n${errorLine}`;
if (sourcePosition.column &&
sourcePosition.currentOffset &&
sourcePosition.endOffset &&
sourcePosition.column > 0 &&
sourcePosition.endOffset > sourcePosition.currentOffset) {
const startColumnOnLine = sourcePosition.column - 1;
const errorTokenLength = sourcePosition.endOffset - sourcePosition.currentOffset;
if (startColumnOnLine >= 0 &&
errorTokenLength > 0 &&
startColumnOnLine <= errorLine.length) {
const padding = " ".repeat(startColumnOnLine);
const carets = "^".repeat(errorTokenLength);
issueString += `\n${padding}${carets}\n\`\`\``;
}
}
}
}
formattedOutput.push(issueString);
}
return formattedOutput.join("\n\n");
}
function validateRulesTool(productName) {
return (0, tool_js_1.tool)({
name: "validate_rules",
description: `Checks the provided ${productName} Rules source for syntax and validation errors. Provide EITHER the source code to validate OR a path to a source file.`,
inputSchema: zod_1.z.object({
source: zod_1.z
.string()
.optional()
.describe("the rules source code to check. provide either this OR a path"),
source_file: zod_1.z
.string()
.optional()
.describe("a file path, relative to the project root, to a file containing the rules source you want to validate. provide this OR source, not both"),
}),
annotations: {
title: `Validate ${productName} Rules`,
readOnlyHint: true,
},
_meta: {
requiresProject: true,
requiresAuth: true,
},
}, async ({ source, source_file }, { projectId, config, host }) => {
var _a, _b;
if (source && source_file) {
return (0, util_js_1.mcpError)("Must supply `source` or `source_file`, not both.");
}
let rulesSourceContent;
if (source_file) {
try {
const filePath = (0, path_1.resolve)(source_file, host.cachedProjectRoot);
if (filePath.includes("../"))
return (0, util_js_1.mcpError)("Cannot read files outside of the project directory.");
rulesSourceContent = config.readProjectFile(source_file);
}
catch (e) {
return (0, util_js_1.mcpError)(`Failed to read source_file '${source_file}': ${e.message}`);
}
}
else if (source) {
rulesSourceContent = source;
}
else {
rulesSourceContent = "";
}
const result = await (0, rules_js_1.testRuleset)(projectId, [
{ name: "test.rules", content: rulesSourceContent },
]);
if ((_b = (_a = result.body) === null || _a === void 0 ? void 0 : _a.issues) === null || _b === void 0 ? void 0 : _b.length) {
const issues = result.body.issues;
let out = `Found ${issues.length} issues in rules source:\n\n`;
out += formatRulesetIssues(issues, rulesSourceContent);
return (0, util_js_1.toContent)(out);
}
return (0, util_js_1.toContent)("OK: No errors detected.");
});
}
exports.validateRulesTool = validateRulesTool;