From 57657c0860b1016fae4531fb8fb08bb8cd297eb1 Mon Sep 17 00:00:00 2001
From: Shigma <1700011071@pku.edu.cn>
Date: Mon, 3 Feb 2020 12:31:09 +0800
Subject: [PATCH] feat(utils): getDateNumber receive number

---
 packages/koishi-utils/src/date.ts        | 3 ++-
 packages/koishi-utils/tests/date.spec.ts | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/packages/koishi-utils/src/date.ts b/packages/koishi-utils/src/date.ts
index 21077075e4..8131415c28 100644
--- a/packages/koishi-utils/src/date.ts
+++ b/packages/koishi-utils/src/date.ts
@@ -1,4 +1,5 @@
-export function getDateNumber (date = new Date()) {
+export function getDateNumber (date: number | Date = new Date()) {
+  if (typeof date === 'number') date = new Date(date)
   return Math.floor(Date.UTC(date.getFullYear(), date.getMonth(), date.getDate()) / 86400000)
 }
 
diff --git a/packages/koishi-utils/tests/date.spec.ts b/packages/koishi-utils/tests/date.spec.ts
index d4d522d825..0df05f29cb 100644
--- a/packages/koishi-utils/tests/date.spec.ts
+++ b/packages/koishi-utils/tests/date.spec.ts
@@ -4,6 +4,7 @@ describe('Date Manipulations', () => {
   test('getDateNumber', () => {
     getDateNumber() /* make coverage happy */
     expect(getDateNumber(new Date(2020, 0))).toBe(18262)
+    expect(getDateNumber(1577808000000)).toBe(18262)
   })
 
   test('fromDateNumber', () => {