Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(eslint-plugin): add prefer-enum-initializers rule #2326

59 changes: 45 additions & 14 deletions packages/eslint-plugin/src/rules/prefer-enum-initializers.ts
@@ -1,41 +1,72 @@
import { TSESTree } from '@typescript-eslint/experimental-utils';
import * as util from '../util';
import { RuleFix } from '@typescript-eslint/experimental-utils/dist/ts-eslint';

export default util.createRule({
type Option = '0-based' | '1-based' | 'key-name';
type MessageIds = 'defineInitializer' | 'defineInitializerSuggestion';

export default util.createRule<Option[], MessageIds>({
name: 'prefer-enum-initializers',
meta: {
type: 'suggestion',
docs: {
description: 'Prefer initializing each enums member value',
category: 'Best Practices',
recommended: false,
suggestion: true,
},
messages: {
defineInitializer:
"The value of the member '{{ name }}' should be explicitly defined",
defineInitializerSuggestion:
'can be suggestion fixed to {{ name }} = {{ suggestedValue }}',
},
schema: [],
schema: [
{
enum: ['0-based', '1-based', 'key-name'],
},
],
},
defaultOptions: [],
create(context) {
const sourceCode = context.getSourceCode();
const config = context.options[0];

return {
TSEnumDeclaration(node: TSESTree.TSEnumDeclaration): void {
const { members } = node;
const violatingMembers = members.filter(member => {
return member.initializer == null;
members.forEach((member, index) => {
if (member.initializer == null) {
const name = sourceCode.getText(member);
context.report({
node: member,
messageId: 'defineInitializer',
data: {
name,
},
suggest: [
{
messageId: 'defineInitializerSuggestion',
data: { name },
fix: (fixer): RuleFix | null => {
if (config === '0-based') {
return fixer.replaceText(member, `${name} = ${index}`);
} else if (config === '1-based') {
return fixer.replaceText(
member,
`${name} = ${index + 1}`,
);
} else if (config === 'key-name') {
return fixer.replaceText(member, `${name} = '${name}'`);
} else {
return null;
}
},
},
],
bradzacher marked this conversation as resolved.
Show resolved Hide resolved
});
}
});

for (const member of violatingMembers) {
context.report({
node: member,
messageId: 'defineInitializer',
data: {
name: sourceCode.getText(member),
},
});
}
},
};
},
Expand Down
133 changes: 114 additions & 19 deletions packages/eslint-plugin/tests/rules/prefer-enum-initializers.test.ts
Expand Up @@ -25,12 +25,6 @@ enum Direction {
enum Direction {
Up = 'Up',
Down = 'Down',
}
`,
`
enum Direction {
Up = 1,
Down = 'Up',
}
`,
],
Expand All @@ -41,11 +35,24 @@ enum Direction {
Up,
}
`,
options: ['key-name'],
errors: [
{
messageId: 'defineInitializer',
data: { name: 'Up' },
line: 3,
suggestions: [
{
messageId: 'defineInitializerSuggestion',
/* eslint-disable @typescript-eslint/internal/plugin-test-formatting */
output: `
enum Direction {
Up = 'Up',
}
`,
/* eslint-enable @typescript-eslint/internal/plugin-test-formatting */
},
],
},
],
},
Expand All @@ -56,76 +63,164 @@ enum Direction {
Down,
}
`,
options: ['key-name'],
errors: [
{
messageId: 'defineInitializer',
data: { name: 'Up' },
line: 3,
suggestions: [
{
messageId: 'defineInitializerSuggestion',
/* eslint-disable @typescript-eslint/internal/plugin-test-formatting */
output: `
enum Direction {
Up = 'Up',
Down,
}
`,
/* eslint-enable @typescript-eslint/internal/plugin-test-formatting */
},
],
},
{
messageId: 'defineInitializer',
data: { name: 'Down' },
line: 4,
suggestions: [
{
messageId: 'defineInitializerSuggestion',
/* eslint-disable @typescript-eslint/internal/plugin-test-formatting */
output: `
enum Direction {
Up,
Down = 'Down',
}
`,
/* eslint-enable @typescript-eslint/internal/plugin-test-formatting */
},
],
},
],
},
{
code: `
enum Direction {
Up = 1,
Down,
Up,
}
`,
options: ['0-based'],
errors: [
{
messageId: 'defineInitializer',
data: { name: 'Down' },
line: 4,
data: { name: 'Up' },
line: 3,
suggestions: [
{
messageId: 'defineInitializerSuggestion',
/* eslint-disable @typescript-eslint/internal/plugin-test-formatting */
output: `
enum Direction {
Up = 0,
}
`,
/* eslint-enable @typescript-eslint/internal/plugin-test-formatting */
},
],
},
],
},
{
code: `
enum Direction {
Up,
Down = 2,
Down,
}
`,
options: ['0-based'],
errors: [
{
messageId: 'defineInitializer',
data: { name: 'Up' },
line: 3,
},
],
},
{
code: `
suggestions: [
{
messageId: 'defineInitializerSuggestion',
/* eslint-disable @typescript-eslint/internal/plugin-test-formatting */
output: `
enum Direction {
Up = 'Up',
Up = 0,
Down,
}
`,
errors: [
/* eslint-enable @typescript-eslint/internal/plugin-test-formatting */
},
],
},
{
messageId: 'defineInitializer',
data: { name: 'Down' },
line: 4,
suggestions: [
{
messageId: 'defineInitializerSuggestion',
/* eslint-disable @typescript-eslint/internal/plugin-test-formatting */
output: `
enum Direction {
Up,
Down = 1,
}
`,
/* eslint-enable @typescript-eslint/internal/plugin-test-formatting */
},
],
},
],
},
{
code: `
enum Direction {
Up,
Down = 'Down',
Down,
}
`,
options: ['1-based'],
errors: [
{
messageId: 'defineInitializer',
data: { name: 'Up' },
line: 3,
suggestions: [
{
messageId: 'defineInitializerSuggestion',
/* eslint-disable @typescript-eslint/internal/plugin-test-formatting */
output: `
enum Direction {
Up = 1,
Down,
}
`,
/* eslint-enable @typescript-eslint/internal/plugin-test-formatting */
},
],
},
{
messageId: 'defineInitializer',
data: { name: 'Down' },
line: 4,
suggestions: [
{
messageId: 'defineInitializerSuggestion',
/* eslint-disable @typescript-eslint/internal/plugin-test-formatting */
output: `
enum Direction {
Up,
Down = 2,
}
`,
/* eslint-enable @typescript-eslint/internal/plugin-test-formatting */
},
],
},
],
},
Expand Down