feat: support svelte:document (#355)

Author: dummdidumm

CHANGELOG.md

@@ -2,7 +2,8 @@
 
 ## 2.10.0 (unreleased)
 
--   (feat) support `requirePragma` and `insertPragma` options
+-   (feat) support `requirePragma` and `insertPragma` options ([#350](https://github.com/sveltejs/prettier-plugin-svelte/issues/350))
+-   (feat) support `<svelte:document>`
 
 ## 2.9.0
 

package-lock.json

@@ -37,7 +37,7 @@
         "prettier": "^2.7.1",
         "rollup": "2.36.0",
         "rollup-plugin-typescript": "1.0.1",
-        "svelte": "^3.54.0",
+        "svelte": "^3.57.0",
         "ts-node": "^9.1.1",
         "tslib": "^2.0.3",
         "typescript": "4.1.3"

package.json

@@ -2,6 +2,7 @@ import {
     ASTNode,
     AttributeNode,
     BodyNode,
+    DocumentNode,
     ElementNode,
     HeadNode,
     InlineComponentNode,
@@ -82,6 +83,7 @@ export function getAttributeLine(
         | StyleNode
         | ScriptNode
         | BodyNode
+        | DocumentNode
         | OptionsNode
         | SlotTemplateNode,
     options: ParserOptions,
@@ -111,6 +113,7 @@ export function printWithPrependedAttributeLine(
         | StyleNode
         | ScriptNode
         | BodyNode
+        | DocumentNode
         | OptionsNode
         | SlotTemplateNode,
     options: ParserOptions,

src/print/helpers.ts

@@ -412,6 +412,21 @@ export function print(path: FastPath, options: ParserOptions, print: PrintFn): D
                 ),
                 ...[bracketSameLine ? ' ' : '', '/>'],
             ]);
+        case 'Document':
+            return groupConcat([
+                '<',
+                node.name,
+                indent(
+                    groupConcat([
+                        ...path.map(
+                            printWithPrependedAttributeLine(node, options, print),
+                            'attributes',
+                        ),
+                        bracketSameLine ? '' : dedent(line),
+                    ]),
+                ),
+                ...[bracketSameLine ? ' ' : '', '/>'],
+            ]);
         case 'Identifier':
             return node.name;
         case 'AttributeShorthand': {

src/print/index.ts

@@ -259,6 +259,12 @@ export interface BodyNode extends BaseNode {
     attributes: Node[];
 }
 
+export interface DocumentNode extends BaseNode {
+    type: 'Document';
+    name: string;
+    attributes: Node[];
+}
+
 export interface OptionsNode extends BaseNode {
     type: 'Options';
     name: string;
@@ -318,6 +324,7 @@ export type Node =
     | InstanceScriptNode
     | ModuleScriptNode
     | BodyNode
+    | DocumentNode
     | OptionsNode
     | SlotTemplateNode
     | ConstTagNode;

src/print/nodes.ts

@@ -0,0 +1 @@
+<svelte:document on:event={handler} />