blob: b940ffd4eed38ece034dab0ee872b41e2a67fe60 [file] [log] [blame]
"use strict";
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;
};
Object.defineProperty(exports, "__esModule", { value: true });
const ts = __importStar(require("typescript"));
const util = __importStar(require("../util"));
exports.default = util.createRule({
name: 'restrict-plus-operands',
meta: {
type: 'problem',
docs: {
description: 'When adding two variables, operands must both be of type number or of type string',
recommended: 'error',
requiresTypeChecking: true,
},
messages: {
notNumbers: "Operands of '+' operation must either be both strings or both numbers.",
notStrings: "Operands of '+' operation must either be both strings or both numbers. Consider using a template literal.",
notBigInts: "Operands of '+' operation must be both bigints.",
notValidAnys: "Operands of '+' operation with any is possible only with string, number, bigint or any",
notValidTypes: "Operands of '+' operation must either be one of string, number, bigint or any (if allowed by option)",
},
schema: [
{
type: 'object',
additionalProperties: false,
properties: {
checkCompoundAssignments: {
type: 'boolean',
},
allowAny: {
type: 'boolean',
},
},
},
],
},
defaultOptions: [
{
checkCompoundAssignments: false,
allowAny: false,
},
],
create(context, [{ checkCompoundAssignments, allowAny }]) {
const service = util.getParserServices(context);
const typeChecker = service.program.getTypeChecker();
/**
* Helper function to get base type of node
*/
function getBaseTypeOfLiteralType(type) {
if (type.isNumberLiteral()) {
return 'number';
}
if (type.isStringLiteral() ||
util.isTypeFlagSet(type, ts.TypeFlags.TemplateLiteral)) {
return 'string';
}
// is BigIntLiteral
if (type.flags & ts.TypeFlags.BigIntLiteral) {
return 'bigint';
}
if (type.isUnion()) {
const types = type.types.map(getBaseTypeOfLiteralType);
return types.every(value => value === types[0]) ? types[0] : 'invalid';
}
if (type.isIntersection()) {
const types = type.types.map(getBaseTypeOfLiteralType);
return types.some(value => value === 'string') ? 'string' : 'invalid';
}
const stringType = typeChecker.typeToString(type);
if (stringType === 'number' ||
stringType === 'string' ||
stringType === 'bigint' ||
stringType === 'any') {
return stringType;
}
return 'invalid';
}
/**
* Helper function to get base type of node
* @param node the node to be evaluated.
*/
function getNodeType(node) {
const tsNode = service.esTreeNodeToTSNodeMap.get(node);
const type = util.getConstrainedTypeAtLocation(typeChecker, tsNode);
return getBaseTypeOfLiteralType(type);
}
function checkPlusOperands(node) {
const leftType = getNodeType(node.left);
const rightType = getNodeType(node.right);
if (leftType === rightType) {
if (leftType === 'invalid') {
context.report({
node,
messageId: 'notValidTypes',
});
}
if (!allowAny && leftType === 'any') {
context.report({
node,
messageId: 'notValidAnys',
});
}
return;
}
if (leftType === 'any' || rightType === 'any') {
if (!allowAny || leftType === 'invalid' || rightType === 'invalid') {
context.report({
node,
messageId: 'notValidAnys',
});
}
return;
}
if (leftType === 'string' || rightType === 'string') {
return context.report({
node,
messageId: 'notStrings',
});
}
if (leftType === 'bigint' || rightType === 'bigint') {
return context.report({
node,
messageId: 'notBigInts',
});
}
if (leftType === 'number' || rightType === 'number') {
return context.report({
node,
messageId: 'notNumbers',
});
}
}
return Object.assign({ "BinaryExpression[operator='+']": checkPlusOperands }, (checkCompoundAssignments && {
"AssignmentExpression[operator='+=']"(node) {
checkPlusOperands(node);
},
}));
},
});
//# sourceMappingURL=restrict-plus-operands.js.map