feat: 🎸 includeClasses option

bddvlpr · Jan 25, 2023 · c528d0a · c528d0a
1 parent 43bec88
commit c528d0a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/utils/events.ts b/src/utils/events.ts
@@ -22,7 +22,11 @@ const createCalendar = async (timetable: Lesson[], options: FormatOptions) => {
   return createEvents([
     ...timetable
       .filter((lesson) => {
-        if (options?.excludeClasses) {
+        if (options?.includeClasses) {
+          return lesson.su.some((su) =>
+            options.includeClasses?.includes(su.id)
+          );
+        } else if (options?.excludeClasses) {
           return !lesson.su.some((su) =>
             options.excludeClasses?.includes(su.id)
           );

diff --git a/src/utils/time.ts b/src/utils/time.ts
@@ -6,6 +6,7 @@ import WebUntis, { Holiday, Lesson } from "webuntis";
 
 interface FormatOptions {
   offsetHours?: number;
+  includeClasses?: number[];
   excludeClasses?: number[];
   notifyBefore?: number;
 }