Handle yield token

src/index.ts

@@ -660,6 +660,10 @@ export const plugin: Plugin = {
 							result = printIndent(previousToken, result, indent, indentLevel);
 							result += '-';
 							break;
+						case 'yield':
+							result = printIndent(previousToken, result, indent, indentLevel);
+							result += 'yield';
+							break;
 						default:
 							throw new Error('Unhandled token: ' + JSON.stringify(token));
 					}

src/pug-lexer.d.ts

@@ -46,7 +46,8 @@ declare module 'pug-lexer' {
 			| 'text-html'
 			| 'text'
 			| 'when'
-			| 'while';
+			| 'while'
+			| 'yield';
 
 		export interface LexToken<Type extends LexTokenType> {
 			type: Type;
@@ -197,6 +198,8 @@ declare module 'pug-lexer' {
 
 		export type BlockcodeToken = LexToken<'blockcode'>;
 
+		export type YieldToken = LexToken<'yield'>;
+
 		export type Token =
 			| AndAttributesToken
 			| AttributeToken
@@ -238,7 +241,8 @@ declare module 'pug-lexer' {
 			| TextHtmlToken
 			| TextToken
 			| WhenToken
-			| WhileToken;
+			| WhileToken
+			| YieldToken;
 
 		export type LexerFunction = (type: string, exp?: any) => boolean;
 		export interface LexerOptions {

test/yield/formatted.pug

@@ -0,0 +1,4 @@
+head
+  script(src="/jquery.js")
+  yield
+  script(src="/jquery.ui.js")

test/yield/unformatted.pug

@@ -0,0 +1,4 @@
+head
+  script(src="/jquery.js")
+  yield
+  script(src="/jquery.ui.js")

test/yield/yield.test.ts

@@ -0,0 +1,14 @@
+import { readFileSync } from 'fs';
+import { resolve } from 'path';
+import { format } from 'prettier';
+import { plugin } from './../../src/index';
+
+describe('Yield', () => {
+	test('should handle yield token', () => {
+		const expected: string = readFileSync(resolve(__dirname, 'formatted.pug'), 'utf8');
+		const code: string = readFileSync(resolve(__dirname, 'unformatted.pug'), 'utf8');
+		const actual: string = format(code, { parser: 'pug' as any, plugins: [plugin] });
+
+		expect(actual).toBe(expected);
+	});
+});