import { useEffect, useState } from "react";

export default function useApplyDepartment(props, query) {
  const [departmentTree, setDepartmentTree] = useState([]);

  const getDepartmentTree = async () => {
    if (!props.entrance) {
      const { data: currentDepartmentTree } = await props["departmentListTree"]();
      const { data: stakeholderDepartmentTree } = await props["departmentListAllTreeByCorpType"]({ enterpriseType: [3] });
      setDepartmentTree([...currentDepartmentTree, ...stakeholderDepartmentTree]);
    }
    else if (props.entrance === "stakeholder") {
      const { data: currentDepartmentTree } = await props["departmentListTree"]();
      setDepartmentTree(currentDepartmentTree);
    }
    else if (props.entrance === "supervision") {
      const { data: currentDepartmentTree } = await props["departmentListTree"]({ eqCorpinfoId: query.corpinfoId });
      setDepartmentTree(currentDepartmentTree);
    }
  };

  useEffect(() => {
    getDepartmentTree();
  }, []);

  return { departmentTree };
}