esrap 1.4.0 → 1.4.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "esrap",
-  "version": "1.4.0",
+  "version": "1.4.2",
   "description": "Parse in reverse",
   "repository": {
     "type": "git",

package/src/handlers.js

@@ -1,5 +1,5 @@
 /** @import { TSESTree } from '@typescript-eslint/types' */
-/** @import { Chunk, Command, Dedent, Handlers, Indent, Newline, NodeWithComments, Sequence, State, TypeAnnotationNodes } from './types' */
+/** @import { Command, Dedent, Handlers, Location, Indent, Newline, NodeWithComments, State, TypeAnnotationNodes } from './types' */
 
 /** @type {Newline} */
 const newline = { type: 'Newline' };
@@ -11,11 +11,10 @@ const indent = { type: 'Indent' };
 const dedent = { type: 'Dedent' };
 
 /**
- * @param {Command[]} children
- * @returns {Sequence}
+ * @returns {Command[]}
  */
-function create_sequence(...children) {
-	return { type: 'Sequence', children };
+function create_sequence() {
+	return [];
 }
 
 /**
@@ -30,11 +29,11 @@ function measure(commands, from, to = commands.length) {
 		const command = commands[i];
 		if (typeof command === 'string') {
 			total += command.length;
-		} else if (command.type === 'Chunk') {
-			total += command.content.length;
-		} else if (command.type === 'Sequence') {
-			// assume this is ', '
-			total += 2;
+		} else if (Array.isArray(command)) {
+			total +=
+				command.length === 0
+					? 2 // assume this is ', '
+					: measure(command, 0);
 		}
 	}
 
@@ -66,17 +65,32 @@ export function handle(node, state) {
 	}
 }
 
+/**
+ * @param {number} line
+ * @param {number} column
+ * @returns {Location}
+ */
+function l(line, column) {
+	return {
+		type: 'Location',
+		line,
+		column
+	};
+}
+
 /**
  * @param {string} content
  * @param {TSESTree.Node} node
- * @returns {Chunk}
+ * @returns {string | Command[]}
  */
 function c(content, node) {
-	return {
-		type: 'Chunk',
-		content,
-		loc: node?.loc ?? null
-	};
+	return node.loc
+		? [
+				l(node.loc.start.line, node.loc.start.column),
+				content,
+				l(node.loc.end.line, node.loc.end.column)
+			]
+		: content;
 }
 
 /**
@@ -101,7 +115,26 @@ function prepend_comments(comments, state, newlines) {
  * @param {'\'' | '"'} char
  */
 function quote(string, char) {
-	return char + string.replaceAll(char, '\\' + char) + char;
+	let out = char;
+	let escaped = false;
+
+	for (const c of string) {
+		if (escaped) {
+			out += c;
+			escaped = false;
+		} else if (c === '\\') {
+			out += '\\\\';
+			escaped = true;
+		} else if (c === char) {
+			out += '\\' + c;
+		} else if (c === '\n') {
+			out += '\\n';
+		} else {
+			out += c;
+		}
+	}
+
+	return out + char;
 }
 
 const OPERATOR_PRECEDENCE = {
@@ -288,7 +321,7 @@ const handle_body = (nodes, state) => {
 				grouped_expression_types.includes(last_statement.type)) &&
 				last_statement.type !== statement.type)
 		) {
-			margin.children.push('\n');
+			margin.push('\n');
 		}
 
 		let add_newline = false;
@@ -332,11 +365,11 @@ const handle_var_declaration = (node, state) => {
 
 	if (multiline) {
 		state.multiline = true;
-		if (node.declarations.length > 1) open.children.push(indent);
-		join.children.push(',', newline);
+		if (node.declarations.length > 1) open.push(indent);
+		join.push(',', newline);
 		if (node.declarations.length > 1) state.commands.push(dedent);
 	} else {
-		join.children.push(', ');
+		join.push(', ');
 	}
 };
 
@@ -408,13 +441,13 @@ function sequence(nodes, state, spaces, fn, separator = ',') {
 	if (multiline) {
 		state.multiline = true;
 
-		open.children.push(indent, newline);
-		join.children.push(newline);
-		close.children.push(dedent, newline);
+		open.push(indent, newline);
+		join.push(newline);
+		close.push(dedent, newline);
 	} else {
-		if (spaces) open.children.push(' ');
-		join.children.push(' ');
-		if (spaces) close.children.push(' ');
+		if (spaces) open.push(' ');
+		join.push(' ');
+		if (spaces) close.push(' ');
 	}
 }
 
@@ -623,16 +656,26 @@ const shared = {
 	 * @param {State} state
 	 */
 	'BlockStatement|ClassBody': (node, state) => {
-		if (node.body.length === 0) {
-			state.commands.push('{}');
-			return;
+		if (node.loc) {
+			const { line, column } = node.loc.start;
+			state.commands.push(l(line, column), '{', l(line, column + 1));
+		} else {
+			state.commands.push('{');
 		}
 
-		state.multiline = true;
+		if (node.body.length > 0) {
+			state.multiline = true;
+			state.commands.push(indent, newline);
+			handle_body(node.body, state);
+			state.commands.push(dedent, newline);
+		}
 
-		state.commands.push('{', indent, newline);
-		handle_body(node.body, state);
-		state.commands.push(dedent, newline, '}');
+		if (node.loc) {
+			const { line, column } = node.loc.end;
+			state.commands.push(l(line, column - 1), '}', l(line, column));
+		} else {
+			state.commands.push('}');
+		}
 	},
 
 	/**
@@ -710,11 +753,11 @@ const shared = {
 		}
 
 		if (multiline) {
-			open.children.push(indent, newline);
-			join.children.push(',', newline);
-			close.children.push(dedent, newline);
+			open.push(indent, newline);
+			join.push(',', newline);
+			close.push(dedent, newline);
 		} else {
-			join.children.push(', ');
+			join.push(', ');
 		}
 	},
 
@@ -906,12 +949,12 @@ const handlers = {
 		const multiline = child_state.multiline;
 
 		if (multiline) {
-			if_true.children.push(indent, newline, '? ');
-			if_false.children.push(newline, ': ');
+			if_true.push(indent, newline, '? ');
+			if_false.push(newline, ': ');
 			state.commands.push(dedent);
 		} else {
-			if_true.children.push(' ? ');
-			if_false.children.push(' : ');
+			if_true.push(' ? ');
+			if_false.push(' : ');
 		}
 	},
 

package/src/index.js

@@ -79,30 +79,21 @@ export function print(node, opts = {}) {
 			return;
 		}
 
-		switch (command.type) {
-			case 'Chunk':
-				const loc = command.loc;
-
-				if (loc) {
-					current_line.push([
-						current_column,
-						0, // source index is always zero
-						loc.start.line - 1,
-						loc.start.column
-					]);
-				}
-
-				append(command.content);
-
-				if (loc) {
-					current_line.push([
-						current_column,
-						0, // source index is always zero
-						loc.end.line - 1,
-						loc.end.column
-					]);
-				}
+		if (Array.isArray(command)) {
+			for (let i = 0; i < command.length; i += 1) {
+				run(command[i]);
+			}
+			return;
+		}
 
+		switch (command.type) {
+			case 'Location':
+				current_line.push([
+					current_column,
+					0, // source index is always zero
+					command.line - 1,
+					command.column
+				]);
 				break;
 
 			case 'Newline':
@@ -117,13 +108,6 @@ export function print(node, opts = {}) {
 				newline = newline.slice(0, -indent.length);
 				break;
 
-			case 'Sequence':
-				for (let i = 0; i < command.children.length; i += 1) {
-					run(command.children[i]);
-				}
-
-				break;
-
 			case 'Comment':
 				if (command.comment.type === 'Line') {
 					append(`//${command.comment.value}`);

package/src/types.d.ts

@@ -38,13 +38,10 @@ export interface State {
 	quote: "'" | '"';
 }
 
-export interface Chunk {
-	type: 'Chunk';
-	content: string;
-	loc: null | {
-		start: { line: number; column: number };
-		end: { line: number; column: number };
-	};
+export interface Location {
+	type: 'Location';
+	line: number;
+	column: number;
 }
 
 export interface Newline {
@@ -64,17 +61,12 @@ export interface IndentChange {
 	offset: number;
 }
 
-export interface Sequence {
-	type: 'Sequence';
-	children: Command[];
-}
-
 export interface CommentChunk {
 	type: 'Comment';
 	comment: TSESTree.Comment;
 }
 
-export type Command = string | Chunk | Newline | Indent | Dedent | Sequence | CommentChunk;
+export type Command = string | Location | Newline | Indent | Dedent | CommentChunk | Command[];
 
 export interface PrintOptions {
 	sourceMapSource?: string;

package/types/index.d.ts.map

@@ -13,6 +13,6 @@
 		null,
 		null
 	],
-	"mappings": ";kBA8EiBA,YAAYA;;;;;;;;;;iBCvDbC,KAAKA",
+	"mappings": ";kBAsEiBA,YAAYA;;;;;;;;;;iBC/CbC,KAAKA",
 	"ignoreList": []
 }