@cpany/app/src/utils.ts

CPany web app

import { Ref, ref, unref } from 'vue';
import {
  IContest,
  IProblem,
  ParticipantType,
  isCodeforces,
  isAtCoder,
  isNowCoder,
  isPintia,
  isLuogu,
  isHdu
} from '@cpany/types';

import { utcToZonedTime } from 'date-fns-tz';

export function isUndef<T>(object: T | undefined | null): object is undefined | null {
  return object === undefined || object === null;
}

export function isDef<T>(object: T | undefined | null): object is T {
  return object !== undefined && object !== null;
}

function alignNumber(value: number) {
  return (value < 10 ? '0' : '') + value;
}

export function toUTC(date: Date) {
  return Date.UTC(
    date.getUTCFullYear(),
    date.getUTCMonth(),
    date.getUTCDate(),
    date.getUTCHours(),
    date.getUTCMinutes(),
    date.getUTCSeconds(),
    date.getUTCMilliseconds()
  );
}

export function toDate(seconds: number | Ref<number>) {
  const utc = toUTC(new Date(unref(seconds) * 1000));
  const date = utcToZonedTime(utc, 'Asia/Shanghai');
  const prefix = `${date.getFullYear()}-${alignNumber(date.getMonth() + 1)}-${alignNumber(
    date.getDate()
  )} `;
  const hours = alignNumber(date.getHours());
  const minutes = alignNumber(date.getMinutes());
  return ref(prefix + hours + ':' + minutes);
}

export function toDay(seconds: number | Ref<number>) {
  const utc = toUTC(new Date(unref(seconds) * 1000));
  const date = utcToZonedTime(utc, 'Asia/Shanghai');
  return ref(
    `${date.getFullYear()}-${alignNumber(date.getMonth() + 1)}-${alignNumber(date.getDate())} `
  );
}

export function toNumDuration(duration: number | Ref<number>) {
  const seconds = unref(duration);
  const hour = Math.floor(seconds / 3600);
  const minute = Math.floor((seconds % 3600) / 60);
  const second = Math.floor(seconds % 60);
  return ref(`${hour}:${alignNumber(minute)}:${alignNumber(second)}`);
}

export function toDuration(duration: number | Ref<number>) {
  const seconds = unref(duration);
  const hour = Math.floor(seconds / 3600);
  const minute = Math.floor((seconds % 3600) / 60);
  const second = Math.floor(seconds % 60);
  return ref(
    [
      hour > 0 ? hour + ' 小时' : '',
      minute > 0 ? minute + ' 分钟' : '',
      second > 0 ? second + ' 秒' : ''
    ].join(' ')
  );
}

export function debounce(fn: Function, delay = 1000) {
  let timer: any;
  return function () {
    clearTimeout(timer);
    timer = setTimeout(function () {
      fn();
    }, delay);
  };
}

export function createSortBy<T>(...mapFns: Array<(item: T) => number>) {
  return (lhs: T, rhs: T) => {
    for (const fn of mapFns) {
      const l = fn(lhs);
      const r = fn(rhs);
      if (l !== r) {
        return l - r;
      }
    }
    return 0;
  };
}

export function createSortByString<T>(mapFn: (item: T) => string) {
  return (_lhs: T, _rhs: T) => {
    const lhs = mapFn(_lhs);
    const rhs = mapFn(_rhs);
    for (let i = 0; i < lhs.length && i < rhs.length; i++) {
      const result = lhs.charCodeAt(i) - rhs.charCodeAt(i);
      if (result !== 0) {
        return result;
      }
    }
    return lhs.length - rhs.length;
  };
}

export function createRevSortBy<T>(...mapFns: Array<(item: T) => number>) {
  return createSortBy(...mapFns.map((fn) => (item: T) => -fn(item)));
}

export function combineCmp<T>(...cmpFns: Array<(lhs: T, rhs: T) => number>) {
  return (lhs: T, rhs: T) => {
    for (const fn of cmpFns) {
      const result = fn(lhs, rhs);
      if (result !== 0) {
        return result;
      }
    }
    return 0;
  };
}

export const displayPlatform = (type: string) => {
  const wrapper = { type };
  if (isCodeforces(wrapper)) {
    return 'Codeforces';
  } else if (isNowCoder(wrapper)) {
    return '牛客';
  } else if (isHdu(wrapper)) {
    return 'HDu';
  } else if (isLuogu(wrapper)) {
    return '洛谷';
  } else if (isPintia(wrapper)) {
    return '拼题A';
  } else if (isAtCoder(wrapper)) {
    return 'AtCoder';
  } else {
    return type.charAt(0).toUpperCase() + type.slice(1);
  }
};

export const displayContestType = (contest: IContest) => {
  if (isCodeforces(contest)) {
    if (/Round|Div|Hello \d+|Good Bye \d+/.test(contest.name)) {
      return 'Codeforces Round';
    } else if (/gym/.test(contest.type)) {
      return 'Codeforces GYM';
    } else {
      return 'Codeforces';
    }
  } else if (isNowCoder(contest)) {
    return '牛客';
  } else if (isHdu(contest)) {
    return 'HDu';
  } else if (isLuogu(contest)) {
    return '洛谷';
  } else if (isPintia(contest)) {
    return '拼题A';
  } else if (isAtCoder(contest)) {
    return 'AtCoder';
  } else {
    const type = contest.type;
    return type.charAt(0).toUpperCase() + type.slice(1);
  }
};

export const displayProblemType = (problem: IProblem) => {
  return displayPlatform(problem.type);
};

export const displayParticipantType = (type: ParticipantType) => {
  if (type === ParticipantType.CONTESTANT) {
    return '比赛';
  } else if (type === ParticipantType.MANAGER) {
    return '管理';
  } else if (type === ParticipantType.OUT_OF_COMPETITION) {
    return '打星';
  } else if (type === ParticipantType.PRACTICE) {
    return '练习';
  } else if (type === ParticipantType.VIRTUAL) {
    return '虚拟';
  } else {
    return '';
  }
};