better-mock/src/random/text.ts

Forked from Mockjs. Generate random data & Intercept ajax request. Support miniprogram.

import { isDef } from '../utils'
import * as basic from './basic'
import * as helper from './helper'
import stringToArray from "../utils/string-to-array";

const _range = function (defaultMin: number, defaultMax: number, min?: number, max?: number): number {
  return !isDef(min)
    ? basic.natural(defaultMin, defaultMax)
    : !isDef(max)
      ? min!
      : basic.natural(parseInt(min!.toString(), 10), parseInt(max!.toString(), 10)) // ( min, max )
}

// 随机生成一段文本。
export const paragraph = function (min?: number, max?: number): string {
  const len = _range(3, 7, min, max)
  const result: string[] = []
  for (let i = 0; i < len; i++) {
    result.push(sentence())
  }
  return result.join(' ')
}

export const cparagraph = function (min?: number, max?: number): string {
  const len = _range(3, 7, min, max)
  const result: string[] = []
  for (let i = 0; i < len; i++) {
    result.push(csentence())
  }
  return result.join('')
}

// 随机生成一个句子，第一个单词的首字母大写。
export const sentence = function (min?: number, max?: number): string {
  const len = _range(12, 18, min, max)
  const result: string[] = []
  for (let i = 0; i < len; i++) {
    result.push(word())
  }
  return helper.capitalize(result.join(' ')) + '.'
}

// 随机生成一个中文句子。
export const csentence = function (min?: number, max?: number): string {
  const len = _range(12, 18, min, max)
  const result: string[] = []
  for (let i = 0; i < len; i++) {
    result.push(cword())
  }

  return result.join('') + '。'
}

// 随机生成一个单词。
export const word = function (min?: number, max?: number): string {
  const len = _range(3, 10, min, max)
  let result = ''
  for (let i = 0; i < len; i++) {
    result += basic.character('lower')
  }
  return result
}

// 随机生成一个或多个汉字。
export const cword = function (pool: string = '', min?: number, max?: number): string {
  // 最常用的 500 个汉字 http://baike.baidu.com/view/568436.htm
  const cnWords = '的一是在不了有和人这中大为上个国我以要他时来用们生到作地于出就分对成会可主发年动同工也能下过子说产种面而方后多定行学法所民得经十三之进着等部度家电力里如水化高自二理起小物现实加量都两体制机当使点从业本去把性好应开它合还因由其些然前外天政四日那社义事平形相全表间样与关各重新线内数正心反你明看原又么利比或但质气第向道命此变条只没结解问意建月公无系军很情者最立代想已通并提直题党程展五果料象员革位入常文总次品式活设及管特件长求老头基资边流路级少图山统接知较将组见计别她手角期根论运农指几九区强放决西被干做必战先回则任取据处队南给色光门即保治北造百规热领七海口东导器压志世金增争济阶油思术极交受联什认六共权收证改清己美再采转更单风切打白教速花带安场身车例真务具万每目至达走积示议声报斗完类八离华名确才科张信马节话米整空元况今集温传土许步群广石记需段研界拉林律叫且究观越织装影算低持音众书布复容儿须际商非验连断深难近矿千周委素技备半办青省列习响约支般史感劳便团往酸历市克何除消构府称太准精值号率族维划选标写存候毛亲快效斯院查江型眼王按格养易置派层片始却专状育厂京识适属圆包火住调满县局照参红细引听该铁价严龙飞'

  let len
  switch (arguments.length) {
    case 0: // ()
      pool = cnWords
      len = 1
      break
    case 1: // ( pool )
      if (typeof arguments[0] === 'string') {
        len = 1
      } else {
        // ( length )
        len = pool
        pool = cnWords
      }
      break
    case 2:
      // ( pool, length )
      if (typeof arguments[0] === 'string') {
        len = min
      } else {
        // ( min, max )
        len = basic.natural(parseInt(pool, 10), min)
        pool = cnWords
      }
      break
    case 3:
      len = basic.natural(min, max)
      break
  }

  let result = ''
  for (let i = 0; i < len; i++) {
    result += pool.charAt(basic.natural(0, pool.length - 1))
  }
  return result
}

