feat(relay): Add outputFileExtension option

packages/relay/README.md

@@ -63,3 +63,24 @@ const swcConfig = require("./.swcrc.js")
 ```
 
 > Note: We're using a `.swcrc.js` file extension up above and importing the config directly because Relay needs access to `__dirname`, which can't be derived from the default JSON parsed from `.swcrc`.
+
+#### Output import paths
+
+By default, `@swc/plugin-relay` will transpile import paths based on the `language` option.
+You can use `outputFileExtension` to change the file extension of the generated import paths.
+
+```js
+plugins: [
+    [
+        "@swc/plugin-relay",
+        {
+            rootDir: __dirname,
+            artifactDirectory: "src/__generated__",
+            language: "typescript",
+            eagerEsModules: true,
+            outputFileExtension: "js",
+        },
+    ],
+],
+```
+In this example typescript graphql files will output transpiled import path of `javascript` ending with `.js`.

packages/relay/src/lib.rs

@@ -6,7 +6,7 @@ use swc_core::{
     ecma::{ast::Program, visit::FoldWith},
     plugin::{plugin_transform, proxies::TransformPluginProgramMetadata},
 };
-use swc_relay::{relay, Config, RelayLanguageConfig};
+use swc_relay::{relay, Config, RelayLanguageConfig, OutputFileExtension};
 
 #[plugin_transform]
 fn relay_plugin_transform(program: Program, metadata: TransformPluginProgramMetadata) -> Program {
@@ -40,6 +40,9 @@ fn relay_plugin_transform(program: Program, metadata: TransformPluginProgramMeta
     let language = plugin_config["language"]
         .as_str()
         .map_or(RelayLanguageConfig::TypeScript, |v| v.try_into().unwrap());
+    let output_file_extension = plugin_config["outputFileExtension"]
+        .as_str()
+        .map_or(OutputFileExtension::Undefined, |v| v.try_into().unwrap());
     let eager_es_modules = plugin_config["eagerEsModules"]
         .as_bool()
         .unwrap_or_default();
@@ -48,6 +51,7 @@ fn relay_plugin_transform(program: Program, metadata: TransformPluginProgramMeta
         artifact_directory,
         language,
         eager_es_modules,
+        output_file_extension
     };
 
     let mut relay = relay(

packages/relay/transform/src/lib.rs

@@ -21,6 +21,14 @@ pub enum RelayLanguageConfig {
     Flow,
 }
 
+#[derive(Copy, Clone, Debug, Deserialize)]
+#[serde(rename_all = "lowercase")]
+pub enum OutputFileExtension {
+    TypeScript,
+    JavaScript,
+    Undefined
+}
+
 impl<'a> TryFrom<&'a str> for RelayLanguageConfig {
     type Error = String;
 
@@ -33,12 +41,31 @@ impl<'a> TryFrom<&'a str> for RelayLanguageConfig {
     }
 }
 
+
+impl<'a> TryFrom<&'a str> for OutputFileExtension {
+    type Error = String;
+
+    fn try_from(value: &'a str) -> Result<Self, Self::Error> {
+        match value {
+            "ts" => Ok(Self::TypeScript),
+            "js" => Ok(Self::JavaScript),
+            _ => Err(format!("Unexpected output file extension value '{}'", value)),
+        }
+    }
+}
+
 impl Default for RelayLanguageConfig {
     fn default() -> Self {
         Self::Flow
     }
 }
 
+impl Default for OutputFileExtension {
+    fn default() -> Self {
+        Self::Undefined
+    }
+}
+
 #[derive(Debug, Clone)]
 struct RelayImport {
     path: JsWord,
@@ -81,6 +108,8 @@ pub struct Config {
     pub language: RelayLanguageConfig,
     #[serde(default)]
     pub eager_es_modules: bool,
+    #[serde(default)]
+    pub output_file_extension: OutputFileExtension,
 }
 
 fn pull_first_operation_name_from_tpl(tpl: &TaggedTpl) -> Option<String> {
@@ -159,14 +188,14 @@ impl<'a> Relay<'a> {
         real_file_name: &Path,
         definition_name: &str,
     ) -> Result<PathBuf, BuildRequirePathError> {
-        let filename = match &self.config.language {
-            RelayLanguageConfig::Flow => format!("{}.graphql.js", definition_name),
-            RelayLanguageConfig::TypeScript => {
-                format!("{}.graphql.ts", definition_name)
-            }
-            RelayLanguageConfig::JavaScript => {
-                format!("{}.graphql.js", definition_name)
-            }
+        let filename = match &self.config.output_file_extension {
+            OutputFileExtension::JavaScript => format!("{}.graphql.js", definition_name),
+            OutputFileExtension::TypeScript => format!("{}.graphql.ts", definition_name),
+            OutputFileExtension::Undefined => match &self.config.language {
+                RelayLanguageConfig::Flow => format!("{}.graphql.js", definition_name),
+                RelayLanguageConfig::TypeScript => format!("{}.graphql.ts", definition_name),
+                RelayLanguageConfig::JavaScript => format!("{}.graphql.js", definition_name),
+            },
         };
 
         if let Some(artifact_directory) = &self.config.artifact_directory {

packages/relay/transform/tests/fixture.rs

@@ -2,7 +2,7 @@ use std::path::PathBuf;
 
 use swc_common::FileName;
 use swc_ecma_transforms_testing::test_fixture;
-use swc_relay::{relay, Config, RelayLanguageConfig};
+use swc_relay::{relay, Config, RelayLanguageConfig, OutputFileExtension};
 
 #[testing::fixture("tests/fixture/simple/**/input.js")]
 fn fixture(input: PathBuf) {
@@ -16,6 +16,7 @@ fn fixture(input: PathBuf) {
                     artifact_directory: None,
                     language: RelayLanguageConfig::TypeScript,
                     eager_es_modules: false,
+                    output_file_extension: OutputFileExtension::Undefined,
                 },
                 FileName::Real("file.js".parse().unwrap()),
                 Default::default(),
@@ -41,6 +42,59 @@ fn fixture_es_modules(input: PathBuf) {
                     artifact_directory: None,
                     language: RelayLanguageConfig::TypeScript,
                     eager_es_modules: true,
+                    output_file_extension: OutputFileExtension::Undefined,
+                },
+                FileName::Real("file.js".parse().unwrap()),
+                Default::default(),
+                None,
+                None,
+            )
+        },
+        &input,
+        &output,
+        Default::default(),
+    );
+}
+
+#[testing::fixture("tests/fixture/outputFileExtension/javascript/**/input.js")]
+fn fixture_output_file_extension_javascript(input: PathBuf) {
+    let output = input.parent().unwrap().join("output.js");
+
+    test_fixture(
+        Default::default(),
+        &|_| {
+            relay(
+                &Config {
+                    artifact_directory: None,
+                    language: RelayLanguageConfig::TypeScript,
+                    eager_es_modules: true,
+                    output_file_extension: OutputFileExtension::JavaScript,
+                },
+                FileName::Real("file.js".parse().unwrap()),
+                Default::default(),
+                None,
+                None,
+            )
+        },
+        &input,
+        &output,
+        Default::default(),
+    );
+}
+
+#[testing::fixture("tests/fixture/outputFileExtension/typescript/**/input.js")]
+fn fixture_output_file_extension_typescript(input: PathBuf) {
+    let output = input.parent().unwrap().join("output.js");
+
+    test_fixture(
+        Default::default(),
+        &|_| {
+            relay(
+                &Config {
+                    artifact_directory: None,
+                    language: RelayLanguageConfig::JavaScript,
+                    eager_es_modules: true,
+                    output_file_extension: OutputFileExtension::TypeScript,
                 },
                 FileName::Real("file.js".parse().unwrap()),
                 Default::default(),

packages/relay/transform/tests/fixture/outputFileExtension/javascript/input.js

@@ -0,0 +1,10 @@
+const myFragment = graphql`
+  fragment FooFragment on Bar {
+    id
+  }
+`
+useQuery(graphql`
+  query FooQuery {
+    id
+  }
+`)

packages/relay/transform/tests/fixture/outputFileExtension/javascript/output.js

@@ -0,0 +1,4 @@
+import __FooFragment from "__generated__/FooFragment.graphql.js";
+import __FooQuery from "__generated__/FooQuery.graphql.js";
+const myFragment = __FooFragment;
+useQuery(__FooQuery);

packages/relay/transform/tests/fixture/outputFileExtension/typescript/input.js

@@ -0,0 +1,10 @@
+const myFragment = graphql`
+  fragment FooFragment on Bar {
+    id
+  }
+`
+useQuery(graphql`
+  query FooQuery {
+    id
+  }
+`)

packages/relay/transform/tests/fixture/outputFileExtension/typescript/output.js

@@ -0,0 +1,4 @@
+import __FooFragment from "__generated__/FooFragment.graphql.ts";
+import __FooQuery from "__generated__/FooQuery.graphql.ts";
+const myFragment = __FooFragment;
+useQuery(__FooQuery);