feat(startOf): add startOf()

iamkun · Apr 11, 2018 · 6443b7f · 6443b7f
1 parent db66598
commit 6443b7f
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/src/index.js b/src/index.js
@@ -7,6 +7,7 @@ class Dayjs {
 
   parseConfig(config) {
     if (!config) return new Date()
+    if (config instanceof Date) return config
     if (/^(\d){8}$/.test(config)) {
       this.utc = true
       const y = config.substr(0, 4)
@@ -33,6 +34,17 @@ class Dayjs {
   toString() {
     return this.date.toUTCString()
   }
+
+  startOf(arg) {
+    switch (arg) {
+      case 'year':
+        return new Dayjs(new Date(this.year(), 0, 1))
+      case 'month':
+        return new Dayjs(new Date(this.year(), this.month(), 1))
+      default:
+        return this
+    }
+  }
 }
 
 export default config => (

diff --git a/test/manipulate.test.js b/test/manipulate.test.js
@@ -0,0 +1,15 @@
+import moment from 'moment'
+import dayjs from '../src'
+
+test('StartOfYear', () => {
+  expect(dayjs().startOf('year').unix()).toBe(moment().startOf('year').unix())
+})
+
+test('StartOfMonth', () => {
+  expect(dayjs().startOf('month').unix()).toBe(moment().startOf('month').unix())
+})
+
+test('StartOfOther', () => {
+  expect(dayjs().startOf('otherString').unix()).toBe(moment().startOf('otherString').unix())
+})
+