@sketch-hq/sketch-assistant-utils
Version:
Utility functions and types for Sketch Assistants.
166 lines (165 loc) • 7.65 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.runAssistant = void 0;
const p_map_1 = __importDefault(require("p-map"));
const p_timeout_1 = __importStar(require("p-timeout"));
const sketch_assistant_types_1 = require("@sketch-hq/sketch-assistant-types");
const rule_utils_1 = require("../../rule-utils");
const assistant_config_1 = require("../../assistant-config");
const ignore_1 = require("../ignore");
/**
* Given a set of violations and rule errors, determine the grade. A "fail"
* means severe error-level violations are present, whereas a "pass"
* means only info or warn-level violations are present. An "unknown" grade
* means an indeterminate result - for example, if one or more rules timed-out
* then we can't give a definitive grade.
*/
const getGrade = (violations, ruleErrors) => {
const hasTimedOutRules = !!ruleErrors.find((ruleError) => ruleError.code === 'timeout');
if (hasTimedOutRules)
return 'unknown';
const hasErrorLevelViolations = !!violations.find((violation) => violation.severity === sketch_assistant_types_1.ViolationSeverity.error);
return hasErrorLevelViolations ? 'fail' : 'pass';
};
/**
* Rule invocation errors are errors thrown by rule functions while they are
* executing. Rule errors could be thrown by the Assistant architecture code,
* or by mistakes in 3rd party rules. In either case, they are caught in the
* runAssistant function and collated into plain RuleError objects to be
* incorporated into the overall result.
*/
class RuleInvocationError extends Error {
constructor(cause, assistantName, ruleName, code) {
super(`Error thrown during invocation of rule "${ruleName}" on assistant "${assistantName}": ${cause.message}`);
this.cause = cause;
this.assistantName = assistantName;
this.ruleName = ruleName;
this.name = 'RuleInvocationError';
this.code = code;
Object.setPrototypeOf(this, new.target.prototype);
}
}
/**
* Run a single assistant, catching and returning any errors encountered during rule invocation.
*/
const runAssistant = (file, assistant, env, cancelToken, getImageMetadata, ignoreConfig, ruleTimeout) => __awaiter(void 0, void 0, void 0, function* () {
const violations = [];
const createUtils = rule_utils_1.createRuleUtilsCreator(file, violations, assistant, cancelToken, getImageMetadata, ignoreConfig);
const context = {
env,
file,
assistant,
cancelToken,
getImageMetadata,
};
const activeRules = assistant.rules
.filter((rule) => assistant_config_1.isRuleActive(assistant.config, rule.name)) // Rule turned on in config
.filter((rule) => (rule.runtime ? rule.runtime === env.runtime : true)); // Rule platform is supported
const profile = {
ruleTimings: {},
};
const metadata = {
assistant: {
name: assistant.name,
config: assistant.config,
},
rules: activeRules.reduce((acc, rule) => {
const ruleConfig = assistant_config_1.getRuleConfig(assistant.config, rule.name);
const configTitle = assistant_config_1.getRuleTitle(assistant.config, rule.name);
const title = configTitle
? configTitle
: ruleConfig && 'active' in ruleConfig && typeof rule.title === 'function'
? rule.title(ruleConfig)
: rule.title;
const description = ruleConfig && 'active' in ruleConfig && typeof rule.description === 'function'
? rule.description(ruleConfig)
: rule.description;
return Object.assign(Object.assign({}, acc), { [rule.name]: {
name: rule.name,
title,
description,
debug: rule.debug,
runtime: rule.runtime,
} });
}, {}),
};
// Filter out ignored rules to find the final set of rules to invoke
const rulesToRun = activeRules.filter((rule) => !ignore_1.isRuleFullIgnored(ignoreConfig, assistant.name, rule.name));
try {
yield p_map_1.default(rulesToRun, (rule) => __awaiter(void 0, void 0, void 0, function* () {
if (cancelToken.cancelled)
return;
const timeoutToken = { timedOut: false };
const { rule: ruleFunction, name: ruleName } = rule;
const ruleContext = Object.assign(Object.assign({}, context), { utils: createUtils(ruleName, timeoutToken) });
const start = Date.now();
try {
const rulePromise = ruleFunction(ruleContext);
yield (ruleTimeout === Infinity
? rulePromise
: p_timeout_1.default(rulePromise, ruleTimeout, `Rule timed-out after ${ruleTimeout} milliseconds`));
}
catch (error) {
const timedOut = error instanceof p_timeout_1.TimeoutError;
timeoutToken.timedOut = timedOut;
throw new RuleInvocationError(error, assistant.name, ruleName, timedOut ? 'timeout' : 'error');
}
profile.ruleTimings[ruleName] = Date.now() - start;
}), { concurrency: 1, stopOnError: false });
}
catch (errors) {
const ruleErrors = Array.from(errors).map((error) => ({
assistantName: error.assistantName,
ruleName: error.ruleName,
message: error.cause.message,
stack: error.cause.stack || '',
code: error.code,
}));
return {
grade: getGrade(violations, ruleErrors),
violations,
ruleErrors,
metadata,
profile,
};
}
return {
grade: getGrade(violations, []),
violations,
ruleErrors: [],
metadata,
profile,
};
});
exports.runAssistant = runAssistant;