##// END OF EJS Templates
Strip non color related ANSI escape sequences from notebook output....
Jack Feser -
Show More
@@ -1,288 +1,294
1 //----------------------------------------------------------------------------
1 //----------------------------------------------------------------------------
2 // Copyright (C) 2008-2012 The IPython Development Team
2 // Copyright (C) 2008-2012 The IPython Development Team
3 //
3 //
4 // Distributed under the terms of the BSD License. The full license is in
4 // Distributed under the terms of the BSD License. The full license is in
5 // the file COPYING, distributed as part of this software.
5 // the file COPYING, distributed as part of this software.
6 //----------------------------------------------------------------------------
6 //----------------------------------------------------------------------------
7
7
8 //============================================================================
8 //============================================================================
9 // Utilities
9 // Utilities
10 //============================================================================
10 //============================================================================
11
11
12 IPython.namespace('IPython.utils');
12 IPython.namespace('IPython.utils');
13
13
14 IPython.utils = (function (IPython) {
14 IPython.utils = (function (IPython) {
15
15
16 //============================================================================
16 //============================================================================
17 // Cross-browser RegEx Split
17 // Cross-browser RegEx Split
18 //============================================================================
18 //============================================================================
19
19
20 // This code has been MODIFIED from the code licensed below to not replace the
20 // This code has been MODIFIED from the code licensed below to not replace the
21 // default browser split. The license is reproduced here.
21 // default browser split. The license is reproduced here.
22
22
23 // see http://blog.stevenlevithan.com/archives/cross-browser-split for more info:
23 // see http://blog.stevenlevithan.com/archives/cross-browser-split for more info:
24 /*!
24 /*!
25 * Cross-Browser Split 1.1.1
25 * Cross-Browser Split 1.1.1
26 * Copyright 2007-2012 Steven Levithan <stevenlevithan.com>
26 * Copyright 2007-2012 Steven Levithan <stevenlevithan.com>
27 * Available under the MIT License
27 * Available under the MIT License
28 * ECMAScript compliant, uniform cross-browser split method
28 * ECMAScript compliant, uniform cross-browser split method
29 */
29 */
30
30
31 /**
31 /**
32 * Splits a string into an array of strings using a regex or string
32 * Splits a string into an array of strings using a regex or string
33 * separator. Matches of the separator are not included in the result array.
33 * separator. Matches of the separator are not included in the result array.
34 * However, if `separator` is a regex that contains capturing groups,
34 * However, if `separator` is a regex that contains capturing groups,
35 * backreferences are spliced into the result each time `separator` is
35 * backreferences are spliced into the result each time `separator` is
36 * matched. Fixes browser bugs compared to the native
36 * matched. Fixes browser bugs compared to the native
37 * `String.prototype.split` and can be used reliably cross-browser.
37 * `String.prototype.split` and can be used reliably cross-browser.
38 * @param {String} str String to split.
38 * @param {String} str String to split.
39 * @param {RegExp|String} separator Regex or string to use for separating
39 * @param {RegExp|String} separator Regex or string to use for separating
40 * the string.
40 * the string.
41 * @param {Number} [limit] Maximum number of items to include in the result
41 * @param {Number} [limit] Maximum number of items to include in the result
42 * array.
42 * array.
43 * @returns {Array} Array of substrings.
43 * @returns {Array} Array of substrings.
44 * @example
44 * @example
45 *
45 *
46 * // Basic use
46 * // Basic use
47 * regex_split('a b c d', ' ');
47 * regex_split('a b c d', ' ');
48 * // -> ['a', 'b', 'c', 'd']
48 * // -> ['a', 'b', 'c', 'd']
49 *
49 *
50 * // With limit
50 * // With limit
51 * regex_split('a b c d', ' ', 2);
51 * regex_split('a b c d', ' ', 2);
52 * // -> ['a', 'b']
52 * // -> ['a', 'b']
53 *
53 *
54 * // Backreferences in result array
54 * // Backreferences in result array
55 * regex_split('..word1 word2..', /([a-z]+)(\d+)/i);
55 * regex_split('..word1 word2..', /([a-z]+)(\d+)/i);
56 * // -> ['..', 'word', '1', ' ', 'word', '2', '..']
56 * // -> ['..', 'word', '1', ' ', 'word', '2', '..']
57 */
57 */
58 var regex_split = function (str, separator, limit) {
58 var regex_split = function (str, separator, limit) {
59 // If `separator` is not a regex, use `split`
59 // If `separator` is not a regex, use `split`
60 if (Object.prototype.toString.call(separator) !== "[object RegExp]") {
60 if (Object.prototype.toString.call(separator) !== "[object RegExp]") {
61 return split.call(str, separator, limit);
61 return split.call(str, separator, limit);
62 }
62 }
63 var output = [],
63 var output = [],
64 flags = (separator.ignoreCase ? "i" : "") +
64 flags = (separator.ignoreCase ? "i" : "") +
65 (separator.multiline ? "m" : "") +
65 (separator.multiline ? "m" : "") +
66 (separator.extended ? "x" : "") + // Proposed for ES6
66 (separator.extended ? "x" : "") + // Proposed for ES6
67 (separator.sticky ? "y" : ""), // Firefox 3+
67 (separator.sticky ? "y" : ""), // Firefox 3+
68 lastLastIndex = 0,
68 lastLastIndex = 0,
69 // Make `global` and avoid `lastIndex` issues by working with a copy
69 // Make `global` and avoid `lastIndex` issues by working with a copy
70 separator = new RegExp(separator.source, flags + "g"),
70 separator = new RegExp(separator.source, flags + "g"),
71 separator2, match, lastIndex, lastLength;
71 separator2, match, lastIndex, lastLength;
72 str += ""; // Type-convert
72 str += ""; // Type-convert
73
73
74 compliantExecNpcg = typeof(/()??/.exec("")[1]) === "undefined"
74 compliantExecNpcg = typeof(/()??/.exec("")[1]) === "undefined"
75 if (!compliantExecNpcg) {
75 if (!compliantExecNpcg) {
76 // Doesn't need flags gy, but they don't hurt
76 // Doesn't need flags gy, but they don't hurt
77 separator2 = new RegExp("^" + separator.source + "$(?!\\s)", flags);
77 separator2 = new RegExp("^" + separator.source + "$(?!\\s)", flags);
78 }
78 }
79 /* Values for `limit`, per the spec:
79 /* Values for `limit`, per the spec:
80 * If undefined: 4294967295 // Math.pow(2, 32) - 1
80 * If undefined: 4294967295 // Math.pow(2, 32) - 1
81 * If 0, Infinity, or NaN: 0
81 * If 0, Infinity, or NaN: 0
82 * If positive number: limit = Math.floor(limit); if (limit > 4294967295) limit -= 4294967296;
82 * If positive number: limit = Math.floor(limit); if (limit > 4294967295) limit -= 4294967296;
83 * If negative number: 4294967296 - Math.floor(Math.abs(limit))
83 * If negative number: 4294967296 - Math.floor(Math.abs(limit))
84 * If other: Type-convert, then use the above rules
84 * If other: Type-convert, then use the above rules
85 */
85 */
86 limit = typeof(limit) === "undefined" ?
86 limit = typeof(limit) === "undefined" ?
87 -1 >>> 0 : // Math.pow(2, 32) - 1
87 -1 >>> 0 : // Math.pow(2, 32) - 1
88 limit >>> 0; // ToUint32(limit)
88 limit >>> 0; // ToUint32(limit)
89 while (match = separator.exec(str)) {
89 while (match = separator.exec(str)) {
90 // `separator.lastIndex` is not reliable cross-browser
90 // `separator.lastIndex` is not reliable cross-browser
91 lastIndex = match.index + match[0].length;
91 lastIndex = match.index + match[0].length;
92 if (lastIndex > lastLastIndex) {
92 if (lastIndex > lastLastIndex) {
93 output.push(str.slice(lastLastIndex, match.index));
93 output.push(str.slice(lastLastIndex, match.index));
94 // Fix browsers whose `exec` methods don't consistently return `undefined` for
94 // Fix browsers whose `exec` methods don't consistently return `undefined` for
95 // nonparticipating capturing groups
95 // nonparticipating capturing groups
96 if (!compliantExecNpcg && match.length > 1) {
96 if (!compliantExecNpcg && match.length > 1) {
97 match[0].replace(separator2, function () {
97 match[0].replace(separator2, function () {
98 for (var i = 1; i < arguments.length - 2; i++) {
98 for (var i = 1; i < arguments.length - 2; i++) {
99 if (typeof(arguments[i]) === "undefined") {
99 if (typeof(arguments[i]) === "undefined") {
100 match[i] = undefined;
100 match[i] = undefined;
101 }
101 }
102 }
102 }
103 });
103 });
104 }
104 }
105 if (match.length > 1 && match.index < str.length) {
105 if (match.length > 1 && match.index < str.length) {
106 Array.prototype.push.apply(output, match.slice(1));
106 Array.prototype.push.apply(output, match.slice(1));
107 }
107 }
108 lastLength = match[0].length;
108 lastLength = match[0].length;
109 lastLastIndex = lastIndex;
109 lastLastIndex = lastIndex;
110 if (output.length >= limit) {
110 if (output.length >= limit) {
111 break;
111 break;
112 }
112 }
113 }
113 }
114 if (separator.lastIndex === match.index) {
114 if (separator.lastIndex === match.index) {
115 separator.lastIndex++; // Avoid an infinite loop
115 separator.lastIndex++; // Avoid an infinite loop
116 }
116 }
117 }
117 }
118 if (lastLastIndex === str.length) {
118 if (lastLastIndex === str.length) {
119 if (lastLength || !separator.test("")) {
119 if (lastLength || !separator.test("")) {
120 output.push("");
120 output.push("");
121 }
121 }
122 } else {
122 } else {
123 output.push(str.slice(lastLastIndex));
123 output.push(str.slice(lastLastIndex));
124 }
124 }
125 return output.length > limit ? output.slice(0, limit) : output;
125 return output.length > limit ? output.slice(0, limit) : output;
126 };
126 };
127
127
128 //============================================================================
128 //============================================================================
129 // End contributed Cross-browser RegEx Split
129 // End contributed Cross-browser RegEx Split
130 //============================================================================
130 //============================================================================
131
131
132
132
133 var uuid = function () {
133 var uuid = function () {
134 // http://www.ietf.org/rfc/rfc4122.txt
134 // http://www.ietf.org/rfc/rfc4122.txt
135 var s = [];
135 var s = [];
136 var hexDigits = "0123456789ABCDEF";
136 var hexDigits = "0123456789ABCDEF";
137 for (var i = 0; i < 32; i++) {
137 for (var i = 0; i < 32; i++) {
138 s[i] = hexDigits.substr(Math.floor(Math.random() * 0x10), 1);
138 s[i] = hexDigits.substr(Math.floor(Math.random() * 0x10), 1);
139 }
139 }
140 s[12] = "4"; // bits 12-15 of the time_hi_and_version field to 0010
140 s[12] = "4"; // bits 12-15 of the time_hi_and_version field to 0010
141 s[16] = hexDigits.substr((s[16] & 0x3) | 0x8, 1); // bits 6-7 of the clock_seq_hi_and_reserved to 01
141 s[16] = hexDigits.substr((s[16] & 0x3) | 0x8, 1); // bits 6-7 of the clock_seq_hi_and_reserved to 01
142
142
143 var uuid = s.join("");
143 var uuid = s.join("");
144 return uuid;
144 return uuid;
145 };
145 };
146
146
147
147
148 //Fix raw text to parse correctly in crazy XML
148 //Fix raw text to parse correctly in crazy XML
149 function xmlencode(string) {
149 function xmlencode(string) {
150 return string.replace(/\&/g,'&'+'amp;')
150 return string.replace(/\&/g,'&'+'amp;')
151 .replace(/</g,'&'+'lt;')
151 .replace(/</g,'&'+'lt;')
152 .replace(/>/g,'&'+'gt;')
152 .replace(/>/g,'&'+'gt;')
153 .replace(/\'/g,'&'+'apos;')
153 .replace(/\'/g,'&'+'apos;')
154 .replace(/\"/g,'&'+'quot;')
154 .replace(/\"/g,'&'+'quot;')
155 .replace(/`/g,'&'+'#96;');
155 .replace(/`/g,'&'+'#96;');
156 }
156 }
157
157
158
158
159 //Map from terminal commands to CSS classes
159 //Map from terminal commands to CSS classes
160 ansi_colormap = {
160 ansi_colormap = {
161 "30":"ansiblack", "31":"ansired",
161 "30":"ansiblack", "31":"ansired",
162 "32":"ansigreen", "33":"ansiyellow",
162 "32":"ansigreen", "33":"ansiyellow",
163 "34":"ansiblue", "35":"ansipurple","36":"ansicyan",
163 "34":"ansiblue", "35":"ansipurple","36":"ansicyan",
164 "37":"ansigrey", "01":"ansibold"
164 "37":"ansigrey", "01":"ansibold"
165 };
165 };
166
166
167 // Transform ANSI color escape codes into HTML <span> tags with css
167 // Transform ANSI color escape codes into HTML <span> tags with css
168 // classes listed in the above ansi_colormap object. The actual color used
168 // classes listed in the above ansi_colormap object. The actual color used
169 // are set in the css file.
169 // are set in the css file.
170 function fixConsole(txt) {
170 function fixConsole(txt) {
171 txt = xmlencode(txt);
171 txt = xmlencode(txt);
172 var re = /\033\[([\dA-Fa-f;]*?)m/;
172 var re = /\033\[([\dA-Fa-f;]*?)m/;
173 var opened = false;
173 var opened = false;
174 var cmds = [];
174 var cmds = [];
175 var opener = "";
175 var opener = "";
176 var closer = "";
176 var closer = "";
177
178 // Strip all ANSI codes that are not color related. Matches
179 // all ANSI codes that do not end with "m".
180 var ignored_re = /(?=(\033\[[\d;=]*[a-ln-zA-Z]{1}))\1(?!m)/g;
181 txt = txt.replace(ignored_re, "");
182
177 while (re.test(txt)) {
183 while (re.test(txt)) {
178 var cmds = txt.match(re)[1].split(";");
184 var cmds = txt.match(re)[1].split(";");
179 closer = opened?"</span>":"";
185 closer = opened?"</span>":"";
180 opened = cmds.length > 1 || cmds[0] != 0;
186 opened = cmds.length > 1 || cmds[0] != 0;
181 var rep = [];
187 var rep = [];
182 for (var i in cmds)
188 for (var i in cmds)
183 if (typeof(ansi_colormap[cmds[i]]) != "undefined")
189 if (typeof(ansi_colormap[cmds[i]]) != "undefined")
184 rep.push(ansi_colormap[cmds[i]]);
190 rep.push(ansi_colormap[cmds[i]]);
185 opener = rep.length > 0?"<span class=\""+rep.join(" ")+"\">":"";
191 opener = rep.length > 0?"<span class=\""+rep.join(" ")+"\">":"";
186 txt = txt.replace(re, closer + opener);
192 txt = txt.replace(re, closer + opener);
187 }
193 }
188 if (opened) txt += "</span>";
194 if (opened) txt += "</span>";
189 return txt;
195 return txt;
190 }
196 }
191
197
192 // Remove chunks that should be overridden by the effect of
198 // Remove chunks that should be overridden by the effect of
193 // carriage return characters
199 // carriage return characters
194 function fixCarriageReturn(txt) {
200 function fixCarriageReturn(txt) {
195 tmp = txt;
201 tmp = txt;
196 do {
202 do {
197 txt = tmp;
203 txt = tmp;
198 tmp = txt.replace(/\r+\n/gm, '\n'); // \r followed by \n --> newline
204 tmp = txt.replace(/\r+\n/gm, '\n'); // \r followed by \n --> newline
199 tmp = tmp.replace(/^.*\r+/gm, ''); // Other \r --> clear line
205 tmp = tmp.replace(/^.*\r+/gm, ''); // Other \r --> clear line
200 } while (tmp.length < txt.length);
206 } while (tmp.length < txt.length);
201 return txt;
207 return txt;
202 }
208 }
203
209
204 // Locate URLs in plain text and wrap them in spaces so that they can be
210 // Locate URLs in plain text and wrap them in spaces so that they can be
205 // better picked out by autoLinkUrls even after the text has been
211 // better picked out by autoLinkUrls even after the text has been
206 // converted to HTML
212 // converted to HTML
207 function wrapUrls(txt) {
213 function wrapUrls(txt) {
208 // Note this regexp is a modified version of one from
214 // Note this regexp is a modified version of one from
209 // Markdown.Converter For now it only supports http(s) and ftp URLs,
215 // Markdown.Converter For now it only supports http(s) and ftp URLs,
210 // but could easily support others (though file:// should maybe be
216 // but could easily support others (though file:// should maybe be
211 // avoided)
217 // avoided)
212 var url_re = /(^|\W)(https?|ftp)(:\/\/[-A-Z0-9+&@#\/%?=~_|\[\]\(\)!:,\.;]*[-A-Z0-9+&@#\/%=~_|\[\]])($|\W)/gi;
218 var url_re = /(^|\W)(https?|ftp)(:\/\/[-A-Z0-9+&@#\/%?=~_|\[\]\(\)!:,\.;]*[-A-Z0-9+&@#\/%=~_|\[\]])($|\W)/gi;
213 return txt.replace(url_re, "$1 $2$3 $4");
219 return txt.replace(url_re, "$1 $2$3 $4");
214 }
220 }
215
221
216 // Locate a URL with spaces around it and convert that to a anchor tag
222 // Locate a URL with spaces around it and convert that to a anchor tag
217 function autoLinkUrls(txt) {
223 function autoLinkUrls(txt) {
218 return txt.replace(/ ((https?|ftp):[^'">\s]+) /gi,
224 return txt.replace(/ ((https?|ftp):[^'">\s]+) /gi,
219 "<a target=\"_blank\" href=\"$1\">$1</a>");
225 "<a target=\"_blank\" href=\"$1\">$1</a>");
220 }
226 }
221
227
222 grow = function(element) {
228 grow = function(element) {
223 // Grow the cell by hand. This is used upon reloading from JSON, when the
229 // Grow the cell by hand. This is used upon reloading from JSON, when the
224 // autogrow handler is not called.
230 // autogrow handler is not called.
225 var dom = element.get(0);
231 var dom = element.get(0);
226 var lines_count = 0;
232 var lines_count = 0;
227 // modified split rule from
233 // modified split rule from
228 // http://stackoverflow.com/questions/2035910/how-to-get-the-number-of-lines-in-a-textarea/2036424#2036424
234 // http://stackoverflow.com/questions/2035910/how-to-get-the-number-of-lines-in-a-textarea/2036424#2036424
229 var lines = dom.value.split(/\r|\r\n|\n/);
235 var lines = dom.value.split(/\r|\r\n|\n/);
230 lines_count = lines.length;
236 lines_count = lines.length;
231 if (lines_count >= 1) {
237 if (lines_count >= 1) {
232 dom.rows = lines_count;
238 dom.rows = lines_count;
233 } else {
239 } else {
234 dom.rows = 1;
240 dom.rows = 1;
235 }
241 }
236 };
242 };
237
243
238 // some keycodes that seem to be platform/browser independant
244 // some keycodes that seem to be platform/browser independant
239 var keycodes ={
245 var keycodes ={
240 BACKSPACE: 8,
246 BACKSPACE: 8,
241 TAB : 9,
247 TAB : 9,
242 ENTER : 13,
248 ENTER : 13,
243 SHIFT : 16,
249 SHIFT : 16,
244 CTRL : 17,
250 CTRL : 17,
245 CONTROL : 17,
251 CONTROL : 17,
246 ALT : 18,
252 ALT : 18,
247 ESC : 27,
253 ESC : 27,
248 SPACE : 32,
254 SPACE : 32,
249 PGUP : 33,
255 PGUP : 33,
250 PGDOWN : 34,
256 PGDOWN : 34,
251 LEFT_ARROW: 37,
257 LEFT_ARROW: 37,
252 LEFTARROW: 37,
258 LEFTARROW: 37,
253 LEFT : 37,
259 LEFT : 37,
254 UP_ARROW : 38,
260 UP_ARROW : 38,
255 UPARROW : 38,
261 UPARROW : 38,
256 UP : 38,
262 UP : 38,
257 RIGHT_ARROW:39,
263 RIGHT_ARROW:39,
258 RIGHTARROW:39,
264 RIGHTARROW:39,
259 RIGHT : 39,
265 RIGHT : 39,
260 DOWN_ARROW: 40,
266 DOWN_ARROW: 40,
261 DOWNARROW: 40,
267 DOWNARROW: 40,
262 DOWN : 40,
268 DOWN : 40,
263 };
269 };
264
270
265
271
266 points_to_pixels = function (points) {
272 points_to_pixels = function (points) {
267 // A reasonably good way of converting between points and pixels.
273 // A reasonably good way of converting between points and pixels.
268 var test = $('<div style="display: none; width: 10000pt; padding:0; border:0;"></div>');
274 var test = $('<div style="display: none; width: 10000pt; padding:0; border:0;"></div>');
269 $(body).append(test);
275 $(body).append(test);
270 var pixel_per_point = test.width()/10000;
276 var pixel_per_point = test.width()/10000;
271 test.remove();
277 test.remove();
272 return Math.floor(points*pixel_per_point);
278 return Math.floor(points*pixel_per_point);
273 }
279 }
274
280
275
281
276 return {
282 return {
277 regex_split : regex_split,
283 regex_split : regex_split,
278 uuid : uuid,
284 uuid : uuid,
279 fixConsole : fixConsole,
285 fixConsole : fixConsole,
280 keycodes : keycodes,
286 keycodes : keycodes,
281 grow : grow,
287 grow : grow,
282 fixCarriageReturn : fixCarriageReturn,
288 fixCarriageReturn : fixCarriageReturn,
283 wrapUrls : wrapUrls,
289 wrapUrls : wrapUrls,
284 autoLinkUrls : autoLinkUrls,
290 autoLinkUrls : autoLinkUrls,
285 points_to_pixels : points_to_pixels
291 points_to_pixels : points_to_pixels
286 };
292 };
287
293
288 }(IPython));
294 }(IPython));
General Comments 0
You need to be logged in to leave comments. Login now