Merge pull request #5 from Rich-Harris/maranomynet/buble-feature/jsx-…

…pragma-comments

Maranomynet/buble feature/jsx pragma comments

package.json

@@ -63,6 +63,7 @@
     "acorn-jsx": "^3.0.1",
     "acorn-object-spread": "^1.0.0",
     "chalk": "^1.1.3",
+    "vlq": "0.2.1",
     "magic-string": "^0.14.0",
     "minimist": "^1.2.0",
     "os-homedir": "^1.0.1",

src/index.js

@@ -46,17 +46,25 @@ export function target ( target ) {
 
 export function transform ( source, options = {} ) {
 	let ast;
+	let jsx = null;
 
 	try {
 		ast = parse( source, {
 			ecmaVersion: 7,
 			preserveParens: true,
 			sourceType: 'module',
+			onComment: (block, text) => {
+				if ( !jsx ) {
+					let match = /@jsx\s+([^\s]+)/.exec( text );
+					if ( match ) jsx = match[1];
+				}
+			},
 			plugins: {
 				jsx: true,
 				objectSpread: true
 			}
 		});
+		options.jsx = jsx || options.jsx;
 	} catch ( err ) {
 		err.snippet = getSnippet( source, err.loc );
 		err.toString = () => `${err.name}: ${err.message}\n${err.snippet}`;

test/cli/supports-jsx-pragma-comment/command.sh

@@ -0,0 +1 @@
+buble input.js -o actual/output.js --jsx NotReact.createElement

test/cli/supports-jsx-pragma-comment/expected/output.js

@@ -0,0 +1,2 @@
+/* @jsx customPragma */
+var div = customPragma( 'div', null, "Hello" )

test/cli/supports-jsx-pragma-comment/input.js

@@ -0,0 +1,2 @@
+/* @jsx customPragma */
+var div = <div>Hello</div>

test/samples/jsx.js

@@ -224,6 +224,32 @@ module.exports = [
 		`
 	},
 
+	{
+		description: 'supports /* @jsx customPragma */ directives (#195)',
+		input: `
+			/* @jsx customPragma */
+			var div = <div>Hello</div>
+		`,
+		output: `
+			/* @jsx customPragma */
+			var div = customPragma( 'div', null, "Hello" )
+		`
+	},
+
+	{
+		description: 'ignores subsequent /* @jsx customPragma */ directives (#195)',
+		input: `
+			/* @jsx customPragma */
+			/* @jsx customPragmaWannabe */
+			var div = <div>Hello</div>
+		`,
+		output: `
+			/* @jsx customPragma */
+			/* @jsx customPragmaWannabe */
+			var div = customPragma( 'div', null, "Hello" )
+		`
+	},
+
 	{
 		description: 'handles dash-cased value-less props',
 
@@ -233,6 +259,5 @@ module.exports = [
 		output: `
 			React.createElement( Thing, { 'data-foo': true })
 		`
-	},
-
+	}
 ];