@keystrokehq/snowflake 0.0.1

package/dist/tasks.mjs

@@ -0,0 +1,104 @@
+import { executeSql } from "./sql.mjs";
+import { a as quoteLiteral, t as buildFqn } from "./sql-safety-CywdR3EP.mjs";
+import { t as formatOptions } from "./sql-options-DI-OmLU6.mjs";
+import { showTasks } from "./catalog.mjs";
+
+//#region src/tasks.ts
+function taskFqn(ref) {
+	return buildFqn(ref.database, ref.schema, ref.name);
+}
+async function createTask(client, options) {
+	const prefix = `CREATE${options.orReplace ? " OR REPLACE" : ""} TASK${options.ifNotExists ? " IF NOT EXISTS" : ""}`;
+	const copyGrants = options.copyGrants ? " COPY GRANTS" : "";
+	const body = options.options && Object.keys(options.options).length > 0 ? ` ${formatOptions(options.options)}` : "";
+	const after = options.after && options.after.length > 0 ? ` AFTER ${options.after.map(taskFqn).join(", ")}` : "";
+	const when = options.when ? ` WHEN ${options.when}` : "";
+	const comment = options.comment ? ` COMMENT = ${quoteLiteral(options.comment)}` : "";
+	await executeSql(client, {
+		statement: `${prefix} ${taskFqn(options)}${copyGrants}${body}${after}${when}${comment} AS ${options.as}`,
+		warehouse: options.warehouse,
+		role: options.role
+	});
+}
+async function alterTask(client, options) {
+	let tail;
+	if (options.rename) tail = `RENAME TO ${taskFqn(options.rename)}`;
+	else if (options.addAfter && options.addAfter.length > 0) tail = `ADD AFTER ${options.addAfter.map(taskFqn).join(", ")}`;
+	else if (options.removeAfter && options.removeAfter.length > 0) tail = `REMOVE AFTER ${options.removeAfter.map(taskFqn).join(", ")}`;
+	else if (options.modifyAs !== void 0) tail = `MODIFY AS ${options.modifyAs}`;
+	else if (options.modifyWhen !== void 0) tail = `MODIFY WHEN ${options.modifyWhen}`;
+	else if (options.set && Object.keys(options.set).length > 0) tail = `SET ${formatOptions(options.set)}`;
+	else if (options.unset && options.unset.length > 0) tail = `UNSET ${options.unset.map((k) => k.toUpperCase()).join(", ")}`;
+	else if (options.comment !== void 0) tail = `SET COMMENT = ${quoteLiteral(options.comment)}`;
+	else throw new Error("alterTask requires one of: rename / addAfter / removeAfter / modifyAs / modifyWhen / set / unset / comment");
+	await executeSql(client, {
+		statement: `ALTER TASK${options.ifExists ? " IF EXISTS" : ""} ${taskFqn(options)} ${tail}`,
+		warehouse: options.warehouse,
+		role: options.role
+	});
+}
+async function dropTask(client, options) {
+	await executeSql(client, {
+		statement: `DROP TASK${options.ifExists ? " IF EXISTS" : ""} ${taskFqn(options)}`,
+		warehouse: options.warehouse,
+		role: options.role
+	});
+}
+async function describeTask(client, options) {
+	return (await executeSql(client, {
+		statement: `DESCRIBE TASK ${taskFqn(options)}`,
+		warehouse: options.warehouse,
+		role: options.role
+	})).rows;
+}
+async function executeTask(client, options) {
+	const suffix = options.retryLast ? " RETRY LAST" : "";
+	await executeSql(client, {
+		statement: `EXECUTE TASK ${taskFqn(options)}${suffix}`,
+		warehouse: options.warehouse,
+		role: options.role
+	});
+}
+async function resumeTask(client, options) {
+	await executeSql(client, {
+		statement: `ALTER TASK${options.ifExists ? " IF EXISTS" : ""} ${taskFqn(options)} RESUME`,
+		warehouse: options.warehouse,
+		role: options.role
+	});
+}
+async function suspendTask(client, options) {
+	await executeSql(client, {
+		statement: `ALTER TASK${options.ifExists ? " IF EXISTS" : ""} ${taskFqn(options)} SUSPEND`,
+		warehouse: options.warehouse,
+		role: options.role
+	});
+}
+function renderTimestampArg(value) {
+	return value ? `TO_TIMESTAMP_LTZ(${quoteLiteral(value)})` : "NULL";
+}
+async function showTaskHistory(client, options = {}) {
+	const args = [];
+	if (options.scheduledTimeRangeStart !== void 0) args.push(`SCHEDULED_TIME_RANGE_START => ${renderTimestampArg(options.scheduledTimeRangeStart)}`);
+	if (options.scheduledTimeRangeEnd !== void 0) args.push(`SCHEDULED_TIME_RANGE_END => ${renderTimestampArg(options.scheduledTimeRangeEnd)}`);
+	if (options.resultLimit !== void 0) args.push(`RESULT_LIMIT => ${options.resultLimit}`);
+	if (options.taskName) args.push(`TASK_NAME => ${quoteLiteral(taskFqn(options.taskName))}`);
+	if (options.rootTaskName) args.push(`ROOT_TASK_NAME => ${quoteLiteral(taskFqn(options.rootTaskName))}`);
+	if (options.errorOnly !== void 0) args.push(`ERROR_ONLY => ${options.errorOnly ? "TRUE" : "FALSE"}`);
+	return (await executeSql(client, {
+		statement: `SELECT * FROM TABLE(INFORMATION_SCHEMA.TASK_HISTORY(${args.join(", ")}))`,
+		warehouse: options.warehouse,
+		role: options.role
+	})).rows;
+}
+async function showTaskDependents(client, options) {
+	const args = [`TASK_NAME => ${quoteLiteral(taskFqn(options))}`];
+	if (options.recursive !== void 0) args.push(`RECURSIVE => ${options.recursive ? "TRUE" : "FALSE"}`);
+	return (await executeSql(client, {
+		statement: `SELECT * FROM TABLE(INFORMATION_SCHEMA.TASK_DEPENDENTS(${args.join(", ")}))`,
+		warehouse: options.warehouse,
+		role: options.role
+	})).rows;
+}
+
+//#endregion
+export { alterTask, createTask, describeTask, dropTask, executeTask, resumeTask, showTaskDependents, showTaskHistory, showTasks, suspendTask };