1 files changed, 134 insertions, 0 deletions
diff --git a/python/edu/learn-python/src/com/jetbrains/python/edu/actions/StudyReloadCourseAction.java b/python/edu/learn-python/src/com/jetbrains/python/edu/actions/StudyReloadCourseAction.java
new file mode 100644
index 000000000000..beefaaa94f7e
--- /dev/null
+++ b/python/edu/learn-python/src/com/jetbrains/python/edu/actions/StudyReloadCourseAction.java
@@ -0,0 +1,134 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.jetbrains.python.edu.actions;
+
+import com.intellij.ide.projectView.ProjectView;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.actionSystem.Presentation;
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.editor.Document;
+import com.intellij.openapi.fileEditor.FileDocumentManager;
+import com.intellij.openapi.fileEditor.FileEditorManager;
+import com.intellij.openapi.project.DumbAwareAction;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.util.ui.tree.TreeUtil;
+import com.jetbrains.python.edu.StudyTaskManager;
+import com.jetbrains.python.edu.StudyUtils;
+import com.jetbrains.python.edu.course.Course;
+import com.jetbrains.python.edu.course.Lesson;
+import com.jetbrains.python.edu.course.Task;
+import com.jetbrains.python.edu.course.TaskFile;
+import org.jetbrains.annotations.NotNull;
+
+import javax.swing.*;
+import javax.swing.tree.TreePath;
+import java.util.List;
+import java.util.Map;
+
+public class StudyReloadCourseAction extends DumbAwareAction {
+
+  public StudyReloadCourseAction() {
+    super("Reload Course", "Reload Course", null);
+  }
+
+  @Override
+  public void update(@NotNull AnActionEvent e) {
+    Presentation presentation = e.getPresentation();
+    Project project = e.getProject();
+    if (project != null) {
+      Course course = StudyTaskManager.getInstance(project).getCourse();
+      if (course != null) {
+        presentation.setVisible(true);
+        presentation.setEnabled(true);
+      }
+    }
+    presentation.setVisible(false);
+    presentation.setEnabled(false);
+  }
+
+  @Override
+  public void actionPerformed(@NotNull AnActionEvent e) {
+    Project project = e.getProject();
+    if (project == null) {
+      return;
+    }
+    reloadCourse(project);
+  }
+
+  public static void reloadCourse(@NotNull final Project project) {
+    ApplicationManager.getApplication().invokeLater(new Runnable() {
+      @Override
+      public void run() {
+        ApplicationManager.getApplication().runWriteAction(new Runnable() {
+          @Override
+          public void run() {
+            Course course = StudyTaskManager.getInstance(project).getCourse();
+            if (course == null) {
+              return;
+            }
+            for (VirtualFile file : FileEditorManager.getInstance(project).getOpenFiles()) {
+              FileEditorManager.getInstance(project).closeFile(file);
+            }
+            JTree tree = ProjectView.getInstance(project).getCurrentProjectViewPane().getTree();
+            TreePath path = TreeUtil.getFirstNodePath(tree);
+            tree.collapsePath(path);
+            List<Lesson> lessons = course.getLessons();
+            for (Lesson lesson : lessons) {
+              List<Task> tasks = lesson.getTaskList();
+              VirtualFile lessonDir = project.getBaseDir().findChild(Lesson.LESSON_DIR + (lesson.getIndex() + 1));
+              if (lessonDir == null) {
+                continue;
+              }
+              for (Task task : tasks) {
+                VirtualFile taskDir = lessonDir.findChild(Task.TASK_DIR + (task.getIndex() + 1));
+                if (taskDir == null) {
+                  continue;
+                }
+                Map<String, TaskFile> taskFiles = task.getTaskFiles();
+                for (Map.Entry<String, TaskFile> entry : taskFiles.entrySet()) {
+                  String name = entry.getKey();
+                  TaskFile taskFile = entry.getValue();
+                  VirtualFile file = taskDir.findChild(name);
+                  if (file == null) {
+                    continue;
+                  }
+                  Document document = FileDocumentManager.getInstance().getDocument(file);
+                  if (document == null) {
+                    continue;
+                  }
+                  StudyRefreshTaskFileAction.resetTaskFile(document, project, course, taskFile, name, task);
+                }
+              }
+            }
+            Lesson firstLesson = StudyUtils.getFirst(lessons);
+            if (firstLesson == null) {
+              return;
+            }
+            Task firstTask = StudyUtils.getFirst(firstLesson.getTaskList());
+            VirtualFile lessonDir = project.getBaseDir().findChild(Lesson.LESSON_DIR + (firstLesson.getIndex() + 1));
+            if (lessonDir != null) {
+              VirtualFile taskDir = lessonDir.findChild(Task.TASK_DIR + (firstTask.getIndex() + 1));
+              if (taskDir != null) {
+                ProjectView.getInstance(project).select(taskDir, taskDir, true);
+              }
+            }
+          }
+        });
+      }
+    });
+  }
+}