import { assert, describe, it, test } from 'vitest';

import { transform } from '../index';

describe('simple', () => {
  const code = 'export class A<T> {}';

  it('matches output', () => {
    const ret = transform('test.ts', code, { sourcemap: true });
    assert.deepEqual(ret, {
      code: 'export class A {}\n',
      errors: [],
      map: {
        mappings: 'AAAA,OAAO,MAAM,EAAK,CAAE',
        names: [],
        sources: ['test.ts'],
        sourcesContent: ['export class A<T> {}'],
        version: 3,
      },
    });
  });

  it('uses the `lang` option', () => {
    const ret = transform('test.vue', code, { lang: 'ts' });
    assert.equal(ret.code, 'export class A {}\n');
  });

  it('uses the `declaration option`', () => {
    const ret = transform('test.ts', code, { typescript: { declaration: {} } });
    assert.equal(ret.declaration, 'export declare class A<T> {}\n');
  });
});

describe('transform', () => {
  it('should not transform by default', () => {
    const cases = [
      '() => {};',
      'a ** b;',
      'async function foo() {}',
      '({ ...x });',
      'try {} catch {}',
      'a ?? b;',
      'a ||= b;',
      'class foo {\n\tstatic {}\n}',
    ];
    for (const code of cases) {
      const ret = transform('test.ts', code);
      assert.equal(ret.code.trim(), code);
    }
  });
});

describe('target', () => {
  const data = [
    ['es2015', 'a ** b;\n'],
    ['es2016', 'async function foo() {}\n'],
    ['es2017', '({ ...x });\n'],
    ['es2017', 'try {} catch {}\n'],
    ['es2019', 'a?.b;\n'],
    ['es2019', 'a ?? b;\n'],
    ['es2021', 'class foo {\n\tstatic {}\n}\n'],
  ];

  test.each(data)('transform %s', (target, code) => {
    // Also test array syntax.
    const ret = transform('test.js', code, { target: [target] });
    assert(ret.errors.length == 0);
    assert(ret.code);
    assert.notEqual(ret.code, code);
  });

  test.each(data)('no transform esnext: %s', (_target, code) => {
    const ret = transform('test.js', code, { target: 'esnext' });
    assert(ret.errors.length == 0);
    assert(ret.code);
    assert.equal(ret.code, code);
  });

  it('should turn off class propertiers because plugin is not ready', () => {
    const code = 'class Foo {\n\t#a;\n}\n';
    const ret = transform('test.js', code, { target: 'es2015' });
    assert(ret.errors.length == 0);
    assert(ret.code);
    assert.equal(ret.code, code);
  });
});

describe('modules', () => {
  it('should transform export = and import ', () => {
    const code = `
export = function foo (): void {}
import bar = require('bar')
`;
    const ret = transform('test.ts', code, {
      typescript: {
        declaration: {},
      },
    });
    assert.deepEqual(ret, {
      code: 'module.exports = function foo() {};\nconst bar = require("bar");\n',
      declaration: 'declare const _default: () => void;\nexport = _default;\n',
      errors: [],
    });
  });
});

describe('react refresh plugin', () => {
  const code = `import { useState } from "react";
  export const App = () => {
    const [count, setCount] = useState(0);
    return <button onClick={() => setCount(count + 1)}>count is {count}</button>;
  };`;

  it('matches output', () => {
    const ret = transform('test.tsx', code, { jsx: { refresh: {} } });
    assert.equal(
      ret.code,
      `import { useState } from "react";
import { jsxs as _jsxs } from "react/jsx-runtime";
var _s = $RefreshSig$();
export const App = () => {
	_s();
	const [count, setCount] = useState(0);
	return _jsxs("button", {
		onClick: () => setCount(count + 1),
		children: ["count is ", count]
	});
};
_s(App, "oDgYfYHkD9Wkv4hrAPCkI/ev3YU=");
_c = App;
var _c;
$RefreshReg$(_c, "App");
`,
    );
  });
});

describe('define plugin', () => {
  it('matches output', () => {
    const code = 'if (process.env.NODE_ENV === "production") { foo; }';
    const ret = transform('test.tsx', code, {
      define: {
        'process.env.NODE_ENV': '"development"',
      },
    });
    assert.equal(ret.code, '');
  });

  it('handles typescript declare global', () => {
    const code = 'declare let __TEST_DEFINE__: string; console.log({ __TEST_DEFINE__ });';
    const ret = transform('test.ts', code, {
      define: {
        '__TEST_DEFINE__': '"replaced"',
      },
    });
    assert.equal(ret.code, 'console.log({ __TEST_DEFINE__: "replaced" });\n');
  });
});

describe('inject plugin', () => {
  const code = 'let _ = Object.assign';

  it('matches output', () => {
    const ret = transform('test.tsx', code, {
      inject: {
        'Object.assign': 'foo',
      },
    });
    assert.equal(ret.code, 'import $inject_Object_assign from "foo";\nlet _ = $inject_Object_assign;\n');
  });
});