[type] bem (#2698)

This commit is contained in:
neverland 2019-02-07 11:08:57 +08:00 committed by GitHub
parent 0fb1083b68
commit cf441d9326
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -7,21 +7,26 @@
* b(['disabled', 'primary']) // 'button button--disabled button--primary' * b(['disabled', 'primary']) // 'button button--disabled button--primary'
*/ */
type Mod = string | { [key: string]: any };
type Mods = Mod | Mod[];
const ELEMENT = '__'; const ELEMENT = '__';
const MODS = '--'; const MODS = '--';
const join = (name: string, el: string, symbol: string) => (el ? name + symbol + el : name); function join(name: string, el: string, symbol: string): string {
return el ? name + symbol + el : name;
}
const prefix = (name: string, mods: any): any => { function prefix(name: string, mods: Mods): Mods {
if (typeof mods === 'string') { if (typeof mods === 'string') {
return join(name, mods, MODS); return join(name, mods, MODS);
} }
if (Array.isArray(mods)) { if (Array.isArray(mods)) {
return mods.map(item => prefix(name, item)); return mods.map(item => <Mod>prefix(name, item));
} }
const ret: { [key: string]: any } = {}; const ret: Mods = {};
if (mods) { if (mods) {
Object.keys(mods).forEach(key => { Object.keys(mods).forEach(key => {
ret[name + MODS + key] = mods[key]; ret[name + MODS + key] = mods[key];
@ -29,9 +34,9 @@ const prefix = (name: string, mods: any): any => {
} }
return ret; return ret;
}; }
export default (name: string) => (el: any, mods?: any) => { export default (name: string) => (el: Mods, mods?: Mods): Mods => {
if (el && typeof el !== 'string') { if (el && typeof el !== 'string') {
mods = el; mods = el;
el = ''; el = '';