// 随机生成一个或多个 emoji 符号
export const emoji = function (pool?: string | number, min?: number, max?: number) {
  if (!['string', 'number', 'undefined'].includes(typeof pool)){
    return ''
  }
  // 常用的 338 个emoji符号 http://www.fhdq.net/emoji.html
  const emojis = '😀😁😂😃😄😅😆😉😊😋😎😍😘😗😙😚☺😇😐😑😶😏😣😥😮😯😪😫😴😌😛😜😝😒😓😔😕😲😷😖😞😟😤😢😭😦😧😨😬😰😱😳😵😡😠😈👿👹👺💀👻👽👦👧👨👩👴👵👶👱👮👲👳👷👸💂🎅👰👼💆💇🙍🙎🙅🙆💁🙋🙇🙌🙏👤👥🚶🏃👯💃👫👬👭💏💑👪💪👈👉☝👆👇✌✋👌👍👎✊👊👋👏👐✍👣👀👂👃👅👄💋👓👔👕👖👗👘👙👚👛👜👝🎒💼👞👟👠👡👢👑👒🎩🎓💄💅💍🌂🙈🙉🙊🐵🐒🐶🐕🐩🐺🐱😺😸😹😻😼😽🙀😿😾🐈🐯🐅🐆🐴🐎🐮🐂🐃🐄🐷🐖🐗🐽🐏🐑🐐🐪🐫🐘🐭🐁🐀🐹🐰🐇🐻🐨🐼🐾🐔🐓🐣🐤🐥🐦🐧🐸🐊🐢🐍🐲🐉🐳🐋🐬🐟🐠🐡🐙🐚🐌🐛🐜🐝🐞💐🌸💮🌹🌺🌻🌼🌷🌱🌲🌳🌴🌵🌾🌿🍀🍁🍂🍃🌍🌎🌏🌐🌑🌒🌓🌔🌕🌖🌗🌘🌙🌚🌛🌜☀🌝🌞⭐🌟🌠☁⛅☔⚡❄🔥💧🌊💩🍇🍈🍉🍊🍋🍌🍍🍎🍏🍐🍑🍒🍓🍅🍆🌽🍄🌰🍞🍖🍗🍔🍟🍕🍳🍲🍱🍘🍙🍚🍛🍜🍝🍠🍢🍣🍤🍥🍡🍦🍧🍨🍩🍪🎂🍰🍫🍬🍭🍮🍯🍼☕🍵🍶🍷🍸🍹🍺🍻🍴'
  let array = stringToArray(emojis)
  if (typeof pool === 'string') {  // emoji("😀😁😂"), emoji("😀😂", 2), emoji("😀😂", 2, 3)
    array = stringToArray(pool)
  } else if (typeof pool === 'number') {  // emoji(2), emoji(2, 3)
    max = min
    min = pool
  }
  if (min === undefined || min < 2){  // emoji("😀😁😂"), emoji()
    return helper.pick(array)  // pick(['1', '2']) => "2", pick(['1', '2'], 1) => "2"
  }
  return helper.pick(array, min, max).join('')
}

// 随机生成一句标题，其中每个单词的首字母大写。
export const title = function (min?: number, max?: number): string {
  const len = _range(3, 7, min, max)
  const result: string[] = []
  for (let i = 0; i < len; i++) {
    result.push(helper.capitalize(word()))
  }
  return result.join(' ')
}

// 随机生成一句中文标题。
export const ctitle = function (min?: number, max?: number): string {
  const len = _range(3, 7, min, max)
  const result: string[] = []
  for (let i = 0; i < len; i++) {
    result.push(cword())
  }
  return result.join('')
}