import { usePublicApi } from '..';

import type { Menu } from '../types';

export function listMenu(codes: string): Promise<Menu>;

export function listMenu(codes: Array<string>): Promise<Array<Menu>>;

export function listMenu(codes: Array<string> | string) {
  if (Array.isArray(codes)) {
    return usePublicApi<Array<Menu>>({
      endpoint: '/publics/template/menus/',
      params: {
        codes,
      },
    });
  }

  return usePublicApi<Array<Menu>>({
    endpoint: '/publics/template/menus/',
    params: {
      codes: [codes],
    },
  }).then(data => data[0]);
}