feat: add ability to specify which typescript to use (#49)

* feat: add ability to specify which typescript to use

* refactor: fix coverage

Author: ikatyang

src/__tests__/remap-snapshot.ts

@@ -1,5 +1,6 @@
 import * as fs from 'fs';
 import * as path from 'path';
+import * as ts from 'typescript';
 import { remap_snapshot } from '../remap-snapshot';
 
 const load_content = (relative_path: string) =>
@@ -11,7 +12,9 @@ const load_content = (relative_path: string) =>
 it('should remap correctly with snapshot-content string', () => {
   const source_content = load_content('example.ts');
   const snapshot_content = load_content('__snapshots__/example.ts.snap');
-  expect(remap_snapshot(snapshot_content, source_content)).toMatchSnapshot();
+  expect(
+    remap_snapshot(snapshot_content, source_content, undefined, ts),
+  ).toMatchSnapshot();
 });
 
 it('should remap correctl with snapshot-content object', () => {
@@ -22,5 +25,7 @@ it('should remap correctl with snapshot-content object', () => {
   const snapshot_content = {
     'Math.max(1, 2, 3) 1': '"number"',
   };
-  expect(remap_snapshot(snapshot_content, source_content)).toMatchSnapshot();
+  expect(
+    remap_snapshot(snapshot_content, source_content, undefined, ts),
+  ).toMatchSnapshot();
 });

src/__tests__/setup.ts

@@ -15,7 +15,7 @@ it('should setup correctly', () => {
 
 it('should setup correctly with specified tsconfig', () => {
   const raw_config: RawConfig = {
-    tsconfig: '<cwd>/fixtures/snapshots/tsconfig.json',
+    tsconfig: '<rootDir>/fixtures/snapshots/tsconfig.json',
   };
   expect(setup(filename, raw_config, targets)).toMatchSnapshot();
 });

src/__tests__/transform-actual.ts

@@ -12,10 +12,11 @@ const transform_fixture = (
   const source_filename = path.resolve(__dirname, relative_path);
   const source_text = fs.readFileSync(source_filename, 'utf8');
   const jest_config: JestConfig = {
+    rootDir: process.cwd(),
     globals: { _dts_jest_: { tsconfig } },
-    _dts_jest_debug_: debug,
+    _dts_jest_internal_test_: debug,
   };
-  return transform_actual(source_text, source_filename, jest_config);
+  return transform_actual(source_text, source_filename, jest_config as any);
 };
 
 it('should transform correctly', () => {

src/__tests__/transform.ts

@@ -1,11 +1,16 @@
 import * as fs from 'fs';
 import * as path from 'path';
+import { JestConfig } from '../definitions';
 import { transform } from '../transform';
 
 const transform_fixture = (relative_path: string) => {
   const source_filename = path.resolve(__dirname, relative_path);
   const source_text = fs.readFileSync(source_filename, 'utf8');
-  return transform(source_text, source_filename);
+  const jest_config: JestConfig = {
+    rootDir: process.cwd(),
+    globals: {},
+  };
+  return transform(source_text, source_filename, jest_config as any);
 };
 
 it('should transform correctly', () => {

src/definitions.ts

@@ -1,11 +1,13 @@
-import * as ts from 'typescript';
+import * as _ts from 'typescript';
 
 // tslint:disable-next-line:no-var-requires
 export const package_name = require('../package.json').name;
 
 export const config_namespace = '_dts_jest_';
 export const runtime_namespace = '_dts_jest_runtime_';
 
+export const env_root_dir = 'DTS_JEST_ROOT_DIR';
+
 export const trigger_regex = /^\s*\/\/\s*@dts-jest\b(:?\S*)\s*(.+)?\s*$/;
 export enum TriggerMatchIndex {
   Input,
@@ -78,11 +80,20 @@ export interface Expected extends Trigger {
 export interface Result extends Target, Snapshot {}
 
 export interface JestConfig {
+  rootDir: string;
   globals: { [K in typeof config_namespace]?: RawConfig };
-  _dts_jest_debug_?: boolean;
+  _dts_jest_internal_test_?: boolean;
 }
 
 export interface RawConfig {
-  tsconfig?: string | ts.CompilerOptions;
-  type_format?: ts.TypeFormatFlags;
+  tsconfig?: string | _ts.CompilerOptions;
+  type_format?: _ts.TypeFormatFlags;
+  typescript?: string;
+}
+
+export interface FormattedConfig {
+  tsconfig: _ts.CompilerOptions;
+  type_format: _ts.TypeFormatFlags;
+  typescript: typeof _ts;
+  typescript_path: string;
 }

src/remap-snapshot-cli.ts

@@ -1,5 +1,6 @@
 import * as fs from 'fs';
 import * as path from 'path';
+import * as ts from 'typescript';
 import { remap_snapshot } from './remap-snapshot';
 
 export = (args: string[]) => {
@@ -17,5 +18,7 @@ export = (args: string[]) => {
     ),
     'utf8',
   );
-  process.stdout.write(remap_snapshot(snapshot_content, source_content));
+  process.stdout.write(
+    remap_snapshot(snapshot_content, source_content, undefined, ts),
+  );
 };

src/remap-snapshot.ts

@@ -1,5 +1,5 @@
 import require_from_string = require('require-from-string');
-import * as ts from 'typescript';
+import * as _ts from 'typescript';
 import { Trigger } from './definitions';
 import { create_triggers } from './utils/create-triggers';
 import { default_to } from './utils/default-to';
@@ -8,7 +8,8 @@ import { get_formatted_description } from './utils/get-formatted-description';
 export const remap_snapshot = (
   snapshot_content: string | Record<string, string>,
   source_content: string,
-  snapshot_filename?: string,
+  snapshot_filename: string | undefined,
+  ts: typeof _ts,
 ) => {
   const snapshot_data =
     typeof snapshot_content === 'string'
@@ -24,7 +25,7 @@ export const remap_snapshot = (
     ts.ScriptTarget.Latest,
     false,
   );
-  const triggers = create_triggers(source_file);
+  const triggers = create_triggers(source_file, ts);
 
   const source_content_lines = source_content.split('\n');
 

src/setup.ts

@@ -1,24 +1,26 @@
-import * as ts from 'typescript';
 import { RawConfig, Result, Target } from './definitions';
 import { Runtime } from './runtime';
 import { create_snapshots } from './utils/create-snapshots';
 import { default_to } from './utils/default-to';
-import { get_tsconfig } from './utils/get-tsconfig';
+import { get_config } from './utils/get-config';
+import { get_root_dir } from './utils/root-dir';
 
 export const setup = (
   filename: string,
   raw_config: RawConfig,
   targets: Target[],
 ) => {
+  const { typescript: ts, tsconfig } = get_config(raw_config, get_root_dir());
+
   const lines = targets.map(target => target.line);
 
   const flag = default_to(
     raw_config.type_format,
     ts.TypeFormatFlags.NoTruncation,
   );
-  const program = ts.createProgram([filename], get_tsconfig(raw_config));
+  const program = ts.createProgram([filename], tsconfig);
 
-  const snapshots = create_snapshots(program, filename, lines, flag);
+  const snapshots = create_snapshots(program, filename, lines, flag, ts);
   const results = targets.reduce<{ [line: number]: Result }>(
     (current_results, target) => ({
       ...current_results,

src/transform-actual.ts

@@ -1,26 +1,29 @@
-import * as ts from 'typescript';
-import { AssertionFlag, JestConfig, RawConfig } from './definitions';
+import { config_namespace, AssertionFlag, JestConfig } from './definitions';
 import { create_expecteds } from './utils/create-expecteds';
 import { create_triggers } from './utils/create-triggers';
-import { default_to } from './utils/default-to';
+import { get_config } from './utils/get-config';
 import { get_formatted_description } from './utils/get-formatted-description';
-import { get_tsconfig } from './utils/get-tsconfig';
 import { rewrite_expecteds_method } from './utils/rewrite-expecteds-method';
 
-export const transform_actual = (
-  source_text: string,
-  source_filename: string,
+export const transform_actual: jest.Transformer['process'] = (
+  source_text,
+  source_filename,
   jest_config: JestConfig,
 ) => {
+  const { typescript: ts, tsconfig } = get_config(
+    jest_config.globals[config_namespace],
+    jest_config.rootDir,
+  );
+
   const source_file = ts.createSourceFile(
     source_filename,
     source_text,
     ts.ScriptTarget.Latest,
     false,
   );
-  const triggers = create_triggers(source_file);
+  const triggers = create_triggers(source_file, ts);
 
-  const expecteds = create_expecteds(triggers, source_file);
+  const expecteds = create_expecteds(triggers, source_file, ts);
   rewrite_expecteds_method(expecteds);
 
   let transformed = source_text;
@@ -44,13 +47,16 @@ export const transform_actual = (
     )}`;
   });
 
-  if (jest_config._dts_jest_debug_ === true) {
+  if (jest_config._dts_jest_internal_test_ === true) {
     return transformed;
   }
 
-  const tsconfig: ts.CompilerOptions = {
-    ...get_tsconfig(default_to<RawConfig>(jest_config.globals._dts_jest_, {})),
-    inlineSourceMap: true,
-  };
-  return ts.transpile(transformed, tsconfig, source_filename);
+  return ts.transpile(
+    transformed,
+    {
+      ...tsconfig,
+      inlineSourceMap: true,
+    },
+    source_filename,
+  );
 };

src/transform.ts

@@ -1,15 +1,23 @@
-import * as ts from 'typescript';
-import { Group } from './definitions';
+import { config_namespace, Group, JestConfig } from './definitions';
 import { create_group_expression } from './utils/create-group-expression';
 import { create_setup_expression } from './utils/create-setup-expression';
 import { create_test_expression } from './utils/create-test-expression';
 import { create_triggers } from './utils/create-triggers';
+import { get_config } from './utils/get-config';
+import { set_root_dir } from './utils/root-dir';
 
-export const transform = (
-  source_text: string,
-  source_filename: string,
-  _jest_config?: any,
+export const transform: jest.Transformer['process'] = (
+  source_text,
+  source_filename,
+  jest_config: JestConfig,
 ) => {
+  const { typescript: ts } = get_config(
+    jest_config.globals[config_namespace],
+    jest_config.rootDir,
+  );
+
+  set_root_dir(jest_config.rootDir);
+
   const source_file = ts.createSourceFile(
     source_filename,
     source_text,
@@ -18,7 +26,7 @@ export const transform = (
   );
 
   let last_group: Group | undefined;
-  const triggers = create_triggers(source_file);
+  const triggers = create_triggers(source_file, ts);
   return triggers
     .reduce<string[]>((transformed_line_texts, trigger, index) => {
       const is_diff_group = trigger.group !== last_group;