Skip to content

Commit

Permalink
fix: robustify migration script (#12019)
Browse files Browse the repository at this point in the history
* handle multiple slots of same kind correctly

* put props below imports

* migrate $$slots, better error position for slot-children-conflict error

* handle event braces+modifier case

* migrate jsdoc type with curly braces correctly

* handle comments above types, fixes #11508

* changeset
  • Loading branch information
dummdidumm committed Jun 13, 2024
1 parent 4cdc371 commit bf23f69
Show file tree
Hide file tree
Showing 16 changed files with 178 additions and 76 deletions.
5 changes: 5 additions & 0 deletions .changeset/clean-cats-wave.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
---
"svelte": patch
---

fix: robustify migration script
120 changes: 78 additions & 42 deletions packages/svelte/src/compiler/migrate/index.js
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,8 @@ export function migrate(source) {

if (state.props.length > 0 || analysis.uses_rest_props || analysis.uses_props) {
const has_many_props = state.props.length > 3;
const props_separator = has_many_props ? `\n${indent}${indent}` : ' ';
const newline_separator = `\n${indent}${indent}`;
const props_separator = has_many_props ? newline_separator : ' ';
let props = '';
if (analysis.uses_props) {
props = `...${state.props_name}`;
Expand Down Expand Up @@ -99,11 +100,12 @@ export function migrate(source) {
if (analysis.uses_props || analysis.uses_rest_props) {
type = `interface ${type_name} { [key: string]: any }`;
} else {
type = `interface ${type_name} {${props_separator}${state.props
type = `interface ${type_name} {${newline_separator}${state.props
.map((prop) => {
return `${prop.exported}${prop.optional ? '?' : ''}: ${prop.type}`;
const comment = prop.comment ? `${prop.comment}${newline_separator}` : '';
return `${comment}${prop.exported}${prop.optional ? '?' : ''}: ${prop.type};`;
})
.join(`,${props_separator}`)}${has_many_props ? `\n${indent}` : ' '}}`;
.join(newline_separator)}\n${indent}}`;
}
} else {
if (analysis.uses_props || analysis.uses_rest_props) {
Expand Down Expand Up @@ -162,7 +164,7 @@ export function migrate(source) {
* str: MagicString;
* analysis: import('../phases/types.js').ComponentAnalysis;
* indent: string;
* props: Array<{ local: string; exported: string; init: string; bindable: boolean; optional: boolean; type: string }>;
* props: Array<{ local: string; exported: string; init: string; bindable: boolean; slot_name?: string; optional: boolean; type: string; comment?: string }>;
* props_insertion_point: number;
* has_props_rune: boolean;
* props_name: string;
Expand Down Expand Up @@ -190,8 +192,11 @@ const instance_script = {
}
next();
},
Identifier(node, { state }) {
handle_identifier(node, state);
Identifier(node, { state, path }) {
handle_identifier(node, state, path);
},
ImportDeclaration(node, { state }) {
state.props_insertion_point = node.end ?? state.props_insertion_point;
},
ExportNamedDeclaration(node, { state, next }) {
if (node.declaration) {
Expand Down Expand Up @@ -299,8 +304,8 @@ const instance_script = {
)
: '',
optional: !!declarator.init,
type: extract_type(declarator, state.str, path),
bindable: binding.mutated || binding.reassigned
bindable: binding.mutated || binding.reassigned,
...extract_type_and_comment(declarator, state.str, path)
});
state.props_insertion_point = /** @type {number} */ (declarator.end);
state.str.update(
Expand Down Expand Up @@ -423,8 +428,8 @@ const instance_script = {

/** @type {import('zimmerframe').Visitors<import('../types/template.js').SvelteNode, State>} */
const template = {
Identifier(node, { state }) {
handle_identifier(node, state);
Identifier(node, { state, path }) {
handle_identifier(node, state, path);
},
RegularElement(node, { state, next }) {
handle_events(node, state);
Expand Down Expand Up @@ -468,14 +473,15 @@ const template = {
},
SlotElement(node, { state, next }) {
let name = 'children';
let slot_name = 'default';
let slot_props = '{ ';

for (const attr of node.attributes) {
if (attr.type === 'SpreadAttribute') {
slot_props += `...${state.str.original.substring(/** @type {number} */ (attr.expression.start), attr.expression.end)}, `;
} else if (attr.type === 'Attribute') {
if (attr.name === 'name') {
name = state.scope.generate(/** @type {any} */ (attr.value)[0].data);
slot_name = /** @type {any} */ (attr.value)[0].data;
} else {
const value =
attr.value !== true
Expand All @@ -494,14 +500,24 @@ const template = {
slot_props = '';
}

state.props.push({
local: name,
exported: name,
init: '',
bindable: false,
optional: true,
type: `import('svelte').${slot_props ? 'Snippet<[any]>' : 'Snippet'}`
});
const existing_prop = state.props.find((prop) => prop.slot_name === slot_name);
if (existing_prop) {
name = existing_prop.local;
} else if (slot_name !== 'default') {
name = state.scope.generate(slot_name);
}

if (!existing_prop) {
state.props.push({
local: name,
exported: name,
init: '',
bindable: false,
optional: true,
slot_name,
type: `import('svelte').${slot_props ? 'Snippet<[any]>' : 'Snippet'}`
});
}

if (node.fragment.nodes.length > 0) {
next();
Expand All @@ -528,37 +544,46 @@ const template = {
* @param {MagicString} str
* @param {import('#compiler').SvelteNode[]} path
*/
function extract_type(declarator, str, path) {
function extract_type_and_comment(declarator, str, path) {
const parent = path.at(-1);

// Try to find jsdoc above the declaration
let comment_node = /** @type {import('estree').Node} */ (parent)?.leadingComments?.at(-1);
if (comment_node?.type !== 'Block') comment_node = undefined;

const comment_start = /** @type {any} */ (comment_node)?.start;
const comment_end = /** @type {any} */ (comment_node)?.end;
const comment = comment_node && str.original.substring(comment_start, comment_end);

if (comment_node) {
str.update(comment_start, comment_end, '');
}

if (declarator.id.typeAnnotation) {
let start = declarator.id.typeAnnotation.start + 1; // skip the colon
while (str.original[start] === ' ') {
start++;
}
return str.original.substring(start, declarator.id.typeAnnotation.end);
return { type: str.original.substring(start, declarator.id.typeAnnotation.end), comment };
}

// try to find a comment with a type annotation, hinting at jsdoc
const parent = path.at(-1);
if (parent?.type === 'ExportNamedDeclaration' && parent.leadingComments) {
const last = parent.leadingComments[parent.leadingComments.length - 1];
if (last.type === 'Block') {
const match = /@type {([^}]+)}/.exec(last.value);
if (match) {
str.update(/** @type {any} */ (last).start, /** @type {any} */ (last).end, '');
return match[1];
}
if (parent?.type === 'ExportNamedDeclaration' && comment_node) {
const match = /@type {(.+)}/.exec(comment_node.value);
if (match) {
return { type: match[1] };
}
}

// try to infer it from the init
if (declarator.init?.type === 'Literal') {
const type = typeof declarator.init.value;
if (type === 'string' || type === 'number' || type === 'boolean') {
return type;
return { type, comment };
}
}

return 'any';
return { type: 'any', comment };
}

/**
Expand Down Expand Up @@ -694,14 +719,16 @@ function handle_events(node, state) {
}

const needs_curlies = last.expression.body.type !== 'BlockStatement';
state.str.prependRight(
/** @type {number} */ (pos) + (needs_curlies ? 0 : 1),
`${needs_curlies ? '{' : ''}${prepend}${state.indent}`
);
state.str.appendRight(
/** @type {number} */ (last.expression.body.end) - (needs_curlies ? 0 : 1),
`\n${needs_curlies ? '}' : ''}`
);
const end = /** @type {number} */ (last.expression.body.end) - (needs_curlies ? 0 : 1);
pos = /** @type {number} */ (pos) + (needs_curlies ? 0 : 1);
if (needs_curlies && state.str.original[pos - 1] === '(') {
// Prettier does something like on:click={() => (foo = true)}, we need to remove the braces in this case
state.str.update(pos - 1, pos, `{${prepend}${state.indent}`);
state.str.update(end, end + 1, '\n}');
} else {
state.str.prependRight(pos, `${needs_curlies ? '{' : ''}${prepend}${state.indent}`);
state.str.appendRight(end, `\n${needs_curlies ? '}' : ''}`);
}
} else {
state.str.update(
/** @type {number} */ (last.expression.start),
Expand Down Expand Up @@ -763,8 +790,12 @@ function generate_event_name(last, state) {
/**
* @param {import('estree').Identifier} node
* @param {State} state
* @param {any[]} path
*/
function handle_identifier(node, state) {
function handle_identifier(node, state, path) {
const parent = path.at(-1);
if (parent?.type === 'MemberExpression' && parent.property === node) return;

if (state.analysis.uses_props) {
if (node.name === '$$props' || node.name === '$$restProps') {
// not 100% correct for $$restProps but it'll do
Expand All @@ -785,6 +816,11 @@ function handle_identifier(node, state) {
/** @type {number} */ (node.end),
state.rest_props_name
);
} else if (node.name === '$$slots' && state.analysis.uses_slots) {
if (parent?.type === 'MemberExpression') {
state.str.update(/** @type {number} */ (node.start), parent.property.start, '');
}
// else passed as identifier, we don't know what to do here, so let it error
}
}

Expand Down
3 changes: 2 additions & 1 deletion packages/svelte/src/compiler/phases/2-analyze/index.js
Original file line number Diff line number Diff line change
Expand Up @@ -554,7 +554,8 @@ export function analyze_component(root, source, options) {
}

if (analysis.uses_render_tags && (analysis.uses_slots || analysis.slot_names.size > 0)) {
e.slot_snippet_conflict(analysis.slot_names.values().next().value);
const pos = analysis.slot_names.values().next().value ?? analysis.source.indexOf('$$slot');
e.slot_snippet_conflict(pos);
}

if (analysis.css.ast) {
Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
<button on:click={() => console.log('hi')} on:click>click me</button>
<button on:click={() => console.log('before')} on:click on:click={() => console.log('after')}>click me</button>
<button on:click={() => console.log('before')} on:click on:click={() => console.log('after')}
>click me</button
>
<button on:click on:click={foo}>click me</button>
<button on:click>click me</button>

Expand All @@ -9,6 +11,7 @@
<button on:custom-event-bubble>click me</button>

<button on:click|preventDefault={() => ''}>click me</button>
<button on:click|preventDefault={() => (searching = true)}>click me</button>
<button on:click|stopPropagation={() => {}}>click me</button>
<button on:click|stopImmediatePropagation={() => ''}>click me</button>
<button on:click|capture={() => ''}>click me</button>
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,9 @@

onclick?.(event);
console.log('after')
}}>click me</button>
}}
>click me</button
>
<button onclick={(event) => {
onclick?.(event);

Expand All @@ -30,6 +32,10 @@
event.preventDefault();
''
}}>click me</button>
<button onclick={(event) => {
event.preventDefault();
searching = true
}}>click me</button>
<button onclick={(event) => {
event.stopPropagation();

Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
<script lang="ts">
interface Props { class?: string }
interface Props {
class?: string;
}
let { class: klass = '' }: Props = $props();
Expand Down
9 changes: 5 additions & 4 deletions packages/svelte/tests/migrate/samples/props-ts/input.svelte
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
<script lang="ts">
export let readonly: number;
export let optional = 'foo';
export let binding: string;
export let bindingOptional: string | undefined = 'bar';
/** some comment */
export let readonly: number;
export let optional = 'foo';
export let binding: string;
export let bindingOptional: string | undefined = 'bar';
</script>

{readonly}
Expand Down
26 changes: 14 additions & 12 deletions packages/svelte/tests/migrate/samples/props-ts/output.svelte
Original file line number Diff line number Diff line change
@@ -1,17 +1,19 @@
<script lang="ts">
interface Props {
readonly: number,
optional?: string,
binding: string,
bindingOptional?: string | undefined
}
interface Props {
/** some comment */
readonly: number;
optional?: string;
binding: string;
bindingOptional?: string | undefined;
}
let {
readonly,
optional = 'foo',
binding = $bindable(),
bindingOptional = $bindable('bar')
}: Props = $props();
let {
readonly,
optional = 'foo',
binding = $bindable(),
bindingOptional = $bindable('bar')
}: Props = $props();
</script>

{readonly}
Expand Down
9 changes: 5 additions & 4 deletions packages/svelte/tests/migrate/samples/props/input.svelte
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
<script>
export let readonly;
export let optional = 'foo';
export let binding;
export let bindingOptional = 'bar';
/** @type {Record<string, { href: string; title: string; }[]>} */
export let readonly;
export let optional = 'foo';
export let binding;
export let bindingOptional = 'bar';
</script>

{readonly}
Expand Down
15 changes: 8 additions & 7 deletions packages/svelte/tests/migrate/samples/props/output.svelte
Original file line number Diff line number Diff line change
@@ -1,11 +1,12 @@
<script>
/** @type {{readonly: any, optional?: string, binding: any, bindingOptional?: string}} */
let {
readonly,
optional = 'foo',
binding = $bindable(),
bindingOptional = $bindable('bar')
} = $props();
/** @type {{readonly: Record<string, { href: string; title: string; }[]>, optional?: string, binding: any, bindingOptional?: string}} */
let {
readonly,
optional = 'foo',
binding = $bindable(),
bindingOptional = $bindable('bar')
} = $props();
</script>

{readonly}
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
<script>
import Foo from './Foo.svelte';
</script>

<slot />
<Foo />
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
<script>
import Foo from './Foo.svelte';
/** @type {{children?: import('svelte').Snippet}} */
let { children } = $props();
</script>

{@render children?.()}
<Foo />
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
<button><slot /></button>
<button><slot /></button>

<slot name="dashed-name" />
<slot name="dashed-name" />
Loading

0 comments on commit bf23f69

Please sign in to comment.