feat: add quarter dates to pivot dates

- this quarter start - this quarter end - last quarter start - last quarter end
telostat · Nov 16, 2023 · a6f4d2f · a6f4d2f
1 parent d6633d3
commit a6f4d2f
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/src/-temporal.test.ts b/src/-temporal.test.ts
@@ -24,6 +24,10 @@ describe('PivotDate queries', () => {
     expect(getPivotDate(PivotDate.THIS_YEAR_END, asofStr)).toBe('2022-12-31');
     expect(getPivotDate(PivotDate.NEXT_YEAR_START, asofStr)).toBe('2023-01-01');
     expect(getPivotDate(PivotDate.NEXT_YEAR_END, asofStr)).toBe('2023-12-31');
+    expect(getPivotDate(PivotDate.THIS_QUARTER_START, asofStr)).toBe('2022-07-01');
+    expect(getPivotDate(PivotDate.THIS_QUARTER_END, asofStr)).toBe('2022-09-30');
+    expect(getPivotDate(PivotDate.LAST_QUARTER_START, asofStr)).toBe('2022-04-01');
+    expect(getPivotDate(PivotDate.LAST_QUARTER_END, asofStr)).toBe('2022-06-30');
   });
 
   test('over Dayjs values', () => {
@@ -45,6 +49,10 @@ describe('PivotDate queries', () => {
     expect(getPivotDate(PivotDate.THIS_YEAR_END, asofDjs)).toStrictEqual(dayjs('2022-12-31').endOf('day'));
     expect(getPivotDate(PivotDate.NEXT_YEAR_START, asofDjs)).toStrictEqual(dayjs('2023-01-01').endOf('day'));
     expect(getPivotDate(PivotDate.NEXT_YEAR_END, asofDjs)).toStrictEqual(dayjs('2023-12-31').endOf('day'));
+    expect(getPivotDate(PivotDate.THIS_QUARTER_START, asofDjs)).toStrictEqual(dayjs('2022-07-01').endOf('day'));
+    expect(getPivotDate(PivotDate.THIS_QUARTER_END, asofDjs)).toStrictEqual(dayjs('2022-09-30').endOf('day'));
+    expect(getPivotDate(PivotDate.LAST_QUARTER_START, asofDjs)).toStrictEqual(dayjs('2022-04-01').endOf('day'));
+    expect(getPivotDate(PivotDate.LAST_QUARTER_END, asofDjs)).toStrictEqual(dayjs('2022-06-30').endOf('day'));
   });
 
   test('over empty values', () => {

diff --git a/src/-temporal.ts b/src/-temporal.ts
@@ -75,6 +75,10 @@ export enum PivotDate {
   THIS_YEAR_END = 'this year end',
   NEXT_YEAR_START = 'next year start',
   NEXT_YEAR_END = 'next year end',
+  THIS_QUARTER_START = 'this quarter start',
+  THIS_QUARTER_END = 'this quarter end',
+  LAST_QUARTER_START = 'last quarter start',
+  LAST_QUARTER_END = 'last quarter end',
 }
 
 /* Signature of getPivotDate specialized over Dayjs argument. */
@@ -151,6 +155,18 @@ export function getPivotDate(p: PivotDate, x?: Dayjs | SDate): Dayjs | SDate {
     case PivotDate.NEXT_YEAR_END:
       pivot = asof.add(1, 'year').endOf('year');
       break;
+    case PivotDate.THIS_QUARTER_START:
+      pivot = asof.startOf('quarter');
+      break;
+    case PivotDate.THIS_QUARTER_END:
+      pivot = asof.endOf('quarter');
+      break;
+    case PivotDate.LAST_QUARTER_START:
+      pivot = asof.subtract(1, 'quarter').startOf('quarter');
+      break;
+    case PivotDate.LAST_QUARTER_END:
+      pivot = asof.subtract(1, 'quarter').endOf('quarter');
+      break;
     default:
       throw Error('Unreachable case');
   }