// 判断登录来源的工具函数

/**
 * 获取登录页面来源
 * @returns {string} 返回 'login' 或 'loginTMS'
 */
export function getLoginPageSource() {
  return sessionStorage.getItem('loginPage') || 'login';
}

/**
 * 判断是否来自TMS登录页面
 * @returns {boolean} 如果是TMS登录页面返回true，否则返回false
 */
export function isFromTMSLogin() {
  return getLoginPageSource() === 'loginTMS';
}

/**
 * 判断是否来自普通登录页面
 * @returns {boolean} 如果是普通登录页面返回true，否则返回false
 */
export function isFromNormalLogin() {
  return getLoginPageSource() === 'login';
}

/**
 * 根据登录来源执行不同的逻辑
 * @param {Function} tmsCallback - 来自TMS登录时的回调函数
 * @param {Function} normalCallback - 来自普通登录时的回调函数
 */
export function executeByLoginSource(tmsCallback, normalCallback) {
  if (isFromTMSLogin()) {
    tmsCallback && tmsCallback();
  } else {
    normalCallback && normalCallback();
  }
}