@rex0220/kintone-sql-tools 3.38.0 → 3.39.0

package/dist-cli/ksql.js

@@ -1175,13 +1175,38 @@ var Parser = class {
   parseDeclareVariable() {
     this.advance();
     const variable = this.expect("VARIABLE" /* VARIABLE */, "DECLARE \u306E\u5F8C\u306B\u306F\u5909\u6570\u540D\uFF08\u4F8B: @name\uFF09\u304C\u5FC5\u8981\u3067\u3059");
+    const relativeDate = this.peek().kind === "IDENT" /* IDENT */ && this.peek().value.toUpperCase() === "RELATIVE_DATE";
+    if (relativeDate) this.advance();
     this.expect("=" /* EQ */);
+    if (relativeDate) {
+      return {
+        type: "DECLARE_VARIABLE",
+        name: variable.value.slice(1).toLowerCase(),
+        annotation: "RELATIVE_DATE",
+        default: this.parseRelativeDateVariableToken()
+      };
+    }
     const expr = this.parseScalarExpr("DECLARE", false);
     if (expr.type === "SCALAR_SUBQUERY") {
       throw new ParseError("DECLARE \u306E\u65E2\u5B9A\u5024\u306B\u30B9\u30AB\u30E9\u30FC\u30B5\u30D6\u30AF\u30A8\u30EA\u306F\u4F7F\u7528\u3067\u304D\u307E\u305B\u3093", this.peek());
     }
     return { type: "DECLARE_VARIABLE", name: variable.value.slice(1).toLowerCase(), default: expr };
   }
+  /** RELATIVE_DATE 宣言専用。WHERE と同じ関数パーサーを使い、日付系14個だけを許可する。 */
+  parseRelativeDateVariableToken() {
+    const tok = this.peek();
+    const contextualFunction = PARSER_CONTEXTUAL_FUNCTION_TOKEN_MAP[tok.kind];
+    if (contextualFunction === "TODAY" || contextualFunction === "NOW") {
+      return this.parseSqlValue();
+    }
+    if (tok.kind === "IDENT" /* IDENT */ && this.peekAt(1).kind === "(" /* LPAREN */ && isRelativeDateFunctionName(tok.value.toUpperCase())) {
+      return this.parseRelativeDateFunction();
+    }
+    throw new ParseError(
+      "RELATIVE_DATE \u306E\u65E2\u5B9A\u5024\u306B\u306F\u30B5\u30DD\u30FC\u30C8\u5BFE\u8C61\u306E\u76F8\u5BFE\u65E5\u4ED8\u95A2\u6570\u30C8\u30FC\u30AF\u30F3\u304C\u5FC5\u8981\u3067\u3059",
+      tok
+    );
+  }
   /** SET / DECLARE RHS 専用。既存式パーサーで構文を読み、フィールド参照を明示的に拒否する。 */
   parseScalarExpr(context, allowScalarSubquery) {
     const tok = this.peek();
@@ -6807,9 +6832,10 @@ function collectRefs(node, tempRefs, appIds) {
     for (const v of Object.values(obj)) collectRefs(v, tempRefs, appIds);
   }
 }
-function collectVariableRefs(node, refs) {
+var RELATIVE_DATE_COMPARISON_OPS = /* @__PURE__ */ new Set(["=", "!=", "<>", ">", "<", ">=", "<="]);
+function collectVariableRefs(node, refs, inWhere = false) {
   if (Array.isArray(node)) {
-    for (const v of node) collectVariableRefs(v, refs);
+    for (const v of node) collectVariableRefs(v, refs, inWhere);
     return;
   }
   if (node !== null && typeof node === "object") {
@@ -6818,13 +6844,27 @@ function collectVariableRefs(node, refs) {
     if ((type === "VARIABLE" || type === "VARIABLE_COL" || type === "VARIABLE_IN_LIST") && typeof obj["name"] === "string") {
       refs.push({
         name: obj["name"],
-        kind: type === "VARIABLE" ? "scalar" : type === "VARIABLE_COL" ? "select-column" : "array-in-list"
+        kind: type === "VARIABLE" ? "scalar" : type === "VARIABLE_COL" ? "select-column" : "array-in-list",
+        relativeDateAllowed: inWhere
       });
       return;
     }
-    for (const v of Object.values(obj)) collectVariableRefs(v, refs);
+    const preservesWhereContext = type === "BINARY" || type === "LOGICAL" || type === "NOT" || type === "GROUP" || type === "NULL_CHECK" || type === "BOOLEAN";
+    for (const [key, value] of Object.entries(obj)) {
+      const directVariable = value !== null && typeof value === "object" && value["type"] === "VARIABLE";
+      const directComparisonRight = inWhere && type === "BINARY" && key === "right" && directVariable && RELATIVE_DATE_COMPARISON_OPS.has(String(obj["op"]));
+      collectVariableRefs(
+        value,
+        refs,
+        key === "where" ? true : type === "BINARY" && key === "right" && directVariable ? directComparisonRight : preservesWhereContext ? inWhere : false
+      );
+    }
   }
 }
+function isRelativeDateDmlUse(stmt) {
+  const target = stmt.type === "EXPLAIN" ? stmt.query : stmt;
+  return isDmlType(target.type) || target.type === "VALIDATE";
+}
 function validateGroupingStaticQueries(node) {
   if (Array.isArray(node)) {
     for (const value of node) validateGroupingStaticQueries(value);
@@ -6919,6 +6959,18 @@ function analyzeBatch(statements) {
           index
         );
       }
+      if (def.relativeDate && isRelativeDateDmlUse(stmt)) {
+        throw new BatchAnalysisError(
+          `ArgumentError: RELATIVE_DATE variable @${use.name} cannot be used in DML or VALIDATE statements.`,
+          index
+        );
+      }
+      if (def.relativeDate && !use.relativeDateAllowed) {
+        throw new BatchAnalysisError(
+          `ArgumentError: RELATIVE_DATE variable @${use.name} can only be used as a WHERE comparison right operand or BETWEEN boundary.`,
+          index
+        );
+      }
       if (!referencedThisStatement.has(use.name)) {
         def.referencedBy.push(index);
         referencedThisStatement.add(use.name);
@@ -6931,6 +6983,7 @@ function analyzeBatch(statements) {
       variableDefs.set(stmt.name, {
         index,
         kind: stmt.type === "SET_VARIABLE" && stmt.expr.type === "ARRAY" ? "array" : "scalar",
+        relativeDate: stmt.type === "DECLARE_VARIABLE" && stmt.annotation === "RELATIVE_DATE",
         referencedBy: []
       });
       variableOrder.push(stmt.name);
@@ -16123,6 +16176,7 @@ async function executeBatch(sql, client, options = {}) {
     throw new Error("UnsupportedError: APPLY mutation requires allowApplyMutation=true");
   }
   const injectedVariables = validateDeclaredBatchVariables(statements, options.variables);
+  const relativeDateVariables = prepareRelativeDateVariables(statements, injectedVariables);
   const batchOptions = { ...options, variables: injectedVariables };
   if (options.continueOnError && analysis.containsDml) {
     throw new Error("ArgumentError: continueOnError is not allowed for batches containing DML.");
@@ -16194,7 +16248,16 @@ async function executeBatch(sql, client, options = {}) {
         );
         const cursorScope = wrapClientWithCursorScope(statementClient);
         const outcome = await runWithDeadline(
-          executeBatchStatement(statements[i], info, cursorScope.client, stmtOptions, cacheContext, tempTables, variables),
+          executeBatchStatement(
+            statements[i],
+            info,
+            cursorScope.client,
+            stmtOptions,
+            cacheContext,
+            tempTables,
+            variables,
+            relativeDateVariables
+          ),
           remaining,
           cursorScope.closeActive
         );
@@ -16239,7 +16302,7 @@ function statementHasApplyMutation(statement) {
   }
   return statement.type === "UPSERT" && statement.validateOnly !== true && Boolean(statement.onInsertApplyBlocks?.length || statement.onUpdateApplyBlocks?.length);
 }
-async function executeBatchStatement(stmt, info, client, options, cacheContext, tempTables, variables) {
+async function executeBatchStatement(stmt, info, client, options, cacheContext, tempTables, variables, relativeDateVariables) {
   if (stmt.type === "SET_VARIABLE") {
     const resolvedStmt2 = resolveBatchVariableReferences(stmt, variables);
     validateStatementStatic(resolvedStmt2);
@@ -16284,6 +16347,13 @@ async function executeBatchStatement(stmt, info, client, options, cacheContext,
     return {};
   }
   if (stmt.type === "DECLARE_VARIABLE") {
+    if (stmt.annotation === "RELATIVE_DATE") {
+      variables.set(stmt.name, {
+        type: "relative-date",
+        value: relativeDateVariables.get(stmt.name)
+      });
+      return {};
+    }
     const injected = options.variables ?? {};
     if (Object.prototype.hasOwnProperty.call(injected, stmt.name)) {
       variables.set(stmt.name, { type: "string", value: injected[stmt.name] });
@@ -16507,6 +16577,32 @@ function parseSqlBatch(sql, enableImport = false) {
   const tokens = new Lexer(sql).tokenize();
   return new Parser(tokens, { import: enableImport }).parseStatements();
 }
+function parseRelativeDateVariableValue(name, value) {
+  try {
+    const statements = parseSqlBatch(`DECLARE @__b111 RELATIVE_DATE = ${value}`);
+    const declaration = statements[0];
+    if (statements.length !== 1 || declaration?.type !== "DECLARE_VARIABLE" || declaration.annotation !== "RELATIVE_DATE") {
+      throw new Error("token was not consumed as one RELATIVE_DATE declaration");
+    }
+    return declaration.default;
+  } catch (error) {
+    const detail = error instanceof Error ? ` ${error.message}` : "";
+    throw new Error(
+      `ArgumentError: RELATIVE_DATE variable @${name} requires one supported relative-date function token.${detail}`
+    );
+  }
+}
+function prepareRelativeDateVariables(statements, injectedVariables) {
+  const prepared = /* @__PURE__ */ new Map();
+  for (const stmt of statements) {
+    if (stmt.type !== "DECLARE_VARIABLE" || stmt.annotation !== "RELATIVE_DATE") continue;
+    prepared.set(
+      stmt.name,
+      Object.prototype.hasOwnProperty.call(injectedVariables, stmt.name) ? parseRelativeDateVariableValue(stmt.name, injectedVariables[stmt.name]) : stmt.default
+    );
+  }
+  return prepared;
+}
 function evaluateScalarExpr(expr) {
   switch (expr.type) {
     case "STRING":
@@ -16543,7 +16639,8 @@ function resolveBatchVariableReferencesInternal(node, variables, numericArithmet
       if (value.type === "array") {
         throw new Error(`ParseError: array variable @${obj["name"]} can only be used as IN @${obj["name"]}.`);
       }
-      if (numericArithmeticOperand && value.type !== "number" && value.placeholder !== true) {
+      if (value.type === "relative-date") return value.value;
+      if (numericArithmeticOperand && value.type !== "number" && !(value.type === "string" && value.placeholder === true)) {
         throw new Error(
           `ArgumentError: variable @${obj["name"]} is not numeric and cannot be used in arithmetic.`
         );
@@ -16554,6 +16651,9 @@ function resolveBatchVariableReferencesInternal(node, variables, numericArithmet
       const value = variables.get(obj["name"]);
       if (value === void 0) throw new Error(`ParseError: variable @${obj["name"]} is not defined in this batch.`);
       if (value.type === "array") throw new Error(`ParseError: array variable @${obj["name"]} cannot be used as a SELECT column.`);
+      if (value.type === "relative-date") {
+        throw new Error(`InternalError: RELATIVE_DATE variable @${obj["name"]} reached a SELECT column.`);
+      }
       const aliasDisplay = typeof obj["aliasDisplay"] === "string" ? { aliasDisplay: obj["aliasDisplay"] } : {};
       return value.type === "number" ? { type: "ARITH_COL", expr: { type: "NUMBER", value: value.value, raw: value.raw ?? String(value.value) }, alias: obj["alias"], ...aliasDisplay } : { type: "LITERAL_COL", value: value.value, alias: obj["alias"], ...aliasDisplay };
     }
@@ -22673,7 +22773,8 @@ async function buildBatchExplainPlans(sql, client, injectedVariables, cacheConte
   try {
     const statements = parseSqlBatch(sql, enableImport);
     const analysis = analyzeBatch(statements);
-    validateDeclaredBatchVariables(statements, injectedVariables);
+    const normalizedInjectedVariables = validateDeclaredBatchVariables(statements, injectedVariables);
+    const relativeDateVariables = prepareRelativeDateVariables(statements, normalizedInjectedVariables);
     const variables = /* @__PURE__ */ new Map();
     const plans = [];
     for (let i = 0; i < statements.length; i++) {
@@ -22706,7 +22807,7 @@ async function buildBatchExplainPlans(sql, client, injectedVariables, cacheConte
         plan: statementPlan.length === 0 ? metadataPlan : [statementPlan[0], ...metadataPlan, ...statementPlan.slice(1)]
       });
       if (stmt.type === "SET_VARIABLE" || stmt.type === "DECLARE_VARIABLE") {
-        variables.set(stmt.name, stmt.type === "SET_VARIABLE" && stmt.expr.type === "ARRAY" ? { type: "array", elements: stmt.expr.elements.map((element) => ({ type: "string", value: element.value })) } : { type: "string", value: `@${stmt.name}`, placeholder: true });
+        variables.set(stmt.name, stmt.type === "DECLARE_VARIABLE" && stmt.annotation === "RELATIVE_DATE" ? { type: "relative-date", value: relativeDateVariables.get(stmt.name) } : stmt.type === "SET_VARIABLE" && stmt.expr.type === "ARRAY" ? { type: "array", elements: stmt.expr.elements.map((element) => ({ type: "string", value: element.value })) } : { type: "string", value: `@${stmt.name}`, placeholder: true });
       }
     }
     return { statementCount: statements.length, statements: plans };
@@ -22745,6 +22846,12 @@ function buildBatchStatementPlan(stmt, info, capabilities, orderPlans, dmlMaxRow
     ];
   }
   if (stmt.type === "DECLARE_VARIABLE") {
+    if (stmt.annotation === "RELATIVE_DATE") {
+      return [
+        `DECLARE @${stmt.name} RELATIVE_DATE = <relative-date token>`,
+        "  value:         \u5916\u90E8\u6CE8\u5165\u304C\u3042\u308C\u3070\u63A1\u7528\u3001\u306A\u3051\u308C\u3070\u65E2\u5B9A\u30C8\u30FC\u30AF\u30F3\u3092\u4F7F\u7528\uFF08\u5024\u306F\u975E\u516C\u958B\uFF09"
+      ];
+    }
     return [
       `DECLARE @${stmt.name} = <default scalar expression>`,
       "  value:         \u5916\u90E8\u6CE8\u5165\u304C\u3042\u308C\u3070\u63A1\u7528\u3001\u306A\u3051\u308C\u3070\u65E2\u5B9A\u5024\u3092\u5B9F\u884C\u6642\u306B1\u56DE\u8A55\u4FA1\uFF08\u5024\u306F\u975E\u516C\u958B\uFF09"
@@ -25374,13 +25481,17 @@ function restoreSqlDiagnosticValue(value, bindings, options = {}) {
         return `${dmlTarget[1]}${target}`;
       }
     }
-    let restored = value;
-    for (const binding of bindings.values()) {
-      const internal = `APP${binding.mappedAppId}`;
+    if (bindings.size === 0) return value;
+    const mappedIds = [...bindings.keys()].sort((a, b) => String(b).length - String(a).length);
+    const internalApp = new RegExp(
+      `APP(${mappedIds.join("|")})(?!\\d)(\\s+AS\\s+[^\\s()]+)?(?:\\s+\\(\\1\\))?`,
+      "g"
+    );
+    return value.replace(internalApp, (_match, mappedIdText, alias) => {
+      const binding = bindings.get(Number(mappedIdText));
       const display = binding.source === "logical" ? displayMode === "physical" ? `LAPP_${binding.logicalName} -> APP${binding.appId}` : `LAPP_${binding.logicalName}@${binding.profile}` : displayMode === "physical" ? `APP${binding.appId}` : `APP${binding.appId}@${binding.profile}`;
-      restored = restored.split(`${internal} (${binding.mappedAppId})`).join(display).split(internal).join(display);
-    }
-    return restored;
+      return `${display}${alias ?? ""}`;
+    });
   }
   if (Array.isArray(value)) {
     return value.map((item) => restoreSqlDiagnosticValue(item, bindings, options));