python/edu/course-creator/src/org/jetbrains/plugins/coursecreator/format/Lesson.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

package org.jetbrains.plugins.coursecreator.format;

import com.google.gson.annotations.Expose;
import com.intellij.psi.PsiDirectory;
import org.jetbrains.annotations.NotNull;

import java.util.*;

public class Lesson implements Comparable{
  @Expose public String name;
  @Expose public List<Task> task_list = new ArrayList<Task>();

  public int myIndex;
  public Map<String, Task> myTasksMap = new HashMap<String, Task>();

  public Lesson() {}

  public Lesson(@NotNull final String name) {
    this.name = name;
  }

  public void addTask(@NotNull final Task task, PsiDirectory taskDirectory) {
    myTasksMap.put(taskDirectory.getName(), task);
    task_list.add(task);
  }

  public void setName(String name) {
    this.name = name;
  }

  public Task getTask(@NotNull final String name) {
    return myTasksMap.get(name);
  }

  public List<Task> getTaskList() {
    return task_list;
  }

  public void setIndex(int index) {
    myIndex = index;
  }

  public int getIndex() {
    return myIndex;
  }

  public Map<String, Task> getTasksMap() {
    return myTasksMap;
  }

  public void init() {
    task_list.clear();
    for (Task task : myTasksMap.values()) {
      task_list.add(task);
    }
    Collections.sort(task_list);
  }

  @Override
  public int compareTo(@NotNull Object o) {
    Lesson lesson = (Lesson) o;
    return myIndex - lesson.getIndex();
  }
}