const locales = {};

declare global {
    interface String {
        format: (...args: Array<any>) => string;
        rawformat: (object: any) => string;
        translate: (language: string | undefined) => string;
    }
}

String.prototype.format = function formatStr(...args) {
    let result = this;
    if (args.length > 0) {
        if (args.length === 1 && typeof (args[0]) === 'object') {
            for (const key in args) {
                if (args[key] !== undefined) {
                    const reg = new RegExp(`(\\{${key}\\})`, 'g');
                    result = result.replace(reg, args[key]);
                }
            }
        } else {
            for (let i = 0; i < args.length; i++) {
                if (args[i] !== undefined) {
                    const reg = new RegExp(`(\\{)${i}(\\})`, 'g');
                    result = result.replace(reg, args[i]);
                }
            }
        }
    }
    return result;
};

String.prototype.rawformat = function rawFormat(object) {
    const res = this.split('{@}');
    return [res[0], object, res[1]].join();
};

String.prototype.translate = function translate(language = 'zh_CN') {
    if (locales[language]) {
        if (locales[language][this] !== undefined) return locales[language][this];
    }
    return this;
};

export default function load(data) {
    for (const i in data) {
        if (!locales[i]) locales[i] = data[i];
        else locales[i] = Object.assign(locales[i], data[i]);
    }
}

global.Hydro.locales = locales;
global.Hydro.lib.i18n = load;