From 3f788d4fb678a1c31fe8cd277b1411083038402e Mon Sep 17 00:00:00 2001
From: wonderful-panda <iwata0303@gmail.com>
Date: Thu, 18 Jan 2018 11:20:55 +0900
Subject: [PATCH] fix: *.vue with `lang='tsx'` is not formatted (#3763)

---
 src/language-vue/embed.js                     |  2 +-
 .../__snapshots__/jsfmt.spec.js.snap          | 53 +++++++++++++++++++
 tests/multiparser_vue/lang-ts.vue             | 10 ++++
 tests/multiparser_vue/lang-tsx.vue            |  7 +++
 4 files changed, 71 insertions(+), 1 deletion(-)
 create mode 100644 tests/multiparser_vue/lang-ts.vue
 create mode 100644 tests/multiparser_vue/lang-tsx.vue

diff --git a/src/language-vue/embed.js b/src/language-vue/embed.js
index f1c48fa2..2f73eb75 100644
--- a/src/language-vue/embed.js
+++ b/src/language-vue/embed.js
@@ -28,7 +28,7 @@ function embed(path, print, textToDoc, options) {
     const langAttr = node.attrs.find(attr => attr.name === "lang");
     if (!langAttr) {
       parser = "babylon";
-    } else if (langAttr.value === "ts") {
+    } else if (langAttr.value === "ts" || langAttr.value === "tsx") {
       parser = "typescript";
     }
   }
diff --git a/tests/multiparser_vue/__snapshots__/jsfmt.spec.js.snap b/tests/multiparser_vue/__snapshots__/jsfmt.spec.js.snap
index 8a428a27..bdb5d7c0 100644
--- a/tests/multiparser_vue/__snapshots__/jsfmt.spec.js.snap
+++ b/tests/multiparser_vue/__snapshots__/jsfmt.spec.js.snap
@@ -1,5 +1,58 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
+exports[`lang-ts.vue 1`] = `
+<template>
+  <div>{{foo}}</div>
+</template>
+
+<script lang="ts">
+export default {
+  computed: {  foo( ): string {  return "foo";  },  },
+}
+</script>
+
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+<template>
+  <div>{{foo}}</div>
+</template>
+
+<script lang="ts">
+export default {
+  computed: {
+    foo(): string {
+      return "foo";
+    }
+  }
+};
+</script>
+
+`;
+
+exports[`lang-tsx.vue 1`] = `
+<script lang="tsx">
+import {VNode} from "vue"
+export default {
+  computed: {  foo( ):string { return "foo" }, },
+  render(h):VNode {  return <div>{ this.foo }</div> },
+}
+</script>
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+<script lang="tsx">
+import { VNode } from "vue";
+export default {
+  computed: {
+    foo(): string {
+      return "foo";
+    }
+  },
+  render(h): VNode {
+    return <div>{this.foo}</div>;
+  }
+};
+</script>
+
+`;
+
 exports[`template-bind.vue 1`] = `
 <template>
     <div v-bind:id=" 'list-'   +  id "></div>
diff --git a/tests/multiparser_vue/lang-ts.vue b/tests/multiparser_vue/lang-ts.vue
new file mode 100644
index 00000000..c9db0056
--- /dev/null
+++ b/tests/multiparser_vue/lang-ts.vue
@@ -0,0 +1,10 @@
+<template>
+  <div>{{foo}}</div>
+</template>
+
+<script lang="ts">
+export default {
+  computed: {  foo( ): string {  return "foo";  },  },
+}
+</script>
+
diff --git a/tests/multiparser_vue/lang-tsx.vue b/tests/multiparser_vue/lang-tsx.vue
new file mode 100644
index 00000000..665cd968
--- /dev/null
+++ b/tests/multiparser_vue/lang-tsx.vue
@@ -0,0 +1,7 @@
+<script lang="tsx">
+import {VNode} from "vue"
+export default {
+  computed: {  foo( ):string { return "foo" }, },
+  render(h):VNode {  return <div>{ this.foo }</div> },
+}
+</script>