|
| 1 | +import { computed, ref } from 'vue' |
| 2 | + |
| 3 | +import type { MaybeRef } from '../types' |
| 4 | +import { toValue, tryOnScopeDispose } from '../utils' |
| 5 | + |
| 6 | +export interface UseSmsCountdownOptions { |
| 7 | + /** |
| 8 | + * 倒计时总共的时间(s) |
| 9 | + * @default 60s |
| 10 | + */ |
| 11 | + totalSecond?: number |
| 12 | + /** 是否可发送 |
| 13 | + * @default true |
| 14 | + */ |
| 15 | + sendAble?: MaybeRef<boolean> |
| 16 | + /** |
| 17 | + * 开始倒计时的文本 |
| 18 | + * @default '获取验证码' |
| 19 | + */ |
| 20 | + startText?: string |
| 21 | + /** |
| 22 | + * 倒计时期间的提示语,必须带有字符 "%s",这里的 "%s",将会被倒计的秒数替代 |
| 23 | + * @default '%s秒后重发' |
| 24 | + */ |
| 25 | + durationText?: string |
| 26 | +} |
| 27 | + |
| 28 | +export function useSmsCountdown(options?: UseSmsCountdownOptions) { |
| 29 | + const { totalSecond = 60, sendAble = true, startText = '获取验证码', durationText = 'x秒后重发' } = options || {} |
| 30 | + |
| 31 | + if (totalSecond <= 0 && totalSecond % 1 !== 0) |
| 32 | + throw new Error('倒计时的时间应该为一个正整数!') |
| 33 | + |
| 34 | + const counts = ref(totalSecond) |
| 35 | + |
| 36 | + /** 是否可以发送验证码,由外部转入的 sendAble 和当前的秒数共同确定 */ |
| 37 | + const canSend = computed(() => { |
| 38 | + return toValue(sendAble) && counts.value === totalSecond |
| 39 | + }) |
| 40 | + |
| 41 | + const text = computed(() => { |
| 42 | + if (counts.value === totalSecond) |
| 43 | + return startText |
| 44 | + |
| 45 | + return durationText.replace(/%s/i, counts.value.toString()) |
| 46 | + }) |
| 47 | + |
| 48 | + let intervalId: ReturnType<typeof setInterval> | null = null |
| 49 | + |
| 50 | + function startCountdown() { |
| 51 | + if (!canSend.value) |
| 52 | + return |
| 53 | + |
| 54 | + counts.value-- |
| 55 | + intervalId = setInterval(() => { |
| 56 | + counts.value-- |
| 57 | + if (counts.value <= 0) { |
| 58 | + counts.value = totalSecond |
| 59 | + clearInterval(intervalId!) |
| 60 | + } |
| 61 | + }, 1000) |
| 62 | + } |
| 63 | + |
| 64 | + /** |
| 65 | + * 停止计时 |
| 66 | + */ |
| 67 | + function stopCountdown() { |
| 68 | + intervalId && clearInterval(intervalId) |
| 69 | + intervalId = null |
| 70 | + counts.value = totalSecond |
| 71 | + } |
| 72 | + tryOnScopeDispose(stopCountdown) |
| 73 | + |
| 74 | + return { |
| 75 | + counts, |
| 76 | + canSend, |
| 77 | + text, |
| 78 | + startCountdown, |
| 79 | + stopCountdown, |
| 80 | + } |
| 81 | +} |
0 commit comments