vite-plugin-antd-dayjs/src/antd-plugin.js

vite plugin for replacing moment.js with Day.js

/**
 * fork from antd-dayjs-webpack-plugin
 */
let localeMap = {
  en_GB: "en-gb",
  en_US: "en",
  zh_CN: "zh-cn",
  zh_TW: "zh-tw",
};

let parseLocale = function parseLocale(locale) {
  let mapLocale = localeMap[locale];
  return mapLocale || locale.split("_")[0];
};

module.exports = function (option, dayjsClass, dayjsFactory) {
  let oldLocale = dayjsClass.prototype.locale;
  dayjsClass.prototype.locale = function (arg) {
    if (typeof arg === "string") {
      arg = parseLocale(arg);
    }
    return oldLocale.call(this, arg);
  };
};