not overflow when use formatter

Author: hyperz111

packages/prompts/src/note.ts

@@ -19,17 +19,22 @@ export interface NoteOptions extends CommonOptions {
 
 const defaultNoteFormatter = (line: string): string => color.dim(line);
 
+const wrapWithFormat = (message: string, width: number, format: NoteOptions["format"]): string => {
+  const wrapMsg = wrapAnsi(message, width).split("\n");
+  const maxWidthNormal = wrapMsg.reduce((sum, ln) => Math.max(strip(ln).length, sum), 0);
+  const maxWidthFormat = wrapMsg.map(format).reduce((sum, ln) => Math.max(strip(ln).length, sum), 0);
+  const wrapWidth = maxWidthNormal - (maxWidthFormat - maxWidthNormal);
+  return wrapAnsi(message, wrapWidth);
+}
+
 export const note = (message = '', title = '', opts?: NoteOptions) => {
   const output: Writable = opts?.output ?? process.stdout;
   const format = opts?.format ?? defaultNoteFormatter;
-  const wrapMsg = wrapAnsi(message, output.columns - 6);
+  const wrapMsg = wrapWithFormat(message, output.columns - 6, format);
   const lines = ['', ...wrapMsg.split('\n').map(format), ''];
   const titleLen = strip(title).length;
   const len = Math.max(
-    lines.reduce((sum, ln) => {
-      const line = strip(ln); 
-      return line.length > sum ? line.length : sum;
-    }, 0),
+    lines.reduce((sum, ln) => Math.max(strip(ln).length, sum), 0),
     titleLen
   );
   const header = `${color.green(S_STEP_SUBMIT)}  ${color.reset(title)} ${color.gray(S_BAR_H.repeat(len - titleLen + 1) + S_CORNER_TOP_RIGHT)}`;

packages/prompts/test/__snapshots__/note.test.ts.snap

@@ -31,6 +31,38 @@ exports[`note (isCI = false) > don't overflow 1`] = `
 ]
 `;
 
+exports[`note (isCI = false) > don't overflow with formatter 1`] = `
+[
+  "│
+◇  title ─────────────────────────────────────────────────────────────────╮
+│                                                                         │
+│  * Eiusmod sint officia proident qui ex tempor laborum ut. Voluptate *  │
+│  * dolore excepteur sunt adipisicing laborum non ea elit. *             │
+│  * Consectetur exercitation ad laboris consectetur nisi minim labore *  │
+│  * culpa enim. Eiusmod duis minim velit mollit nisi. Laboris magna *    │
+│  * qui fugiat sit minim. Aliqua id aute sunt proident do tempor *       │
+│  * mollit nulla veniam exercitation est ipsum occaecat nostrud. *       │
+│  * Exercitation ullamco proident ex ut et in ut. *                      │
+│  * Dolore magna proident id in magna quis. Sunt et laboris Lorem *      │
+│  * adipisicing nisi id proident qui enim laboris proident do. Fugiat *  │
+│  * eu commodo exercitation ipsum in aliquip ipsum magna consequat *     │
+│  * laborum. Id aute enim enim officia in do aliqua voluptate aute *     │
+│  * nulla proident. Mollit pariatur exercitation sit mollit in *         │
+│  * exercitation culpa deserunt sunt laboris aliquip cillum mollit *     │
+│  * ad. Aliquip exercitation adipisicing qui laboris ea do occaecat *    │
+│  * elit magna commodo sint adipisicing. *                               │
+│  * Fugiat veniam non nostrud reprehenderit tempor quis officia ad *     │
+│  * reprehenderit. Veniam ut est irure est. Laboris reprehenderit *      │
+│  * dolor dolor id laborum nulla velit et in sunt aliquip. Culpa *       │
+│  * voluptate dolore enim ut incididunt dolor sint cupidatat enim. *     │
+│  * Irure sint consectetur minim velit aliquip consequat enim ex in *    │
+│  * sunt. *                                                              │
+│                                                                         │
+├─────────────────────────────────────────────────────────────────────────╯
+",
+]
+`;
+
 exports[`note (isCI = false) > formatter which adds colors works 1`] = `
 [
   "│
@@ -48,13 +80,16 @@ exports[`note (isCI = false) > formatter which adds colors works 1`] = `
 exports[`note (isCI = false) > formatter which adds length works 1`] = `
 [
   "│
-◇  title ──────╮
-│              │
-│  * line 0 *  │
-│  * line 1 *  │
-│  * line 2 *  │
-│              │
-├──────────────╯
+◇  title ────╮
+│            │
+│  * line *  │
+│  * 0 *     │
+│  * line *  │
+│  * 1 *     │
+│  * line *  │
+│  * 2 *     │
+│            │
+├────────────╯
 ",
 ]
 `;
@@ -115,6 +150,38 @@ exports[`note (isCI = true) > don't overflow 1`] = `
 ]
 `;
 
+exports[`note (isCI = true) > don't overflow with formatter 1`] = `
+[
+  "│
+◇  title ─────────────────────────────────────────────────────────────────╮
+│                                                                         │
+│  * Eiusmod sint officia proident qui ex tempor laborum ut. Voluptate *  │
+│  * dolore excepteur sunt adipisicing laborum non ea elit. *             │
+│  * Consectetur exercitation ad laboris consectetur nisi minim labore *  │
+│  * culpa enim. Eiusmod duis minim velit mollit nisi. Laboris magna *    │
+│  * qui fugiat sit minim. Aliqua id aute sunt proident do tempor *       │
+│  * mollit nulla veniam exercitation est ipsum occaecat nostrud. *       │
+│  * Exercitation ullamco proident ex ut et in ut. *                      │
+│  * Dolore magna proident id in magna quis. Sunt et laboris Lorem *      │
+│  * adipisicing nisi id proident qui enim laboris proident do. Fugiat *  │
+│  * eu commodo exercitation ipsum in aliquip ipsum magna consequat *     │
+│  * laborum. Id aute enim enim officia in do aliqua voluptate aute *     │
+│  * nulla proident. Mollit pariatur exercitation sit mollit in *         │
+│  * exercitation culpa deserunt sunt laboris aliquip cillum mollit *     │
+│  * ad. Aliquip exercitation adipisicing qui laboris ea do occaecat *    │
+│  * elit magna commodo sint adipisicing. *                               │
+│  * Fugiat veniam non nostrud reprehenderit tempor quis officia ad *     │
+│  * reprehenderit. Veniam ut est irure est. Laboris reprehenderit *      │
+│  * dolor dolor id laborum nulla velit et in sunt aliquip. Culpa *       │
+│  * voluptate dolore enim ut incididunt dolor sint cupidatat enim. *     │
+│  * Irure sint consectetur minim velit aliquip consequat enim ex in *    │
+│  * sunt. *                                                              │
+│                                                                         │
+├─────────────────────────────────────────────────────────────────────────╯
+",
+]
+`;
+
 exports[`note (isCI = true) > formatter which adds colors works 1`] = `
 [
   "│
@@ -132,13 +199,16 @@ exports[`note (isCI = true) > formatter which adds colors works 1`] = `
 exports[`note (isCI = true) > formatter which adds length works 1`] = `
 [
   "│
-◇  title ──────╮
-│              │
-│  * line 0 *  │
-│  * line 1 *  │
-│  * line 2 *  │
-│              │
-├──────────────╯
+◇  title ────╮
+│            │
+│  * line *  │
+│  * 0 *     │
+│  * line *  │
+│  * 1 *     │
+│  * line *  │
+│  * 2 *     │
+│            │
+├────────────╯
 ",
 ]
 `;

packages/prompts/test/note.test.ts

@@ -77,4 +77,19 @@ describe.each(['true', 'false'])('note (isCI = %s)', (isCI) => {
 
 		expect(output.buffer).toMatchSnapshot();
 	});
+
+	test('don\'t overflow with formatter', () => {
+	  const lorem = [
+	    "Eiusmod sint officia proident qui ex tempor laborum ut. Voluptate dolore excepteur sunt adipisicing laborum non ea elit. Consectetur exercitation ad laboris consectetur nisi minim labore culpa enim. Eiusmod duis minim velit mollit nisi. Laboris magna qui fugiat sit minim. Aliqua id aute sunt proident do tempor mollit nulla veniam exercitation est ipsum occaecat nostrud. Exercitation ullamco proident ex ut et in ut.",
+	    "Dolore magna proident id in magna quis. Sunt et laboris Lorem adipisicing nisi id proident qui enim laboris proident do. Fugiat eu commodo exercitation ipsum in aliquip ipsum magna consequat laborum. Id aute enim enim officia in do aliqua voluptate aute nulla proident. Mollit pariatur exercitation sit mollit in exercitation culpa deserunt sunt laboris aliquip cillum mollit ad. Aliquip exercitation adipisicing qui laboris ea do occaecat elit magna commodo sint adipisicing.",
+	    "Fugiat veniam non nostrud reprehenderit tempor quis officia ad reprehenderit. Veniam ut est irure est. Laboris reprehenderit dolor dolor id laborum nulla velit et in sunt aliquip. Culpa voluptate dolore enim ut incididunt dolor sint cupidatat enim. Irure sint consectetur minim velit aliquip consequat enim ex in sunt.",
+	  ];
+		prompts.note(lorem.join("\n"), 'title', {
+		  format: (line) => colors.red(`* ${colors.cyan(line)} *`),
+			input,
+			output: Object.assign(output, { columns: 75 }),
+		});
+
+		expect(output.buffer).toMatchSnapshot();
+	});
 });