Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit c045410

Browse files
xiaoxiangmoexiaoxiangmoe
committedSep 26, 2023
feat: use enum to replace const enum
1 parent b8fc18c commit c045410

34 files changed

+143
-105
lines changed
 

Diff for: ‎.eslintrc.cjs

+21-3
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,14 @@
11
const DOMGlobals = ['window', 'document']
22
const NodeGlobals = ['module', 'require']
33

4+
const banConstEnum = {
5+
selector: 'TSEnumDeclaration[const=true]',
6+
message: 'Please use enums, instead',
7+
}
8+
9+
/**
10+
* @type {import('eslint-define-config').ESLintConfig}
11+
*/
412
module.exports = {
513
parser: '@typescript-eslint/parser',
614
parserOptions: {
@@ -20,6 +28,7 @@ module.exports = {
2028

2129
'no-restricted-syntax': [
2230
'error',
31+
banConstEnum,
2332
// since we target ES2015 for baseline support, we need to forbid object
2433
// rest spread usage in destructure as it compiles into a verbose helper.
2534
'ObjectPattern > RestElement',
@@ -61,23 +70,32 @@ module.exports = {
6170
],
6271
rules: {
6372
'no-restricted-globals': ['error', ...DOMGlobals],
64-
'no-restricted-syntax': 'off'
73+
'no-restricted-syntax': [
74+
'error',
75+
banConstEnum,
76+
]
6577
}
6678
},
6779
// Private package, browser only + no syntax restrictions
6880
{
6981
files: ['packages/template-explorer/**', 'packages/sfc-playground/**'],
7082
rules: {
7183
'no-restricted-globals': ['error', ...NodeGlobals],
72-
'no-restricted-syntax': 'off'
84+
'no-restricted-syntax': [
85+
'error',
86+
banConstEnum,
87+
]
7388
}
7489
},
7590
// Node scripts
7691
{
7792
files: ['scripts/**', '*.{js,ts}', 'packages/**/index.js'],
7893
rules: {
7994
'no-restricted-globals': 'off',
80-
'no-restricted-syntax': 'off'
95+
'no-restricted-syntax': [
96+
'error',
97+
banConstEnum,
98+
]
8199
}
82100
}
83101
]

Diff for: ‎package.json

+1
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@
7676
"esbuild": "^0.19.3",
7777
"esbuild-plugin-polyfill-node": "^0.3.0",
7878
"eslint": "^8.49.0",
79+
"eslint-define-config": "^1.23.0",
7980
"eslint-plugin-jest": "^27.4.0",
8081
"estree-walker": "^2.0.2",
8182
"execa": "^8.0.1",

Diff for: ‎packages/compiler-core/src/ast.ts

+4-4
Original file line numberDiff line numberDiff line change
@@ -21,11 +21,11 @@ import { ImportItem, TransformContext } from './transform'
2121
// compilers.
2222
export type Namespace = number
2323

24-
export const enum Namespaces {
24+
export enum Namespaces {
2525
HTML
2626
}
2727

28-
export const enum NodeTypes {
28+
export enum NodeTypes {
2929
ROOT,
3030
ELEMENT,
3131
TEXT,
@@ -59,7 +59,7 @@ export const enum NodeTypes {
5959
JS_RETURN_STATEMENT
6060
}
6161

62-
export const enum ElementTypes {
62+
export enum ElementTypes {
6363
ELEMENT,
6464
COMPONENT,
6565
SLOT,
@@ -202,7 +202,7 @@ export interface DirectiveNode extends Node {
202202
* Higher levels implies lower levels. e.g. a node that can be stringified
203203
* can always be hoisted and skipped for patch.
204204
*/
205-
export const enum ConstantTypes {
205+
export enum ConstantTypes {
206206
NOT_CONSTANT = 0,
207207
CAN_SKIP_PATCH,
208208
CAN_HOIST,

Diff for: ‎packages/compiler-core/src/compat/compatConfig.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ export interface CompilerCompatOptions {
1313
compatConfig?: CompilerCompatConfig
1414
}
1515

16-
export const enum CompilerDeprecationTypes {
16+
export enum CompilerDeprecationTypes {
1717
COMPILER_IS_ON_ELEMENT = 'COMPILER_IS_ON_ELEMENT',
1818
COMPILER_V_BIND_SYNC = 'COMPILER_V_BIND_SYNC',
1919
COMPILER_V_BIND_PROP = 'COMPILER_V_BIND_PROP',

Diff for: ‎packages/compiler-core/src/errors.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ export function createCompilerError<T extends number>(
3737
return error
3838
}
3939

40-
export const enum ErrorCodes {
40+
export enum ErrorCodes {
4141
// parse errors
4242
ABRUPT_CLOSING_OF_EMPTY_COMMENT,
4343
CDATA_IN_HTML_CONTENT,

Diff for: ‎packages/compiler-core/src/options.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ export type HoistTransform = (
7373
parent: ParentNode
7474
) => void
7575

76-
export const enum BindingTypes {
76+
export enum BindingTypes {
7777
/**
7878
* returned from data()
7979
*/

Diff for: ‎packages/compiler-core/src/parse.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ export const defaultParserOptions: MergedParserOptions = {
8080
comments: __DEV__
8181
}
8282

83-
export const enum TextModes {
83+
export enum TextModes {
8484
// | Elements | Entities | End sign | Inside of
8585
DATA, // | ✔ | ✔ | End tags of ancestors |
8686
RCDATA, // | ✘ | ✔ | End tag of the parent | <textarea>
@@ -502,7 +502,7 @@ function parseElement(
502502
return element
503503
}
504504

505-
const enum TagType {
505+
enum TagType {
506506
Start,
507507
End
508508
}

Diff for: ‎packages/compiler-core/src/utils.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ const nonIdentifierRE = /^\d|[^\$\w]/
6464
export const isSimpleIdentifier = (name: string): boolean =>
6565
!nonIdentifierRE.test(name)
6666

67-
const enum MemberExpLexState {
67+
enum MemberExpLexState {
6868
inMemberExp,
6969
inBrackets,
7070
inParens,

Diff for: ‎packages/compiler-dom/src/errors.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ export function createDOMCompilerError(
2020
) as DOMCompilerError
2121
}
2222

23-
export const enum DOMErrorCodes {
23+
export enum DOMErrorCodes {
2424
X_V_HTML_NO_EXPRESSION = 53 /* ErrorCodes.__EXTEND_POINT__ */,
2525
X_V_HTML_WITH_CHILDREN,
2626
X_V_TEXT_NO_EXPRESSION,
@@ -36,7 +36,7 @@ export const enum DOMErrorCodes {
3636
}
3737

3838
if (__TEST__) {
39-
// esbuild cannot infer const enum increments if first value is from another
39+
// esbuild cannot infer enum increments if first value is from another
4040
// file, so we have to manually keep them in sync. this check ensures it
4141
// errors out if there are collisions.
4242
if (DOMErrorCodes.X_V_HTML_NO_EXPRESSION < ErrorCodes.__EXTEND_POINT__) {

Diff for: ‎packages/compiler-dom/src/parserOptions.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ const isRawTextContainer = /*#__PURE__*/ makeMap(
1515
true
1616
)
1717

18-
export const enum DOMNamespaces {
18+
export enum DOMNamespaces {
1919
HTML = 0 /* Namespaces.HTML */,
2020
SVG,
2121
MATH_ML

Diff for: ‎packages/compiler-dom/src/transforms/stringifyStatic.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ import {
3333
} from '@vue/shared'
3434
import { DOMNamespaces } from '../parserOptions'
3535

36-
export const enum StringifyThresholds {
36+
export enum StringifyThresholds {
3737
ELEMENT_WITH_BINDING_COUNT = 5,
3838
NODE_COUNT = 20
3939
}

Diff for: ‎packages/compiler-sfc/src/style/cssVars.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ export function parseCssVars(sfc: SFCDescriptor): string[] {
7070
return vars
7171
}
7272

73-
const enum LexerState {
73+
enum LexerState {
7474
inParens,
7575
inSingleQuoteString,
7676
inDoubleQuoteString

Diff for: ‎packages/compiler-ssr/src/errors.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -16,14 +16,14 @@ export function createSSRCompilerError(
1616
return createCompilerError(code, loc, SSRErrorMessages) as SSRCompilerError
1717
}
1818

19-
export const enum SSRErrorCodes {
19+
export enum SSRErrorCodes {
2020
X_SSR_UNSAFE_ATTR_NAME = 65 /* DOMErrorCodes.__EXTEND_POINT__ */,
2121
X_SSR_NO_TELEPORT_TARGET,
2222
X_SSR_INVALID_AST_NODE
2323
}
2424

2525
if (__TEST__) {
26-
// esbuild cannot infer const enum increments if first value is from another
26+
// esbuild cannot infer enum increments if first value is from another
2727
// file, so we have to manually keep them in sync. this check ensures it
2828
// errors out if there are collisions.
2929
if (SSRErrorCodes.X_SSR_UNSAFE_ATTR_NAME < DOMErrorCodes.__EXTEND_POINT__) {

Diff for: ‎packages/reactivity/src/operations.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
// using literal strings instead of numbers so that it's easier to inspect
22
// debugger events
33

4-
export const enum TrackOpTypes {
4+
export enum TrackOpTypes {
55
GET = 'get',
66
HAS = 'has',
77
ITERATE = 'iterate'
88
}
99

10-
export const enum TriggerOpTypes {
10+
export enum TriggerOpTypes {
1111
SET = 'set',
1212
ADD = 'add',
1313
DELETE = 'delete',

Diff for: ‎packages/reactivity/src/reactive.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ import {
1313
} from './collectionHandlers'
1414
import type { UnwrapRefSimple, Ref, RawSymbol } from './ref'
1515

16-
export const enum ReactiveFlags {
16+
export enum ReactiveFlags {
1717
SKIP = '__v_skip',
1818
IS_REACTIVE = '__v_isReactive',
1919
IS_READONLY = '__v_isReadonly',
@@ -34,7 +34,7 @@ export const shallowReactiveMap = new WeakMap<Target, any>()
3434
export const readonlyMap = new WeakMap<Target, any>()
3535
export const shallowReadonlyMap = new WeakMap<Target, any>()
3636

37-
const enum TargetType {
37+
enum TargetType {
3838
INVALID = 0,
3939
COMMON = 1,
4040
COLLECTION = 2

Diff for: ‎packages/runtime-core/src/compat/compatConfig.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ import {
1010
} from '../component'
1111
import { warn } from '../warning'
1212

13-
export const enum DeprecationTypes {
13+
export enum DeprecationTypes {
1414
GLOBAL_MOUNT = 'GLOBAL_MOUNT',
1515
GLOBAL_MOUNT_CONTAINER = 'GLOBAL_MOUNT_CONTAINER',
1616
GLOBAL_EXTEND = 'GLOBAL_EXTEND',

Diff for: ‎packages/runtime-core/src/componentOptions.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -586,7 +586,7 @@ export type OptionTypesType<
586586
Defaults: Defaults
587587
}
588588

589-
const enum OptionTypes {
589+
enum OptionTypes {
590590
PROPS = 'Props',
591591
DATA = 'Data',
592592
COMPUTED = 'Computed',

Diff for: ‎packages/runtime-core/src/componentProps.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ export type ExtractPublicPropTypes<O> = {
164164
[K in keyof Pick<O, PublicOptionalKeys<O>>]?: InferPropType<O[K]>
165165
}
166166

167-
const enum BooleanFlags {
167+
enum BooleanFlags {
168168
shouldCast,
169169
shouldCastTrue
170170
}

Diff for: ‎packages/runtime-core/src/componentPublicInstance.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -278,7 +278,7 @@ if (__COMPAT__) {
278278
installCompatInstanceProperties(publicPropertiesMap)
279279
}
280280

281-
const enum AccessTypes {
281+
enum AccessTypes {
282282
OTHER,
283283
SETUP,
284284
DATA,

Diff for: ‎packages/runtime-core/src/components/Teleport.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ export const TeleportImpl = {
269269
hydrate: hydrateTeleport
270270
}
271271

272-
export const enum TeleportMoveTypes {
272+
export enum TeleportMoveTypes {
273273
TARGET_CHANGE,
274274
TOGGLE, // enable / disable
275275
REORDER // moved in the main view

Diff for: ‎packages/runtime-core/src/devtools.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ interface AppRecord {
1010
types: Record<string, string | Symbol>
1111
}
1212

13-
const enum DevtoolsHooks {
13+
enum DevtoolsHooks {
1414
APP_INIT = 'app:init',
1515
APP_UNMOUNT = 'app:unmount',
1616
COMPONENT_UPDATED = 'component:updated',

Diff for: ‎packages/runtime-core/src/enums.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
export const enum LifecycleHooks {
1+
export enum LifecycleHooks {
22
BEFORE_CREATE = 'bc',
33
CREATED = 'c',
44
BEFORE_MOUNT = 'bm',

Diff for: ‎packages/runtime-core/src/errorHandling.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import { LifecycleHooks } from './enums'
66

77
// contexts where user provided function may be executed, in addition to
88
// lifecycle hooks.
9-
export const enum ErrorCodes {
9+
export enum ErrorCodes {
1010
SETUP_FUNCTION,
1111
RENDER_FUNCTION,
1212
WATCH_GETTER,

Diff for: ‎packages/runtime-core/src/hydration.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ export type RootHydrateFunction = (
3030
container: (Element | ShadowRoot) & { _vnode?: VNode }
3131
) => void
3232

33-
const enum DOMNodeTypes {
33+
enum DOMNodeTypes {
3434
ELEMENT = 1,
3535
TEXT = 3,
3636
COMMENT = 8

Diff for: ‎packages/runtime-core/src/renderer.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,7 @@ export type SetupRenderEffectFn = (
264264
optimized: boolean
265265
) => void
266266

267-
export const enum MoveType {
267+
export enum MoveType {
268268
ENTER,
269269
LEAVE,
270270
REORDER

Diff for: ‎packages/runtime-test/src/nodeOps.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
import { markRaw } from '@vue/reactivity'
22

3-
export const enum TestNodeTypes {
3+
export enum TestNodeTypes {
44
TEXT = 'text',
55
ELEMENT = 'element',
66
COMMENT = 'comment'
77
}
88

9-
export const enum NodeOpTypes {
9+
export enum NodeOpTypes {
1010
CREATE = 'create',
1111
INSERT = 'insert',
1212
REMOVE = 'remove',

Diff for: ‎packages/shared/src/patchFlags.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
* Check the `patchElement` function in '../../runtime-core/src/renderer.ts' to see how the
1717
* flags are handled during diff.
1818
*/
19-
export const enum PatchFlags {
19+
export enum PatchFlags {
2020
/**
2121
* Indicates an element with dynamic textContent (children fast path)
2222
*/

Diff for: ‎packages/shared/src/shapeFlags.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
export const enum ShapeFlags {
1+
export enum ShapeFlags {
22
ELEMENT = 1,
33
FUNCTIONAL_COMPONENT = 1 << 1,
44
STATEFUL_COMPONENT = 1 << 2,

Diff for: ‎packages/shared/src/slotFlags.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
export const enum SlotFlags {
1+
export enum SlotFlags {
22
/**
33
* Stable slots that only reference slot props or context state. The slot
44
* can fully capture its own dependencies so when passed down the parent won't

Diff for: ‎packages/vue/macros.d.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ import {
1010

1111
export declare const RefType: unique symbol
1212

13-
export declare const enum RefTypes {
13+
export declare enum RefTypes {
1414
Ref = 1,
1515
ComputedRef = 2,
1616
WritableComputedRef = 3

Diff for: ‎pnpm-lock.yaml

+8
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

Diff for: ‎rollup.config.js

+2-2
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ import terser from '@rollup/plugin-terser'
1212
import esbuild from 'rollup-plugin-esbuild'
1313
import alias from '@rollup/plugin-alias'
1414
import { entries } from './scripts/aliases.js'
15-
import { constEnum } from './scripts/const-enum.js'
15+
import { simplifyEnum } from './scripts/simplify-enum.js'
1616

1717
if (!process.env.TARGET) {
1818
throw new Error('TARGET package must be specified via --environment flag.')
@@ -32,7 +32,7 @@ const pkg = require(resolve(`package.json`))
3232
const packageOptions = pkg.buildOptions || {}
3333
const name = packageOptions.filename || path.basename(packageDir)
3434

35-
const [enumPlugin, enumDefines] = constEnum()
35+
const [enumPlugin, enumDefines] = simplifyEnum()
3636

3737
const outputConfigs = {
3838
'esm-bundler': {

Diff for: ‎scripts/build.js

+1-1
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ import { execa, execaSync } from 'execa'
2626
import { cpus } from 'node:os'
2727
import { createRequire } from 'node:module'
2828
import { targets as allTargets, fuzzyMatchTarget } from './utils.js'
29-
import { scanEnums } from './const-enum.js'
29+
import { scanEnums } from './simplify-enum.js'
3030
import prettyBytes from 'pretty-bytes'
3131

3232
const require = createRequire(import.meta.url)

Diff for: ‎scripts/const-enum.js renamed to ‎scripts/simplify-enum.js

+73-62
Original file line numberDiff line numberDiff line change
@@ -26,26 +26,33 @@ import { parse } from '@babel/parser'
2626
import path from 'node:path'
2727
import MagicString from 'magic-string'
2828

29+
/**
30+
* @typedef {{ id: string, range: [start: number, end: number], members: Array<{ name: string, value: string | number }>}} EnumDeclaration
31+
* @typedef {{ declarations: { [file: string] : Array<EnumDeclaration>}, defines: { [ id_key: `${string}.${string}`]: string }, ids: string[] }} EnumData
32+
*/
33+
2934
const ENUM_CACHE_PATH = 'temp/enum.json'
3035

36+
/**
37+
* @param {string} exp
38+
* @returns {string | number}
39+
*/
3140
function evaluate(exp) {
3241
return new Function(`return ${exp}`)()
3342
}
3443

3544
// this is called in the build script entry once
3645
// so the data can be shared across concurrent Rollup processes
3746
export function scanEnums() {
38-
/**
39-
* @type {{ ranges: Record<string, [number, number][]>, defines: Record<string, string>, ids: string[] }}
40-
*/
47+
/** @type {EnumData} */
4148
const enumData = {
42-
ranges: {},
49+
declarations: {},
4350
defines: {},
4451
ids: []
4552
}
4653

4754
// 1. grep for files with exported const enum
48-
const { stdout } = execaSync('git', ['grep', `export const enum`])
55+
const { stdout } = execaSync('git', ['grep', `export enum`])
4956
const files = [...new Set(stdout.split('\n').map(line => line.split(':')[0]))]
5057

5158
// 2. parse matched files to collect enum info
@@ -63,32 +70,38 @@ export function scanEnums() {
6370
node.declaration &&
6471
node.declaration.type === 'TSEnumDeclaration'
6572
) {
66-
if (file in enumData.ranges) {
67-
// @ts-ignore
68-
enumData.ranges[file].push([node.start, node.end])
69-
} else {
70-
// @ts-ignore
71-
enumData.ranges[file] = [[node.start, node.end]]
72-
}
73-
7473
const decl = node.declaration
74+
const id = decl.id.name
75+
/** @type {string | number | undefined} */
7576
let lastInitialized
77+
78+
/** @type {EnumDeclaration} */
79+
const declaration = {
80+
id,
81+
range: [node.start ?? -1, node.end ?? -1],
82+
members: []
83+
}
84+
if (enumData.ids.includes(id)) {
85+
// Can not throw Error here.
86+
// We need allow same name enum in different file.
87+
// For example: enum ErrorCodes exist in both @vue/compiler-core and @vue/runtime-core
88+
} else {
89+
enumData.ids.push(id)
90+
}
7691
for (let i = 0; i < decl.members.length; i++) {
7792
const e = decl.members[i]
78-
const id = decl.id.name
79-
if (!enumData.ids.includes(id)) {
80-
enumData.ids.push(id)
81-
}
8293
const key = e.id.type === 'Identifier' ? e.id.name : e.id.value
83-
const fullKey = `${id}.${key}`
84-
const saveValue = value => {
94+
const fullKey = /** @type {const} */ (`${id}.${key}`)
95+
const saveValue = (/** @type {string | number} */ value) => {
8596
if (fullKey in enumData.defines) {
8697
throw new Error(`name conflict for enum ${id} in ${file}`)
8798
}
8899
enumData.defines[fullKey] = JSON.stringify(value)
100+
declaration.members.push({ name: key, value })
89101
}
90102
const init = e.initializer
91103
if (init) {
104+
/** @type {string | number | undefined} */
92105
let value
93106
if (
94107
init.type === 'StringLiteral' ||
@@ -99,14 +112,16 @@ export function scanEnums() {
99112

100113
// e.g. 1 << 2
101114
if (init.type === 'BinaryExpression') {
102-
const resolveValue = node => {
115+
const resolveValue = (
116+
/** @type {import('@babel/types').Expression | import('@babel/types').PrivateName} */ node
117+
) => {
103118
if (
104119
node.type === 'NumericLiteral' ||
105120
node.type === 'StringLiteral'
106121
) {
107122
return node.value
108123
} else if (node.type === 'MemberExpression') {
109-
const exp = content.slice(node.start, node.end)
124+
const exp = content.slice(node.start ?? 0, node.end ?? 0)
110125
if (!(exp in enumData.defines)) {
111126
throw new Error(
112127
`unhandled enum initialization expression ${exp} in ${file}`
@@ -158,6 +173,10 @@ export function scanEnums() {
158173
}
159174
}
160175
}
176+
if (!(file in enumData.declarations)) {
177+
enumData.declarations[file] = []
178+
}
179+
enumData.declarations[file].push(declaration)
161180
}
162181
}
163182
}
@@ -174,12 +193,12 @@ export function scanEnums() {
174193
/**
175194
* @returns {[import('rollup').Plugin, Record<string, string>]}
176195
*/
177-
export function constEnum() {
196+
export function simplifyEnum() {
178197
if (!existsSync(ENUM_CACHE_PATH)) {
179198
throw new Error('enum cache needs to be initialized before creating plugin')
180199
}
181200
/**
182-
* @type {{ ranges: Record<string, [number, number][]>, defines: Record<string, string>, ids: string[] }}
201+
* @type {EnumData}
183202
*/
184203
const enumData = JSON.parse(readFileSync(ENUM_CACHE_PATH, 'utf-8'))
185204

@@ -195,50 +214,42 @@ export function constEnum() {
195214
* @type {import('rollup').Plugin}
196215
*/
197216
const plugin = {
198-
name: 'remove-const-enum',
217+
name: 'simplify-enum',
199218
transform(code, id) {
200219
let s
201220

202-
if (id in enumData.ranges) {
221+
if (id in enumData.declarations) {
203222
s = s || new MagicString(code)
204-
for (const [start, end] of enumData.ranges[id]) {
205-
s.remove(start, end)
206-
}
207-
}
223+
for (const declaration of enumData.declarations[id]) {
224+
const {
225+
range: [start, end],
226+
id,
227+
members
228+
} = declaration
229+
s.update(
230+
start,
231+
end,
232+
`export const ${id} = {\n ${members
233+
.flatMap(({ name, value }) => {
234+
const forwardMapping =
235+
JSON.stringify(name) + ': ' + JSON.stringify(value)
236+
const reverseMapping =
237+
JSON.stringify(value.toString()) + ': ' + JSON.stringify(name)
208238
209-
// check for const enum re-exports that must be removed
210-
if (reExportsRE.test(code)) {
211-
s = s || new MagicString(code)
212-
const ast = parse(code, {
213-
plugins: ['typescript'],
214-
sourceType: 'module'
215-
})
216-
for (const node of ast.program.body) {
217-
if (
218-
node.type === 'ExportNamedDeclaration' &&
219-
node.exportKind !== 'type' &&
220-
node.source
221-
) {
222-
for (let i = 0; i < node.specifiers.length; i++) {
223-
const spec = node.specifiers[i]
224-
if (
225-
spec.type === 'ExportSpecifier' &&
226-
spec.exportKind !== 'type' &&
227-
enumData.ids.includes(spec.local.name)
228-
) {
229-
const next = node.specifiers[i + 1]
230-
if (next) {
231-
// @ts-ignore
232-
s.remove(spec.start, next.start)
233-
} else {
234-
// last one
235-
const prev = node.specifiers[i - 1]
236-
// @ts-ignore
237-
s.remove(prev ? prev.end : spec.start, spec.end)
238-
}
239-
}
240-
}
241-
}
239+
// see https://www.typescriptlang.org/docs/handbook/enums.html#reverse-mappings
240+
return typeof value === 'string'
241+
? [
242+
forwardMapping
243+
// string enum members do not get a reverse mapping generated at all
244+
]
245+
: [
246+
forwardMapping,
247+
// other enum members should support enum reverse mapping
248+
reverseMapping
249+
]
250+
})
251+
.join(',\n ')}}`
252+
)
242253
}
243254
}
244255

0 commit comments

Comments
 (0)
Please sign in to comment.