[Timer] Fix children timers, restore active task

Author: Dmitry Yadrikhinsky

src/models/TaskModel.ts

@@ -5,8 +5,8 @@ import { computed, makeObservable, observable } from 'mobx';
 interface ITaskModel extends ITreeItem<ITaskModel> {
   projectId: string;
   checked: boolean;
-  time: number[][];
   active: boolean;
+  time: number[][];
 }
 
 export default class TaskModel extends AbstractModel implements ITaskModel {
@@ -21,6 +21,8 @@ export default class TaskModel extends AbstractModel implements ITaskModel {
   constructor(props: ITaskModel) {
     super();
     this.load(props);
+    this.children = props.children?.map((json) => new TaskModel(json)) || [];
+
     makeObservable(this, {
       key: observable,
       title: observable,

src/services/tasks/TaskStore.ts

@@ -58,6 +58,7 @@ export default class TaskStore {
 
   restore() {
     this.tasks = this.tasksService.getAll();
+    this.findActiveTask();
   }
 
   getCheckedKeys(projectId: string): string[] {
@@ -76,6 +77,32 @@ export default class TaskStore {
     this.tasksService.save(this.tasks);
   }
 
+  private findActiveTask() {
+    Object.keys(this.tasks).find((projectId) => {
+      const found = this.findActiveTaskRecursive(this.tasks[projectId]);
+      if (found) {
+        this.activeTask = found;
+        return true;
+      }
+      return false;
+    });
+  }
+
+  private findActiveTaskRecursive(tasks: TaskModel[]): TaskModel | undefined {
+    return tasks.find((task) => {
+      if (task.active) {
+        return task;
+      }
+      if (Array.isArray(task.children)) {
+        const found = this.findActiveTaskRecursive(task.children);
+        if (found) {
+          return found;
+        }
+      }
+      return false;
+    });
+  }
+
   private getCheckedKeysRecursive(tasks: TaskModel[], checkedIds: string[]) {
     tasks.forEach((task) => {
       if (task.checked) {