"use client";

import { useState } from "react";
import CompanyAuditHeader from "./CompanyAuditHeader";
import CompanyAuditTable from "./CompanyAuditTable";
import AuditStageTable from "./AuditStages/AuditStageTable";
import { AppToaster } from "./../../../../components/AppToaster";

export type ActiveModuleTab = "audits" | "stages";

export default function CompanyAuditsPage() {
  const [activeTab, setActiveTab]   = useState<ActiveModuleTab>("audits");
  const [refreshFlag, setRefreshFlag] = useState(false);
  const refreshData = () => setRefreshFlag((prev) => !prev);

  return (
    <>
      <AppToaster />

      {/* ── Header (contains the module tab switcher + action buttons) ── */}
      <CompanyAuditHeader
        activeTab={activeTab}
        setActiveTab={setActiveTab}
        refreshData={refreshData}
      />

      {/* ── Content ── */}
      {activeTab === "audits" ? (
        <CompanyAuditTable refreshFlag={refreshFlag} />
      ) : (
        <AuditStageTable />
      )}
    </>
  );
}