import { Output, refkey, StatementList } from "@alloy-js/core";
import { expect, it } from "vitest";

import * as ts from "../src/components/index.js";
import { createTSNamePolicy } from "../src/name-policy.js";

it("applies to functions and variables", () => {
  const ref1 = refkey({});
  const ref2 = refkey({});

  const namePolicy = createTSNamePolicy();
  expect(
    <Output namePolicy={namePolicy}>
      <ts.SourceFile path="test.ts">
        <ts.FunctionDeclaration name="foo-bar" refkey={ref1} />
        <hbr />
        <StatementList>
          <ts.VarDeclaration name="one-two" refkey={ref2}>
            "hello"
          </ts.VarDeclaration>
          <ts.Reference refkey={ref1} />
          <ts.Reference refkey={ref2} />
        </StatementList>
      </ts.SourceFile>
    </Output>,
  ).toRenderTo(`
    function fooBar() {}
    const oneTwo = "hello";
    fooBar;
    oneTwo;
  `);
});

it("keeps _ and $ prefix", () => {
  const ref1 = refkey({});
  const ref2 = refkey({});

  const namePolicy = createTSNamePolicy();
  expect(
    <Output namePolicy={namePolicy}>
      <ts.SourceFile path="test.ts">
        <ts.FunctionDeclaration name="_foo-bar" refkey={ref1} />
        <hbr />
        <StatementList>
          <ts.VarDeclaration name="$one-two" refkey={ref2}>
            "hello"
          </ts.VarDeclaration>
          <ts.Reference refkey={ref1} />
          <ts.Reference refkey={ref2} />
        </StatementList>
      </ts.SourceFile>
    </Output>,
  ).toRenderTo(`
    function _fooBar() {}
    const $oneTwo = "hello";
    _fooBar;
    $oneTwo;
  `);
});

it("appends _ to reserved words", () => {
  const ref1 = refkey({});
  const ref2 = refkey({});

  const namePolicy = createTSNamePolicy();
  expect(
    <Output namePolicy={namePolicy}>
      <ts.SourceFile path="test.ts">
        <ts.FunctionDeclaration
          name="default"
          refkey={ref1}
          parameters={[{ name: "await", type: "any" }]}
        />
        <hbr />
        <StatementList>
          <ts.VarDeclaration name="super" refkey={ref2}>
            "hello"
          </ts.VarDeclaration>
          <ts.Reference refkey={ref1} />
          <ts.Reference refkey={ref2} />
        </StatementList>
      </ts.SourceFile>
    </Output>,
  ).toRenderTo(`
    function default_(await_: any) {}
    const super_ = "hello";
    default_;
    super_;
  `);
});